zsh/rc: Display @:hostname when running screen on remote host.

diff --git a/zsh/rc b/zsh/rc
index 3d6b28d5fc94186c5d77ab3a2552252773bf28f3..74011ab9c316f8f9a4e42f0e9c627c0341e9d9e2 100644 (file)
--- a/zsh/rc
+++ b/zsh/rc
@@ -227,8 +227,10 @@ unset red blue green yellow default
 #
 # If a command is run with sudo or if the shell is running as root then a ! is
 # added at the beginning of the command to make this clear. If a command is
-# running on a different computer with ssh a @ is added at the beginning. This
-# only works if the .zshrc on the server also uses this command.
+# running on a different computer with ssh a @ is added at the beginning. If
+# screen is running on the remote machine instead of @screen @:hostname
+# (hostname replaced by the machine's hostname) is displayed. This only works
+# if the .zshrc on the server also uses this command.
 if [[ $TERM == screen* || $TERM == xterm* || $TERM == rxvt* ]]; then
     # Is set to a non empty value to reset the window name in the next
     # precmd() call.
@@ -283,6 +285,12 @@ if [[ $TERM == screen* || $TERM == xterm* || $TERM == rxvt* ]]; then
         # different computer.
         if [[ -n $SSH_CONNECTION ]]; then
             program_name="@$program_name"
+
+            # If screen is running in SSH then display "@:hostname" as title
+            # in the term/outer screen.
+            if [[ $program_name == @screen ]]; then
+                program_name="@:${$(hostname)//.*/}"
+            fi
         fi
 
         # Set the window name to the currently running program.