#!/bin/bash

SSHCOLOURS=~/.ssh/colours

fqdn=${1##*@}
colour=

# Get colour by regex.
if [ -z "$colour" ]; then
  for path in $(ls "$SSHCOLOURS" | lensort -r); do
    re=${path##*/}
    echo "$fqdn" | grep -qs "$re" || continue
    colour=$(readlink "$SSHCOLOURS/$path")
    break
  done
fi

# Get colour by netgroup.
if [ -z "$colour" ]; then
  for netgroup in $(ls "$SSHCOLOURS" | grep ^\@); do
    if netgroups $netgroup ${fqdn%%.*} 2>/dev/null; then
      colour=$(readlink "$SSHCOLOURS/$netgroup")
      break
    fi
  done
fi

if [ ! -z "$colour" ]; then
  colour=${colour##*/}
  colour=${colour##*:}
fi

[ -n "$colour" ] && colour="-bg $colour"
exec ${SSHTERM:-xterm} ${SSHTERM_TITLE:--title} "$1" \
     $colour ${SSHTERM_EXEC:--e} ${SSHTERM_SSH:-ssh} ${1+"$@"}