#!/bin/bash

set -eo pipefail

# To allow overriding the program names
TMUX_CMD="${TMUX_CMD:-tmux}"
MULTITAIL_CMD="${MULTITAIL_CMD:-multitail}"
GANACHE_CMD="${GANACHE_CMD:-ganache-cli}"
PROMETHEUS_CMD="${PROMETHEUS_CMD:-prometheus}"
CTAIL_CMD="${CTAIL_CMD:-ctail}"

TMUX_SESSION_NAME="${TMUX_SESSION_NAME:-nbc-sim}"

WAIT_GENESIS="${WAIT_GENESIS:-no}"

USE_MULTITAIL="${USE_MULTITAIL:-no}"
if [[ "$USE_MULTITAIL" != "no" ]]; then
  type "$MULTITAIL_CMD" &>/dev/null || { echo "${MULTITAIL_CMD}" is missing; USE_MULTITAIL="no"; }
fi

USE_TMUX="${USE_TMUX:-no}"
if [[ "$USE_TMUX" != "no" ]]; then
  type "$TMUX_CMD" &>/dev/null || { echo "${TMUX_CMD}" is missing; USE_TMUX="no"; }
fi

USE_GANACHE="${USE_GANACHE:-yes}"
if [[ "$USE_GANACHE" == "yes" ]]; then
  type "$GANACHE_CMD" &>/dev/null || { echo $GANACHE_CMD is missing; USE_GANACHE="no"; }
fi

USE_PROMETHEUS="${USE_PROMETHEUS:-yes}"
if [[ "$USE_PROMETHEUS" == "yes" ]]; then
  type "$PROMETHEUS_CMD" &>/dev/null || { echo $PROMETHEUS_CMD is missing; USE_PROMETHEUS="no"; }
fi

USE_CTAIL="${USE_CTAIL:-yes}"
if [[ "$USE_CTAIL" == "yes" ]]; then
  type "$CTAIL_CMD" &>/dev/null || { USE_CTAIL="no"; }
fi

# Read in variables
# shellcheck source=/dev/null
source "$(dirname "$0")/vars.sh"

cd "$SIM_ROOT"
mkdir -p "$SIMULATION_DIR"
mkdir -p "$VALIDATORS_DIR"
mkdir -p "$SECRETS_DIR"

cd "$GIT_ROOT"

CUSTOM_NIMFLAGS="${NIMFLAGS} -d:useSysAsserts -d:chronicles_sinks:textlines,json[file] -d:const_preset=mainnet -d:insecure"

# Run with "SLOTS_PER_EPOCH=8 ./start.sh" to change these
DEFS=""
DEFS+="-d:MIN_GENESIS_ACTIVE_VALIDATOR_COUNT=${NUM_VALIDATORS} -d:MIN_GENESIS_TIME=0 "
DEFS+="-d:MAX_COMMITTEES_PER_SLOT=${MAX_COMMITTEES_PER_SLOT:-1} "      # Spec default: 64
DEFS+="-d:SLOTS_PER_EPOCH=${SLOTS_PER_EPOCH:-6} "   # Spec default: 32
DEFS+="-d:SECONDS_PER_SLOT=${SECONDS_PER_SLOT:-6} "  # Spec default: 12

# Windows detection
if uname | grep -qiE "mingw|msys"; then
  MAKE="mingw32-make"
else
  MAKE="make"
fi

mkdir -p "${METRICS_DIR}"
./scripts/make_prometheus_config.sh \
  --nodes ${TOTAL_NODES} \
  --base-metrics-port ${BASE_METRICS_PORT} \
  --config-file "${METRICS_DIR}/prometheus.yml" || true # TODO: this currently fails on macOS,
                                                        # but it can be considered non-critical

COMMANDS=()

if [[ "$USE_GANACHE" == "yes" ]]; then
  if [[ "$USE_TMUX" == "yes" ]]; then
    $TMUX_CMD new-window -d -t $TMUX_SESSION_NAME -n "$GANACHE_CMD" "$GANACHE_CMD --blockTime 5 --gasLimit 100000000 -e 100000 --verbose"
  else
    echo NOTICE: $GANACHE_CMD will be started automatically only with USE_TMUX=1
    USE_GANACHE="no"
  fi
fi

if [[ "$USE_PROMETHEUS" == "yes" ]]; then
  if [[ "$USE_TMUX" == "yes" ]]; then
    rm -rf "${METRICS_DIR}/data"
    mkdir -p "${METRICS_DIR}/data"
    # TODO: Prometheus is not shut down properly on tmux kill-session
    killall prometheus > /dev/null || true
    PROMETHEUS_FLAGS="--config.file=./prometheus.yml --storage.tsdb.path=./data"
    $TMUX_CMD new-window -d -t $TMUX_SESSION_NAME -n "$PROMETHEUS_CMD" "cd '$METRICS_DIR' && $PROMETHEUS_CMD $PROMETHEUS_FLAGS"
  else
    echo NOTICE: $PROMETHEUS_CMD will be started automatically only with USE_TMUX=1
    USE_PROMETHEUS="no"
  fi
fi

$MAKE -j3 --no-print-directory NIMFLAGS="$CUSTOM_NIMFLAGS $DEFS" LOG_LEVEL="${LOG_LEVEL:-DEBUG}" beacon_node validator_client

count_files () {
  { ls -1q $1 2> /dev/null || true ; } | wc -l
}

EXISTING_VALIDATORS=$(count_files "$VALIDATORS_DIR/*/deposit.json")

if [[ $EXISTING_VALIDATORS -lt $NUM_VALIDATORS ]]; then
  rm -rf "$VALIDATORS_DIR"
  rm -rf "$SECRETS_DIR"

  $BEACON_NODE_BIN deposits create \
    --count="${NUM_VALIDATORS}" \
    --non-interactive \
    --out-deposits-dir="$VALIDATORS_DIR" \
    --out-secrets-dir="$SECRETS_DIR" \
    --dont-send

  echo "All deposits prepared"
fi

if [ ! -f "${SNAPSHOT_FILE}" ]; then
  if [[ "${WAIT_GENESIS}" != "yes" ]]; then
    echo Creating testnet genesis...
    $BEACON_NODE_BIN \
      --data-dir="${SIMULATION_DIR}/node-$BOOTSTRAP_NODE" \
      createTestnet \
      --validators-dir="${VALIDATORS_DIR}" \
      --total-validators="${NUM_VALIDATORS}" \
      --output-genesis="${SNAPSHOT_FILE}" \
      --output-bootstrap-file="${NETWORK_BOOTSTRAP_FILE}" \
      --bootstrap-address=127.0.0.1 \
      --bootstrap-port=$(( BASE_P2P_PORT + BOOTSTRAP_NODE )) \
      --genesis-offset=15 # Delay in seconds
  fi
fi

function run_cmd {
  i=$1
  CMD=$2
  bin_name=$3
  if [[ "$USE_TMUX" == "yes" ]]; then
    echo "Starting node $i..."
    $TMUX_CMD select-window -t "${TMUX_SESSION_NAME}:sim"
    $TMUX_CMD split-window -t "${TMUX_SESSION_NAME}" "if ! $CMD; then; read; fi"
    $TMUX_CMD select-layout -t "${TMUX_SESSION_NAME}:sim" tiled
  elif [[ "$USE_MULTITAIL" != "no" ]]; then
    if [[ "$i" == "$BOOTSTRAP_NODE" ]]; then
      SLEEP="0"
    else
      SLEEP="3"
    fi
    # "multitail" closes the corresponding panel when a command exits, so let's make sure it doesn't exit
    COMMANDS+=( " -cT ansi -t '$bin_name #$i' -l 'sleep $SLEEP; $CMD; echo [node execution completed]; while true; do sleep 100; done'" )
  else
    eval "${CMD}" &
  fi
}

if [ "$USE_GANACHE" != "no" ]; then
  make deposit_contract
  echo Deploying the validator deposit contract...
  echo $DEPLOY_DEPOSIT_CONTRACT_BIN deploy $WEB3_ARG
  DEPOSIT_CONTRACT_ADDRESS=$($DEPLOY_DEPOSIT_CONTRACT_BIN deploy $WEB3_ARG)
  echo Contract deployed at $DEPOSIT_CONTRACT_ADDRESS
  echo $DEPOSIT_CONTRACT_ADDRESS > $DEPOSIT_CONTRACT_FILE

  if [[ "$WAIT_GENESIS" == "yes" ]]; then
    run_cmd "(deposit maker)" "$BEACON_NODE_BIN deposits send \
      --non-interactive \
      --deposits-dir='$VALIDATORS_DIR' \
      --min-delay=0 --max-delay=1 \
      $WEB3_ARG \
      --deposit-contract=${DEPOSIT_CONTRACT_ADDRESS}"
  fi
fi

if [[ "$USE_TMUX" == "yes" ]]; then
  $TMUX_CMD select-window -t "${TMUX_SESSION_NAME}:sim"
fi

# Delete any leftover address files from a previous session
if [ -f "${BOOTSTRAP_ENR_FILE}" ]; then
  rm "${BOOTSTRAP_ENR_FILE}"
fi

# Kill child processes on Ctrl-C/SIGTERM/exit, passing the PID of this shell
# instance as the parent and the target process name as a pattern to the
# "pkill" command.
if [[ "$USE_MULTITAIL" == "no" && "$USE_TMUX" != "yes" ]]; then
  trap 'pkill -P $$ beacon_node' SIGINT EXIT
fi

LAST_WAITING_NODE=0

for i in $(seq $BOOTSTRAP_NODE -1 $TOTAL_USER_NODES); do
  if [[ "$i" != "$BOOTSTRAP_NODE" && "$USE_MULTITAIL" == "no" ]]; then
    # Wait for the master node to write out its address file
    while [ ! -f "${BOOTSTRAP_ENR_FILE}" ]; do
      if (( LAST_WAITING_NODE != i )); then
        echo Waiting for $BOOTSTRAP_ENR_FILE to appear...
        LAST_WAITING_NODE=i
      fi
      sleep 0.1
    done
  fi

  run_cmd $i "${SIM_ROOT}/run_node.sh ${i} --verify-finalization" "node"

  if [ "${BN_VC_VALIDATOR_SPLIT:-}" == "yes" ]; then
    # start the VC with a few seconds of delay so that we can connect through RPC
    run_cmd $i "sleep 3 && ${SIM_ROOT}/run_validator.sh ${i}" "validator"
  fi
done

if [[ "$USE_CTAIL" != "no" ]]; then
  if [[ "$USE_TMUX" == "yes" ]]; then
    $TMUX_CMD new-window -d -t $TMUX_SESSION_NAME -n "$CTAIL_CMD" "$CTAIL_CMD tail -q -n +1 -f ${SIMULATION_DIR}/node-*/beacon_node.log"
  else
    echo NOTICE: $CTAIL_CMD will be started automatically only with USE_TMUX=1
    USE_CTAIL="no"
  fi
fi

if [[ "$USE_TMUX" == "yes" ]]; then
  # kill the console window in the pane where the simulation is running
  $TMUX_CMD kill-pane -t $TMUX_SESSION_NAME:sim.0
  $TMUX_CMD select-window -t "${TMUX_SESSION_NAME}:sim"
  $TMUX_CMD select-layout tiled
elif [[ "$USE_MULTITAIL" != "no" ]]; then
  eval $MULTITAIL_CMD -s 3 -M 0 -x \"Nimbus beacon chain\" "${COMMANDS[@]}"
else
  wait # Stop when all nodes have gone down
fi