import importlib.util
import json
import runpy
import sys
from pathlib import Path
from types import ModuleType

import pytest


REPO_ROOT = Path(__file__).resolve().parents[1]
SPEC_CHECKER_PATH = (
    REPO_ROOT / "skills" / "gaokao-spec-checker" / "scripts" / "spec_checker_v2.py"
)
VISUAL_REPORT_PATH = (
    REPO_ROOT
    / "skills"
    / "gaokao-college-advisor"
    / "scripts"
    / "gaokao_visual_report.py"
)


def _load_module(name: str, path: Path) -> ModuleType:
    spec = importlib.util.spec_from_file_location(name, path)
    assert spec is not None and spec.loader is not None
    module = importlib.util.module_from_spec(spec)
    spec.loader.exec_module(module)
    return module


spec_checker = _load_module("spec_checker_v2_coverage", SPEC_CHECKER_PATH)
visual_report = _load_module("gaokao_visual_report_coverage", VISUAL_REPORT_PATH)


GaokaoSpecCheckerV2 = spec_checker.GaokaoSpecCheckerV2


def test_detect_province_alias_branch():
    assert spec_checker.detect_province("湘考生，578分，志愿方案") == "湖南"


def test_detect_province_full_name_branch(monkeypatch):
    trimmed_rules = dict(spec_checker.PROVINCE_RULES)
    trimmed_rules.pop("广东")
    monkeypatch.setattr(spec_checker, "PROVINCE_RULES", trimmed_rules)
    assert spec_checker.detect_province("广东省考生，630分，志愿方案") == "广东"


def test_report_unsupported_province_branch():
    checker = GaokaoSpecCheckerV2(province="火星")
    report = checker.auto_detect_and_check("火星考生志愿方案")
    assert "暂不支持 火星" in report


def test_checker_hits_core_warning_and_validation_branches():
    checker = GaokaoSpecCheckerV2(province="湖南")
    report = checker.auto_detect_and_check(
        "\n".join([
            "湖南2026高考志愿填报方案",
            "本次共填报20个院校专业组志愿。",
            "每组最多6个专业。",
            "全部专业服从调剂。",
            "会计专业普遍要求物化生。",
            "88%录取概率。",
            "2026年位次预计5000名。",
            "请关注退档风险。",
        ])
    )
    assert "调剂范围错误（湖南）" in report
    assert "主观概率估算" in report
    assert "2026年位次" in report
    assert "选科要求一刀切" in report


def test_checker_hits_professional_school_mode_error():
    checker = GaokaoSpecCheckerV2(province="浙江")
    report = checker.auto_detect_and_check(
        "\n".join([
            "浙江省630分志愿方案",
            "本次共填报80个专业+学校志愿。",
            "每个志愿填1个专业。",
            "仍保留专业组概念，并支持组内调剂。",
            "风险提示：注意退档风险。",
        ])
    )
    assert "模式错误（浙江）" in report


def test_checker_can_generate_clean_report():
    checker = GaokaoSpecCheckerV2(province="浙江")
    report = checker.auto_detect_and_check(
        "\n".join([
            "浙江省630分志愿方案",
            "本次共填报80个专业+学校志愿。",
            "每个志愿填1个专业。",
            "无调剂。",
            "风险提示：注意退档与体检限制。",
        ])
    )
    assert "方案基本合规" in report
    assert "问题总数：0 个" in report


def test_checker_hits_volunteer_overflow_branch():
    checker = GaokaoSpecCheckerV2(province="湖南")
    report = checker.auto_detect_and_check(
        "\n".join([
            "湖南2026高考志愿填报方案",
            "本次共填报46个院校专业组志愿。",
            "每组最多6个专业。",
            "组内专业服从调剂。",
            "风险提示：注意退档风险。",
        ])
    )
    assert "志愿数量超标（湖南）" in report


def test_spec_checker_main_without_args_runs_demo_output(capsys, monkeypatch):
    monkeypatch.setattr(sys, "argv", [str(SPEC_CHECKER_PATH)])
    runpy.run_path(str(SPEC_CHECKER_PATH), run_name="__main__")
    output = capsys.readouterr().out
    assert "测试1：湖南方案（错误版）" in output
    assert "测试2：浙江方案（专业+学校模式）" in output
    assert "测试3：山东方案" in output


def test_spec_checker_main_with_file_argument_outputs_report(
    tmp_path, capsys, monkeypatch
):
    plan_path = tmp_path / "plan.txt"
    plan_path.write_text(
        "湖南考生，578分，志愿方案\n每组最多6个专业。\n风险提示：注意退档风险。",
        encoding="utf-8",
    )
    monkeypatch.setattr(sys, "argv", [str(SPEC_CHECKER_PATH), str(plan_path), "湖南"])
    runpy.run_path(str(SPEC_CHECKER_PATH), run_name="__main__")
    output = capsys.readouterr().out
    assert "志愿方案规范检查报告" in output
    assert "检测省份：湖南" in output


def test_visual_report_usage_message(capsys, monkeypatch):
    monkeypatch.setattr(visual_report.sys, "argv", ["gaokao_visual_report.py"])
    with pytest.raises(SystemExit) as exc:
        visual_report.main()
    assert exc.value.code == 1
    output = capsys.readouterr().out
    assert "用法: python3 gaokao_visual_report.py <student_data.json>" in output
    assert "生成示例" in output


def test_visual_report_demo_mode(capsys, monkeypatch):
    monkeypatch.setattr(
        visual_report,
        "generate_visual_report",
        lambda student, volunteers, output_format="all": ["/tmp/demo.md"],
    )
    monkeypatch.setattr(
        visual_report.sys, "argv", ["gaokao_visual_report.py", "--demo"]
    )

    visual_report.main()

    output = capsys.readouterr().out
    assert "生成完成" in output
    assert "/tmp/demo.md" in output


def test_visual_report_json_input(tmp_path, capsys, monkeypatch):
    payload = {
        "student": {"name": "张三", "province": "浙江省", "score": 620},
        "volunteers": [{"school": "浙江大学", "major": "计算机类"}],
    }
    data_path = tmp_path / "student.json"
    data_path.write_text(json.dumps(payload, ensure_ascii=False), encoding="utf-8")

    observed: dict[str, object] = {}

    def fake_generate(student, volunteers, output_format="all"):
        observed["student"] = student
        observed["volunteers"] = volunteers
        observed["output_format"] = output_format
        return ["/tmp/from-json.html"]

    monkeypatch.setattr(visual_report, "generate_visual_report", fake_generate)
    monkeypatch.setattr(
        visual_report.sys, "argv", ["gaokao_visual_report.py", str(data_path)]
    )

    visual_report.main()

    assert observed["student"] == payload["student"]
    assert observed["volunteers"] == payload["volunteers"]
    assert observed["output_format"] == "all"
    output = capsys.readouterr().out
    assert "/tmp/from-json.html" in output


def test_visual_report_main_guard_executes_script(tmp_path, capsys, monkeypatch):
    payload = {
        "student": {"name": "张三", "province": "浙江省", "score": 620},
        "volunteers": [
            {
                "school": "浙江大学",
                "major": "计算机类",
                "match_score": 95,
                "type": "稳",
                "probability": 70,
                "required_subjects": ["数学", "physical"],
            }
        ],
    }
    data_path = tmp_path / "student.json"
    data_path.write_text(json.dumps(payload, ensure_ascii=False), encoding="utf-8")
    monkeypatch.chdir(tmp_path)
    monkeypatch.setattr(sys, "argv", [str(VISUAL_REPORT_PATH), str(data_path)])
    runpy.run_path(str(VISUAL_REPORT_PATH), run_name="__main__")
    output = capsys.readouterr().out
    assert "生成完成" in output