window_preexec() {
# Get the program name with its arguments.
local program_name=$1
+
# When sudo is used use real program name instead, but with an
# exclamation mark at the beginning.
+ local program_sudo=
if [[ $program_name == sudo* ]]; then
- program_name=!${program_name#sudo }
+ program_name=${program_name#sudo }
+ program_sudo=yes
fi
# Remove all arguments from the program name.
program_name=${program_name%% *}
;;
esac
+ # Add an exclamation mark at the beginning if running with sudo.
+ if [[ $program_sudo == yes ]]; then
+ program_name=!$program_name
+ fi
+
+ # Add an at mark at the beginning if running ssh on a different
+ # computer.
+ if [[ -n $SSH_CONNECTION ]]; then
+ program_name="@$program_name"
+ fi
+
# Set the window name to the currently running program.
window_title "$program_name"
[[ -z $window_reset ]] && return
# Reset the window name to 'zsh'.
- window_title "zsh"
+ if [[ -n $SSH_CONNECTION ]]; then
+ window_title "@zsh"
+ else
+ window_title "zsh"
+ fi
# Just reset the name, so no screen reset necessary for the moment.
window_reset=