#!/usr/bin/env bash
# D1 smoke — proves the Day 1 acceptance gate end-to-end.
# Builds all 5 binaries, launches them, polls /health on each until
# ready, then runs the actual probes. Exits 0 on success.
#
# Per Opus + Qwen BLOCK #2 review: replaced the prior `sleep 0.5`
# liveness gate with a poll loop so cold-start CI boxes don't race
# the bind.
#
# Usage:  ./scripts/d1_smoke.sh

set -euo pipefail
cd "$(dirname "$0")/.."

export PATH="$PATH:/usr/local/go/bin"

echo "[d1-smoke] building..."
go build -o bin/ ./cmd/...

PIDS=()
trap 'echo "[d1-smoke] cleanup"; kill ${PIDS[@]} 2>/dev/null || true; wait 2>/dev/null || true' EXIT INT TERM

echo "[d1-smoke] launching..."
for SVC in gateway storaged catalogd ingestd queryd; do
  ./bin/$SVC > /tmp/${SVC}.log 2>&1 &
  PIDS+=($!)
done

# Poll /health on each service until it returns 200 or we hit the
# 5s budget. Cheaper than a fixed sleep AND deterministic — first
# bind error surfaces immediately, slow boxes wait as long as needed.
poll_health() {
  local name="$1" port="$2" deadline=$(($(date +%s) + 5))
  while [ "$(date +%s)" -lt "$deadline" ]; do
    if curl -sS --max-time 1 "http://127.0.0.1:$port/health" >/dev/null 2>&1; then
      return 0
    fi
    sleep 0.05
  done
  echo "  [d1-smoke] $name (:$port) failed to bind within 5s — log:"
  tail -5 "/tmp/${name}.log" | sed 's/^/    /'
  return 1
}

echo "[d1-smoke] waiting for /health (poll up to 5s/svc)..."
for SPEC in "gateway:3110" "storaged:3211" "catalogd:3212" "ingestd:3213" "queryd:3214"; do
  NAME="${SPEC%:*}"; PORT="${SPEC#*:}"
  if ! poll_health "$NAME" "$PORT"; then
    exit 1
  fi
done

echo "[d1-smoke] /health probes:"
FAILED=0
for SPEC in "gateway:3110" "storaged:3211" "catalogd:3212" "ingestd:3213" "queryd:3214"; do
  NAME="${SPEC%:*}"; PORT="${SPEC#*:}"
  RESP="$(curl -sS --max-time 2 "http://127.0.0.1:$PORT/health" || echo FAIL)"
  if echo "$RESP" | grep -q "\"service\":\"$NAME\""; then
    echo "  ✓ $NAME (:$PORT) → $RESP"
  else
    echo "  ✗ $NAME (:$PORT) → $RESP"
    FAILED=1
  fi
done

# Single curl with -i grabs both code + headers in one pass, per Opus
# WARN #6 — was 2 calls per route, doubling load + creating window.
echo "[d1-smoke] gateway 501 stub probes:"
for ROUTE in /v1/ingest /v1/sql; do
  RESP="$(curl -sS -i --max-time 2 -X POST "http://127.0.0.1:3110$ROUTE")"
  # Per Opus 2nd-pass WARN: head -1 fails on 1xx interim lines.
  # awk picks the LAST HTTP/* status line — robust to 100 Continue.
  CODE="$(echo "$RESP" | awk '/^HTTP\//{code=$2} END{print code}')"
  HDR="$(echo "$RESP" | grep -i 'X-Lakehouse-Stub' || true)"
  if [ "$CODE" = "501" ] && [ -n "$HDR" ]; then
    echo "  ✓ POST $ROUTE → 501 + $HDR"
  else
    echo "  ✗ POST $ROUTE → code=$CODE hdr=$HDR"
    FAILED=1
  fi
done

if [ "$FAILED" -ne 0 ]; then
  echo "[d1-smoke] FAILED"
  exit 1
fi
echo "[d1-smoke] D1 acceptance gate: PASSED"