import ast
from pathlib import Path

import pytest

from app.helper.market import extract_plugin_market_repos_from_wiki
from scripts.generate_plugin_market_default import (
    OFFICIAL_PLUGIN_MARKET,
    _generate_plugin_market_default,
)


def _read_plugin_market_default(config_file: Path) -> str:
    tree = ast.parse(config_file.read_text(encoding="utf-8"))
    for node in tree.body:
        if not isinstance(node, ast.ClassDef) or node.name != "ConfigModel":
            continue
        for item in node.body:
            if not isinstance(item, ast.AnnAssign):
                continue
            if isinstance(item.target, ast.Name) and item.target.id == "PLUGIN_MARKET":
                return ast.literal_eval(item.value)
    raise AssertionError("未找到 PLUGIN_MARKET 默认值")


def test_extract_plugin_market_repos_uses_marked_section_and_deduplicates() -> None:
    """
    Wiki 清单解析只读取标记区域，并规范化、去重仓库地址。
    """
    markdown = """
- https://github.com/outside/ignored
<!-- plugin-market-repos:start -->
- https://github.com/jxxghp/MoviePilot-Plugins/
- https://github.com/demo/Market.git
- https://github.com/demo/Market
<!-- plugin-market-repos:end -->
- https://github.com/outside/ignored-again
"""

    assert extract_plugin_market_repos_from_wiki(
        markdown, require_markers=True
    ) == [
        OFFICIAL_PLUGIN_MARKET,
        "https://github.com/demo/Market",
    ]


def test_extract_plugin_market_repos_requires_unique_markers_for_build() -> None:
    """
    构建模式拒绝缺失或重复边界标记的 Wiki 文档。
    """
    with pytest.raises(ValueError, match="唯一且有序的开始和结束标记"):
        extract_plugin_market_repos_from_wiki(
            f"- {OFFICIAL_PLUGIN_MARKET}", require_markers=True
        )

    markdown = f"""
<!-- plugin-market-repos:start -->
<!-- plugin-market-repos:start -->
- {OFFICIAL_PLUGIN_MARKET}
<!-- plugin-market-repos:end -->
"""
    with pytest.raises(ValueError, match="唯一且有序的开始和结束标记"):
        extract_plugin_market_repos_from_wiki(markdown, require_markers=True)


def test_generate_plugin_market_default_updates_assignment_idempotently(
    tmp_path: Path,
) -> None:
    """
    生成脚本只替换 ConfigModel 默认值，并保持重复执行结果一致。
    """
    wiki_file = tmp_path / "plugin.md"
    wiki_file.write_text(
        f"""
<!-- plugin-market-repos:start -->
- {OFFICIAL_PLUGIN_MARKET}
- https://github.com/demo/MoviePilot-Plugins
<!-- plugin-market-repos:end -->
""",
        encoding="utf-8",
    )
    config_file = tmp_path / "config.py"
    config_file.write_text(
        """class ConfigModel(BaseModel):
    PLUGIN_MARKET: str = "https://github.com/old/Market"
    OTHER_SETTING: bool = True
""",
        encoding="utf-8",
    )

    repos = _generate_plugin_market_default(wiki_file, config_file)
    first_result = config_file.read_text(encoding="utf-8")
    _generate_plugin_market_default(wiki_file, config_file)

    assert repos == [
        OFFICIAL_PLUGIN_MARKET,
        "https://github.com/demo/MoviePilot-Plugins",
    ]
    assert _read_plugin_market_default(config_file) == ",".join(repos)
    assert "OTHER_SETTING: bool = True" in first_result
    assert config_file.read_text(encoding="utf-8") == first_result


def test_generate_plugin_market_default_requires_official_repo(
    tmp_path: Path,
) -> None:
    """
    发版默认清单缺少官方仓库时终止生成。
    """
    wiki_file = tmp_path / "plugin.md"
    wiki_file.write_text(
        """
<!-- plugin-market-repos:start -->
- https://github.com/demo/MoviePilot-Plugins
<!-- plugin-market-repos:end -->
""",
        encoding="utf-8",
    )
    config_file = tmp_path / "config.py"
    config_file.write_text(
        """class ConfigModel(BaseModel):
    PLUGIN_MARKET: str = "https://github.com/old/Market"
""",
        encoding="utf-8",
    )

    with pytest.raises(ValueError, match="缺少 MoviePilot 官方插件仓库"):
        _generate_plugin_market_default(wiki_file, config_file)