#!/bin/bash
#
# Simple autossh and screen wrapper.
#

# Parameters
BASENAME="`basename $0`"
DEST="$1"
COMMAND="$2"

# Set window title
# http://stackoverflow.com/questions/899609/gnu-screen-run-script-that-sends-commands-to-the-screen-session-it-is-being-run
function window_title {
  if [ -n "$STY" ]; then
    screen -p $WINDOW -X title "$*"
  else
    if which xtitle &> /dev/null; then
      xtitle "$*"
    fi
  fi
}

# Syntax check
if [ -z "$DEST" ]; then
  exit 1
fi

# Set default screen title
window_title $DEST

# Dispatcher
if [ "$BASENAME" == "shells" ]; then
  # Remote screen shell using autossh
  if [ "$COMMAND" == "root" ]; then
    autossh $DEST -t -- sudo screen -x
  else
    autossh $DEST -t -- screen -x $COMMAND
  fi
else
  if [ -z "$COMMAND" ] && screen -ls $DEST | grep -q "There is a screen on"; then
    # Local existing screen shell
    wscreen $DEST
  elif [ -z "$COMMAND" ] && [ "$DEST" == "root" ]; then
    # Local root shell
    window_title root
    sudo screen -x
  elif [ -z "$COMMAND" ] && [ "$DEST" == "network" ]; then
    # Local root shell
    window_title network
    shell local network
  elif [ "$DEST" == "local" ]; then
    # Local screen shell
    window_title $COMMAND
    wscreen $COMMAND
  else
    # Remote shell using autossh
    autossh $DEST -t -- $COMMAND
  fi
fi

# Restore screen title
window_title terminal