#!/bin/bash
# Connection via libpq env vars (PGHOST/PGPORT/PGUSER/PGDATABASE), override binary with PSQL
# Usage: ./run_bench.sh [iterations]
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
ITERATIONS="${1:-5}"
OUTFILE="$SCRIPT_DIR/results.csv"
PSQL="${PSQL:-psql}"

echo "category,test_name,engine,rows,time_ms,iteration" > "$OUTFILE"

# Build data once; iterations measure regex only
echo "=== Building benchmark data ===" >&2
"$PSQL" -X -q -f "$SCRIPT_DIR/setup.sql" >&2

for i in $(seq 1 "$ITERATIONS"); do
    echo "=== Iteration $i/$ITERATIONS ===" >&2
    "$PSQL" -X -q -f "$SCRIPT_DIR/bench.sql" | while IFS= read -r line; do
        case "$line" in
            match,*|extract,*|extract_all,*|extract_all_unnest,*|replace_one,*|replace_all,*|count_matches,*|idx_btree,*|idx_gin,*)
                echo "${line},${i}" ;;
        esac
    done >> "$OUTFILE"
done

"$PSQL" -X -q -f "$SCRIPT_DIR/teardown.sql" >&2

echo "Results written to $OUTFILE" >&2
column -t -s',' "$OUTFILE" >&2