from __future__ import annotations

import pytest

from conftest import import_any, install_fake_adapter, maybe_import, text_blob, unload_promptboost_modules


API_MODULES = (
    "promptboost.api",
    "promptboost.app",
    "promptboost.main",
    "promptboost.server",
)


def _fake_boost(*_args, **_kwargs):
    return {
        "run_id": "test-run",
        "boosted_prompt": "API boosted prompt without paid API calls.",
        "target_harness": _kwargs.get("target_harness") or _kwargs.get("harness") or "codex",
        "target_model": _kwargs.get("target_model") or _kwargs.get("model") or "test-model",
    }


def _patch_api_boost(monkeypatch):
    for module_name in (
        "promptboost",
        "promptboost.core",
        "promptboost.boost",
        "promptboost.api",
        "promptboost.app",
        "promptboost.main",
        "promptboost.server",
    ):
        module = maybe_import(module_name)
        if module is not None:
            monkeypatch.setattr(module, "boost", _fake_boost, raising=False)


def _load_app(monkeypatch):
    api_module = import_any(*API_MODULES)
    _patch_api_boost(monkeypatch)

    create_app = getattr(api_module, "create_app", None)
    if callable(create_app):
        return create_app()

    app = getattr(api_module, "app", None)
    if app is None:
        pytest.fail(
            "FastAPI module must expose either app or create_app() for route tests."
        )
    return app


def _request_first_existing(client, method: str, paths: list[str], **kwargs):
    for path in paths:
        response = getattr(client, method)(path, **kwargs)
        if response.status_code != 404:
            return path, response
    pytest.fail(f"None of the expected API routes exist: {', '.join(paths)}")


def test_api_health_route(promptboost_db_path, monkeypatch):
    testclient = pytest.importorskip("fastapi.testclient")
    unload_promptboost_modules()
    install_fake_adapter(monkeypatch, "API boosted prompt without paid API calls.")
    app = _load_app(monkeypatch)
    client = testclient.TestClient(app)

    path, response = _request_first_existing(
        client,
        "get",
        ["/health", "/healthz", "/api/health", "/"],
    )

    assert response.status_code == 200, path
    assert any(token in text_blob(response.json()) for token in ("ok", "healthy", "promptboost"))


def test_api_boost_route_returns_boosted_prompt(promptboost_db_path, monkeypatch):
    testclient = pytest.importorskip("fastapi.testclient")
    unload_promptboost_modules()
    install_fake_adapter(monkeypatch, "API boosted prompt without paid API calls.")
    app = _load_app(monkeypatch)
    client = testclient.TestClient(app)

    _path, response = _request_first_existing(
        client,
        "post",
        ["/boost", "/api/boost", "/v1/boost"],
        json={
            "raw_prompt": "Improve this API prompt without calling paid services.",
            "target_harness": "codex",
            "target_model": "test-model",
        },
    )

    assert response.status_code in (200, 201)
    assert "api boosted prompt without paid api calls" in text_blob(response.json())