#!/usr/bin/env sh

usage="usage: bench <year> <day> <language> <testfile> [<warmups>]"
year=${1?$usage}
day=${2?$usage}
lang=${3?$usage}
input=${4?$usage}
warmups=${5:-0}

# get directory of this script
script=$(readlink -f "$0")
script_path=$(dirname "$script")

# find aoc and data paths
aoc_path=$script_path/$year/$day
data_path="$aoc_path/data/$input.txt"

# check solution is correct for this input
$script_path/check $year $day $lang $input >/dev/null
[ ! $? ] && echo "Incorrect solution" && exit 1

case $lang in
"python")
    # Sync to create venv
    uv sync \
        --directory $aoc_path/python
    # Bypass any `uv run` overhead and run .venv directly
    exec="$aoc_path/python/.venv/bin/python $aoc_path/python/main.py"
    ;;
"rust")
    # Compile to /tmp/aoc_rust and run binary
    cargo build \
        --release \
        --target-dir /tmp/aoc_rust \
        --manifest-path $aoc_path/rust/Cargo.toml
    exec="/tmp/aoc_rust/release/aoc_$year-$day"
    ;;
*)
    echo "unknown lang: $lang"
    exit 1
    ;;
esac

hyperfine \
    --shell none \
    --warmup $warmups \
    --input $data_path \
    "$exec"