From 689ad5a124b9387cb064f7aa3911afa123ebc7f8 Mon Sep 17 00:00:00 2001
From: Sungho Moon <sunghom@amazon.com>
Date: Sun, 16 May 2021 23:59:24 +0900
Subject: [PATCH] allow SSHing with default identity file when node has public
 IP

---
 script_ssh.sh | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/script_ssh.sh b/script_ssh.sh
index 25d1e52..fa0cf51 100755
--- a/script_ssh.sh
+++ b/script_ssh.sh
@@ -98,6 +98,20 @@ else [ "$KUBECTL_PLUGINS_LOCAL_FLAG_CLOUD_PROVIDER" == "aws" ]
       esac
     done
   else
-    [[ ! -z "${IP}" ]] && echo "SSHing into Worker Node with IP: ${IP}" && ssh -i ${KUBECTL_PLUGINS_LOCAL_FLAG_IDENTITY_FILE} ${KUBECTL_PLUGINS_LOCAL_FLAG_SSH_USER}@$IP || echo "Can't SSH"; exit 1;
+    if [ ! -z "${IP}" ]
+    then
+      if [ "${KUBECTL_PLUGINS_LOCAL_FLAG_IDENTITY_FILE}" == "" ]
+      then
+        echo "SSHing into Worker Node using default identity file with IP: ${IP}"
+        ssh ${KUBECTL_PLUGINS_LOCAL_FLAG_SSH_USER}@$IP || echo "Can't SSH"; exit 1;
+      elif [ ! -z "${IP}" ]
+      then
+        echo "SSHing into Worker Node with IP: ${IP}"
+        ssh -i ${KUBECTL_PLUGINS_LOCAL_FLAG_IDENTITY_FILE} ${KUBECTL_PLUGINS_LOCAL_FLAG_SSH_USER}@$IP || echo "Can't SSH"; exit 1;
+      fi
+    else
+      echo "Cannot find IP address of node."
+      exit 1;
+    fi
   fi
 fi