#!/usr/bin/env bash
# Verify that tests were actually executed and counts match expectations
# Usage: verify-test-execution.sh <test-specs-file> <test-output-file-or-string>

set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# Count expected tests from test-specs.md or .feature files
count_expected_tests() {
    local input_path="$1"

    if [[ -d "$input_path" ]]; then
        # Directory: count Scenario: lines across .feature files
        local count=0
        local f
        for f in "$input_path"/*.feature; do
            [[ -f "$f" ]] || continue
            local file_count
            file_count=$(grep -cE "^\s*(Scenario:|Scenario Outline:)" "$f" 2>/dev/null) || true
            count=$((count + ${file_count:-0}))
        done
        echo "$count"
        return
    fi

    if [[ ! -f "$input_path" ]]; then
        echo "0"
        return
    fi

    # Legacy: count TS-XXX patterns in test-specs.md
    local count
    count=$(grep -cE "^###[[:space:]]+TS-[0-9]+" "$input_path" 2>/dev/null) || true
    echo "${count:-0}"
}

# Parse test count from common test runner outputs
# Supports: Jest, Vitest, Pytest, Go test, Playwright, Mocha, Behave, Cucumber.js, Godog
parse_test_output() {
    local output="$1"
    local passed=0
    local failed=0
    local total=0

    # Behave: "X features passed, Y failed" or "X scenarios passed, Y failed, Z skipped"
    if echo "$output" | grep -qE "[0-9]+ scenario"; then
        passed=$(echo "$output" | grep -oE "([0-9]+) scenario.*passed" | grep -oE "[0-9]+" | head -1) || passed=0
        failed=$(echo "$output" | grep -oE "([0-9]+) scenario.*failed" | grep -oE "[0-9]+" | head -1) || failed=0
        if [[ "$passed" -eq 0 ]] && [[ "$failed" -eq 0 ]]; then
            # Alternative: "X scenarios passed"
            passed=$(echo "$output" | grep -oE "([0-9]+) scenarios? passed" | grep -oE "[0-9]+" | head -1) || passed=0
        fi
        total=$((passed + failed))
    # Cucumber.js: "X scenarios (Y passed)" or "X scenarios (Y passed, Z failed)"
    elif echo "$output" | grep -qE "[0-9]+ scenarios?"; then
        passed=$(echo "$output" | grep -oE "([0-9]+) passed" | grep -oE "[0-9]+" | tail -1) || passed=0
        failed=$(echo "$output" | grep -oE "([0-9]+) failed" | grep -oE "[0-9]+" | tail -1) || failed=0
        total=$(echo "$output" | grep -oE "([0-9]+) scenarios?" | grep -oE "[0-9]+" | head -1) || total=$((passed + failed))
    # Jest/Vitest: "Tests: X passed, Y failed, Z total"
    elif echo "$output" | grep -qE "Tests:.*passed"; then
        passed=$(echo "$output" | grep -oE "([0-9]+) passed" | grep -oE "[0-9]+" | tail -1) || passed=0
        failed=$(echo "$output" | grep -oE "([0-9]+) failed" | grep -oE "[0-9]+" | tail -1) || failed=0
        total=$((passed + failed))
    # Pytest/pytest-bdd: "X passed" or "X passed, Y failed"
    elif echo "$output" | grep -qE "[0-9]+ passed"; then
        passed=$(echo "$output" | grep -oE "([0-9]+) passed" | grep -oE "[0-9]+" | tail -1) || passed=0
        failed=$(echo "$output" | grep -oE "([0-9]+) failed" | grep -oE "[0-9]+" | tail -1) || failed=0
        total=$((passed + failed))
    # Go test/godog: "ok" or "FAIL" with test counts, or "--- PASS:" counts
    # Note: Use -- to prevent "--- PASS:" from being interpreted as options
    elif echo "$output" | grep -qE -- "(^ok[[:space:]]|^FAIL[[:space:]]|--- PASS:|--- FAIL:)"; then
        passed=$(echo "$output" | grep -cE -- "--- PASS:" 2>/dev/null) || true
        failed=$(echo "$output" | grep -cE -- "--- FAIL:" 2>/dev/null) || true
        passed=${passed:-0}
        failed=${failed:-0}
        total=$((passed + failed))
    # Playwright: "X passed" or "X failed"
    elif echo "$output" | grep -qE "[0-9]+ (passed|failed)"; then
        passed=$(echo "$output" | grep -oE "([0-9]+) passed" | grep -oE "[0-9]+" | tail -1) || passed=0
        failed=$(echo "$output" | grep -oE "([0-9]+) failed" | grep -oE "[0-9]+" | tail -1) || failed=0
        total=$((passed + failed))
    # Mocha: "X passing" "Y failing"
    elif echo "$output" | grep -qE "[0-9]+ passing"; then
        passed=$(echo "$output" | grep -oE "([0-9]+) passing" | grep -oE "[0-9]+" | tail -1) || passed=0
        failed=$(echo "$output" | grep -oE "([0-9]+) failing" | grep -oE "[0-9]+" | tail -1) || failed=0
        total=$((passed + failed))
    fi

    echo "{\"passed\": $passed, \"failed\": $failed, \"total\": $total}"
}

# Verify test execution against expectations
verify_execution() {
    local test_specs_file="$1"
    local test_output="$2"

    local expected
    expected=$(count_expected_tests "$test_specs_file")

    local results
    results=$(parse_test_output "$test_output")

    local actual_total
    actual_total=$(echo "$results" | grep -oE '"total": [0-9]+' | grep -oE '[0-9]+')

    local passed
    passed=$(echo "$results" | grep -oE '"passed": [0-9]+' | grep -oE '[0-9]+')

    local failed
    failed=$(echo "$results" | grep -oE '"failed": [0-9]+' | grep -oE '[0-9]+')

    # Determine status
    local status="UNKNOWN"
    local message=""

    if [[ "$actual_total" -eq 0 ]]; then
        status="NO_TESTS_RUN"
        message="Could not detect any test execution in output"
    elif [[ "$failed" -gt 0 ]]; then
        status="TESTS_FAILING"
        message="$failed tests failing - fix code before proceeding"
    elif [[ "$expected" -gt 0 ]] && [[ "$actual_total" -lt "$expected" ]]; then
        status="INCOMPLETE"
        message="Only $actual_total tests run, expected $expected from test-specs.md"
    elif [[ "$passed" -gt 0 ]] && [[ "$failed" -eq 0 ]]; then
        status="PASS"
        message="All $passed tests passing"
    fi

    cat <<EOF
{
    "status": "$status",
    "message": "$message",
    "expected": $expected,
    "actual": {
        "total": $actual_total,
        "passed": $passed,
        "failed": $failed
    }
}
EOF
}

# Main
case "${1:-help}" in
    count-expected)
        if [[ $# -lt 2 ]]; then
            echo "Usage: $0 count-expected <test-specs-file>"
            exit 1
        fi
        count_expected_tests "$2"
        ;;
    parse-output)
        if [[ $# -lt 2 ]]; then
            echo "Usage: $0 parse-output <test-output-string>"
            exit 1
        fi
        parse_test_output "$2"
        ;;
    verify)
        if [[ $# -lt 3 ]]; then
            echo "Usage: $0 verify <test-specs-file> <test-output-string>"
            exit 1
        fi
        verify_execution "$2" "$3"
        ;;
    help|*)
        echo "Test Execution Verification"
        echo ""
        echo "Commands:"
        echo "  count-expected <test-specs-file>     Count TS-XXX entries in test specs"
        echo "  parse-output <output-string>         Parse test runner output for counts"
        echo "  verify <specs-file> <output>         Compare expected vs actual"
        echo ""
        echo "Supported test runners: Jest, Vitest, Pytest, Go test, Playwright, Mocha"
        ;;
esac