~/bin/ssh wrapper to change Mac OS Terminal profiles when running SSH

ssh

#!/bin/sh

HOSTNAME=`echo $@ | sed s/.*@//`

set_bg () {
  osascript -e "tell application \"Terminal\" to set current settings of window 1 to (first settings set whose name is \"$1\")"
}

on_exit () {
  set_bg "BenDog"
}
trap on_exit EXIT

case $HOSTNAME in
  bendog.org|celeste.bendog.org) set_bg "bendog.org" ;;
  *) set_bg "ssh" ;;
esac

/usr/bin/ssh "$@"