#!/usr/bin/env bash
set -euo pipefail

# JS7 Controller starter

JS7_HOME="$(cd "${0%/*}/../bin/.." && pwd)"
export JS7_HOME
. "$JS7_HOME/bin/internal/set-context.sh"
declare classpathString java  # returned by set-context.sh
declare standardJavaOptions

config=/var/opt/js7/controller/config
data=/var/opt/js7/controller/data
httpPort=4444
httpsPort=4443
controllerOptions=()
javaOptions=(-XX:+UseStringDeduplication)
execOptions=()

for arg in "$@"; do :
  case "$arg" in
    --arg0=*)
      a="${arg#*=}"
      execOptions=("-a" "${arg#*=}")
      ;;
    -J*)
      a="${arg#-J}"
      javaOptions+=("$a")
      ;;
    --java-option=*)
      a="${arg#*=}"
      javaOptions+=("$a")
      ;;
    --directory=*)
      config="${arg#*=}"/config
      data="${arg#*=}"/data
      ;;
    --config-directory=*)
      config="${arg#*=}"
      ;;
    --data-directory=*)
      data="${arg#*=}"
      ;;
    --http-port=*)
      httpPort="${arg#*=}"
      ;;
    --https-port=*)
      httpsPort="${arg#*=}"
      ;;
    --rmx-port=*)
      a="${arg#*=}"
      javaOptions+=(
        "-Dcom.sun.management.jmxremote"
        "-Dcom.sun.management.jmxremote.ssl=false"
        "-Dcom.sun.management.jmxremote.authenticate=false"
        "-Dcom.sun.management.jmxremote.port=$a")
      ;;
    --debug-port=*)
      a="${arg#*=}"
      javaOptions+=("-agentlib:jdwp=transport=dt_socket,server=y,suspend=n,address=$a")
      ;;
    *)
      controllerOptions+=("$arg")
  esac
done

if [ ! -d "$config" ]; then :
  echo "No such directory: --config-directory=$config"
  exit 1
fi
config="$(toSystemPath "$config")"

if [ ! -d "$data" ]; then :
  echo "No such directory: --data-directory=$data"
  exit 1
fi
data="$(toSystemPath "$data")"

logs="$data/logs"
[ -d "$logs" ] || mkdir "$logs"

stateDir="$data/state"
[ -d "$stateDir" ] || mkdir "$stateDir"

workDir="$data/work"
[ -d "$workDir" ] || mkdir "$workDir"

# Copy stdout and stderr to stdouterr.log
if [ -e "$logs/stdouterr.log" ]; then
  echo >>"$logs/stdouterr.log" \
    -e "\n\n━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
fi
exec &> >(exec tee -a "$logs"/stdouterr.log)

javaOptions+=("-Djs7.log4j.directory=$logs")  # Used in logging configuration
if [ -f "$config/log4j2.xml" ]; then :
  javaOptions+=("-Dlog4j2.configurationFile=classpath:js7/log4j2.xml,$config/log4j2.xml")
fi

controllerOptions+=("--config-directory=$config" "--data-directory=$data")
[ -z "$httpPort" ] || controllerOptions+=("--http-port=$httpPort")
[ -z "$httpsPort" ] || controllerOptions+=("--https-port=$httpsPort")

while true; do
  loopJavaOptions=()

  if [[ "$JAVA_VERSION" =~ ([0-9]+).* && ${BASH_REMATCH[1]} -ge 13 ]]; then
    [ -d "$data/cache" ] || mkdir "$data/cache"
    jsaFile="$data/cache/js7-controller-$(cat "$JS7_HOME/VERSION").jsa"
    if [[ "$JAVA_VERSION" =~ ([0-9]+).* && ${BASH_REMATCH[1]} -ge 19 ]]; then
      loopJavaOptions+=("-XX:SharedArchiveFile=$jsaFile" -XX:+AutoCreateSharedArchive)
    elif [ -f "$jsaFile" ]; then
      loopJavaOptions+=("-XX:SharedArchiveFile=$jsaFile" -Xshare:auto)
    else
      loopJavaOptions+=("-XX:ArchiveClassesAtExit=$jsaFile" -Xshare:auto)
    fi
  fi

  execute=(
    "$java"
    "${javaOptions[@]}"
    "${loopJavaOptions[@]}"
    "${standardJavaOptions[@]}"
    -classpath "$classpathString"
    js7.controller.ControllerMain
    "${controllerOptions[@]}")

  log "${execute[@]}"
  exec "${execOptions[@]}" "${execute[@]}" &

  pid=$!
  log "PID $pid"

  terminate() {
    echo "⚡️ $1"
    log "$1: ShutDown JS7 Controller"
    "$JS7_HOME/bin/js7-shutdown" --data-directory="$data" &
    wait $pid
    returnCode=$?
  }

  returnCode=
  trap "terminate SIGTERM" SIGTERM
  trap "terminate SIGINT" SIGINT
  if wait $pid; then
    returnCode=${returnCode:-$?}
  else
    returnCode=${returnCode:-$?}
  fi
  # After a trapped signal, $? is 128+signal, but not the real $returnCode (only in Docker?)
  trap - SIGTERM SIGINT

  if [[ $returnCode -eq 0 ]] || [[ $returnCode -eq 97 ]]; then
    log "JS7 Controller has been shut down"
  else
    log "JS7 Controller exited with exit code $returnCode"
  fi

  if [[ $returnCode -eq 0 ]] || [[ $returnCode -ge 97 ]] && [[ $returnCode -le 99 ]]; then
    log "Clear $workDir"
    find "$workDir" -mindepth 1 -delete ! -type d -printf "Removed %p\n"
  fi

  if [[ $returnCode -eq 97 ]] || [[ $returnCode -eq 98 ]]; then
    log "Starting again"
  else
    exit $returnCode
  fi
done
