fix up completion for ssh functions

[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 772a4262d315ede1b4f831cb12f509c43486aa65..53d189c8bcf8f2c4776547bc1fa65ec846fed85b 100644 (file)
--- a/.zsh/zshrc/parts.d/50-ssh
+++ b/.zsh/zshrc/parts.d/50-ssh
@@ -16,9 +16,24 @@ if [ -r $HOME/.ssh/known_hosts ]; then
   unset _myhosts
 fi
 
-alias rrsh='() { ssh "${@:-}" -t /bin/su - }'
+_ssh_proxy() {
+  local cmd sshopts cmdopts
+  cmd="$1"; shift
+  sshopts="${*%%--*}"; sshopts="${sshopts%% }"
+  cmdopts="${*#$sshopts}"; cmdopts="${cmdopts#*-- }"
+  eval ssh $sshopts -t $cmd $cmdopts
+}
+
+# need functions instead of aliases for completion
+rrsh() { _ssh_proxy "su -" "$@" }
 compdef rrsh=ssh
 
+ttsh() { _ssh_proxy "tmux last" "$@" }
+compdef ttsh=ssh
+
+rtsh() { _ssh_proxy "tmux new su -" "$@" }
+compdef rtsh=ssh
+
 alias kssh='rm -vf -- $VARDIR/ssh/ssh_control_*(=as-60Y1omN)'
 
 # vim:ft=zsh