import pytest

from cibuildwheel.options import ConfigOptionError, ConfigOptions, _dig_first

PYPROJECT_1 = """
[tool.cibuildwheel]
build = "cp39*"
environment = {THING = "OTHER", FOO="BAR"}

test-command = "pyproject"
test-requires = "something"
test-extras = ["one", "two"]

manylinux-x86_64-image = "manylinux1"

[tool.cibuildwheel.macos]
test-requires = "else"

[tool.cibuildwheel.linux]
test-requires = ["other", "many"]
"""


@pytest.fixture(params=["linux", "macos", "windows"])
def platform(request):
    return request.param


@pytest.mark.parametrize("fname", ["pyproject.toml", "cibuildwheel.toml"])
def test_simple_settings(tmp_path, platform, fname):
    with tmp_path.joinpath(fname).open("w") as f:
        f.write(PYPROJECT_1)

    options = ConfigOptions(tmp_path, f"{{package}}/{fname}", platform=platform)

    assert options("build", env_plat=False, sep=" ") == "cp39*"

    assert options("test-command") == "pyproject"
    assert options("archs", sep=" ") == "auto"
    assert (
        options("test-requires", sep=" ")
        == {"windows": "something", "macos": "else", "linux": "other many"}[platform]
    )

    # Also testing options for support for both lists and tables
    assert (
        options("environment", table={"item": '{k}="{v}"', "sep": " "}) == 'THING="OTHER" FOO="BAR"'
    )
    assert (
        options("environment", sep="x", table={"item": '{k}="{v}"', "sep": " "})
        == 'THING="OTHER" FOO="BAR"'
    )
    assert options("test-extras", sep=",") == "one,two"
    assert options("test-extras", sep=",", table={"item": '{k}="{v}"', "sep": " "}) == "one,two"

    assert options("manylinux-x86_64-image") == "manylinux1"
    assert options("manylinux-i686-image") == "manylinux2010"

    with pytest.raises(ConfigOptionError):
        options("environment", sep=" ")

    with pytest.raises(ConfigOptionError):
        options("test-extras", table={"item": '{k}="{v}"', "sep": " "})


def test_envvar_override(tmp_path, platform, monkeypatch):
    monkeypatch.setenv("CIBW_BUILD", "cp38*")
    monkeypatch.setenv("CIBW_MANYLINUX_X86_64_IMAGE", "manylinux2014")
    monkeypatch.setenv("CIBW_TEST_COMMAND", "mytest")
    monkeypatch.setenv("CIBW_TEST_REQUIRES", "docs")
    monkeypatch.setenv("CIBW_TEST_REQUIRES_LINUX", "scod")

    with tmp_path.joinpath("pyproject.toml").open("w") as f:
        f.write(PYPROJECT_1)

    options = ConfigOptions(tmp_path, platform=platform)

    assert options("archs", sep=" ") == "auto"

    assert options("build", sep=" ") == "cp38*"
    assert options("manylinux-x86_64-image") == "manylinux2014"
    assert options("manylinux-i686-image") == "manylinux2010"

    assert (
        options("test-requires", sep=" ")
        == {"windows": "docs", "macos": "docs", "linux": "scod"}[platform]
    )
    assert options("test-command") == "mytest"


def test_project_global_override_default_platform(tmp_path, platform):
    tmp_path.joinpath("pyproject.toml").write_text(
        """
[tool.cibuildwheel]
repair-wheel-command = "repair-project-global"
"""
    )
    options = ConfigOptions(tmp_path, platform=platform)
    assert options("repair-wheel-command") == "repair-project-global"


def test_env_global_override_default_platform(tmp_path, platform, monkeypatch):
    monkeypatch.setenv("CIBW_REPAIR_WHEEL_COMMAND", "repair-env-global")
    options = ConfigOptions(tmp_path, platform=platform)
    assert options("repair-wheel-command") == "repair-env-global"


def test_env_global_override_project_platform(tmp_path, platform, monkeypatch):
    monkeypatch.setenv("CIBW_REPAIR_WHEEL_COMMAND", "repair-env-global")
    tmp_path.joinpath("pyproject.toml").write_text(
        """
[tool.cibuildwheel.linux]
repair-wheel-command = "repair-project-linux"
[tool.cibuildwheel.windows]
repair-wheel-command = "repair-project-windows"
[tool.cibuildwheel.macos]
repair-wheel-command = "repair-project-macos"
"""
    )
    options = ConfigOptions(tmp_path, platform=platform)
    assert options("repair-wheel-command") == "repair-env-global"


def test_global_platform_order(tmp_path, platform):
    tmp_path.joinpath("pyproject.toml").write_text(
        """
[tool.cibuildwheel.linux]
repair-wheel-command = "repair-project-linux"
[tool.cibuildwheel.windows]
repair-wheel-command = "repair-project-windows"
[tool.cibuildwheel.macos]
repair-wheel-command = "repair-project-macos"
[tool.cibuildwheel]
repair-wheel-command = "repair-project-global"
"""
    )
    options = ConfigOptions(tmp_path, platform=platform)
    assert options("repair-wheel-command") == f"repair-project-{platform}"


def test_unexpected_key(tmp_path):
    # Note that platform contents are only checked when running
    # for that platform.
    tmp_path.joinpath("pyproject.toml").write_text(
        """
[tool.cibuildwheel]
repairs-wheel-command = "repair-project-linux"
"""
    )

    with pytest.raises(ConfigOptionError):
        ConfigOptions(tmp_path, platform="linux")


def test_unexpected_table(tmp_path):
    tmp_path.joinpath("pyproject.toml").write_text(
        """
[tool.cibuildwheel.linus]
repair-wheel-command = "repair-project-linux"
"""
    )
    with pytest.raises(ConfigOptionError):
        ConfigOptions(tmp_path, platform="linux")


def test_unsupported_join(tmp_path):
    tmp_path.joinpath("pyproject.toml").write_text(
        """
[tool.cibuildwheel]
build = ["1", "2"]
"""
    )
    options = ConfigOptions(tmp_path, platform="linux")

    assert "1, 2" == options("build", sep=", ")
    with pytest.raises(ConfigOptionError):
        options("build")


def test_disallowed_a(tmp_path):
    tmp_path.joinpath("pyproject.toml").write_text(
        """
[tool.cibuildwheel.windows]
manylinux-x86_64-image = "manylinux1"
"""
    )
    disallow = {"windows": {"manylinux-x86_64-image"}}
    ConfigOptions(tmp_path, platform="linux", disallow=disallow)
    with pytest.raises(ConfigOptionError):
        ConfigOptions(tmp_path, platform="windows", disallow=disallow)


def test_environment_override_empty(tmp_path, monkeypatch):
    tmp_path.joinpath("pyproject.toml").write_text(
        """
[tool.cibuildwheel]
manylinux-i686-image = "manylinux1"
manylinux-x86_64-image = ""
"""
    )

    monkeypatch.setenv("CIBW_MANYLINUX_I686_IMAGE", "")
    monkeypatch.setenv("CIBW_MANYLINUX_AARCH64_IMAGE", "manylinux1")

    options = ConfigOptions(tmp_path, platform="linux")

    assert options("manylinux-x86_64-image") == ""
    assert options("manylinux-i686-image") == ""
    assert options("manylinux-aarch64-image") == "manylinux1"

    assert options("manylinux-x86_64-image", ignore_empty=True) == "manylinux2010"
    assert options("manylinux-i686-image", ignore_empty=True) == "manylinux1"
    assert options("manylinux-aarch64-image", ignore_empty=True) == "manylinux1"


@pytest.mark.parametrize("ignore_empty", (True, False))
def test_dig_first(ignore_empty):
    d1 = {"random": "thing"}
    d2 = {"this": "that", "empty": ""}
    d3 = {"other": "hi"}
    d4 = {"this": "d4", "empty": "not"}

    answer = _dig_first(
        (d1, "empty"),
        (d2, "empty"),
        (d3, "empty"),
        (d4, "empty"),
        ignore_empty=ignore_empty,
    )
    assert answer == ("not" if ignore_empty else "")

    answer = _dig_first(
        (d1, "this"),
        (d2, "this"),
        (d3, "this"),
        (d4, "this"),
        ignore_empty=ignore_empty,
    )
    assert answer == "that"

    with pytest.raises(KeyError):
        _dig_first(
            (d1, "this"),
            (d2, "other"),
            (d3, "this"),
            (d4, "other"),
            ignore_empty=ignore_empty,
        )