#!/usr/bin/env bash
# Smoke tests — classify (no LLM) + minimal chat
set -euo pipefail

STACK_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
cd "$STACK_DIR"

set -a
# shellcheck disable=SC1091
source .env 2>/dev/null || true
set +a

BASE="${TEST_BASE_URL:-http://127.0.0.1:8000}"
LITELLM_BASE="${TEST_LITELLM_URL:-http://127.0.0.1:4000}"
ROUTER_KEY="${ROUTER_API_KEY:?ROUTER_API_KEY required}"

CURL=(curl -sf)
if [[ "$BASE" == https:* ]]; then
  CURL+=( -k )
fi

echo "== Router health =="
"${CURL[@]}" "${BASE}/health" | jq .

echo "== Classify SIMPLE =="
"${CURL[@]}" "${BASE}/classify" \
  -H "Authorization: Bearer ${ROUTER_KEY}" \
  -H "Content-Type: application/json" \
  -d '{"messages":[{"role":"user","content":"Привет, что такое Docker?"}]}' | jq .

echo "== Classify COMPLEX =="
"${CURL[@]}" "${BASE}/classify" \
  -H "Authorization: Bearer ${ROUTER_KEY}" \
  -H "Content-Type: application/json" \
  -d '{"messages":[{"role":"user","content":"Спроектируй архитектуру microservices"}]}' | jq .

echo "== Chat smart-router max_tokens=16 =="
"${CURL[@]}" "${BASE}/v1/chat/completions" \
  -H "Authorization: Bearer ${ROUTER_KEY}" \
  -H "Content-Type: application/json" \
  -d '{"model":"smart-router","max_tokens":16,"messages":[{"role":"user","content":"bash docker service ls"}]}' \
  | jq '.choices[0].message.content, .x_router_meta // empty'

if [[ -n "${SKIP_LITELLM_SMOKE:-}" ]]; then
  echo "SKIP_LITELLM_SMOKE set — skipping LiteLLM checks"
else
  echo "== LiteLLM liveliness =="
  "${CURL[@]}" "${LITELLM_BASE}/health/liveliness" && echo
  echo "== LiteLLM UI =="
  curl -sfI -k "${LITELLM_BASE}/ui" | head -3 || true
fi

echo "All smoke checks passed."