improved known_hosts completion

[etc/ssh.git] / .zsh / zshrc / parts.d / 50-ssh

diff --git a/.zsh/zshrc/parts.d/50-ssh b/.zsh/zshrc/parts.d/50-ssh
index 8ece86be2a64cf9fa547a2945cd6af066207db36..04b9358da5980bb477f90fb2684b0ea07c07ac43 100644 (file)
--- a/.zsh/zshrc/parts.d/50-ssh
+++ b/.zsh/zshrc/parts.d/50-ssh
@@ -11,8 +11,15 @@
 # populate hosts completion with SSH's known_hosts
 if [ -r $HOME/.ssh/known_hosts ]; then
   local _myhosts
 # populate hosts completion with SSH's known_hosts
 if [ -r $HOME/.ssh/known_hosts ]; then
   local _myhosts

-  _myhosts=(${${${${(f)"$(cat $HOME/.ssh/known_hosts)"}:#[0-9]*}%%\ *}%%,*})
-  zstyle ':completion:*' hosts $_myhosts
+  _myhosts=$(while read host rest; do
+    host="${(f)host%%,*}"
+    case "$host" in
+      ([[:alnum:]]*) echo "$host";;
+      (\[[[:alnum:]]*) echo "${${host#\[}%\]*}";;
+    esac
+  done < ~/.ssh/known_hosts)
+  #_myhosts=(${${${${(f)"$(<$HOME/.ssh/known_hosts)"}:#[0-9]*}%%\ *}%%,*})
+  #zstyle ':completion:*' hosts $_myhosts

   unset _myhosts
 fi
 
   unset _myhosts
 fi
 


@@ -20,30 +27,49 @@ _ssh_proxy() {
   local cmd sshopts cmdopts
   cmd="$1"; shift
   sshopts="${*%%--*}"; sshopts="${sshopts%% }"
   local cmd sshopts cmdopts
   cmd="$1"; shift
   sshopts="${*%%--*}"; sshopts="${sshopts%% }"

+  [[ -n "$sshopts" ]] || { ssh; return $? }

   cmdopts="${*#$sshopts}"; cmdopts="${cmdopts#*-- }"
   eval ssh $sshopts -t $cmd $cmdopts
 }
 
 # need functions instead of aliases for completion
   cmdopts="${*#$sshopts}"; cmdopts="${cmdopts#*-- }"
   eval ssh $sshopts -t $cmd $cmdopts
 }
 
 # need functions instead of aliases for completion

-rrsh() { _ssh_proxy "su -" "$@" }
+rrsh() { _ssh_proxy "sudo -i" "$@" }

 compdef rrsh=ssh
 
 ttsh() { _ssh_proxy "tmux last" "$@" }
 compdef ttsh=ssh
 
 compdef rrsh=ssh
 
 ttsh() { _ssh_proxy "tmux last" "$@" }
 compdef ttsh=ssh
 

-rtsh() { _ssh_proxy "tmux new su -" "$@" }
+rtsh() { _ssh_proxy "tmux new sudo -i" "$@" }

 compdef rtsh=ssh
 
 compdef rtsh=ssh
 

-local SOCKDIR=$VARDIR/ssh
+declare _SSH_SOCKDIR="${VARDIR}/ssh"

 kssh() {
 kssh() {

-  local SOCKDIR=$SOCKDIR

   if [[ -z "$1" ]]; then
   if [[ -z "$1" ]]; then

-    rm -vf -- $SOCKDIR/ssh_control_*(=as-60Y1omN)
+    rm -vf -- ${_SSH_SOCKDIR}/*(=as-60Y1omN)

   else
     rm "$@"
   fi
 }
   else
     rm "$@"
   fi
 }

-compdef "_files -W $SOCKDIR" kssh
-unset SOCKDIR
+compdef -e "compset -P ${_SSH_SOCKDIR}/ || PREFIX=${_SSH_SOCKDIR}/; _files -W ${_SSH_SOCKDIR}" kssh
+
+scd() {
+  local args host dir arg
+
+  for arg in "$@"; do
+    case "$arg" in
+      (*:*) host=${arg%%:*}; dir="${arg#*:}";;
+      (*) args="${args:+$args }$arg";;
+    esac
+  done
+
+  if [[ -z $dir ]]; then
+    ssh "$@"
+
+  else
+
+    ssh -t $args $host "cd '${dir:gs/\'/\'\\\'\'}' && exec \$SHELL --login"
+  fi
+}
+compdef scd=scp

 
 # vim:ft=zsh
 
 # vim:ft=zsh