#!/bin/bash
# Copyright (c) Mysten Labs, Inc.
# SPDX-License-Identifier: Apache-2.0

if [ "$1" != "simtest" ]; then
  echo "expected to be invoked via \`cargo simtest\`"
  exit 1
fi

# consume simtest arg
shift

# cargo does not export $CARGO_MANIFEST_DIR to subcommands so we have to find it
# ourselves.
STARTING_DIR=$(pwd)
MANIFEST_DIR="$STARTING_DIR"
while true; do
  if grep -q '^\[workspace\]$' Cargo.toml 2> /dev/null; then
    break
  fi
  cd ..
  MANIFEST_DIR=$(pwd)
done

# cd "$MANIFEST_DIR"
# if ! git diff --quiet Cargo.lock; then
#   echo "Please commit or revert your changes to Cargo.lock before running cargo simtest"
#   exit 1
# fi
cd "$STARTING_DIR"

cleanup () {
  cd "$MANIFEST_DIR"
  git checkout Cargo.lock > /dev/null
  cd "$STARTING_DIR"
}

trap cleanup SIGINT

if [ -z "$MSIM_TEST_SEED" ]; then
  export MSIM_TEST_SEED=1
else
  echo "Using MSIM_TEST_SEED=$MSIM_TEST_SEED from the environment"
fi

RUST_FLAGS=('"--cfg"' '"msim"')

if [ -n "$LOCAL_MSIM_PATH" ]; then
  cargo_patch_args=(
    --config "patch.crates-io.tokio.path = \"$LOCAL_MSIM_PATH/msim-tokio\""
    --config "patch.'https://github.com/MystenLabs/mysten-sim'.msim.path = \"$LOCAL_MSIM_PATH/msim\""
    --config "patch.crates-io.futures-timer.path = \"$LOCAL_MSIM_PATH/mocked-crates/futures-timer\""
  )
else
  cargo_patch_args=(
    --config 'patch.crates-io.tokio.git = "https://github.com/MystenLabs/mysten-sim.git"'
    --config 'patch.crates-io.tokio.rev = "4d9040e5c6bb264bfe14900d09a5da4000154da8"'
    --config 'patch.crates-io.futures-timer.git = "https://github.com/MystenLabs/mysten-sim.git"'
    --config 'patch.crates-io.futures-timer.rev = "4d9040e5c6bb264bfe14900d09a5da4000154da8"'
  )
fi

# Mock out the blst crate to massively speed up the simulation.
# You should not assume that test runs will be repeatable with and without blst mocking,
# as blst samples from the PRNG when not mocked.
if [ -n "$USE_MOCK_CRYPTO" ]; then
  echo "Using mocked crypto crates - no cryptographic verification will occur"
  RUST_FLAGS+=(
    '"--cfg"'
    '"use_mock_crypto"'
  )
  cargo_patch_args+=(
    --config 'patch.crates-io.blst.git = "https://github.com/MystenLabs/mock-blst.git"'
    --config 'patch.crates-io.blst.rev = "630ca4d55de8e199e62c5b6a695c702d95fe6498"'
  )
fi

RUST_FLAGS=$(IFS=, ; echo "${RUST_FLAGS[*]}")

if ! cargo nextest --help > /dev/null 2>&1; then
  echo "nextest (https://nexte.st) does not appear to be installed. Please install before proceeding."
  echo "You can install it by running:"
  echo ""
  echo "   \$ cargo install cargo-nextest --locked"
  echo ""
  echo "or see https://nexte.st for other installation options"
  exit 1
fi

CARGO_COMMAND=(nextest run --cargo-profile simulator)
if [ "$1" = "build" ]; then
  shift
  CARGO_COMMAND=(build --profile simulator)
fi

# Must supply a new temp dir - the test is deterministic and can't choose one randomly itself.
export TMPDIR=$(mktemp -d)

cargo ${CARGO_COMMAND[@]} \
  --config "build.rustflags = [$RUST_FLAGS]" \
  "${cargo_patch_args[@]}" \
  "$@"

STATUS=$?

cleanup

exit $STATUS