Initial working version of ssh-agent-loadkey-menu

1 files changed, 55 insertions, 8 deletions

diff --git a/ssh-agent-loadkey-menu b/ssh-agent-loadkey-menu
index c804d1d..6029729 100755
--- a/ssh-agent-loadkey-menu
+++ b/ssh-agent-loadkey-menu
@@ -13,13 +13,60 @@ if [ ! -d "$KEYS" ]; then
   exit 1
 fi
 
-# Display menu with available keys
-# Check the selected option
-# Check if the selected option has a custom procedure (monkeysphere, keyringer, etc)
-# Grab passphrase and load the key
-#( cd $KEYS && find -name '*.pub' | grep -v decomissioned )
+# Get available keys
+function __query {
+  ( cd $KEYS && find -name '*.pub' | sed -e 's/.pub$//' | grep -v decomissioned )
+}
+
+# List available keys
+function __list {
+  n="0"
+  __query | sort | uniq | while read key; do
+    echo -en "$n. $key"
+    echo ""
+    let ++n
+  done | column -t -c 6
+}
 
 # Display the keys available in the agent
-#echo "Current loaded keys:"
-#echo ""
-#ssh-add -L | cut -d ' ' -f 3 | sed -e 's/^/\t/'
+function __loaded {
+  echo "Current loaded keys:"
+  echo ""
+
+  ssh-add -L | cut -d ' ' -f 3 | sed -e 's/^/\t/'
+}
+
+# Key chooser mennu
+function __chooser {
+  echo "Available keys"
+  echo ""
+  __list | sed -e 's/^/\t/'
+  echo ""
+  __loaded
+  echo ""
+
+  read -rep "Choose key: " n
+
+  # Check the selected option
+  if [ ! -z "$n" ]; then
+    key="$(__list | grep -E "(^$n.| $n:)" | sed -e "s/^[0-9]*. //" | cut -d : -f 1)"
+
+    if [ ! -z "$key" ]; then
+      __load $key
+    fi
+  fi
+}
+
+# Load a key
+function __load {
+  # Check if the selected option has a custom procedure (monkeysphere, keyringer, etc)
+  if [ -x "$KEYS/$1.askpass" ]; then
+    # SSH-ADD(1) says: "Note that on some machines it may be necessary to redirect the input from /dev/null to make this work".
+    SSH_ASKPASS="$KEYS/$1.askpass" ssh-add $HOME/.ssh/keys/$1 < /dev/null
+  else
+    ssh-add $HOME/.ssh/keys/$1
+  fi
+}
+
+# Dispatch
+__chooser