"""Tests for GitHub Copilot support in ``nlm setup add/remove/list``."""

import json
from pathlib import Path
from unittest.mock import patch

from notebooklm_tools.cli.commands.setup import (
    CLIENT_REGISTRY,
    MCP_SERVER_CMD,
    _detect_tool,
    _github_copilot_config_path,
    _is_already_configured,
    _remove_single,
    _setup_github_copilot,
)


class TestGitHubCopilotRegistry:
    """Verify GitHub Copilot is properly registered in CLIENT_REGISTRY."""

    def test_github_copilot_in_registry(self):
        assert "github-copilot" in CLIENT_REGISTRY

    def test_github_copilot_has_auto_setup(self):
        assert CLIENT_REGISTRY["github-copilot"]["has_auto_setup"] is True

    def test_github_copilot_name(self):
        assert CLIENT_REGISTRY["github-copilot"]["name"] == "GitHub Copilot"


class TestGitHubCopilotConfigPath:
    """Verify workspace config path resolution."""

    def test_config_path_is_workspace_vscode_mcp_json(self):
        path = _github_copilot_config_path()
        assert path == Path(".vscode") / "mcp.json"


class TestSetupGitHubCopilot:
    """Test ``_setup_github_copilot()`` writes the correct config format."""

    def test_creates_config_from_scratch(self, tmp_path):
        config_path = tmp_path / ".vscode" / "mcp.json"
        with patch(
            "notebooklm_tools.cli.commands.setup._github_copilot_config_path",
            return_value=config_path,
        ):
            result = _setup_github_copilot()

        assert result is True
        config = json.loads(config_path.read_text())
        assert "servers" in config
        assert "notebooklm-mcp" in config["servers"]
        entry = config["servers"]["notebooklm-mcp"]
        assert entry["command"] == MCP_SERVER_CMD
        assert entry["args"] == []

    def test_preserves_existing_config_keys(self, tmp_path):
        config_path = tmp_path / ".vscode" / "mcp.json"
        existing = {
            "inputs": [{"type": "promptString", "id": "token"}],
            "servers": {"fetch": {"command": "uvx", "args": ["mcp-server-fetch"]}},
        }
        config_path.parent.mkdir(parents=True, exist_ok=True)
        config_path.write_text(json.dumps(existing))

        with patch(
            "notebooklm_tools.cli.commands.setup._github_copilot_config_path",
            return_value=config_path,
        ):
            _setup_github_copilot()

        config = json.loads(config_path.read_text())
        assert config["inputs"] == existing["inputs"]
        assert "fetch" in config["servers"]
        assert "notebooklm-mcp" in config["servers"]

    def test_skips_if_already_configured(self, tmp_path):
        config_path = tmp_path / ".vscode" / "mcp.json"
        config_path.parent.mkdir(parents=True, exist_ok=True)
        config_path.write_text(
            json.dumps({"servers": {"notebooklm-mcp": {"command": MCP_SERVER_CMD, "args": []}}})
        )

        with patch(
            "notebooklm_tools.cli.commands.setup._github_copilot_config_path",
            return_value=config_path,
        ):
            result = _setup_github_copilot()

        assert result is True

    def test_uses_servers_key_not_mcpservers(self, tmp_path):
        config_path = tmp_path / ".vscode" / "mcp.json"
        with patch(
            "notebooklm_tools.cli.commands.setup._github_copilot_config_path",
            return_value=config_path,
        ):
            _setup_github_copilot()

        config = json.loads(config_path.read_text())
        assert "servers" in config
        assert "mcpServers" not in config


class TestIsAlreadyConfigured:
    """Test ``_is_already_configured()`` for GitHub Copilot."""

    def test_detects_notebooklm_key(self, tmp_path):
        config_path = tmp_path / ".vscode" / "mcp.json"
        config_path.parent.mkdir(parents=True, exist_ok=True)
        config_path.write_text(json.dumps({"servers": {"notebooklm": {"command": MCP_SERVER_CMD}}}))
        with patch(
            "notebooklm_tools.cli.commands.setup._github_copilot_config_path",
            return_value=config_path,
        ):
            assert _is_already_configured("github-copilot") is True

    def test_returns_false_when_not_configured(self, tmp_path):
        config_path = tmp_path / ".vscode" / "mcp.json"
        config_path.parent.mkdir(parents=True, exist_ok=True)
        config_path.write_text(json.dumps({"servers": {}}))
        with patch(
            "notebooklm_tools.cli.commands.setup._github_copilot_config_path",
            return_value=config_path,
        ):
            assert _is_already_configured("github-copilot") is False

    def test_returns_false_when_no_config_file(self, tmp_path):
        config_path = tmp_path / ".vscode" / "mcp.json"
        with patch(
            "notebooklm_tools.cli.commands.setup._github_copilot_config_path",
            return_value=config_path,
        ):
            assert _is_already_configured("github-copilot") is False


class TestDetectTool:
    """Test ``_detect_tool()`` for GitHub Copilot."""

    def test_detects_via_which(self):
        with patch("shutil.which", return_value="/usr/bin/code"):
            assert _detect_tool("github-copilot") is True

    def test_detects_via_workspace_directory(self, tmp_path):
        config_path = tmp_path / ".vscode" / "mcp.json"
        config_path.parent.mkdir(parents=True, exist_ok=True)
        with (
            patch("shutil.which", return_value=None),
            patch(
                "notebooklm_tools.cli.commands.setup._github_copilot_config_path",
                return_value=config_path,
            ),
        ):
            assert _detect_tool("github-copilot") is True

    def test_not_detected_when_absent(self, tmp_path):
        config_path = tmp_path / ".vscode" / "mcp.json"
        with (
            patch("shutil.which", return_value=None),
            patch(
                "notebooklm_tools.cli.commands.setup._github_copilot_config_path",
                return_value=config_path,
            ),
        ):
            assert _detect_tool("github-copilot") is False


class TestRemoveGitHubCopilot:
    """Test ``_remove_single()`` for GitHub Copilot."""

    def test_removes_notebooklm_entry(self, tmp_path):
        config_path = tmp_path / ".vscode" / "mcp.json"
        config = {
            "inputs": [{"type": "promptString"}],
            "servers": {
                "notebooklm-mcp": {"command": MCP_SERVER_CMD, "args": []},
                "fetch": {"command": "uvx", "args": ["mcp-server-fetch"]},
            },
        }
        config_path.parent.mkdir(parents=True, exist_ok=True)
        config_path.write_text(json.dumps(config))

        with patch(
            "notebooklm_tools.cli.commands.setup._github_copilot_config_path",
            return_value=config_path,
        ):
            result = _remove_single("github-copilot")

        assert result is True
        updated = json.loads(config_path.read_text())
        assert "notebooklm-mcp" not in updated["servers"]
        assert "fetch" in updated["servers"]
        assert updated["inputs"] == config["inputs"]

    def test_returns_false_when_not_configured(self, tmp_path):
        config_path = tmp_path / ".vscode" / "mcp.json"
        config_path.parent.mkdir(parents=True, exist_ok=True)
        config_path.write_text(json.dumps({"servers": {}}))

        with patch(
            "notebooklm_tools.cli.commands.setup._github_copilot_config_path",
            return_value=config_path,
        ):
            result = _remove_single("github-copilot")

        assert result is False

    def test_returns_false_when_no_config_file(self, tmp_path):
        config_path = tmp_path / ".vscode" / "mcp.json"
        with patch(
            "notebooklm_tools.cli.commands.setup._github_copilot_config_path",
            return_value=config_path,
        ):
            result = _remove_single("github-copilot")

        assert result is False