import asyncio
import base64
import json
from pathlib import Path

from a2a_pack.auth import PlatformUserAuth
from a2a_pack.cli.local import load_local_project
from a2a_pack.context import LocalRunContext

from agent import QuoteJudgeStudioV1, _LOCAL_COMPARISONS, _LOCAL_RECEIPTS


ROOT = Path(__file__).resolve().parents[1]
AUTH = PlatformUserAuth(sub="user-123", user_id=123, email="buyer@example.com")


def run(coro):
    return asyncio.run(coro)


def ctx():
    return LocalRunContext(auth=AUTH, task_id="test-task", caller="pytest")


def test_full_stack_product_contract():
    manifest = (ROOT / "a2a.yaml").read_text(encoding="utf-8")
    source = (ROOT / "agent.py").read_text(encoding="utf-8")
    frontend = (ROOT / "frontend" / "src" / "App.jsx").read_text(encoding="utf-8")
    browser_client = (ROOT / "frontend" / "src" / "a2a.js").read_text(encoding="utf-8")

    assert "frontend:" in manifest and "mount: /app" in manifest
    assert "public: false" in manifest
    assert "resources:" in manifest and "databases:" in manifest
    assert "migrations:" in manifest and "db/migrations" in manifest
    assert "runtime:" in manifest and "max_runtime_seconds: 600" in manifest
    assert "PlatformUserAuth" in source
    assert "AgentPlatformResources" in source and "AgentDatabase(" in source
    assert "LLMProvisioning" not in source and "ctx.llm" not in source
    assert "Skill runner" not in frontend
    assert "QuoteJudge Studio" in frontend and "compare_quotes" in frontend
    assert "authorizeUrl" in browser_client and "loginUrl" in browser_client
    forbidden = ["DATABASE_URL", "A2A_LITELLM", "OPENAI_API_KEY", ".svc.cluster.local"]
    assert not any(secret in frontend for secret in forbidden)

    migrations = list((ROOT / "db" / "migrations").glob("*.sql"))
    assert migrations and all(path.read_text(encoding="utf-8").strip() for path in migrations)

    card = load_local_project(ROOT).agent_cls().card()
    skill_names = {skill.name for skill in card.skills}
    assert {"compare_quotes", "get_comparison", "compare_quotes_from_browser_upload", "compare_quotes_from_file"} <= skill_names
    databases = card.runtime.platform_resources.databases
    assert databases, "live Agent Card must declare its managed database"
    assert databases[0].scope == "user"
    assert databases[0].migrations is not None
    assert databases[0].migrations.path == "db/migrations"
    assert card.runtime.pricing.caller_pays_llm is False


def test_success_failure_reload_and_receipt_contract():
    _LOCAL_COMPARISONS.clear()
    _LOCAL_RECEIPTS.clear()
    agent = QuoteJudgeStudioV1()
    quotes = [
        {"vendor": "Acme Bearings", "quantity": 500, "unit_price": 10.75, "delivery_days": 12, "warranty_months": 12},
        {"vendor": "Beta Industrial", "quantity": 500, "unit_price": 9.1, "delivery_days": 8, "warranty_months": 18},
    ]
    weights = {"price": 50, "delivery": 30, "warranty": 20}

    success = run(agent.local_invoke("compare_quotes", auth=AUTH, comparison_id="studio-quote-v1", quotes=quotes, weights=weights))
    assert success["ok"] is True
    assert success["comparison_id"] == "studio-quote-v1"
    assert success["recommendation"]["vendor"] == "Beta Industrial"
    assert success["receipt"]["receipt_id"].startswith("qjr_")
    assert success["receipt"]["persisted"] is True
    assert _LOCAL_RECEIPTS and _LOCAL_RECEIPTS[-1]["schema"] == "quotejudge.production_receipt.v1"
    assert "tenant_key" not in json.dumps(success)

    reload = run(agent.local_invoke("get_comparison", auth=AUTH, comparison_id="studio-quote-v1"))
    assert reload["ok"] is True
    assert reload["comparison_id"] == "studio-quote-v1"
    assert reload["recommendation"]["vendor"] == "Beta Industrial"
    assert reload["reloaded"] is True

    failure = run(agent.local_invoke(
        "compare_quotes",
        auth=AUTH,
        comparison_id="studio-quote-invalid",
        quotes=[{"vendor": "Only Vendor", "quantity": 1, "unit_price": 10, "delivery_days": 5, "warranty_months": 6}],
        weights=weights,
    ))
    assert failure == {"ok": False, "code": "at_least_two_quotes_required", "comparison_id": "studio-quote-invalid"}


def test_browser_upload_bridge_success_and_invalid_base64():
    _LOCAL_COMPARISONS.clear()
    agent = QuoteJudgeStudioV1()
    payload = base64.b64encode(json.dumps({
        "quotes": [
            {"vendor": "Acme Bearings", "quantity": 500, "unit_price": 10.75, "delivery_days": 12, "warranty_months": 12},
            {"vendor": "Beta Industrial", "quantity": 500, "unit_price": 9.1, "delivery_days": 8, "warranty_months": 18},
        ]
    }).encode()).decode()

    result = run(agent.local_invoke(
        "compare_quotes_from_browser_upload",
        auth=AUTH,
        comparison_id="upload-quote-v1",
        documents=[{"filename": "quotes.json", "media_type": "application/json", "data_base64": payload}],
        weights={"price": 50, "delivery": 30, "warranty": 20},
    ))
    assert result["ok"] is True
    assert result["recommendation"]["vendor"] == "Beta Industrial"
    assert result["source"] == "browser_upload"
    assert result["parsed_documents"] == [{"filename": "quotes.json", "quotes": 2}]

    invalid = run(agent.local_invoke(
        "compare_quotes_from_browser_upload",
        auth=AUTH,
        comparison_id="upload-bad",
        documents=[{"filename": "quotes.json", "media_type": "application/json", "data_base64": "not base64 ***"}],
        weights={"price": 50, "delivery": 30, "warranty": 20},
    ))
    assert invalid["ok"] is False
    assert invalid["code"] == "invalid_base64"


def test_mcp_tool_schemas_are_bounded_and_typed():
    card = load_local_project(ROOT).agent_cls().card()
    skills = {skill.name: skill for skill in card.skills}
    schema = skills["compare_quotes"].input_schema
    assert schema["type"] == "object"
    assert schema["additionalProperties"] is False
    assert set(schema["required"]) == {"comparison_id", "quotes", "weights"}
    assert schema["properties"]["comparison_id"]["type"] == "string"
    assert "quotes" in schema["properties"]
    assert skills["compare_quotes_from_file"].input_schema["properties"]["document"]["x-a2a-file-upload"]["max_bytes"] == 64 * 1024