from app.config.ai_prompt_registry import (
    PROMPT_REGISTRY_MAP,
    PROMPT_FAMILY_ALLOWED,
    get_prompt_registry,
    get_prompt_registry_entry,
    iter_prompt_registry_keys,
)
from app.config.ai_prompts import (
    PROMPT_DECOMPOSE_TASKS,
    PROMPT_FEISHU_PROJECT_SIGNAL_CLASSIFICATION,
    PROMPT_FEISHU_REQUIREMENT_ANALYSIS,
    PROMPT_TASK_BREAKDOWN,
)
from app.config.ai_prompts_catalog import __all__ as PROMPT_CATALOG_EXPORTS


def test_prompt_registry_covers_all_catalog_exports() -> None:
    catalog_keys = set(PROMPT_CATALOG_EXPORTS)
    registry_keys = set(iter_prompt_registry_keys())

    assert registry_keys == catalog_keys
    assert set(PROMPT_REGISTRY_MAP) == catalog_keys

    for key in catalog_keys:
        entry = get_prompt_registry_entry(key)
        assert entry is not None
        assert entry["prompt_key"] == key
        assert entry["prompt_family"] in PROMPT_FAMILY_ALLOWED
        assert entry["responsibility_scope"]
        assert entry["priority"] in {"high", "medium", "low"}
        assert entry["audience"]


def test_prompt_registry_returns_detached_copies() -> None:
    registry = get_prompt_registry()
    original_key = registry[0]["prompt_key"]

    registry[0]["prompt_key"] = "BROKEN_KEY"

    assert PROMPT_REGISTRY_MAP[original_key]["prompt_key"] == original_key


def test_high_frequency_registry_entries_have_complete_metadata() -> None:
    expected_families = {
        "PROMPT_FEISHU_PROJECT_SIGNAL_CLASSIFICATION": "classification",
        "PROMPT_FEISHU_LEVEL_CLASSIFICATION": "classification",
        "PROMPT_FEISHU_SIGNAL_EXTRACTION": "extraction",
        "PROMPT_FEISHU_TASK_ORCHESTRATION": "planning",
        "PROMPT_SIGNAL_PLANNER": "planning",
        "PROMPT_GENERATE_CHARTER": "planning",
        "PROMPT_DECOMPOSE_TASKS": "planning",
        "PROMPT_PROJECT_STATUS_QA": "response",
        "PROMPT_MEMBER_WORKLOAD_QA": "response",
        "PROMPT_AI_PMO_QA": "response",
        "PROMPT_EXTERNAL_TREND_QA": "response",
        "PROMPT_INDUSTRY_TREND_QA": "response",
    }

    for prompt_key, family in expected_families.items():
        entry = get_prompt_registry_entry(prompt_key)
        assert entry is not None
        assert entry["module"]
        assert entry["business_scenario"]
        assert entry["input_contract"]
        assert entry["output_contract"]
        assert isinstance(entry["call_sites"], tuple)
        assert entry["call_sites"], prompt_key
        assert entry["prompt_family"] == family
        assert entry["responsibility_scope"]

    signal_planner_entry = get_prompt_registry_entry("PROMPT_SIGNAL_PLANNER")
    assert signal_planner_entry is not None
    assert "charter_json" in signal_planner_entry["input_contract"]
    assert any("mixin_prompt.py" in site for site in signal_planner_entry["call_sites"])
    assert any("mixin_analyze.py" in site for site in signal_planner_entry["call_sites"])
    assert "规划" in signal_planner_entry["responsibility_scope"]


def test_deprecated_prompt_aliases_point_to_canonical_prompts() -> None:
    registry_keys = set(iter_prompt_registry_keys())

    assert "PROMPT_FEISHU_REQUIREMENT_ANALYSIS" not in registry_keys
    assert "PROMPT_TASK_BREAKDOWN" not in registry_keys
    assert PROMPT_FEISHU_REQUIREMENT_ANALYSIS == PROMPT_FEISHU_PROJECT_SIGNAL_CLASSIFICATION
    assert PROMPT_TASK_BREAKDOWN == PROMPT_DECOMPOSE_TASKS