#!/bin/bash
#
# Simple $AUTOSSH and screen wrapper.
#

# Parameters
#
# We're enforcing a random monitoring por for autossh
# as some systems might deny reading /proc/net/tcp like
# kernels with grsecurity patch.
BASENAME="`basename $0`"
DEST="$1"
COMMAND="$2"
MULTIPLEXER="wscreen"
MONITORING="-M $(($RANDOM + 1024))"
AUTOSSH="autossh $MONITORING"

# 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
    $MULTIPLEXER $DEST
  elif [ -z "$COMMAND" ] && [ "$DEST" == "root" ]; then
    # Local root shell
    window_title root
    sudo screen -x
  elif [ -z "$COMMAND" ] && [ -f "$HOME/.screen/$DEST" ]; then
    # Local root shell
    window_title $DEST
    shell local $DEST
  elif [ "$DEST" == "local" ]; then
    # Local screen shell
    window_title $COMMAND
    $MULTIPLEXER $COMMAND
  else
    # Remote shell using $AUTOSSH
    $AUTOSSH $DEST -t -- $COMMAND
  fi
fi

# Restore screen title
window_title terminal