X-Git-Url: https://git.madduck.net/etc/ssh.git/blobdiff_plain/3bf6f0a97128a150976cafb771a09caf42df5db4..4d6632c9b8585576014e89b84c04628ef4bffd4b:/.zsh/zshrc/parts.d/50-ssh

diff --git a/.zsh/zshrc/parts.d/50-ssh b/.zsh/zshrc/parts.d/50-ssh
index 98f730d..1d840eb 100644
--- a/.zsh/zshrc/parts.d/50-ssh
+++ b/.zsh/zshrc/parts.d/50-ssh
@@ -16,9 +16,35 @@ if [ -r $HOME/.ssh/known_hosts ]; then
   unset _myhosts
 fi
 
-rrsh() { ssh "${@:-}" -t /bin/su - }
+_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 "sudo -i" "$@" }
 compdef rrsh=ssh
 
-alias kssh='rm -vf -- $VARDIR/ssh/ssh_control_*(=as-60Y1omN)'
+ttsh() { _ssh_proxy "tmux last" "$@" }
+compdef ttsh=ssh
+
+rtsh() { _ssh_proxy "tmux new sudo -i" "$@" }
+compdef rtsh=ssh
+
+local SOCKDIR=$VARDIR/ssh
+kssh() {
+  local SOCKDIR=$SOCKDIR
+  if [[ -z "$1" ]]; then
+    rm -vf -- $SOCKDIR/ssh_control_*(=as-60Y1omN)
+  else
+    rm "$@"
+  fi
+}
+compdef "_files -W $SOCKDIR" kssh
+unset SOCKDIR
 
 # vim:ft=zsh