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

# Smoke test deployed examples.
#
# Usage:
#   ./scripts/smoke-test.sh                          # test production URLs
#   ./scripts/smoke-test.sh --preview pr-42           # test PR preview URLs
#
# Checks every deployed example returns HTTP 200 with HTML content.
# Exits non-zero if any check fails.

DOMAIN="vinext.workers.dev"
PREVIEW_ALIAS=""

while [[ $# -gt 0 ]]; do
  case "$1" in
    --preview)
      if [[ -z "${2:-}" ]]; then
        echo "Error: --preview requires an argument (e.g. --preview pr-42)" >&2
        exit 1
      fi
      PREVIEW_ALIAS="$2"
      shift 2
      ;;
    *)
      echo "Unknown option: $1" >&2
      exit 1
      ;;
  esac
done

# Each entry: worker-name path expected-text
# expected-text is a simple string that must appear in the response body.
CHECKS=(
  "app-router-cloudflare         /       vinext"
  "pages-router-cloudflare       /       vinext"
  "app-router-playground         /       Playground"
  "realworld-api-rest            /       vinext"
  "nextra-docs-template          /       Introduction"
  "nextra-docs-template          /about  About"
  "benchmarks                    /       Benchmark"
  "hackernews                    /       Hacker News"
  "vinext-web                    /       Run your Next.js app on Vite"
)

# Content-correctness checks for dynamic routes.
# Verify that pages with dynamic params render the RIGHT data — not
# stale/cached data from a different param value.
# Format: "worker-name  path  must-contain  must-not-contain"
# Both strings are matched case-insensitively against the response body.
CONTENT_CHECKS=(
  # Nested layouts: each section must show its own products
  'app-router-playground  /layouts/electronics  alt="Phone"       alt="Basketball"'
  'app-router-playground  /layouts/sports       alt="Basketball"  alt="Phone"'
  'app-router-playground  /layouts/clothing     alt="Top"         alt="Phone"'
  # Route groups: subcategory pages must show the correct items
  'app-router-playground  /route-groups/clothing/shoes  alt="Shoes"  alt="Shorts"'
)

tmpfile=$(mktemp)
trap "rm -f '$tmpfile'" EXIT

passed=0
failed=0
errors=()

for check in "${CHECKS[@]}"; do
  read -r worker path expected <<< "$check"

  if [[ -n "$PREVIEW_ALIAS" ]]; then
    url="https://${PREVIEW_ALIAS}-${worker}.${DOMAIN}${path}"
  else
    url="https://${worker}.${DOMAIN}${path}"
  fi

  # Fetch with a 10s timeout, follow redirects
  status=$(curl -s -o "$tmpfile" -w "%{http_code}" -L --max-time 10 "$url" 2>/dev/null || echo "000")
  body=$(cat "$tmpfile" 2>/dev/null || echo "")

  if [[ "$status" != "200" ]]; then
    echo "FAIL  ${worker}${path}  (HTTP ${status})"
    errors+=("${worker}${path} returned HTTP ${status}")
    failed=$((failed + 1))
    continue
  fi

  if [[ -n "$expected" ]] && ! echo "$body" | grep -qiF "$expected"; then
    echo "FAIL  ${worker}${path}  (missing '${expected}' in body)"
    errors+=("${worker}${path} missing expected text '${expected}'")
    failed=$((failed + 1))
    continue
  fi

  echo "  OK  ${worker}${path}"
  passed=$((passed + 1))
done

# ---------------------------------------------------------------------------
# Content-correctness checks: right data for the right dynamic route
# ---------------------------------------------------------------------------

for check in "${CONTENT_CHECKS[@]}"; do
  read -r worker path must_contain must_not_contain <<< "$check"

  if [[ -n "$PREVIEW_ALIAS" ]]; then
    url="https://${PREVIEW_ALIAS}-${worker}.${DOMAIN}${path}"
  else
    url="https://${worker}.${DOMAIN}${path}"
  fi

  status=$(curl -s -o "$tmpfile" -w "%{http_code}" -L --max-time 10 "$url" 2>/dev/null || echo "000")
  body=$(cat "$tmpfile" 2>/dev/null || echo "")

  if [[ "$status" != "200" ]]; then
    echo "FAIL  ${worker}${path}  (HTTP ${status})"
    errors+=("${worker}${path} returned HTTP ${status}")
    failed=$((failed + 1))
    continue
  fi

  if ! echo "$body" | grep -qiF "$must_contain"; then
    echo "FAIL  ${worker}${path}  (missing '${must_contain}')"
    errors+=("${worker}${path} missing '${must_contain}' — wrong content rendered")
    failed=$((failed + 1))
    continue
  fi

  if echo "$body" | grep -qiF "$must_not_contain"; then
    echo "FAIL  ${worker}${path}  (found '${must_not_contain}' — wrong section data)"
    errors+=("${worker}${path} contains '${must_not_contain}' — data from wrong dynamic param")
    failed=$((failed + 1))
    continue
  fi

  echo "  OK  ${worker}${path}  (content correct)"
  passed=$((passed + 1))
done

echo ""
echo "${passed} passed, ${failed} failed"

if [[ $failed -gt 0 ]]; then
  echo ""
  echo "Failures:"
  for err in "${errors[@]}"; do
    echo "  - $err"
  done
  exit 1
fi