Set tmux pane title to short hostname on ssh connections

set-tmux-title

ssh() {
    if [ "$(ps -p $(ps -p $$ -o ppid=) -o comm=)" = "tmux" ]; then
        tmux rename-window "$(echo $* | cut -d . -f 1)"
        command ssh "$@"
        tmux set-window-option automatic-rename "on" 1>/dev/null
    else
        command ssh "$@"
    fi
}

Working variant in Termux on Samsung DeX:

# Requires procps package installed.
ssh() {
  if [ "$(ps -p $(ps -p $$ -o ppid=) -o comm= | cut -d : -f1)" = "tmux" ]; then
    tmux rename-window "$(echo -- $* | awk '{print $NF}')"
    command ssh "$@"
    tmux set-window-option automatic-rename "on" 1>/dev/null
  else
    command ssh "$@"
  fi
}

It will go wrong if I run with '-i' option like ssh -i my.pem test@x.x.x.x .

As @int32bit mentioned, options aren't handled in the previous snippets. The following variant handles those cases:

https://gist.github.com/nevesnunes/951f77085116a9beea62c9610dda31e8

Could you please share where would I place this code ? I pasted this in my ~/.zshrc file and ssh to a server but I didn't see remote host name anywhere.

Note: I do have ~/.tmux.config file which have few keybindings.