From a2f3ec58f751a493b7a7dd962b5d82ee379d9548 Mon Sep 17 00:00:00 2001
From: Frank Wittig <frank.wittig@here.com>
Date: Wed, 9 Apr 2014 12:20:12 +0200
Subject: [PATCH] autocomplete hosts from hosts sections in ssh-config, two
 functions to load/unload keys from the agent (e.g. for use in .zshrc)

---
 plugins/ssh/ssh.plugin.zsh | 39 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 plugins/ssh/ssh.plugin.zsh

diff --git a/plugins/ssh/ssh.plugin.zsh b/plugins/ssh/ssh.plugin.zsh
new file mode 100644
index 000000000..a308c09b1
--- /dev/null
+++ b/plugins/ssh/ssh.plugin.zsh
@@ -0,0 +1,39 @@
+############################################################
+# Take all host sections in .ssh/config and offer them for
+# completion as hosts (e.g. for ssh, rsync, scp and the like)
+# Filter out wildcard host sections.
+local hosts
+if [[ -f $HOME/.ssh/config ]]; then
+  hosts=($(cat $HOME/.ssh/config | egrep '^Host.*' | awk '{print $2}' | grep -v '^*' | sed -e 's/\.*\*$//' | grep '\.'))
+  zstyle ':completion:*:hosts' hosts $hosts
+fi
+
+############################################################
+# Remove host key from known hosts based on a host section
+# name from .ssh/config
+function ssh_rmhkey {
+  if [[ "x$1" == "x" ]]; then return; fi
+  ssh-keygen -R $(grep -A10 $1 ~/.ssh/config | grep -i HostName | head -n 1 | awk '{print $2}')
+}
+compctl -k hosts ssh_rmhkey
+
+############################################################
+# Load SSH key into agent
+function ssh_load_key() {
+  local key=$1
+  if [[ "$key" == "" ]]; then return; fi
+  if ( ! ssh-add -l | grep -q $key ); then
+    ssh-add ~/.ssh/$key;
+  fi
+}
+
+############################################################
+# Remove SSH key from agent
+function ssh_unload_key {
+  local key=$1
+  if [[ "$key" == "" ]]; then return; fi
+  if ( ssh-add -l|grep -q $key ); then
+    local keyfile=$(ssh-add -l | grep $key | cut -d ' ' -f 3)
+    ssh-add -d $keyfile
+  fi
+}