from pathlib import Path

import pytest

from astrbot.core.utils.toml_parser import (
    read_pyproject_project_dependencies,
    read_pyproject_project_version,
)


def test_read_pyproject_project_version_reads_project_section(tmp_path: Path) -> None:
    pyproject_path = tmp_path / "pyproject.toml"
    pyproject_path.write_text(
        "\n".join(
            [
                'version = "ignored"',
                "[project]",
                'name = "AstrBot"',
                'version = "1.2.3-beta.4" # release version',
                "[tool.example]",
                'version = "ignored-again"',
            ]
        ),
        encoding="utf-8",
    )

    assert read_pyproject_project_version(pyproject_path) == "1.2.3-beta.4"


@pytest.mark.parametrize(
    ("version_line", "expected"),
    [
        ('version = "1.2.3"', "1.2.3"),
        ("version='1.2.3-beta.4'", "1.2.3-beta.4"),
        ('   version  =  "1.2.3-rc.1"   ', "1.2.3-rc.1"),
    ],
)
def test_read_pyproject_project_version_accepts_simple_variants(
    tmp_path: Path,
    version_line: str,
    expected: str,
) -> None:
    pyproject_path = tmp_path / "pyproject.toml"
    pyproject_path.write_text(
        "\n".join(
            [
                "[project]",
                'name = "AstrBot"',
                version_line,
            ]
        ),
        encoding="utf-8",
    )

    assert read_pyproject_project_version(pyproject_path) == expected


@pytest.mark.parametrize(
    ("version_line", "message"),
    [
        ("version", "Missing value separator for project.version"),
        ('version = "1.2.3', "Unterminated project.version string"),
        ('version = "1.2.3" extra', "Unsupported content after project.version"),
        ('version = ""', "Empty project.version value"),
    ],
)
def test_read_pyproject_project_version_rejects_invalid_values(
    tmp_path: Path,
    version_line: str,
    message: str,
) -> None:
    pyproject_path = tmp_path / "pyproject.toml"
    pyproject_path.write_text(
        "\n".join(
            [
                "[project]",
                'name = "AstrBot"',
                version_line,
            ]
        ),
        encoding="utf-8",
    )

    with pytest.raises(ValueError, match=message):
        read_pyproject_project_version(pyproject_path)


def test_read_pyproject_project_dependencies_reads_multiline_array(
    tmp_path: Path,
) -> None:
    pyproject_path = tmp_path / "pyproject.toml"
    pyproject_path.write_text(
        "\n".join(
            [
                "[project]",
                "dependencies = [",
                '  "aiohttp>=3.11.18",',
                "  \"audioop-lts ; python_full_version >= '3.13'\", # marker",
                "] # end dependencies",
            ]
        ),
        encoding="utf-8",
    )

    assert read_pyproject_project_dependencies(pyproject_path) == [
        "aiohttp>=3.11.18",
        "audioop-lts ; python_full_version >= '3.13'",
    ]


@pytest.mark.parametrize(
    ("dependency_line", "expected"),
    [
        ("dependencies = []", []),
        ('dependencies = ["aiohttp>=3.11.18"]', ["aiohttp>=3.11.18"]),
        (
            'dependencies = ["psutil>=5.8.0,<7.2.0", "httpx[socks]>=0.28.1"]',
            ["psutil>=5.8.0,<7.2.0", "httpx[socks]>=0.28.1"],
        ),
    ],
)
def test_read_pyproject_project_dependencies_accepts_inline_arrays(
    tmp_path: Path,
    dependency_line: str,
    expected: list[str],
) -> None:
    pyproject_path = tmp_path / "pyproject.toml"
    pyproject_path.write_text(
        "\n".join(
            [
                "[project]",
                dependency_line,
            ]
        ),
        encoding="utf-8",
    )

    assert read_pyproject_project_dependencies(pyproject_path) == expected


@pytest.mark.parametrize(
    ("project_lines", "message"),
    [
        (["[project]", 'name = "AstrBot"'], "Missing project.dependencies"),
        (
            ["[project]", "dependencies = ["],
            "Unterminated project.dependencies array",
        ),
        (
            ["[project]", 'dependencies = "aiohttp>=3.11.18"'],
            "Unsupported project.dependencies value",
        ),
        (
            ["[project]", "dependencies = [", "  aiohttp>=3.11.18,", "]"],
            "Unsupported project.dependencies entry value",
        ),
        (
            ["[project]", "dependencies = [", '  "aiohttp>=3.11.18" extra', "]"],
            "Unsupported content after project.dependencies entry",
        ),
        (
            ["[project]", "dependencies = [", '  ""', "]"],
            "Empty project.dependencies entry value",
        ),
    ],
)
def test_read_pyproject_project_dependencies_rejects_invalid_values(
    tmp_path: Path,
    project_lines: list[str],
    message: str,
) -> None:
    pyproject_path = tmp_path / "pyproject.toml"
    pyproject_path.write_text("\n".join(project_lines), encoding="utf-8")

    with pytest.raises(ValueError, match=message):
        read_pyproject_project_dependencies(pyproject_path)


def test_read_pyproject_project_version_raises_when_missing(tmp_path: Path) -> None:
    pyproject_path = tmp_path / "pyproject.toml"
    pyproject_path.write_text('[project]\nname = "AstrBot"\n', encoding="utf-8")

    with pytest.raises(ValueError, match="Missing project.version"):
        read_pyproject_project_version(pyproject_path)