From c9c08f58e0fac97b33059ea8dff9fa29793b45b1 Mon Sep 17 00:00:00 2001
From: AbelAnaya <abelanaya97@gmail.com>
Date: Thu, 12 Oct 2023 00:10:35 +0200
Subject: [PATCH] feat: Add ssh-session plugin

Shows user and hostname of active ssh session if any in the current tmux pane. Connected port
is also available to show under configuration option (default false)
---
 scripts/dracula.sh     |   8 +++-
 scripts/ssh_session.sh | 105 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 112 insertions(+), 1 deletion(-)
 create mode 100755 scripts/ssh_session.sh

diff --git a/scripts/dracula.sh b/scripts/dracula.sh
index 0248ded..2e8bbd7 100755
--- a/scripts/dracula.sh
+++ b/scripts/dracula.sh
@@ -30,6 +30,7 @@ main()
   show_refresh=$(get_tmux_option "@dracula-refresh-rate" 5)
   show_synchronize_panes_label=$(get_tmux_option "@dracula-synchronize-panes-label" "Sync")
   time_format=$(get_tmux_option "@dracula-time-format" "")
+  show_ssh_session_port=$(get_tmux_option "@dracula-show-ssh-session-port" false)
   IFS=' ' read -r -a plugins <<< $(get_tmux_option "@dracula-plugins" "battery network weather")
   show_empty_plugins=$(get_tmux_option "@dracula-show-empty-plugins" true)
 
@@ -146,7 +147,7 @@ main()
       IFS=' ' read -r -a colors  <<< $(get_tmux_option "@dracula-cwd-colors" "dark_gray white")
       tmux set-option -g status-right-length 250
       script="#($current_dir/cwd.sh)"
-    
+
     elif [ $plugin = "fossil" ]; then
       IFS=' ' read -r -a colors  <<< $(get_tmux_option "@dracula-fossil-colors" "green dark_gray")
       tmux set-option -g status-right-length 250
@@ -249,6 +250,11 @@ main()
     elif [ $plugin = "synchronize-panes" ]; then
       IFS=' ' read -r -a colors <<< $(get_tmux_option "@dracula-synchronize-panes-colors" "cyan dark_gray")
       script="#($current_dir/synchronize_panes.sh $show_synchronize_panes_label)"
+
+    elif [ $plugin = "ssh-session" ]; then
+      IFS=' ' read -r -a colors <<< $(get_tmux_option "@dracula-ssh-session-colors" "green dark_gray")
+      script="#($current_dir/ssh_session.sh $show_ssh_session_port)"
+
     else
       continue
     fi
diff --git a/scripts/ssh_session.sh b/scripts/ssh_session.sh
new file mode 100755
index 0000000..6f2a8c3
--- /dev/null
+++ b/scripts/ssh_session.sh
@@ -0,0 +1,105 @@
+#!/usr/bin/env bash
+
+# setting the locale, some users have issues with different locales, this forces the correct one
+export LC_ALL=en_US.UTF-8
+
+show_ssh_session_port=$1
+
+parse_ssh_port() {
+  # Get port from connection
+  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 get info from remote
+  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" ]
+}
+
+main() {
+  hostname=$(get_info hostname)
+  user=$(get_info whoami)
+
+  # Only show port info if ssh session connected (no localhost) and option enabled
+  if $(ssh_connected) && [[ $show_ssh_session_port == "true" ]] ; then
+    port=$(get_info port)
+    echo $user@$hostname:$port
+  else
+    echo $user@$hostname
+  fi
+}
+
+main