104 lines
2.1 KiB
Bash
104 lines
2.1 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
|
||
|
|
get_tmux_option() {
|
||
|
|
local option=$1
|
||
|
|
local default_value=$2
|
||
|
|
local option_value=$(tmux show-option -gqv "$option")
|
||
|
|
if [ -z "$option_value" ]; then
|
||
|
|
echo "$default_value"
|
||
|
|
else
|
||
|
|
echo "$option_value"
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
set_tmux_option() {
|
||
|
|
local option=$1
|
||
|
|
local value=$2
|
||
|
|
tmux set-option -gq "$option" "$value"
|
||
|
|
}
|
||
|
|
|
||
|
|
parse_ssh_port() {
|
||
|
|
# If there is a port get it
|
||
|
|
local port=$(echo $1|grep -Eo '\-p\s*([0-9]+)'|sed 's/-p\s*//')
|
||
|
|
|
||
|
|
if [ -z $port ]; then
|
||
|
|
local port=22
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo $port
|
||
|
|
}
|
||
|
|
|
||
|
|
get_ssh_user() {
|
||
|
|
local ssh_user=$(whoami)
|
||
|
|
|
||
|
|
for ssh_config in `awk '
|
||
|
|
$1 == "Host" {
|
||
|
|
gsub("\\\\.", "\\\\.", $2);
|
||
|
|
gsub("\\\\*", ".*", $2);
|
||
|
|
host = $2;
|
||
|
|
next;
|
||
|
|
}
|
||
|
|
$1 == "User" {
|
||
|
|
$1 = "";
|
||
|
|
sub( /^[[:space:]]*/, "" );
|
||
|
|
printf "%s|%s\n", host, $0;
|
||
|
|
}' .ssh/config`; do
|
||
|
|
local host_regex=${ssh_config%|*}
|
||
|
|
local host_user=${ssh_config#*|}
|
||
|
|
if [[ "$1" =~ $host_regex ]]; then
|
||
|
|
ssh_user=$host_user
|
||
|
|
break
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
|
||
|
|
echo $ssh_user
|
||
|
|
}
|
||
|
|
|
||
|
|
get_remote_info() {
|
||
|
|
local command=$1
|
||
|
|
|
||
|
|
# First get the current pane command pid to get the full command with arguments
|
||
|
|
local cmd=$({ pgrep -flaP `tmux display-message -p "#{pane_pid}"` ; ps -o command -p `tmux display-message -p "#{pane_pid}"` ; } | xargs -I{} echo {} | grep ssh | sed -E 's/^[0-9]*[[:blank:]]*ssh //')
|
||
|
|
|
||
|
|
local port=$(parse_ssh_port "$cmd")
|
||
|
|
|
||
|
|
local cmd=$(echo $cmd|sed 's/\-p\s*'"$port"'//g')
|
||
|
|
local user=$(echo $cmd | awk '{print $NF}'|cut -f1 -d@)
|
||
|
|
local host=$(echo $cmd | awk '{print $NF}'|cut -f2 -d@)
|
||
|
|
|
||
|
|
if [ $user == $host ]; then
|
||
|
|
local user=$(get_ssh_user $host)
|
||
|
|
fi
|
||
|
|
|
||
|
|
case "$1" in
|
||
|
|
"whoami")
|
||
|
|
echo $user
|
||
|
|
;;
|
||
|
|
"hostname")
|
||
|
|
echo $host
|
||
|
|
;;
|
||
|
|
"port")
|
||
|
|
echo $port
|
||
|
|
;;
|
||
|
|
*)
|
||
|
|
echo "$user@$host:$port"
|
||
|
|
;;
|
||
|
|
esac
|
||
|
|
}
|
||
|
|
|
||
|
|
get_info() {
|
||
|
|
# If command is ssh do some magic
|
||
|
|
if ssh_connected; then
|
||
|
|
echo $(get_remote_info $1)
|
||
|
|
else
|
||
|
|
echo $($1)
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
ssh_connected() {
|
||
|
|
# Get current pane command
|
||
|
|
local cmd=$(tmux display-message -p "#{pane_current_command}")
|
||
|
|
|
||
|
|
[ $cmd = "ssh" ] || [ $cmd = "sshpass" ]
|
||
|
|
}
|