#!/bin/bash -e

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

echo --- downloading snap from build artifacts
buildkite-agent artifact download "solana_*.snap" .

# shellcheck disable=SC1091
source ci/upload_ci_artifact.sh

[[ -n $ITERATION_WAIT ]] || ITERATION_WAIT=300
[[ -n $NUMBER_OF_NODES ]] || NUMBER_OF_NODES="10 25 50 100"
[[ -n $LEADER_CPU_MACHINE_TYPE ]] ||
  LEADER_CPU_MACHINE_TYPE="n1-standard-16 --accelerator count=2,type=nvidia-tesla-v100"
[[ -n $CLIENT_COUNT ]] || CLIENT_COUNT=2
[[ -n $TESTNET_TAG ]] || TESTNET_TAG=testnet-automation
[[ -n $TESTNET_ZONE ]] || TESTNET_ZONE=us-west1-b

launchTestnet() {
  declare nodeCount=$1
  echo --- setup "$nodeCount" node test
  net/gce.sh create \
    -n "$nodeCount" -c "$CLIENT_COUNT" \
    -G "$LEADER_CPU_MACHINE_TYPE" \
    -p "$TESTNET_TAG" -z "$TESTNET_ZONE"

  echo --- configure database
  net/init-metrics.sh -e

  echo --- start "$nodeCount" node test
  net/net.sh start -o noValidatorSanity -S solana_*.snap

  echo --- wait "$ITERATION_WAIT" seconds to complete test
  sleep "$ITERATION_WAIT"

  declare q_mean_tps='
    SELECT round(mean("sum_count")) AS "mean_tps" FROM (
      SELECT sum("count") AS "sum_count"
        FROM "testnet-automation"."autogen"."counter-banking_stage-process_transactions"
        WHERE time > now() - 300s GROUP BY time(1s)
    )'

  declare q_max_tps='
    SELECT round(max("sum_count")) AS "max_tps" FROM (
      SELECT sum("count") AS "sum_count"
        FROM "testnet-automation"."autogen"."counter-banking_stage-process_transactions"
        WHERE time > now() - 300s GROUP BY time(1s)
    )'

  declare q_mean_finality='
    SELECT round(mean("duration_ms")) as "mean_finality"
      FROM "testnet-automation"."autogen"."leader-finality"
      WHERE time > now() - 300s'

  declare q_max_finality='
    SELECT round(max("duration_ms")) as "max_finality"
      FROM "testnet-automation"."autogen"."leader-finality"
      WHERE time > now() - 300s'

  declare q_99th_finality='
    SELECT round(percentile("duration_ms", 99)) as "99th_finality"
      FROM "testnet-automation"."autogen"."leader-finality"
      WHERE time > now() - 300s'

  curl -G "https://metrics.solana.com:8086/query?u=${INFLUX_USERNAME}&p=${INFLUX_PASSWORD}" \
    --data-urlencode "db=$INFLUX_DATABASE" \
    --data-urlencode "q=$q_mean_tps;$q_max_tps;$q_mean_finality;$q_max_finality;$q_99th_finality" |
    python ci/testnet-automation-json-parser.py >>TPS"$nodeCount".log

  upload_ci_artifact TPS"$nodeCount".log
}

# This is needed, because buildkite doesn't let us define an array of numbers.
# The array is defined as a space separated string of numbers
# shellcheck disable=SC2206
nodes_count_array=($NUMBER_OF_NODES)

for n in "${nodes_count_array[@]}"; do
  launchTestnet "$n"
done