X-Git-Url: https://git.madduck.net/etc/ssh.git/blobdiff_plain/6b87f90ed79ac94c90e92e6cb01b0fbd0bf9e577..20e73c623c763198a060514e91d37a64571157e9:/.zsh/zshrc/parts.d/50-ssh

diff --git a/.zsh/zshrc/parts.d/50-ssh b/.zsh/zshrc/parts.d/50-ssh
index 8ece86b..ab151c0 100644
--- a/.zsh/zshrc/parts.d/50-ssh
+++ b/.zsh/zshrc/parts.d/50-ssh
@@ -20,30 +20,29 @@ _ssh_proxy() {
   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
-rrsh() { _ssh_proxy "su -" "$@" }
+rrsh() { _ssh_proxy "sudo -i" "$@" }
 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
 
-local SOCKDIR=$VARDIR/ssh
+declare _SSH_SOCKDIR="${VARDIR}/ssh"
 kssh() {
-  local SOCKDIR=$SOCKDIR
   if [[ -z "$1" ]]; then
-    rm -vf -- $SOCKDIR/ssh_control_*(=as-60Y1omN)
+    rm -vf -- ${_SSH_SOCKDIR}/*(=as-60Y1omN)
   else
     rm "$@"
   fi
 }
-compdef "_files -W $SOCKDIR" kssh
-unset SOCKDIR
+compdef -e "compset -P ${_SSH_SOCKDIR}/ || PREFIX=${_SSH_SOCKDIR}/; _files -W ${_SSH_SOCKDIR}" kssh
 
 # vim:ft=zsh