#!/bin/bash -e

cd "$(dirname "$0")"/../..

deployMethod="$1"
nodeType="$2"
leaderIp="$3"
numNodes="$4"
setupArgs="$5"
RUST_LOG="$6"

cat > deployConfig <<EOF
deployMethod="$deployMethod"
leaderIp="$leaderIp"
numNodes="$numNodes"
EOF

[[ -n $deployMethod ]] || exit
[[ -n $nodeType ]] || exit
[[ -n $leaderIp ]] || exit

source net/common.sh
loadConfigFile

scripts/install-earlyoom.sh


case $deployMethod in
snap)
  SECONDS=0
  rsync -vPr "$leaderIp:~/solana/solana.snap" .
  sudo snap install solana.snap --devmode --dangerous

  commonNodeConfig="\
    leader-ip=$leaderIp \
    metrics-config=$SOLANA_METRICS_CONFIG \
    rust-log=$RUST_LOG \
    setup-args=$setupArgs \
  "

  if [[ -e /dev/nvidia0 ]]; then
    commonNodeConfig="$commonNodeConfig enable-cuda=1"
  fi

  if [[ $nodeType = leader ]]; then
    nodeConfig="mode=leader+drone $commonNodeConfig"
  else
    nodeConfig="mode=validator $commonNodeConfig"
  fi

  logmarker="solana deploy $(date)/$RANDOM"
  logger "$logmarker"

  # shellcheck disable=SC2086 # Don't want to double quote "$nodeConfig"
  sudo snap set solana $nodeConfig
  snap info solana
  sudo snap get solana
  echo Slight delay to get more syslog output
  sleep 2
  sudo grep -Pzo "$logmarker(.|\\n)*" /var/log/syslog

  echo "Succeeded in ${SECONDS} seconds"
  ;;
local)
  PATH="$HOME"/.cargo/bin:"$PATH"
  export USE_INSTALL=1
  export RUST_LOG
  if [[ -e /dev/nvidia0 ]]; then
    export SOLANA_CUDA=1
  fi

  ./fetch-perf-libs.sh
  scripts/oom-monitor.sh  > oom-monitor.log 2>&1 &

  case $nodeType in
  leader)
    # shellcheck disable=SC2086 # Don't want to double quote "$setupArgs"
    ./multinode-demo/setup.sh -t leader -p $setupArgs
    ./multinode-demo/drone.sh > drone.log 2>&1 &
    ./multinode-demo/leader.sh > leader.log 2>&1 &
    ;;
  validator)
    rsync -vPr "$leaderIp:~/.cargo/bin/solana*" ~/.cargo/bin/

    # shellcheck disable=SC2086 # Don't want to double quote "$setupArgs"
    ./multinode-demo/setup.sh -t validator -p $setupArgs
    ./multinode-demo/validator.sh "$leaderIp":~/solana "$leaderIp" >validator.log 2>&1 &
    ;;
  *)
    echo "Error: unknown node type: $nodeType"
    exit 1
    ;;
  esac
  ;;
*)
  echo "Unknown deployment method: $deployMethod"
  exit 1
esac