Merge pull request #1 from EstebanMonge/main

add support for key files

Author: itachi1621

README.md

@@ -95,6 +95,7 @@ sudo cp ssh-manager.sh /usr/local/bin/ssh-manager
 - Ubuntu Server 22.04 LTS
 - Ubuntu 22.04 LTS
 - Ubuntu Server 20.04 LTS
+- Debian 12
 
 ## License
 This script is licensed under the MIT License.

ssh-manager.sh

@@ -164,6 +164,44 @@ fi
 
  }
 
+addNewServerKeyFile(){
+
+
+read -p "Enter the path to the key file or leave blank if not used: " keyfile
+
+if ! [ -z "$keyfile" ]
+then
+    #check if the key file exists
+    if ! [ -f "$keyfile" ]
+    then
+        printf "%s${warning}File does not exist.${reset}\n"
+        addNewServerKeyFile
+    fi
+fi
+}
+
+ editServerKeyFile(){
+
+    printf "Current key file is %s \n " "$keyfile"
+    read -p "Enter the path to the key file or leave blank to use current key file: " newKeyFile
+
+    if [ -z "$newKeyFile" ]
+    then
+        keyfile=${keyfile//\//\\/}
+    else
+        #check if the key file exists
+        if ! [ -f "$newKeyFile" ]
+        then
+            printf "%s${warning}File does not exist.${reset}\n"
+            editServerKeyFile
+        else
+            keyfile=${newKeyFile//\//\\/}
+        fi
+    fi
+
+ }
+
+
  createNewSSHCredentials(){
 
     printf "%s${info}===========================${reset}\n"
@@ -174,9 +212,10 @@ fi
     addNewServerIp
     addNewServerPort
     addNewServerUser
+    addNewServerKeyFile
 
 
-    echo "$name,$ip,$port,$user" >> "$cfg_file_name"
+    echo "$name,$ip,$port,$user,$keyfile" >> "$cfg_file_name"
 
     echo -e "${success}SSH Connection added successfully${reset}"
     read -p "Do you want to connect to the added SSH connection now? (y/n) " selection
@@ -196,8 +235,8 @@ fi
         echo -e "${info} Saved SSH Connections ${reset}"
         printf "%s${info}===========================${reset}\n"
         #Now to use awk to list the servers in a nice format 1 , 2 , 3 etc
-        printf "%s${info}#  Name IP/Host \tPort Username${reset}\n"
-        awk -F, '{print NR " " $1 " " $2 " " $3 " " $4}' "$cfg_file_name" | column -t # -t is used to align the columns,  using awk is always awkward .... but it works
+        printf "%s${info}#  Name IP/Host \tPort Username\tKey File${reset}\n"
+        awk -F, '{print NR " " $1 " " $2 " " $3 " " $4 " " $5}' "$cfg_file_name" | column -t # -t is used to align the columns,  using awk is always awkward .... but it works
 
         printf "%s${warning}Enter the number of the SSH connection you want to edit or enter 0 to cancel : ${reset}"
         read -p "" serverNumber
@@ -220,15 +259,17 @@ fi
         ip=$(awk -F, -v serverNumber="$serverNumber" 'NR==serverNumber {print $2}' "$cfg_file_name")
         port=$(awk -F, -v serverNumber="$serverNumber" 'NR==serverNumber {print $3}' "$cfg_file_name")
         user=$(awk -F, -v serverNumber="$serverNumber" 'NR==serverNumber {print $4}' "$cfg_file_name")
+        keyfile=$(awk -F, -v serverNumber="$serverNumber" 'NR==serverNumber {print $5}' "$cfg_file_name")
 
         editServer
         editServerIp
         editServerPort
         editServerUser
+        editServerKeyFile
 
         #Now to replace the selected lines info with the updated info to the file
 
-        sed -i "${serverNumber}s/.*/$name,$ip,$port,$user/" "$cfg_file_name"
+        sed -i "${serverNumber}s/.*/$name,$ip,$port,$user,$keyfile/" "$cfg_file_name"
         printf "%s${success}SSH Connection has been edited${reset}\n"
         menu
 
@@ -243,8 +284,8 @@ fi
     echo -e "${info} Saved SSH Connections ${reset}"
     printf "%s${info}===========================${reset}\n"
     #Now to use awk to list the servers in a nice format 1 , 2 , 3 etc in a table format starting with the header but starting the numbering at from the second line
-    printf "%s${info}#  Name IP/Host Port Username${reset}\n"
-    awk -F, '{print NR " " $1 " " $2 " " $3 " " $4}' "$cfg_file_name" | column -t # -t is used to align the columns,  using awk is always awkward .... but it works
+    printf "%s${info}#  Name IP/Host Port Username Key file${reset}\n"
+    awk -F, '{print NR " " $1 " " $2 " " $3 " " $4 " " $5}' "$cfg_file_name" | column -t # -t is used to align the columns,  using awk is always awkward .... but it works
 
     menu
 
@@ -261,15 +302,16 @@ fi
         serverIp=$ip
         serverPort=$port
         serverUser=$user
+        serverKeyFile=$keyfile
     else
 
     fileEmptyCheck
     printf "%s${info}===========================${reset}\n"
      echo -e "${info} Saved SSH Connections ${reset}"
      printf "%s${info}===========================${reset}\n"
     #Now to use awk to list the servers in a nice format 1 , 2 , 3 etc
-    printf "%s${info}#  Name \t IP/Host \tPort Username${reset}\n"
-    awk -F, '{print NR " " $1 " " $2 " " $3 " " $4}' "$cfg_file_name" | column -t # -t is used to align the columns,  using awk is always awkward .... but it works
+    printf "%s${info}#  Name \t IP/Host \tPort Username\tKey file${reset}\n"
+    awk -F, '{print NR " " $1 " " $2 " " $3 " " $4 " " $5}' "$cfg_file_name" | column -t # -t is used to align the columns,  using awk is always awkward .... but it works
 
     printf "%s${info}Enter the number of the SSH connection you want to connect to or enter 0 to cancel : ${reset}"
     read -p "" serverNumber
@@ -301,14 +343,19 @@ fi
     serverIp=$(awk -F, -v serverNumber="$serverNumber" 'NR==serverNumber {print $2}' "$cfg_file_name")
     serverPort=$(awk -F, -v serverNumber="$serverNumber" 'NR==serverNumber {print $3}' "$cfg_file_name")
     serverUser=$(awk -F, -v serverNumber="$serverNumber" 'NR==serverNumber {print $4}' "$cfg_file_name")
+    serverKeyFile=$(awk -F, -v serverNumber="$serverNumber" 'NR==serverNumber {print $5}' "$cfg_file_name")
 
 
 fi
     #echo $serverPort
     #Here we go connecting to the server
     printf "%s${success}Connecting to ${serverName} ...${reset}\n"
-    ssh -p "$serverPort" "$serverUser""@""$serverIp"
-
+    if [ -z $serverKeyFile ]
+    then
+        ssh -p "$serverPort" "$serverUser""@""$serverIp"
+    else
+        ssh -i "$serverKeyFile" -p "$serverPort" "$serverUser""@""$serverIp"
+    fi
     menu