"""Tests for the smart enhancement dispatcher (enhance_command.py)."""

import argparse
import sys

import pytest


# ---------------------------------------------------------------------------
# Helpers
# ---------------------------------------------------------------------------


def _make_args(**kwargs):
    """Build a fake Namespace with sensible defaults."""
    defaults = {
        "skill_directory": "output/react",
        "target": None,
        "api_key": None,
        "dry_run": False,
        "agent": None,
        "agent_cmd": None,
        "interactive_enhancement": False,
        "background": False,
        "daemon": False,
        "no_force": False,
        "timeout": 600,
    }
    defaults.update(kwargs)
    return argparse.Namespace(**defaults)


def _make_skill_dir(tmp_path):
    skill_dir = tmp_path / "test_skill"
    skill_dir.mkdir()
    (skill_dir / "SKILL.md").write_text("# Test", encoding="utf-8")
    return skill_dir


def _stub_sdk_available(monkeypatch, available=True):
    """Make auto-detect mode picking independent of locally installed SDKs."""
    monkeypatch.setattr(
        "skill_seekers.cli.enhance_command.api_sdk_available",
        lambda _target: available,
    )


# ---------------------------------------------------------------------------
# _is_root
# ---------------------------------------------------------------------------


class TestIsRoot:
    def test_returns_bool(self):
        from skill_seekers.cli.enhance_command import _is_root

        assert isinstance(_is_root(), bool)

    def test_not_root_when_monkeypatched(self, monkeypatch):
        import os

        monkeypatch.setattr(os, "getuid", lambda: 1000)
        from skill_seekers.cli.enhance_command import _is_root

        assert _is_root() is False

    def test_root_when_uid_zero(self, monkeypatch):
        import os

        monkeypatch.setattr(os, "getuid", lambda: 0)
        from skill_seekers.cli.enhance_command import _is_root

        assert _is_root() is True

    def test_windows_no_getuid(self, monkeypatch):
        """On Windows (no os.getuid), _is_root should return False."""
        import os

        if hasattr(os, "getuid"):
            monkeypatch.delattr(os, "getuid")
        from skill_seekers.cli.enhance_command import _is_root

        assert _is_root() is False


# ---------------------------------------------------------------------------
# _pick_mode — explicit --target flag
# ---------------------------------------------------------------------------


class TestPickModeExplicitTarget:
    def test_target_gemini_forces_api(self, monkeypatch):
        monkeypatch.delenv("ANTHROPIC_API_KEY", raising=False)
        monkeypatch.delenv("GOOGLE_API_KEY", raising=False)
        monkeypatch.delenv("OPENAI_API_KEY", raising=False)

        from skill_seekers.cli.enhance_command import _pick_mode

        args = _make_args(target="gemini")
        mode, target = _pick_mode(args)
        assert mode == "api"
        assert target == "gemini"

    def test_target_openai_forces_api(self, monkeypatch):
        monkeypatch.delenv("ANTHROPIC_API_KEY", raising=False)

        from skill_seekers.cli.enhance_command import _pick_mode

        args = _make_args(target="openai")
        mode, target = _pick_mode(args)
        assert mode == "api"
        assert target == "openai"

    def test_target_claude_forces_api(self, monkeypatch):
        monkeypatch.delenv("ANTHROPIC_API_KEY", raising=False)

        from skill_seekers.cli.enhance_command import _pick_mode

        args = _make_args(target="claude")
        mode, target = _pick_mode(args)
        assert mode == "api"
        assert target == "claude"


# ---------------------------------------------------------------------------
# _pick_mode — auto-detection from env vars
# ---------------------------------------------------------------------------


class TestPickModeAutoDetect:
    def test_anthropic_key_selects_claude(self, monkeypatch):
        monkeypatch.setenv("ANTHROPIC_API_KEY", "sk-ant-test")
        monkeypatch.delenv("GOOGLE_API_KEY", raising=False)
        monkeypatch.delenv("OPENAI_API_KEY", raising=False)

        from skill_seekers.cli.enhance_command import _pick_mode

        mode, target = _pick_mode(_make_args())
        assert mode == "api"
        assert target == "claude"

    def test_google_key_selects_gemini(self, monkeypatch):
        monkeypatch.delenv("ANTHROPIC_API_KEY", raising=False)
        monkeypatch.delenv("ANTHROPIC_AUTH_TOKEN", raising=False)
        monkeypatch.setenv("GOOGLE_API_KEY", "AIza-test")
        monkeypatch.delenv("OPENAI_API_KEY", raising=False)
        _stub_sdk_available(monkeypatch)  # gemini SDK is an optional dep

        from skill_seekers.cli.enhance_command import _pick_mode

        mode, target = _pick_mode(_make_args())
        assert mode == "api"
        assert target == "gemini"

    def test_openai_key_selects_openai(self, monkeypatch):
        monkeypatch.delenv("ANTHROPIC_API_KEY", raising=False)
        monkeypatch.delenv("ANTHROPIC_AUTH_TOKEN", raising=False)
        monkeypatch.delenv("GOOGLE_API_KEY", raising=False)
        monkeypatch.setenv("OPENAI_API_KEY", "sk-proj-test")
        _stub_sdk_available(monkeypatch)  # openai SDK is an optional dep

        from skill_seekers.cli.enhance_command import _pick_mode

        mode, target = _pick_mode(_make_args())
        assert mode == "api"
        assert target == "openai"

    def test_moonshot_key_selects_kimi(self, monkeypatch):
        """Regression (ENH-12): a Moonshot-only user must reach API mode, not
        be silently dropped to LOCAL — provided the kimi SDK is installed."""
        monkeypatch.delenv("ANTHROPIC_API_KEY", raising=False)
        monkeypatch.delenv("ANTHROPIC_AUTH_TOKEN", raising=False)
        monkeypatch.delenv("GOOGLE_API_KEY", raising=False)
        monkeypatch.delenv("OPENAI_API_KEY", raising=False)
        monkeypatch.setenv("MOONSHOT_API_KEY", "sk-moonshot-test")
        _stub_sdk_available(monkeypatch)  # kimi rides the optional openai SDK

        from skill_seekers.cli.enhance_command import _pick_mode

        mode, target = _pick_mode(_make_args())
        assert mode == "api"
        assert target == "kimi"

    def test_no_keys_falls_back_to_local(self, monkeypatch):
        monkeypatch.delenv("ANTHROPIC_API_KEY", raising=False)
        monkeypatch.delenv("ANTHROPIC_AUTH_TOKEN", raising=False)
        monkeypatch.delenv("GOOGLE_API_KEY", raising=False)
        monkeypatch.delenv("OPENAI_API_KEY", raising=False)
        monkeypatch.delenv("MOONSHOT_API_KEY", raising=False)

        from skill_seekers.cli.enhance_command import _pick_mode

        mode, target = _pick_mode(_make_args())
        assert mode == "local"
        assert target is None

    def test_anthropic_takes_priority_over_google(self, monkeypatch):
        """ANTHROPIC_API_KEY should win when both are set."""
        monkeypatch.setenv("ANTHROPIC_API_KEY", "sk-ant-test")
        monkeypatch.setenv("GOOGLE_API_KEY", "AIza-test")
        monkeypatch.delenv("OPENAI_API_KEY", raising=False)

        from skill_seekers.cli.enhance_command import _pick_mode

        mode, target = _pick_mode(_make_args())
        assert mode == "api"
        assert target == "claude"


# ---------------------------------------------------------------------------
# _pick_mode — SDK availability gate on auto-detection
# ---------------------------------------------------------------------------


class TestPickModeSdkFallback:
    """Regression: with ONLY MOONSHOT_API_KEY set, auto-detection routed to API
    mode targeting kimi, but KimiAdaptor needs the optional `openai` SDK — when
    it isn't installed the run hard-failed with exit 1 instead of falling back
    to LOCAL mode (the pre-PR behavior for this env)."""

    def _moonshot_only_env(self, monkeypatch):
        monkeypatch.delenv("ANTHROPIC_API_KEY", raising=False)
        monkeypatch.delenv("ANTHROPIC_AUTH_TOKEN", raising=False)
        monkeypatch.delenv("GOOGLE_API_KEY", raising=False)
        monkeypatch.delenv("OPENAI_API_KEY", raising=False)
        monkeypatch.setenv("MOONSHOT_API_KEY", "sk-moonshot-test")

    def test_moonshot_key_without_openai_sdk_falls_back_to_local(self, monkeypatch, capsys):
        """Detected target whose SDK is missing → LOCAL mode with a notice."""
        from skill_seekers.cli import enhance_command

        self._moonshot_only_env(monkeypatch)
        # Point the kimi requirement at a guaranteed-missing module so the
        # real find_spec path runs regardless of locally installed SDKs.
        monkeypatch.setitem(enhance_command.TARGET_SDK_MODULES, "kimi", "skill_seekers_no_such_sdk")

        mode, target = enhance_command._pick_mode(_make_args())
        assert mode == "local"
        assert target is None
        out = capsys.readouterr().out
        assert "kimi" in out
        assert "LOCAL" in out

    def test_explicit_target_not_gated_by_sdk_check(self, monkeypatch):
        """--target forces API mode even if the SDK is missing — the adaptor
        reports its own error for an explicit user choice."""
        from skill_seekers.cli import enhance_command

        self._moonshot_only_env(monkeypatch)
        monkeypatch.setitem(enhance_command.TARGET_SDK_MODULES, "kimi", "skill_seekers_no_such_sdk")

        mode, target = enhance_command._pick_mode(_make_args(target="kimi"))
        assert mode == "api"
        assert target == "kimi"

    def test_api_sdk_available_claude_true(self):
        """anthropic is a core dependency, so claude must always be available."""
        from skill_seekers.cli.enhance_command import api_sdk_available

        assert api_sdk_available("claude") is True

    def test_api_sdk_available_unknown_target_true(self):
        """Unknown targets are not gated — the adaptor surfaces its own error."""
        from skill_seekers.cli.enhance_command import api_sdk_available

        assert api_sdk_available("minimax") is True

    def test_api_sdk_available_missing_module(self, monkeypatch):
        from skill_seekers.cli import enhance_command

        monkeypatch.setitem(enhance_command.TARGET_SDK_MODULES, "kimi", "skill_seekers_no_such_sdk")
        assert enhance_command.api_sdk_available("kimi") is False

    def test_api_sdk_available_missing_parent_package(self, monkeypatch):
        """find_spec on a dotted name raises when the parent package is missing
        (e.g. no `google` namespace at all) — that must count as unavailable."""
        from skill_seekers.cli import enhance_command

        monkeypatch.setitem(
            enhance_command.TARGET_SDK_MODULES, "gemini", "skill_seekers_no_such_pkg.sub"
        )
        assert enhance_command.api_sdk_available("gemini") is False


# ---------------------------------------------------------------------------
# _pick_mode — config default_agent
# ---------------------------------------------------------------------------


class TestPickModeConfigAgent:
    def _patch_config(self, monkeypatch, agent: str | None):
        """Patch get_config_manager to return a stub with get_default_agent()."""
        monkeypatch.setattr(
            "skill_seekers.cli.enhance_command._get_config_default_agent",
            lambda: agent,
        )

    def test_config_gemini_with_key_uses_gemini(self, monkeypatch):
        self._patch_config(monkeypatch, "gemini")
        monkeypatch.delenv("ANTHROPIC_API_KEY", raising=False)
        monkeypatch.delenv("ANTHROPIC_AUTH_TOKEN", raising=False)
        monkeypatch.setenv("GOOGLE_API_KEY", "AIza-test")

        from skill_seekers.cli.enhance_command import _pick_mode

        mode, target = _pick_mode(_make_args())
        assert mode == "api"
        assert target == "gemini"

    def test_config_gemini_without_key_falls_to_autodetect(self, monkeypatch):
        """Config says gemini but no GOOGLE_API_KEY → auto-detect."""
        self._patch_config(monkeypatch, "gemini")
        monkeypatch.delenv("ANTHROPIC_API_KEY", raising=False)
        monkeypatch.delenv("ANTHROPIC_AUTH_TOKEN", raising=False)
        monkeypatch.delenv("GOOGLE_API_KEY", raising=False)
        monkeypatch.delenv("OPENAI_API_KEY", raising=False)

        from skill_seekers.cli.enhance_command import _pick_mode

        mode, target = _pick_mode(_make_args())
        assert mode == "local"

    def test_config_agent_overridden_by_explicit_target(self, monkeypatch):
        """--target flag takes priority over config."""
        self._patch_config(monkeypatch, "gemini")
        monkeypatch.delenv("ANTHROPIC_API_KEY", raising=False)

        from skill_seekers.cli.enhance_command import _pick_mode

        args = _make_args(target="openai")
        mode, target = _pick_mode(args)
        assert mode == "api"
        assert target == "openai"


# ---------------------------------------------------------------------------
# CLI argument parsing
# ---------------------------------------------------------------------------


class TestEnhanceArgumentParsing:
    """Test that the enhance parser exposes all expected arguments."""

    def _parse(self, argv, tmp_path):
        import argparse as _ap
        from skill_seekers.cli.arguments.enhance import add_enhance_arguments

        parser = _ap.ArgumentParser()
        add_enhance_arguments(parser)
        return parser.parse_args(argv)

    def test_target_gemini(self, tmp_path):
        args = self._parse(["output/react", "--target", "gemini"], tmp_path)
        assert args.target == "gemini"

    def test_target_openai(self, tmp_path):
        args = self._parse(["output/react", "--target", "openai"], tmp_path)
        assert args.target == "openai"

    def test_api_key_stored(self, tmp_path):
        args = self._parse(["output/react", "--api-key", "test-key-123"], tmp_path)
        assert args.api_key == "test-key-123"

    def test_dry_run(self, tmp_path):
        args = self._parse(["output/react", "--dry-run"], tmp_path)
        assert args.dry_run is True

    def test_no_target_defaults_none(self, tmp_path):
        args = self._parse(["output/react"], tmp_path)
        assert args.target is None

    def test_invalid_target_rejected(self, tmp_path):
        import argparse as _ap
        from skill_seekers.cli.arguments.enhance import add_enhance_arguments

        parser = _ap.ArgumentParser()
        add_enhance_arguments(parser)
        with pytest.raises(SystemExit):
            parser.parse_args(["output/react", "--target", "notaplatform"])

    def test_target_minimax_accepted(self, tmp_path):
        """MiniMax (and other OpenAI-compatible adaptors) must be valid targets."""
        args = self._parse(["output/react", "--target", "minimax"], tmp_path)
        assert args.target == "minimax"

    def test_model_flag_stored(self, tmp_path):
        args = self._parse(
            ["output/react", "--target", "minimax", "--model", "MiniMax-M2.7"], tmp_path
        )
        assert args.model == "MiniMax-M2.7"

    def test_model_defaults_none(self, tmp_path):
        args = self._parse(["output/react"], tmp_path)
        assert args.model is None


# ---------------------------------------------------------------------------
# main() CLI integration — dry-run + root detection
# ---------------------------------------------------------------------------


class TestEnhanceCommandMain:
    def test_dry_run_no_ai_call(self, tmp_path):
        skill_dir = _make_skill_dir(tmp_path)
        sys_argv_backup = sys.argv.copy()
        sys.argv = ["enhance_command.py", str(skill_dir), "--dry-run"]
        try:
            from skill_seekers.cli.enhance_command import main

            rc = main()
            assert rc == 0
        finally:
            sys.argv = sys_argv_backup

    def test_missing_dir_returns_error(self, tmp_path):
        sys_argv_backup = sys.argv.copy()
        sys.argv = ["enhance_command.py", str(tmp_path / "nonexistent")]
        try:
            from skill_seekers.cli.enhance_command import main

            rc = main()
            assert rc == 1
        finally:
            sys.argv = sys_argv_backup

    def test_root_local_mode_blocked(self, monkeypatch, tmp_path):
        import os

        skill_dir = _make_skill_dir(tmp_path)
        monkeypatch.setattr(os, "getuid", lambda: 0)
        monkeypatch.delenv("ANTHROPIC_API_KEY", raising=False)
        monkeypatch.delenv("ANTHROPIC_AUTH_TOKEN", raising=False)
        monkeypatch.delenv("GOOGLE_API_KEY", raising=False)
        monkeypatch.delenv("OPENAI_API_KEY", raising=False)

        sys_argv_backup = sys.argv.copy()
        sys.argv = ["enhance_command.py", str(skill_dir)]
        try:
            from skill_seekers.cli.enhance_command import main

            rc = main()
            assert rc == 1
        finally:
            sys.argv = sys_argv_backup

    def test_root_api_mode_allowed(self, monkeypatch, tmp_path):
        """Even as root, API mode should be selected (not blocked)."""
        import os

        skill_dir = _make_skill_dir(tmp_path)
        monkeypatch.setattr(os, "getuid", lambda: 0)
        monkeypatch.setenv("ANTHROPIC_API_KEY", "sk-ant-test")

        # Patch _run_api_mode to avoid real API call
        monkeypatch.setattr(
            "skill_seekers.cli.enhance_command._run_api_mode",
            lambda *_: 0,
        )

        sys_argv_backup = sys.argv.copy()
        sys.argv = ["enhance_command.py", str(skill_dir)]
        try:
            from skill_seekers.cli.enhance_command import main

            rc = main()
            assert rc == 0
        finally:
            sys.argv = sys_argv_backup


# ---------------------------------------------------------------------------
# _run_api_mode — API key selection
# ---------------------------------------------------------------------------


class TestRunApiModeKeySelection:
    """_run_api_mode resolves the env key for the selected target via the
    agent_client provider registry (no hand-maintained target→key map)."""

    def _captured_argv(self, monkeypatch, tmp_path, target, env=None):
        from skill_seekers.cli import enhance_skill
        from skill_seekers.cli.agent_client import API_PROVIDERS
        from skill_seekers.cli.enhance_command import _run_api_mode

        for p in API_PROVIDERS:
            for var in p["env_vars"]:
                monkeypatch.delenv(var, raising=False)
        for var, value in (env or {}).items():
            monkeypatch.setenv(var, value)
        captured = {}
        monkeypatch.setattr(enhance_skill, "main", lambda: captured.update(argv=sys.argv.copy()))
        args = _make_args(skill_directory=str(_make_skill_dir(tmp_path)))
        assert _run_api_mode(args, target) == 0
        return captured["argv"]

    def test_registry_key_passed_for_target(self, monkeypatch, tmp_path):
        env = {"MOONSHOT_API_KEY": "sk-moon-test"}
        argv = self._captured_argv(monkeypatch, tmp_path, "kimi", env)
        assert argv[argv.index("--api-key") + 1] == "sk-moon-test"

    def test_aliased_env_var_passed_for_target(self, monkeypatch, tmp_path):
        env = {"ANTHROPIC_AUTH_TOKEN": "sk-ant-alias"}
        argv = self._captured_argv(monkeypatch, tmp_path, "claude", env)
        assert argv[argv.index("--api-key") + 1] == "sk-ant-alias"

    def test_no_key_omits_flag(self, monkeypatch, tmp_path):
        argv = self._captured_argv(monkeypatch, tmp_path, "claude")
        assert "--api-key" not in argv


# ---------------------------------------------------------------------------
# Config manager — get_default_agent
# ---------------------------------------------------------------------------


class TestConfigManagerDefaultAgent:
    def test_get_default_agent_none_by_default(self, tmp_path, monkeypatch):
        from skill_seekers.cli.config_manager import ConfigManager

        monkeypatch.setattr(ConfigManager, "CONFIG_DIR", tmp_path / "cfg")
        monkeypatch.setattr(ConfigManager, "CONFIG_FILE", tmp_path / "cfg" / "config.json")
        monkeypatch.setattr(ConfigManager, "PROGRESS_DIR", tmp_path / "prog")

        mgr = ConfigManager()
        assert mgr.get_default_agent() is None

    def test_set_and_get_default_agent(self, tmp_path, monkeypatch):
        from skill_seekers.cli.config_manager import ConfigManager

        monkeypatch.setattr(ConfigManager, "CONFIG_DIR", tmp_path / "cfg")
        monkeypatch.setattr(ConfigManager, "CONFIG_FILE", tmp_path / "cfg" / "config.json")
        monkeypatch.setattr(ConfigManager, "PROGRESS_DIR", tmp_path / "prog")

        mgr = ConfigManager()
        mgr.set_default_agent("gemini")
        assert mgr.get_default_agent() == "gemini"

    def test_set_default_agent_persisted(self, tmp_path, monkeypatch):
        from skill_seekers.cli.config_manager import ConfigManager

        monkeypatch.setattr(ConfigManager, "CONFIG_DIR", tmp_path / "cfg")
        config_file = tmp_path / "cfg" / "config.json"
        monkeypatch.setattr(ConfigManager, "CONFIG_FILE", config_file)
        monkeypatch.setattr(ConfigManager, "PROGRESS_DIR", tmp_path / "prog")

        mgr = ConfigManager()
        mgr.set_default_agent("openai")

        # Re-instantiate to verify persistence
        mgr2 = ConfigManager()
        assert mgr2.get_default_agent() == "openai"