#!/bin/sh
# =====================================
# Shared test harness for builtin tests
# =====================================
# Source this file from individual builtin test scripts.
# Each test file must set TEST_PREFIX before sourcing.

# Portable timeout: macOS has gtimeout (from coreutils) or no timeout
if command -v timeout >/dev/null 2>&1; then
    run_with_timeout() { timeout "$@"; }
elif command -v gtimeout >/dev/null 2>&1; then
    run_with_timeout() { gtimeout "$@"; }
else
    run_with_timeout() {
        _rwt_secs="$1"; shift
        _rwt_tmp=$(mktemp /tmp/bensch_timeout.XXXXXX)
        ( "$@" ) > "$_rwt_tmp" 2>&1 &
        _rwt_pid=$!
        ( sleep "$_rwt_secs"; kill "$_rwt_pid" 2>/dev/null ) &
        _rwt_wd=$!
        wait "$_rwt_pid" 2>/dev/null
        _rwt_rc=$?
        kill "$_rwt_wd" 2>/dev/null
        wait "$_rwt_wd" 2>/dev/null
        cat "$_rwt_tmp"
        rm -f "$_rwt_tmp"
        return $_rwt_rc
    }
fi

# Colors (POSIX-compliant way)
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

CURRENT_SECTION=""
TEST_NUM=0

PASSED=0
FAILED=0
SKIPPED=0
FAILED_TESTS_LIST=""

# Get script directory (POSIX way)
SCRIPT_DIR=$(cd "$(dirname "$0")" && pwd)
SHELL_BIN="${SHELL_BIN:?ERROR: SHELL_BIN must be set}"

# Reference bash for expected output — must be bash 4+ for assoc arrays, case transforms
# macOS ships bash 3.2 which lacks these; override with BASH_REF=/opt/homebrew/bin/bash
BASH_REF="${BASH_REF:-bash}"

# Check if shell binary exists
if [ ! -x "$SHELL_BIN" ]; then
    printf "${RED}ERROR${NC}: shell binary not found at $SHELL_BIN\n"
    printf "Please set SHELL_BIN or set SHELL_BIN environment variable\n"
    exit 1
fi

# Temp directory for test files
TEST_TMPDIR=$(mktemp -d)
cleanup() { rm -rf "$TEST_TMPDIR" 2>/dev/null; }
trap cleanup EXIT INT TERM

pass() {
    TEST_NUM=$((TEST_NUM + 1))
    printf "${GREEN}✓ PASS${NC} ${TEST_PREFIX} ${CURRENT_SECTION}.${TEST_NUM}: %s\n" "$1"
    PASSED=$((PASSED + 1))
}

fail() {
    TEST_NUM=$((TEST_NUM + 1))
    TEST_ID="${TEST_PREFIX} ${CURRENT_SECTION}.${TEST_NUM}"
    printf "${RED}✗ FAIL${NC} ${TEST_ID}: %s\n" "$1"
    FAILED_TESTS_LIST="${FAILED_TESTS_LIST}  ${TEST_ID}: $1\n"
    if [ -n "$2" ]; then printf "  expected: %s\n" "$2"; fi
    if [ -n "$3" ]; then printf "  got:      %s\n" "$3"; fi
    FAILED=$((FAILED + 1))
}

skip() {
    TEST_NUM=$((TEST_NUM + 1))
    printf "${YELLOW}⊘ SKIP${NC} ${TEST_PREFIX} ${CURRENT_SECTION}.${TEST_NUM}: %s - %s\n" "$1" "$2"
    SKIPPED=$((SKIPPED + 1))
}

section() {
    CURRENT_SECTION=$(echo "$1" | grep -oE '^[0-9]+' || echo "0")
    TEST_NUM=0
    printf "\n${BLUE}==========================================\n%s\n==========================================${NC}\n" "$1"
}

# Normalize shell name prefixes in error messages for comparison
normalize_shell_name() {
    printf '%s\n' "$1" | sed -e 's/[a-z]*sh[0-9]*: line [0-9]*: /SHELL: /g' \
                              -e  \
                              -e 's/[a-z]*sh[0-9]*: /SHELL: /g' \
                              -e 's/[a-z]*sh[0-9]*: /SHELL: /g'
}

# Per-test timeout (seconds) — prevents hanging tests from blocking CI
TEST_TIMEOUT="${TEST_TIMEOUT:-10}"

# Helper: compare output against bash
compare_output() {
    test_name="$1"; command="$2"
    expected=$(run_with_timeout "$TEST_TIMEOUT" "$BASH_REF" -c "$command" 2>&1)
    actual=$(run_with_timeout "$TEST_TIMEOUT" "$SHELL_BIN" -c "$command" 2>&1)
    norm_expected=$(normalize_shell_name "$expected")
    norm_actual=$(normalize_shell_name "$actual")
    if [ "$norm_expected" = "$norm_actual" ]; then pass "$test_name"
    else fail "$test_name" "$expected" "$actual"; fi
}

# Helper: compare exit code only
compare_exit() {
    test_name="$1"; command="$2"
    run_with_timeout "$TEST_TIMEOUT" "$BASH_REF" -c "$command" >/dev/null 2>&1; expected=$?
    run_with_timeout "$TEST_TIMEOUT" "$SHELL_BIN" -c "$command" >/dev/null 2>&1; actual=$?
    if [ "$expected" = "$actual" ]; then pass "$test_name"
    else fail "$test_name" "exit $expected" "exit $actual"; fi
}

# Helper: compare both output and exit code
compare_both() {
    test_name="$1"; command="$2"
    expected_out=$(run_with_timeout "$TEST_TIMEOUT" "$BASH_REF" -c "$command" 2>&1); expected_exit=$?
    actual_out=$(run_with_timeout "$TEST_TIMEOUT" "$SHELL_BIN" -c "$command" 2>&1); actual_exit=$?
    norm_expected=$(normalize_shell_name "$expected_out")
    norm_actual=$(normalize_shell_name "$actual_out")
    if [ "$norm_expected" = "$norm_actual" ] && [ "$expected_exit" = "$actual_exit" ]; then
        pass "$test_name"
    else
        fail "$test_name" "out='$expected_out' exit=$expected_exit" "out='$actual_out' exit=$actual_exit"
    fi
}

# Helper: check fortsh output matches expected string
check_output() {
    test_name="$1"; command="$2"; expected="$3"
    actual=$(timeout "$TEST_TIMEOUT" "$SHELL_BIN" -c "$command" 2>&1)
    if [ "$actual" = "$expected" ]; then pass "$test_name"
    else fail "$test_name" "$expected" "$actual"; fi
}

# Helper: check fortsh exit code matches expected
check_exit() {
    test_name="$1"; command="$2"; expected="$3"
    timeout "$TEST_TIMEOUT" "$SHELL_BIN" -c "$command" >/dev/null 2>&1; actual=$?
    if [ "$actual" = "$expected" ]; then pass "$test_name"
    else fail "$test_name" "exit $expected" "exit $actual"; fi
}

# Print summary — call at end of each test file
print_summary() {
    printf "\n==========================================\n"
    printf "%s TEST RESULTS\n" "$TEST_PREFIX"
    printf "==========================================\n"
    printf "Passed:  %d\n" "$PASSED"
    printf "Failed:  %d\n" "$FAILED"
    printf "Skipped: %d\n" "$SKIPPED"
    printf "Total:   %d\n" "$((PASSED + FAILED + SKIPPED))"
    printf "==========================================\n"

    if [ $((PASSED + FAILED)) -gt 0 ]; then
        PASS_RATE=$((PASSED * 100 / (PASSED + FAILED)))
        printf "Pass rate: %d%%\n" "$PASS_RATE"
    fi

    if [ "$FAILED" -gt 0 ]; then
        printf "\nFailed tests:\n"
        printf "%b" "$FAILED_TESTS_LIST"
        printf "==========================================\n"
    fi

    if [ "$FAILED" -eq 0 ]; then
        printf "ALL %s TESTS PASSED!\n" "$TEST_PREFIX"
        exit 0
    else
        exit 1
    fi
}