import os
import sys
import tempfile
from pathlib import Path
from typing import Generator

import pytest

sys.path.append(os.path.join(os.path.dirname(__file__), "helpers"))

pytest_plugins = [
    "api",
    "temporary_dir",
    "clan_flake",
    "root",
    "age_keys",
    "sshd",
    "command",
    "ports",
    "host_group",
]


@pytest.fixture(scope="module")
def monkeymodule() -> Generator[pytest.MonkeyPatch, None, None]:
    with pytest.MonkeyPatch.context() as mp:
        yield mp


# fixture for the example flake located under ./example_flake
# The flake is a template that is copied to a temporary location.
# Variables like __CLAN_NIXPKGS__ are replaced with the value of the
# CLAN_NIXPKGS environment variable.
@pytest.fixture(scope="module")
def machine_flake(monkeymodule: pytest.MonkeyPatch) -> Generator[Path, None, None]:
    CLAN_NIXPKGS = os.environ.get("CLAN_NIXPKGS", "")
    if CLAN_NIXPKGS == "":
        raise Exception("CLAN_NIXPKGS not set")
    template = Path(__file__).parent / "machine_flake"
    # copy the template to a new temporary location
    with tempfile.TemporaryDirectory() as tmpdir_:
        flake = Path(tmpdir_)
        for path in template.glob("**/*"):
            if path.is_dir():
                (flake / path.relative_to(template)).mkdir()
            else:
                (flake / path.relative_to(template)).write_text(path.read_text())
        # in the flake.nix file replace the string __CLAN_URL__ with the the clan flake
        # provided by get_clan_flake_toplevel
        flake_nix = flake / "flake.nix"
        flake_nix.write_text(
            flake_nix.read_text().replace("__CLAN_NIXPKGS__", CLAN_NIXPKGS)
        )
        # check that an empty config is returned if no json file exists
        monkeymodule.chdir(flake)
        # monkeypatch get_clan_flake_toplevel to return the flake
        monkeymodule.setattr("clan_cli.dirs.get_clan_flake_toplevel", lambda: flake)
        yield flake