from __future__ import annotations

import subprocess
import sys
from pathlib import Path

import opensquilla.tool_boundary as tool_boundary
from opensquilla.engine.types import ToolCall
from opensquilla.tools.dispatch import build_tool_handler
from opensquilla.tools.registry import get_default_registry
from opensquilla.tools.types import CallerKind, ToolContext

REMOVED_TOOL_NAMES = {"generate_image", "spawn_subagent", "send_message"}
CANONICAL_TOOL_NAMES = {
    "image_generate",
    "sessions_spawn",
    "sessions_send",
    "web_search",
    "web_discover",
}
OWNER_ONLY_TOOL_NAMES = {"http_request", "git_commit"}


def test_tool_call_boundary_has_canonical_and_stable_exports() -> None:
    from opensquilla.engine import ToolHandler as EngineToolHandler
    from opensquilla.engine.types import ToolResult as EngineToolResult
    from opensquilla.tools.boundary import ToolCall as ToolsToolCall

    assert tool_boundary.ToolCall is ToolCall
    assert tool_boundary.ToolResult is EngineToolResult
    assert ToolsToolCall is ToolCall
    assert EngineToolHandler is tool_boundary.AgentToolHandler


def test_engine_types_import_does_not_register_builtin_tools() -> None:
    script = (
        "import sys; "
        "import opensquilla.engine.types; "
        "assert 'opensquilla.tools.builtin' not in sys.modules, "
        "sorted(k for k in sys.modules if k.startswith('opensquilla.tools'))"
    )

    result = subprocess.run(
        [sys.executable, "-c", script],
        text=True,
        stdout=subprocess.PIPE,
        stderr=subprocess.STDOUT,
        check=False,
    )

    assert result.returncode == 0, result.stdout


def test_default_registry_public_surface_uses_canonical_tool_names() -> None:
    import opensquilla.tools.builtin  # noqa: F401

    registry = get_default_registry()
    owner_names = {
        tool.name
        for tool in registry.to_tool_definitions(
            ToolContext(is_owner=True, caller_kind=CallerKind.AGENT)
        )
    }
    channel_names = {
        tool.name
        for tool in registry.to_tool_definitions(
            ToolContext(is_owner=False, caller_kind=CallerKind.CHANNEL)
        )
    }

    assert REMOVED_TOOL_NAMES.isdisjoint(owner_names)
    assert REMOVED_TOOL_NAMES.isdisjoint(channel_names)
    assert CANONICAL_TOOL_NAMES <= owner_names
    assert "research_search" not in owner_names
    assert "research_search" not in channel_names
    assert "web_search" in channel_names
    assert "web_discover" in channel_names
    assert OWNER_ONLY_TOOL_NAMES <= owner_names
    assert OWNER_ONLY_TOOL_NAMES.isdisjoint(channel_names)


async def test_removed_tools_are_not_dispatchable_by_name() -> None:
    import opensquilla.tools.builtin  # noqa: F401

    handler = build_tool_handler(
        get_default_registry(),
        ToolContext(is_owner=True, caller_kind=CallerKind.AGENT),
    )

    for name in REMOVED_TOOL_NAMES:
        result = await handler(ToolCall(tool_use_id=f"tc-{name}", tool_name=name, arguments={}))
        assert result.is_error is True
        assert '"error_class": "ToolNotFound"' in result.content


def test_web_ui_tool_icon_map_avoids_removed_wrapper_tools() -> None:
    source = Path("src/opensquilla/gateway/static/js/views/chat.js").read_text(
        encoding="utf-8"
    )
    start = source.index("const _TOOL_EMOJI = {")
    end = source.index("  function _toolEmoji", start)
    tool_display_map = source[start:end]

    for name in REMOVED_TOOL_NAMES:
        assert name not in tool_display_map
    assert "http_request" in tool_display_map