"""
End-to-end smoke test for the production API.

Usage:
    python scripts/smoke_test.py                          # test local (localhost:7860)
    python scripts/smoke_test.py https://my-space.hf.space  # test remote
"""

from __future__ import annotations

import json
import sys
import time

import requests


def smoke_test(base_url: str) -> bool:
    """Run smoke tests against the API. Returns True if all pass."""
    base = base_url.rstrip("/")
    passed = 0
    failed = 0
    results = []

    def check(name: str, method: str, path: str, expected_keys=None, body=None):
        nonlocal passed, failed
        url = f"{base}{path}"
        try:
            if method == "GET":
                resp = requests.get(url, timeout=15)
            else:
                resp = requests.post(url, json=body, timeout=30)

            if resp.status_code == 200:
                data = resp.json()
                if expected_keys:
                    missing = [k for k in expected_keys if k not in data]
                    if missing:
                        results.append(f"  WARN  {name}: missing keys {missing}")
                        passed += 1  # partial pass
                        return
                results.append(f"  PASS  {name}")
                passed += 1
            else:
                results.append(f"  FAIL  {name}: HTTP {resp.status_code} — {resp.text[:100]}")
                failed += 1
        except Exception as exc:
            results.append(f"  FAIL  {name}: {exc}")
            failed += 1

    print(f"\nSmoke testing: {base}\n{'='*60}")

    check("Health check",
          "GET", "/api/health",
          expected_keys=["status", "uptime_seconds"])

    check("Weather current",
          "GET", "/api/weather/current",
          expected_keys=["air_temperature_c", "ghi_w_m2"])

    check("Weather history",
          "GET", "/api/weather/history")

    check("Sensor snapshot",
          "GET", "/api/sensors/snapshot")

    check("Energy current",
          "GET", "/api/energy/current")

    check("Energy daily",
          "GET", "/api/energy/daily/2026-03-17")

    check("Photosynthesis current",
          "GET", "/api/photosynthesis/current")

    check("Control status",
          "GET", "/api/control/status")

    check("Control plan",
          "GET", "/api/control/plan")

    check("Biology rules",
          "GET", "/api/biology/rules",
          expected_keys=["rules"])

    check("Biology phenology",
          "GET", "/api/biology/phenology")

    check("Chatbot message",
          "POST", "/api/chatbot/message",
          expected_keys=["message"],
          body={"message": "What is the current temperature?", "session_id": "smoke-test"})

    # Print results
    print()
    for r in results:
        print(r)
    print(f"\n{'='*60}")
    print(f"Results: {passed} passed, {failed} failed out of {passed + failed}")

    return failed == 0


if __name__ == "__main__":
    url = sys.argv[1] if len(sys.argv) > 1 else "http://localhost:7860"
    ok = smoke_test(url)
    sys.exit(0 if ok else 1)