colorize ssh

iTerm2-ssh.zsh

# tabc <profile name> do the profile change
function tabc() {
  NAME=$1; if [ -z "$NAME" ]; then NAME="Default"; fi
  # if you have trouble with this, change
  # "Default" to the name of your default theme
  echo -e "\033]50;SetProfile=$NAME\a"
}

# reset the terminal profile to Default  when exit from the ssh session
function tab-reset() {
    NAME="Default"
    echo -e "\033]50;SetProfile=$NAME\a"
}

# selecting different terminal profile according to ssh'ing host
# tabc <profile name> do the profile change
#   1. Production profile to production server (ssh eranga@production_box)
#   2. Staging profile to staging server(ssh eranga@staging_box)
#   3. Other profile to any other server(test server, amazon box etc)
function colorssh() {
    if [[ -n "$ITERM_SESSION_ID" ]]; then
        trap "tab-reset" INT EXIT
        if [[ "$*" =~ ".*prod.*" ]]; then
            tabc Production
        elif [[ "$*" =~ ".*staging.*" ]] || [[ "$*" =~ ".*stage.*" ]]; then
            tabc Staging
        else
            tabc Other
        fi
    fi
    command ssh $*
}


# creates an alias to ssh
# when execute ssh from the terminal it calls to colorssh function
alias ssh="colorssh"
compdef colorssh='ssh'