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

ROOT_OVERRIDE="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
export ROOT_OVERRIDE
# shellcheck disable=SC1091
source "${ROOT_OVERRIDE}/scripts/common.sh"

init_log selftest-native
print_header "selftest-native: START"

need_cmd curl
need_cmd python
need_cmd ollama
need_cmd grep
load_env
check_for_legacy_docker_ollama

[[ -x "$SQLAI_BIN" ]] || fail "bin/sqlai missing or not executable at ${SQLAI_BIN}" 60

log "selftest-native: waiting for Ollama API at ${OLLAMA_URL}"
wait_for_ollama_api "$OLLAMA_URL" 120 || fail "Ollama API not reachable at ${OLLAMA_URL}" 61
ok "selftest-native: Ollama API reachable"

tags="$(fetch_tags)"
[[ -n "$tags" ]] || fail "/api/tags returned empty body" 62
printf '%s' "$tags" | is_json || fail "/api/tags did not return valid JSON" 63
ok "selftest-native: /api/tags returned valid JSON"

if model_in_tags "$EXPERT_MODEL" "$tags"; then
  ok "selftest-native: expert model present: ${EXPERT_MODEL}"
else
  fail "expert model missing: ${EXPERT_MODEL}" 64
fi

if [[ -n "${BASE_MODEL}" ]]; then
  if model_in_tags "$BASE_MODEL" "$tags"; then
    ok "selftest-native: base model present: ${BASE_MODEL}"
  else
    fail "base model missing: ${BASE_MODEL}" 65
  fi
fi

log "selftest-native: warmup request via sqlai"
set +e
"$SQLAI_BIN" ask --text "Warmup: reply with exactly 'OK'." --no-metrics
rc=$?
set -e
[[ "$rc" -eq 0 ]] || warn "warmup failed rc=${rc}"

log "selftest-native: real query via sqlai"
set +e
out="$("$SQLAI_BIN" ask --text "Give a concise checklist to troubleshoot parameter sniffing in SQL Server 2022. Keep it technical." --no-metrics 2>&1)"
rc=$?
set -e
[[ "$rc" -eq 0 ]] || fail "real query failed rc=${rc}" 66
[[ -n "${out//[[:space:]]/}" ]] || fail "real query returned empty output" 67
ok "selftest-native: real query returned non-empty output"

print_footer "selftest-native: END status=OK log=${LOG_FILE}"