#!/bin/bash

#====================================== GLOBAL VARIABLES ======================================#

INSTALL_TARGET_DIR="/opt"
WYN_INSTALL_DIR="$INSTALL_TARGET_DIR/Wyn"

CLUSTER_MODE=false

MODULE_SERVER="Server"
MODULE_REPORTING_WORKER="ReportingWorker"
MODULE_COT_WORKER="CotWorker"
MODULE_DASHBOARD_WORKER="DashboardWorker"

MODULES=""
WYN_VERSION=""
SILENT_INSTALLATION=false
LOG_FILE="install.log"
LICENSE=""
CONFIGURE_SCRIPT_FILE="/opt/Wyn/Monitor/configure.sh"

EXIT_CODE_SUCCESS=0
EXIT_CODE_SHOW_HELP=1
EXIT_CODE_INVALID_ARGUMENTS=2
EXIT_CODE_INVALID_DATABASE_PROVIDER=3
EXIT_CODE_INVALID_DATABASE_CONNECTION_STRING=4
EXIT_CODE_INVALID_MYSQL_REQUIREMENT=5
EXIT_CODE_INVALID_MODULES_DETECTED=6
EXIT_CODE_NO_MODULES_SELECTED=7
EXIT_CODE_NO_BACKEND_SERVER_ADDRESS=8
EXIT_CODE_START_WYN_FAILED=9
EXIT_CODE_REGISTER_LICENSE_FAILED=10
EXIT_CODE_INVALID_PASSWORD=11
EXIT_CODE_UNSUPPORTED_POSTGRES_VERSION=12
EXIT_CODE_NO_CONFIGURE_SCRIPT=30
EXIT_CODE_CTRL_C=100

#====================================== FUNCTIONS ======================================#

prtMsg() {
  for msg in "$@"
  do
    if $SILENT_INSTALLATION ; then
      echo "$(date) $msg" >> $LOG_FILE
    else
      echo "$msg"
    fi
  done
}

# catch "ctrl-c" and exit the installation process
trap_ctrlc() {
  prtMsg ""
  prtMsg "Ctrl-C caught, canceling the installation..."
  stty echo
  exit $EXIT_CODE_CTRL_C
}

parseArguments() {
  OPTIND=1
  local ret=0
  while getopts sd:c:om:if:b:k:a:v:t:elh opt; do
    case $opt in
      s)
        SILENT_INSTALLATION=true
        ;;
      d)
        DB_PROVIDER="$OPTARG"
        ;;
      c)
        DATABASE_CONNECTIONSTRING="$OPTARG"
        ;;
      o)
        SINGLE_DATABASE_MODE=true
        ;;
      m)
        MODULES="$OPTARG"
        ;;
      i)
        IMPORT_SAMPLES=true
        ;;
      f)
        SAMPLE_FILE="$OPTARG"
        ;;
      b)
        BACKEND_SERVER_ADDRESS="$OPTARG"
        ;;
      k)
        LICENSE="$OPTARG"
        ;;
      a)
        ADMIN_PASSWORD="$OPTARG"
        ;;
      v)
        WYN_VERSION="$OPTARG"
        ;;
      t)
        SECONDARY_NODE_SECRET="$OPTARG"
        ;;
      e)
        ENCRYPT_DATABASE_CONNECTION_STRING=true
        ;;
      l)
        CLUSTER_MODE=true
        CONFIG_FILE_PATH="$WYN_INSTALL_DIR/$MODULE_MONITOR/conf/$CONFIG_FILE_NAME"
        ;;
      h)
        ret=$EXIT_CODE_SHOW_HELP
        ;;
      *)
        ret=$EXIT_CODE_INVALID_ARGUMENTS
        ;;
    esac
  done
  return $ret
}

showUsage() {
  prtMsg "Usage: ${0##*/} [-s] [-d {DATABASE_PROVIDER} -c {DATABASE_CONNECTION_STRING}] [-o] [-m {MODULES}] [-i [-f {SAMPLE_FILE}]] [-b {BACKEND_SERVER_ADDRESS}] [-k {LICENSE_KEY}] [-a {ADMIN_PASSWORD}] [-v {VERSION}] [-t {SECONDARY_NODE_SECRET}] [-e {ENCRYPT_DATABASE_CONNECTION_STRING}] [-l {NON_SINGLE_PROCESS_MODE}] [-h]
  -s    Silent installation.
  -d    Database provider, available values: 'Postgres', 'SqlServer', 'MySql' and 'Oracle'.
  -c    Database server connection string.
  -o    Single database mode.
  -m    Modules to be installed, including '$MODULE_SERVER', '$MODULE_REPORTING_WORKER', '$MODULE_COT_WORKER' and '$MODULE_DASHBOARD_WORKER'.
  -i    Import samples.
  -f    The location of the user specified sample data.
  -b    The back-end server address.
  -k    The license key to register to Wyn system.
  -a    The password of the administrator user 'admin'.
  -v    Wyn version.
  -t    The secret for secondary node.
  -e    Encrypt the database connection string.
  -l    Non single-process mode.
  -h    Help information."
}

# input: $1 - Wyn version
installWyn() {
  local wynVersion=$1
  # add repository information
  if [ ! -f "/etc/apt/sources.list.d/wyn-8.0.list" ]; then
    sudo bash -c 'echo "deb https://cdn.wynenterprise.io/BI/installation/apt/8.0/ /" > /etc/apt/sources.list.d/wyn-8.0.list'
  fi
  if $SILENT_INSTALLATION ; then
    wget -q -O - https://cdn.wynenterprise.io/BI/installation/apt/wyn.enterprise.pubkey | sudo apt-key add - >> $LOG_FILE 2>&1
    sudo apt-get update >> $LOG_FILE 2>&1
    if [ "" == "$wynVersion" ]; then
      sudo apt-get install -y wyn-enterprise >> $LOG_FILE 2>&1
    else
      sudo apt-get install -y wyn-enterprise=$wynVersion >> $LOG_FILE 2>&1
    fi
  else
    wget -q -O - https://cdn.wynenterprise.io/BI/installation/apt/wyn.enterprise.pubkey | sudo apt-key add -
    sudo apt-get update
    if [ "" == "$wynVersion" ]; then
      sudo apt-get install wyn-enterprise
    else
      sudo apt-get install wyn-enterprise=$wynVersion
    fi
  fi
}

#====================================== INSTALLATION ======================================#

trap "trap_ctrlc" INT

# parse and check the parameters
parseArguments "$@"
exit_code=$?
if [ "0" != "$exit_code" ]; then
  showUsage
  exit $exit_code
fi

if $SILENT_INSTALLATION ; then
  prtMsg "-------------------- INSTALLATION START --------------------"
fi

# install Wyn
installWyn $WYN_VERSION

# configure service
ret=$EXIT_CODE_SUCCESS

sudo systemctl is-active -q wyn > /dev/null 2>/dev/null
if [ "0" != "$?" ]; then
  if [ -f "$CONFIGURE_SCRIPT_FILE" ]; then
    sudo bash "$CONFIGURE_SCRIPT_FILE" "$@"
    ret=$?
  else
    ret=$EXIT_CODE_NO_CONFIGURE_SCRIPT
  fi
fi

exit $ret