import base64

import httpx
import pytest
import respx

from mcp_docugen.main import build_app
from mcp_docugen.models import TemplateFrontmatter, TemplateVariable


def _openrouter_success(text: str = "# Generated"):
    return httpx.Response(
        200,
        json={
            "id": "g",
            "choices": [{"message": {"role": "assistant", "content": text}}],
            "model": "anthropic/claude-sonnet-4",
            "usage": {
                "prompt_tokens": 50,
                "completion_tokens": 100,
                "total_cost": 0.02,
            },
        },
    )


@pytest.fixture
async def app_env(tmp_path, monkeypatch):
    monkeypatch.setenv("API_KEY", "test-api-key")
    monkeypatch.setenv("OPENROUTER_API_KEY", "sk-or-test")
    monkeypatch.setenv("PUBLIC_BASE_URL", "https://mcp.test.com")
    monkeypatch.setenv("DATA_DIR", str(tmp_path))
    app = await build_app()
    return app, tmp_path


@respx.mock
async def test_full_generation_flow_records_and_serves_asset(app_env):
    app, tmp_path = app_env
    respx.post("https://openrouter.ai/api/v1/chat/completions").mock(
        return_value=_openrouter_success("# Done")
    )

    template_store = app.state.template_store
    renderer = app.state.renderer
    generation_store = app.state.generation_store

    fm = TemplateFrontmatter(
        name="report",
        description="x",
        required_variables=[TemplateVariable(name="foto", type="image")],
    )
    await template_store.create(
        name="report", frontmatter=fm, body="![foto]({{foto}})"
    )

    png = b"\x89PNG\r\n\x1a\n" + b"\x00" * 50
    img_var = {
        "kind": "image",
        "data_b64": base64.b64encode(png).decode(),
        "mime": "image/png",
    }

    result = await renderer.generate(
        template_name="report",
        content_md="content",
        variables={"foto": img_var},
        instructions=None,
    )

    assert result.markdown == "# Done"
    stats = await generation_store.get_stats()
    assert stats["success"] == 1

    gen_files = list((tmp_path / "generated" / result.generation_id).iterdir())
    assert len(gen_files) == 1