#!/bin/bash
# Run package_check tests against the app in the working directory on a Vagrant virtual machine.

# Fail on first error
set -e

# Configuration constants
APP_DIR="/vagrant"

function _usage() {
  echo "Run package_check tests against the app in the working directory on a Vagrant virtual machine."
  echo "Usage: test.sh [--verbose] [--help]"
}

# Configuration arguments
function _parse_args() {
  VERBOSE=false
  VERBOSE_OPT=''
  while [ "$1" != "" ]; do
    case $1 in
      "-v" | "--verbose")
        shift
        VERBOSE=true
        VERBOSE_OPT='--verbose';;
      "--help")
        _usage
        exit;;
      *)
        _usage
        exit 1;;
    esac
  done
}

# Execute an ssh command on the vagrant box
function _vagrant_ssh() {
  local command="$1"
  local tty_output=$([ $VERBOSE ] && echo '/dev/stdout' || echo '/dev/null')

  [ $VERBOSE == true ] && echo "vagrant ssh -c \"$command\""

  vagrant ssh -c "$command" \
    > $tty_output \
    2> >(grep --invert-match 'Connection to 127.0.0.1 closed.' 1>&2) # Filter out the SSH deconnection message printed on stderr
  local exit_code=$?
  return $exit_code
}

function setup() {
  echo "--- Setting up Vagrant VM ---"
  vagrant up --provision
}

function test_package_check() {
  echo "--- Running package_check ---"
  _vagrant_ssh "package_check/package_check.sh --bash-mode '$APP_DIR'"
}

function abort() {
  trap - SIGINT SIGTERM
  echo "--- Aborting ---"
  if (vagrant status | grep -q "running"); then
    echo "Removing package_check lock…"
    _vagrant_ssh "rm -f package_check/pcheck.lock"
  fi
  teardown
}
trap abort SIGINT SIGTERM

function teardown() {
  echo "--- Cleaning up ---"
}

_parse_args $*
setup
test_package_check
teardown