#!/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##*:}
  echo -e "\033]49;$colour\007"
  clear
fi

exec ${SSHTERM_SSH:-ssh} ${1+"$@"}