#!/usr/bin/env bash
# Verify the Atlas Stack is actually serving.
set -uo pipefail

pass=0; fail=0

check() {
  local name="$1" url="$2"
  local code
  code=$(curl -s -o /dev/null -w '%{http_code}' --max-time 8 "$url" || echo 000)
  if [ "$code" != "000" ] && [ "$code" -lt 500 ]; then
    printf '  %-10s OK   (%s)\n' "$name" "$code"; pass=$((pass+1))
  else
    printf '  %-10s FAIL (%s) %s\n' "$name" "$code" "$url"; fail=$((fail+1))
  fi
}

echo "Checking services..."
check ollama     "http://localhost:11434/"
check tgi        "http://localhost:8080/"
check vllm       "http://localhost:8000/"
check localai    "http://localhost:8081/"
check qdrant     "http://localhost:6333/"
check chroma     "http://localhost:8002/"
check milvus     "http://localhost:19530/"

echo
echo "passed=$pass failed=$fail"
[ "$fail" -eq 0 ] || echo "some services are not up - see RUNBOOK.md"
exit 0