from unittest.mock import AsyncMock

import pytest

from mcp_docugen.generation_store import GenerationStore
from mcp_docugen.llm_client import LLMResponse
from mcp_docugen.mcp_tools import build_mcp_server
from mcp_docugen.renderer import Renderer
from mcp_docugen.template_store import TemplateStore


@pytest.fixture
async def mcp_env(tmp_path):
    template_store = TemplateStore(base_dir=tmp_path / "templates")
    generation_store = GenerationStore(
        db_path=tmp_path / "gen.db",
        generated_dir=tmp_path / "generated",
    )
    await generation_store.init()
    llm = AsyncMock()
    llm.chat.return_value = LLMResponse(
        text="# Out",
        model="m",
        tokens_in=5,
        tokens_out=10,
        cost_usd=0.01,
        latency_ms=50,
    )
    renderer = Renderer(
        template_store=template_store,
        generation_store=generation_store,
        llm=llm,
        public_base_url="https://mcp.example.com",
        default_model="m",
        asset_ttl_days=30,
        max_image_size_mb=10,
    )
    mcp = build_mcp_server(template_store, renderer)
    return mcp, template_store


async def test_template_create_and_get(mcp_env):
    mcp, _ = mcp_env
    await mcp.tools["template_create"](
        name="demo",
        frontmatter={"name": "demo", "description": "d"},
        body="body",
        assets=None,
    )
    got = await mcp.tools["template_get"](name="demo")
    assert got["name"] == "demo"
    assert got["body"] == "body"


async def test_template_create_duplicate_errors(mcp_env):
    mcp, _ = mcp_env
    await mcp.tools["template_create"](
        name="demo",
        frontmatter={"name": "demo", "description": "x"},
        body="b",
        assets=None,
    )
    with pytest.raises(Exception):
        await mcp.tools["template_create"](
            name="demo",
            frontmatter={"name": "demo", "description": "x"},
            body="b",
            assets=None,
        )


async def test_template_list_returns_summaries(mcp_env):
    mcp, _ = mcp_env
    await mcp.tools["template_create"](
        name="a",
        frontmatter={"name": "a", "description": "A"},
        body="b",
        assets=None,
    )
    await mcp.tools["template_create"](
        name="b",
        frontmatter={"name": "b", "description": "B"},
        body="b",
        assets=None,
    )
    result = await mcp.tools["template_list"]()
    names = sorted(t["name"] for t in result)
    assert names == ["a", "b"]


async def test_template_update(mcp_env):
    mcp, _ = mcp_env
    await mcp.tools["template_create"](
        name="u",
        frontmatter={"name": "u", "description": "d"},
        body="old",
        assets=None,
    )
    await mcp.tools["template_update"](
        name="u",
        frontmatter={"name": "u", "description": "new"},
        body="new",
        assets=None,
    )
    got = await mcp.tools["template_get"](name="u")
    assert got["body"] == "new"
    assert got["frontmatter"]["description"] == "new"


async def test_template_delete(mcp_env):
    mcp, _ = mcp_env
    await mcp.tools["template_create"](
        name="d",
        frontmatter={"name": "d", "description": "x"},
        body="b",
        assets=None,
    )
    await mcp.tools["template_delete"](name="d")
    with pytest.raises(Exception):
        await mcp.tools["template_get"](name="d")


async def test_document_generate_happy_path(mcp_env):
    mcp, _ = mcp_env
    await mcp.tools["template_create"](
        name="g",
        frontmatter={
            "name": "g",
            "description": "x",
            "required_variables": [{"name": "cliente", "type": "string"}],
        },
        body="Cliente: {{cliente}}",
        assets=None,
    )
    result = await mcp.tools["document_generate"](
        template_name="g",
        content_md="# Ordine",
        variables={"cliente": "ACME"},
        instructions=None,
    )
    assert result["markdown"] == "# Out"
    assert result["model"] == "m"
    assert result["tokens"]["input"] == 5
    assert "generation_id" in result