#!/bin/bash

usage()
{
cat << EOF

usage: $0 [options] command [arguments...]

This script runs the leap platform deploy tests

OPTIONS

  -c|--config file        specify config file   
  -a|--all                run command on all nodes
  -V|--versions           show versions/git revision of leap_cli and leap_platoform in provider dir
  -h|--help               show help

COMMANDS
  bootstrap    <node(s)>  bootstrap node(s):
                            - leap local start
                            - leap node int
                            - sets up hostname and runs apt-get dist-upgrade
                            - leap local save
  create_provider         creates a provider instance
  deploy       <node(s)>  deploy node(s)
  reset_deploy <node(s)>  reset and deploy node(s)
  
EOF
}

add_nodes() {
  for i in $@
  do
    node=${i%:*}
    services=${i#*:}

    case $services in
      openvpn)
        config='openvpn.gateway_address:10.5.5.98 openvpn.second_gateway_address:10.5.5.99'
        ;;
      *)
        config=''
        ;;
    esac

    $LEAP_CMD node add --local $node $config services:$services
  done
}


bootstrap_nodes() {
  for vm in $@
  do
    $LEAP_CMD $OPTS local start $vm
    wait_for_node $vm
    $LEAP_CMD $OPTS node init $vm
    # set hostname + do dist-upgrade
    $LEAP_CMD $OPTS deploy $vm --tags site_apt::dist_upgrade,site_config::hosts
    $LEAP_CMD $OPTS local save $vm
  done
}

create_provider() {
  if [ -e $PROVIDERDIR ]
  then
    echo $PROVIDERDIR exists - exiting
    exit 1
  fi

  mkdir -p  $PROVIDERDIR
  cd  $PROVIDERDIR

  $LEAP_CMD $OPTS new --contacts $CONTACTS --domain $DOMAIN --name $PROVIDER --platform=$PLATFORMDIR .
  cd $PLATFORMDIR 
  git checkout $PLATFORM_BRANCH 
  git submodule sync 
  git submodule update --init
  cd $PROVIDERDIR

  # for now, we use the vagrant pubkey until https://leap.se/code/issues/2039 is solved
  $LEAP_CMD $OPTS add-user --self --ssh-pub-key=$SSHKEY
  $LEAP_CMD $OPTS cert ca && leap cert csr

  $LEAP_CMD $OPTS cert dh
  add_nodes $NODES
}

deploy() {

  # remove colors until #1751 is fixed
  RMCOLORS='s/\x1B\[([0-9]{1,2}(;[0-9]{1,2})?)?[m|K]//g'

  # we need to deploy with verbose level 2, and filter out unwanted stuff
  # until puppet errors show up in verbose level 0 +1 (#1750)
  FILTER_CLI='= read|= loading|= no change|   - executing| = executing| = applying| = ran git| = checking| = synching| = skipping file_path|   - cd .*; rsync -|   - hiera| = created hiera/|   - cd /root/|   - rolling backexecuting|   - files/ca/client_ca.crt|\[bin,puppet\] ->|] Hostname updated.| = Updating submodule puppet/modules|Warning: Permanently added.*to the list of known hosts.| = leap command v| = leap platform v|   - \[.*\] ok'

  FILTER_PUPPET='] notice: |] No change to hostname|] Puppet apply complete|] warning: Dynamic lookup|] warning: Scope\(Class'
  FILTER_ALL="($FILTER_CLI|$FILTER_PUPPET)"


  for vm in $@
  do
    [ -e $LOGDIR ] || mkdir -p $LOGDIR
    date=`date +"%F-%H%M%S"`
    LOG1="$LOGDIR/deploy-$vm.log"
    LOG2="$LOGDIR/deploy-$vm-$date.log"
    ERRLOG1="$LOGDIR/deploy-$vm-$date-error.log"
    ERRLOG2="$LOGDIR/deploy-error.log"
    echo "Deploying \"$vm\" on `date`"|tee -a $LOG1 $LOG2

    $LEAP_CMD $OPTS -v 2 deploy $vm 2>&1 |  sed -r "$RMCOLORS" | tee -a $LOG1 $LOG2 | egrep -v "$FILTER_ALL" | tee -a $ERRLOG1 $ERRLOG2 > /dev/null

    if [ -s $ERRLOG1 ]
    then
      versions=`versions`
      msg="From: $MAIL_FROM\nTo: $MAIL_TO\nSubject: Deploy to \"$vm\" had errors !\n\n"
      msg="$msg \nOutput of error log below:\n `cat $ERRLOG1` \n\n"
      msg="$msg\n Tested with following versions/git commit IDs: \n$versions"
      echo "Deploy to \"$vm\" on `date` had errors." | tee -a $LOG1 $LOG2 $ERRLOG2
      echo
      cat $ERRLOG1 
      
      if [ "$MAIL_TO" != '' ]; then
        echo "Sending this mail to $MAIL_TO:"
        printf "$msg" | /usr/sbin/sendmail -t
      fi

    else
      echo "Deploy to $vm on `date` went fine."| tee -a $LOG1 $LOG2
      rm $ERRLOG1
    fi
  done
} 

get_ip () {
  grep ip_address $PROVIDERDIR/nodes/$1.json |cut -f 2 -d:|sed 's/[",]//g'
}

reset_deploy() {

  update_platform
  update_leap_cli

  cd $PROVIDERDIR

  log_start

  echo "Starting deploy_nodes for nodes $@ as background tasks on `date`"

  for i in $@
  do
    $LEAP_CMD $OPTS local reset $i
    wait_for_node $i
    deploy $i &
  done

  # needed in a detached screen session, otherwise it would terminate before deploy jobs 
  # have finished
  echo "Waiting for last deploy process has finished..."
  wait
}

log_start() {
  echo
  echo "Starting $0 on `date`"
}

ip_pingable () {
  ping -q -W10 -c1 $1 >/dev/null 2>&1
  return $?
}

wait_for_node() {
  vm=$1
  ip=`get_ip $vm`
  online=0
  echo "Waiting for VM $vm (IP: $ip) to come up..."
  while [ $online -eq 0 ]
  do
      ip_pingable $ip && online=1
      sleep 1
  done
}

update_leap_cli () {
  cd $LEAP_SRC
  git pull
}


update_platform () {
  cd $PLATFORMDIR
  git pull
}

versions () {
  cd $PLATFORMDIR
  platform_head=`git rev-parse HEAD`
  platform_commit=`git show | head -5`

  cd $LEAP_SRC
  cli_head=`git rev-parse HEAD`
  cli_commit=`git show $cli_head | head -5`

  cd $PROVIDERDIR
  [ -d .git ] && provider_head=`git rev-parse HEAD`
  if [ -n $provider_head ]
  then
    provider_head='not under version control'
  fi  

  echo "Provider:"
  echo 
  echo "provider ($PROVIDERDIR): $provider_head"
  echo

  echo "leap_cli:"
  echo
  echo leap cli: $cli_head
  $LEAP_CMD -v 2 list | grep ' = leap command v'
  echo "$cli_commit"
  echo
  
  echo "leap_platform:"
  $LEAP_CMD -v 2 list | grep ' = leap platform v'
  echo "$platform_commit"
  echo
  echo
}





config=""
all=false
print_versions=false

if ! options=$(getopt -o avVc:h -l all,verbose,versions,config:,help -- "$@")
then 
  # something went wrong, getopt will put out an error message for us
  usage
  exit 1
fi  

eval set -- "$options"

while [ $# -gt 0 ]
do
  case $1 in
    -h|--help)     usage; exit 1;;
    -c|--config)   config=$2; shift ;;
    -a|--all)      all=true;;
    -v|--verbose)  VERBOSE=1 ;;
    -V|--versions)  print_versions=true;; 
    (--) shift; break;;
    (-*) echo "$0: error - unrecognized option $1" 1>&2; exit 1;;
    (*) break;;
  esac
  shift
done


cmd=$1
shift
nodelist=$@

#echo "config:   $config"
#echo "cmd:      $cmd"
#echo "nodelist: $nodelist"
#echo "all:      $all"

if [ -z $config ]
then
  usage
  echo "Please provide a config file"
  exit 1
fi

# provider specific config
source $config
# common config for all providers
source /etc/leap/platform-test-common.cfg


[ -e $PROVIDERDIR ] && cd  $PROVIDERDIR

if $print_versions
then
  versions=`versions`
  echo "$versions"
  echo
  exit 0
fi

if [ -z $LEAP_CMD -o -z $LEAP_SRC ] 
then
  echo "please provide a path to the leap_cli binary and the source in the config file, using the LEAP_CMD and LEAP_SRC var."
  exit 1
fi

if $all ; then
  # use NODES variable from the config file
  nodes=$NODES

  # strip services from nodes_services
  # i.e. nodes_services='redevcouchdb1:couchdb redevcouchdb2:couchdb'
  # -> nodes='redevcouchdb1 redevcouchdb2'
  nodes=`echo "$NODES" | sed 's/:[[:alnum:]]*//g'`
else
  # use nodelist provided via cmdline parameters
  nodes=$nodelist
fi

# check if another process is already running for given nodes
for i in $nodes; do
  procs=`pgrep -fc "leap-platform-test.*$i"`
  if [ $procs -gt 1 ]; then
    echo "Another process is already running for node $i - exiting." 
    exit 1
  fi
done


case $cmd in 
  add_nodes)        add_nodes "$nodes";;
  bootstrap)        bootstrap_nodes "$nodes";;
  create_provider)  create_provider;;
  deploy)           deploy "$nodes";;
  reset_deploy)     reset_deploy "$nodes";;
  (*)               usage; echo "Please specify a command."; exit 1;;
esac