aboutsummaryrefslogtreecommitdiff
path: root/ssh-agent-loadkey
blob: d3a6d37c205e2a7215c3d8c45aadc98b2486acfa (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
#!/bin/bash
#
# Load a key from a menu.
#

# Parameters
BASENAME="`basename $0`"
KEYS="$HOME/.ssh/keys"

# Check
if [ ! -d "$KEYS" ]; then
  echo "$BASENAME: folder not found: $KEYS"
  exit 1
fi

# Get available keys
function __query {
  (
  cd $KEYS && find -name '*.pub' | sed -e 's/.pub$//' | grep -v decomissioned | while read line; do
    handle="`echo $line | cut -d '/' -f 3`"
    type="`echo $line | cut -d '/' -f 2`"
    echo "$handle ($type)"
  done
  )
}

# 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
function __loaded {
  #ssh-add -L | cut -d ' ' -f 3 | sed -e 's/^/\t/'

  ssh-add -L | while read line; do
    type="`echo $line | cut -d ' ' -f 1 | sed -e 's/^ssh-//'`"
    handle="`echo $line | cut -d ' ' -f 3-`"

    if [ -e "$handle" ]; then
      handle="`basename $handle`"
    fi

    echo "$handle ($type)"
  done | column -t -c 6
}

# Key chooser mennu
function __chooser {
  echo "Usage: $BASENAME <keytype> <handle>"
  echo ""
  echo "Available keys"
  echo ""
  __list | sed -e 's/^/\t/'
  echo ""

  if [ "`LC_ALL=C ssh-add -L`" != "The agent has no identities." ]; then
    echo "Current loaded keys:"
    echo ""
    __loaded | sed -e 's/^/\t/'
    echo ""
  fi

  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 {
  HANDLE="$1"
  TYPE="`echo $2 | sed -e 's/(//' -e 's/)//'`"
  KEY="$KEYS/$TYPE/$HANDLE"

  # Check if the selected option has a custom procedure (monkeysphere, keyringer, etc)
  if [ -x "$KEY.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="$KEY.askpass" ssh-add $KEY < /dev/null
  else
    ssh-add $KEY
  fi

  # UX
  if [ "$?" == "0" ]; then
    if which awesome-client &> /dev/null; then
      echo "naughty.notify({title = \"SSH:\", text =\"Loaded $HANDLE ($TYPE)\", timeout = 2})" | awesome-client
    fi
  fi
}

# Dispatch
if [ ! -z "$2" ]; then
  __load $*
else
  __chooser
fi