#!/usr/bin/env bash
set -euo pipefail

root="$(cd "$(dirname "$0")/.." && pwd)"
golden="$root/tests/golden/typescript/typescript-fetch"

if [ ! -d "$golden" ]; then
  echo "error: golden dir not found: $golden" >&2
  exit 1
fi

toolchain="${1:-}"
if [ "$toolchain" != "tsc" ] && [ "$toolchain" != "vp" ]; then
  echo "usage: $0 <tsc|vp>" >&2
  exit 1
fi

total=0; passed=0; failed=0; fails=()

for test_dir in "$golden"/*/; do
  [ -f "$test_dir/tsconfig.json.golden" ] || continue

  # Select tests by toolchain: vp tests have vite.config.ts, tsc tests don't
  has_vp_config=0
  [ -f "$test_dir/vite.config.ts.golden" ] && has_vp_config=1

  if [ "$toolchain" = "vp" ] && [ "$has_vp_config" = "0" ]; then
    continue
  fi
  if [ "$toolchain" = "tsc" ] && [ "$has_vp_config" = "1" ]; then
    continue
  fi

  total=$((total + 1))
  name=$(basename "$test_dir")
  printf "[%d] %s... " "$total" "$name"

  tmp=$(mktemp -d)
  trap 'rm -rf "$tmp"' EXIT

  while IFS= read -r -d '' f; do
    rel="${f#"$test_dir"}"
    dst="$tmp/${rel%.golden}"
    mkdir -p "$(dirname "$dst")"
    cp "$f" "$dst"
  done < <(find "$test_dir" -type f -name '*.golden' -print0)

  log=$(mktemp)
  if [ "$toolchain" = "vp" ]; then
    if (cd "$tmp" && vp install && vp check --no-fmt && vp pack) >>"$log" 2>&1; then
      echo "ok"
      passed=$((passed + 1))
    else
      echo "FAIL"
      failed=$((failed + 1))
      fails+=("$name")
      sed 's/^/    /' "$log" || true
    fi
  else
    if (cd "$tmp" && tsc --noEmit) >>"$log" 2>&1; then
      echo "ok"
      passed=$((passed + 1))
    else
      echo "FAIL"
      failed=$((failed + 1))
      fails+=("$name")
      sed 's/^/    /' "$log" || true
    fi
  fi

  rm -rf "$tmp" "$log"
  trap - EXIT
done

echo
echo "TypeScript ($toolchain): $passed/$total passed"
if [ "$failed" -gt 0 ]; then
  echo "failed:"
  for n in "${fails[@]}"; do echo "  - $n"; done
  exit 1
fi