Terminal color adjustment for ssh into different environments

gistfile1.txt

ssh() {
  # Change profile for a specific host
  if [[ "$1" == "prod-server" ]]; then
    # Set profile to Red (or any custom profile name you created)
    echo -e "\033]50;SetTerminalHeader=PROD SERVER\a"
    osascript -e 'tell application "Terminal" to set current settings of selected tab of front window to settings set "Pro"'
    
    # Run SSH
    command ssh "$@"
    
    # Reset back to default profile on exit
    osascript -e 'tell application "Terminal" to set current settings of selected tab of front window to settings set "Basic"'
  else
    command ssh "$@"
  fi
}