#!/bin/bash
set -euo pipefail

export TEST_TRACE=0

export GIT_TRACE=1
export RUST_LOG=trace
export RUST_BACKTRACE=1

# Scaling factor for iterations (set via environment variable, default to 1)
# Higher values increase the number of iterations for more thorough testing
ITERATION_SCALE_FACTOR=${ITERATION_SCALE_FACTOR:-1}

# Base iteration counts (multiplied by ITERATION_SCALE_FACTOR)
BASE_PUSH_ITERATIONS=150
BASE_REMOVE_ITERATIONS=150
BASE_PRUNE_ITERATIONS=100
BASE_ADD_ITERATIONS=199

# Calculate actual iteration counts based on scale factor
NUM_PUSH_ITERATIONS=$((BASE_PUSH_ITERATIONS * ITERATION_SCALE_FACTOR))
NUM_REMOVE_ITERATIONS=$((BASE_REMOVE_ITERATIONS * ITERATION_SCALE_FACTOR))
NUM_PRUNE_ITERATIONS=$((BASE_PRUNE_ITERATIONS * ITERATION_SCALE_FACTOR))
NUM_ADD_ITERATIONS=$((BASE_ADD_ITERATIONS * ITERATION_SCALE_FACTOR))

# Concurrency settings (NOT affected by scale factor)
CONCURRENT_ADDERS=4
CONCURRENT_PUSHERS=2
CONCURRENT_REMOVERS=2
CONCURRENT_PRUNERS=2

script_dir=$(unset CDPATH; cd "$(dirname "$0")" > /dev/null; pwd -P)
# shellcheck source=test/common.sh
source "$script_dir/common.sh"

cd "$(mktemp -d)"
root=$(pwd)

mkdir upstream
pushd upstream
git init --bare
popd

git clone "$root/upstream" work
pushd work
git commit --allow-empty -m 'test commit'
git push

# Add initial measurement to evade corner cases of for example git-perf remove with no measurements on the remote.
# These are not useful to test in this context.
git perf add -m 'seedmeasurement' 1
git perf push


# Function to handle cleanup when script is interrupted
# shellcheck disable=SC2317
cleanup() {
    echo ""
    echo "Received interrupt signal. Terminating all jobs..."
    # Kill all jobs in the current process group
    kill "$(jobs -p)" 2>/dev/null || true
    echo "Cleanup complete. Exiting."
    exit 1
}

# Set up trap for CTRL+C and other termination signals
trap cleanup SIGINT SIGTERM

# Function to run git-perf push command (without measurement parameter)
run_remove_test() {
    local log_prefix=$1
    shift  # Remove first parameter (prefix) from the argument list
    local iterations=$NUM_REMOVE_ITERATIONS

    local cmd_display="git-perf $*"
    echo "Starting $iterations iterations of '$cmd_display'..."

    for (( i=1; i<=iterations; i++ )); do
        # Run the push command without additional parameters
        # 7 days is used as a reasonable default for test cleanup. Adjust if needed for other scenarios.
        git-perf "$@" --older-than '7d' 2>&1 | perl -pe "s/^/[$log_prefix] /" || (echo "Failed to remove" && exit 1)

        # Every 10 iterations, print a status update, and back off a bit
        if (( i % 10 == 0 )); then
            echo "[$log_prefix] Completed $i/$iterations iterations"
            # Back off a little bit
            sleep 1
        fi
    done

    echo "[$log_prefix] All $iterations iterations completed"
    return 0
}

run_prune_test() {
    local log_prefix=$1
    shift  # Remove first parameter (prefix) from the argument list
    local iterations=$NUM_PRUNE_ITERATIONS

    local cmd_display="git-perf $*"
    echo "Starting $iterations iterations of '$cmd_display'..."

    for (( i=1; i<=iterations; i++ )); do
        # Run the push command without additional parameters
        # 7 days is used as a reasonable default for test cleanup. Adjust if needed for other scenarios.
        git-perf "$@" 2>&1 | perl -pe "s/^/[$log_prefix] /" || (echo "Failed to prune" && exit 1)

        # Every 10 iterations, print a status update, and back off a bit
        if (( i % 10 == 0 )); then
            echo "[$log_prefix] Completed $i/$iterations iterations"
            # Back off a little bit
            sleep 1
        fi
    done

    echo "[$log_prefix] All $iterations iterations completed"
    return 0
}

# Function to run git-perf push command (without measurement parameter)
run_push_test() {
    local log_prefix=$1
    shift  # Remove first parameter (prefix) from the argument list
    local iterations=$NUM_PUSH_ITERATIONS

    local cmd_display="git-perf $*"
    echo "Starting $iterations iterations of '$cmd_display'..."

    for (( i=1; i<=iterations; i++ )); do
        # Run the push command without additional parameters
        git-perf "$@" 2>&1 | perl -pe "s/^/[$log_prefix] /" || (echo "Failed to push" && exit 1)

        # Every 10 iterations, print a status update, and back off a bit
        if (( i % 10 == 0 )); then
            echo "[$log_prefix] Completed $i/$iterations iterations"
            # Back off a little bit
            sleep 1
        fi
    done

    echo "[$log_prefix] All $iterations iterations completed"
    return 0
}

# Function to run git-perf add command (with measurement parameter)
run_add_test() {
    local log_prefix=$1
    shift  # Remove first parameter (prefix) from the argument list
    local iterations=$NUM_ADD_ITERATIONS

    local cmd_display="git-perf $* --measurement test-$log_prefix <random>"
    echo "Starting $iterations iterations of '$cmd_display'..."

    for (( i=1; i<=iterations; i++ )); do
        # Generate a random integer
        local random_value=$((RANDOM + RANDOM * i + $(date +%s) % 10000))

        # Run the add command with measurement parameter
        # Adding separate measurements for each adder as the random values could overlap
        git-perf "$@" --measurement "test-$log_prefix" $random_value 2>&1 | perl -pe "s/^/[$log_prefix] /" || (echo "Failed to add" && exit 1)

        # Every 100 iterations, print a status update
        if (( i % 10 == 0 )); then
            echo "[$log_prefix] Completed $i/$iterations iterations"
        fi
    done

    echo "[$log_prefix] All $iterations iterations completed"
    return 0
}

# Run the tests in parallel
echo "Starting test harness..."
echo "Press CTRL+C at any time to abort the test"

PUSH_PIDS=()
for i in $(seq 1 $CONCURRENT_PUSHERS); do
  run_push_test "PUSH_$i" push &
  PUSH_PIDS+=($!)
done

REMOVE_PIDS=()
for i in $(seq 1 $CONCURRENT_REMOVERS); do
  run_remove_test "REMOVE_$i" remove &
  REMOVE_PIDS+=($!)
done

ADD_PIDS=()
for i in $(seq 1 $CONCURRENT_ADDERS); do
  run_add_test "ADD_$i" add &
  ADD_PIDS+=($!)
done

PRUNE_PIDS=()
for i in $(seq 1 $CONCURRENT_PRUNERS); do
  run_prune_test "PRUNE_$i" prune &
  PRUNE_PIDS+=($!)
done

# Wait for both processes to complete
echo "Waiting for all tests to complete..."
set +e

PUSH_STATUS=0
if [ $CONCURRENT_PUSHERS -gt 0 ]; then
  wait "${PUSH_PIDS[@]}"
  PUSH_STATUS=$?
fi

REMOVE_STATUS=0
if [ $CONCURRENT_REMOVERS -gt 0 ]; then
  wait "${REMOVE_PIDS[@]}"
  REMOVE_STATUS=$?
fi

ADD_STATUS=0
if [ $CONCURRENT_ADDERS -gt 0 ]; then
  wait "${ADD_PIDS[@]}"
  ADD_STATUS=$?
fi

PRUNE_STATUS=0
if [ $CONCURRENT_PRUNERS -gt 0 ]; then
  wait "${PRUNE_PIDS[@]}"
  PRUNE_STATUS=$?
fi

# Reset trap for normal completion
trap - SIGINT SIGTERM

exit_code=0

# Check if all processes completed successfully
if [ $PUSH_STATUS -ne 0 ] || [ $ADD_STATUS -ne 0 ] || [ $REMOVE_STATUS -ne 0 ] || [ $PRUNE_STATUS -ne 0 ]; then
    echo "ERROR: One or more tests failed!"
    [ $PUSH_STATUS -ne 0 ] && echo "  'push' test failed with exit code $PUSH_STATUS"
    [ $REMOVE_STATUS -ne 0 ] && echo "  'remove' test failed with exit code $REMOVE_STATUS"
    [ $ADD_STATUS -ne 0 ] && echo "  'add' test failed with exit code $ADD_STATUS"
    [ $PRUNE_STATUS -ne 0 ] && echo "  'prune' test failed with exit code $PRUNE_STATUS"
    exit_code=1
fi

# Verify the results
echo "Verifying results..."
LINE_COUNT=$(git-perf report -o - | wc -l)
# First seed measurement + measurements from adders + header line
EXPECTED_COUNT=$((NUM_ADD_ITERATIONS * CONCURRENT_ADDERS + 1 + 1))

if [[ $LINE_COUNT -eq $EXPECTED_COUNT ]]; then
    echo "SUCCESS: Verification passed. Found exactly $EXPECTED_COUNT lines in the report."
else
    echo "ERROR: Verification failed. Expected $EXPECTED_COUNT lines but found $LINE_COUNT."
    exit_code=1
fi

test_stats
exit $exit_code