From 53a9ea38e425f24af0d757950c39538d3c54cbbd Mon Sep 17 00:00:00 2001
From: Anatoly Kopyl <akopyl@radner.ru>
Date: Sat, 3 Apr 2021 15:37:23 +0300
Subject: [PATCH] Turned everything ito a function

---
 sshukh.plugin.zsh | 23 ++++++++++++-----------
 1 file changed, 12 insertions(+), 11 deletions(-)

diff --git a/sshukh.plugin.zsh b/sshukh.plugin.zsh
index 1455f81..a705415 100755
--- a/sshukh.plugin.zsh
+++ b/sshukh.plugin.zsh
@@ -1,14 +1,15 @@
-#!/bin/sh
-output=$(ssh "$@")
-if [ $? -eq 255 ];
-then
+sshukh () {
+  output=$(ssh "$@")
+  if [ $? -eq 255 ];
+  then
     host=$(cut -d'@' -f2 <<< $1)
     while true; do
-        read -p "Update known_hosts? [y/n] " yn
-        case $yn in
-            [Yy]* ) ssh-keygen -R $host && ssh "$@"; break;;
-            [Nn]* ) exit;;
-            * ) echo "Please answer y or n.";;
-        esac
+      read -p "Update known_hosts? [y/n] " yn
+      case $yn in
+        [Yy]* ) ssh-keygen -R $host && ssh "$@"; break;;
+        [Nn]* ) exit;;
+        * ) echo "Please answer y or n.";;
+      esac
     done
-fi
+  fi
+}