import importlib.util
import sys
import uuid
from pathlib import Path

import pytest


MODULE_PATH = Path(__file__).resolve().parents[1] / "scripts" / "local_setup.py"


def load_local_setup_module():
    """以隔离模块名加载本地安装脚本，避免测试间共享模块状态。"""
    module_name = f"moviepilot_local_setup_resources_{uuid.uuid4().hex}"
    spec = importlib.util.spec_from_file_location(module_name, MODULE_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


@pytest.mark.parametrize(
    ("gil_disabled", "suffix"),
    [
        (0, ""),
        (1, "t"),
    ],
)
def test_python_version_tag_distinguishes_free_threaded_abi(
    monkeypatch, gil_disabled, suffix
):
    module = load_local_setup_module()
    monkeypatch.setattr(
        module.sysconfig,
        "get_config_var",
        lambda name: gil_disabled if name == "Py_GIL_DISABLED" else None,
    )

    expected = f"cp{sys.version_info.major}{sys.version_info.minor}{suffix}"
    assert module._get_python_version_tag() == expected


@pytest.mark.parametrize(
    ("platform_tag", "machine", "python_version", "expected"),
    [
        (
            "linux",
            "x86_64",
            "cp314",
            "sites.cpython-314-x86_64-linux-gnu.so",
        ),
        (
            "linux",
            "x86_64",
            "cp314t",
            "sites.cpython-314t-x86_64-linux-gnu.so",
        ),
        (
            "linux",
            "aarch64",
            "cp314",
            "sites.cpython-314-aarch64-linux-gnu.so",
        ),
        (
            "linux",
            "aarch64",
            "cp314t",
            "sites.cpython-314t-aarch64-linux-gnu.so",
        ),
        ("darwin", "arm64", "cp314", "sites.cpython-314-darwin.so"),
        ("darwin", "arm64", "cp314t", "sites.cpython-314t-darwin.so"),
        ("windows", "amd64", "cp314", "sites.cp314-win_amd64.pyd"),
        ("windows", "amd64", "cp314t", "sites.cp314t-win_amd64.pyd"),
    ],
)
def test_filter_resources_files_selects_exact_runtime_artifact(
    tmp_path, platform_tag, machine, python_version, expected
):
    module = load_local_setup_module()
    filenames = {
        "user.sites.v3.bin",
        "sites.pyi",
        "sites.cpython-314-x86_64-linux-gnu.so",
        "sites.cpython-314-aarch64-linux-gnu.so",
        "sites.cpython-314t-x86_64-linux-gnu.so",
        "sites.cpython-314t-aarch64-linux-gnu.so",
        "sites.cpython-314-darwin.so",
        "sites.cpython-314t-darwin.so",
        "sites.cp314-win_amd64.pyd",
        "sites.cp314t-win_amd64.pyd",
    }
    for filename in filenames:
        (tmp_path / filename).write_bytes(filename.encode())

    matched = module._filter_resources_files(
        tmp_path,
        platform_tag,
        machine,
        python_version,
    )

    assert [path.name for path in matched] == ["user.sites.v3.bin", expected]


def test_copy_resource_files_keeps_only_current_runtime_artifact(
    monkeypatch, tmp_path
):
    module = load_local_setup_module()
    source_dir = tmp_path / "resources.v3"
    target_dir = tmp_path / "site"
    source_dir.mkdir()
    target_dir.mkdir()
    for filename in (
        "user.sites.v3.bin",
        "sites.cpython-314-x86_64-linux-gnu.so",
        "sites.cpython-314-aarch64-linux-gnu.so",
        "sites.cpython-314t-x86_64-linux-gnu.so",
    ):
        (source_dir / filename).write_text(f"source:{filename}", encoding="utf-8")
    (target_dir / "sites.pyi").write_text("typing", encoding="utf-8")
    (target_dir / "sites.cpython-313-x86_64-linux-gnu.so").write_text(
        "stale", encoding="utf-8"
    )
    (target_dir / "sites.cp314-win_amd64.pyd").write_text(
        "stale", encoding="utf-8"
    )
    (target_dir / "sites.cpython-314t-darwin.so").write_text(
        "stale", encoding="utf-8"
    )
    (target_dir / "sites.legacy.dylib").write_text("stale", encoding="utf-8")
    monkeypatch.setattr(module, "SITE_RESOURCE_DIR", target_dir)
    monkeypatch.setattr(module, "_get_platform_tag", lambda: ("linux", "x86_64"))
    monkeypatch.setattr(module, "_get_python_version_tag", lambda: "cp314t")

    copied = module.copy_resource_files(source_dir)

    assert copied == [
        "user.sites.v3.bin",
        "sites.cpython-314t-x86_64-linux-gnu.so",
    ]
    assert sorted(path.name for path in target_dir.iterdir()) == [
        "sites.cpython-314t-x86_64-linux-gnu.so",
        "sites.pyi",
        "user.sites.v3.bin",
    ]


def test_copy_resource_files_rejects_partial_source_before_changing_target(
    monkeypatch, tmp_path
):
    module = load_local_setup_module()
    source_dir = tmp_path / "resources.v3"
    target_dir = tmp_path / "site"
    source_dir.mkdir()
    target_dir.mkdir()
    (source_dir / "user.sites.v3.bin").write_bytes(b"new-data")
    old_native = target_dir / "sites.cpython-314-darwin.so"
    old_data = target_dir / "user.sites.v3.bin"
    old_native.write_bytes(b"old-native")
    old_data.write_bytes(b"old-data")
    monkeypatch.setattr(module, "SITE_RESOURCE_DIR", target_dir)
    monkeypatch.setattr(module, "_get_platform_tag", lambda: ("darwin", "arm64"))
    monkeypatch.setattr(module, "_get_python_version_tag", lambda: "cp314t")

    with pytest.raises(RuntimeError, match="sites.cpython-314t-darwin.so"):
        module.copy_resource_files(source_dir)

    assert old_native.read_bytes() == b"old-native"
    assert old_data.read_bytes() == b"old-data"


def test_copy_resource_files_keeps_target_when_staging_fails(monkeypatch, tmp_path):
    module = load_local_setup_module()
    source_dir = tmp_path / "resources.v3"
    target_dir = tmp_path / "site"
    source_dir.mkdir()
    target_dir.mkdir()
    (source_dir / "user.sites.v3.bin").write_bytes(b"new-data")
    new_native_name = "sites.cpython-314t-darwin.so"
    (source_dir / new_native_name).write_bytes(b"new-native")
    old_native = target_dir / "sites.cpython-314-darwin.so"
    old_data = target_dir / "user.sites.v3.bin"
    old_native.write_bytes(b"old-native")
    old_data.write_bytes(b"old-data")
    real_copy2 = module.shutil.copy2

    def fail_native_staging(source, target):
        if Path(source).name == new_native_name:
            raise OSError("injected native staging failure")
        return real_copy2(source, target)

    monkeypatch.setattr(module, "SITE_RESOURCE_DIR", target_dir)
    monkeypatch.setattr(module, "_get_platform_tag", lambda: ("darwin", "arm64"))
    monkeypatch.setattr(module, "_get_python_version_tag", lambda: "cp314t")
    monkeypatch.setattr(module.shutil, "copy2", fail_native_staging)

    with pytest.raises(OSError, match="injected native staging failure"):
        module.copy_resource_files(source_dir)

    assert old_native.read_bytes() == b"old-native"
    assert old_data.read_bytes() == b"old-data"
    assert not (target_dir / new_native_name).exists()
    assert not list(target_dir.glob(".sites-sync-*"))


def test_copy_resource_files_rolls_back_when_commit_fails(monkeypatch, tmp_path):
    module = load_local_setup_module()
    source_dir = tmp_path / "resources.v3"
    target_dir = tmp_path / "site"
    source_dir.mkdir()
    target_dir.mkdir()
    (source_dir / "user.sites.v3.bin").write_bytes(b"new-data")
    new_native_name = "sites.cpython-314t-darwin.so"
    (source_dir / new_native_name).write_bytes(b"new-native")
    old_native = target_dir / "sites.cpython-314-darwin.so"
    old_data = target_dir / "user.sites.v3.bin"
    old_native.write_bytes(b"old-native")
    old_data.write_bytes(b"old-data")
    real_replace = module.os.replace

    def fail_native_commit(source, target):
        if Path(source).parent.name == "staging" and Path(target).name == new_native_name:
            raise OSError("injected native commit failure")
        return real_replace(source, target)

    monkeypatch.setattr(module, "SITE_RESOURCE_DIR", target_dir)
    monkeypatch.setattr(module, "_get_platform_tag", lambda: ("darwin", "arm64"))
    monkeypatch.setattr(module, "_get_python_version_tag", lambda: "cp314t")
    monkeypatch.setattr(module.os, "replace", fail_native_commit)

    with pytest.raises(OSError, match="injected native commit failure"):
        module.copy_resource_files(source_dir)

    assert old_native.read_bytes() == b"old-native"
    assert old_data.read_bytes() == b"old-data"
    assert not (target_dir / new_native_name).exists()


def test_local_resource_status_requires_current_native_artifact(
    monkeypatch, tmp_path
):
    module = load_local_setup_module()
    monkeypatch.setattr(module, "SITE_RESOURCE_DIR", tmp_path)
    monkeypatch.setattr(module, "_get_platform_tag", lambda: ("darwin", "arm64"))
    monkeypatch.setattr(module, "_get_python_version_tag", lambda: "cp314t")
    (tmp_path / "user.sites.v3.bin").write_bytes(b"data")
    (tmp_path / "sites.pyi").write_text("typing", encoding="utf-8")
    (tmp_path / "sites.cpython-314-darwin.so").write_bytes(b"standard")

    assert module.local_resource_status() is False

    (tmp_path / "sites.cpython-314t-darwin.so").write_bytes(b"free-threaded")

    assert module.local_resource_status() is True