"""Agent-facing MCP self-service tools: search catalog, install/remove servers.

Allows Butler to discover and install MCP integrations on user request
without requiring CLI access. Install/remove operations require human
approval via the existing human_gate mechanism.
"""

from __future__ import annotations

import json
import logging
import os
from typing import Any, Callable, cast
from butler.tools.mcp_self_service_ops import tool_json_loud
from butler.registry.mcp_catalog import McpCatalogService
from butler.tools.mcp_self_service_ops import attach_post_install_verify_safe, resolve_project_workspace_safe, tool_json_loud
from butler.registry.mcp_install import install_catalog_server
from butler.tools.mcp_self_service_ops import resolve_project_workspace_safe, tool_json_loud
from butler.registry.mcp_install import remove_mcp_server
from butler.defaults.env_defaults import (
    MCP_ENABLED_DEFAULT,
    MCP_SELF_SERVICE_DEFAULT,
)

logger = logging.getLogger(__name__)


def _mcp_self_service_enabled() -> bool:
    if os.getenv("BUTLER_MCP_ENABLED", MCP_ENABLED_DEFAULT).strip().lower() in ("0", "false", "no", "off"):
        return False
    return os.getenv("BUTLER_MCP_SELF_SERVICE", MCP_SELF_SERVICE_DEFAULT).strip() in ("1", "true")


def _tool_mcp_catalog_search(query: str = "", **_: Any) -> str:
    """Search the MCP server catalog."""

    def _run() -> str:

        svc = McpCatalogService()
        entries = svc.search(query.strip(), limit=10)
        installed = set(svc.list_installed_ids())

        results = []
        for e in entries:
            results.append({
                "id": e.id,
                "title": e.title,
                "description": e.description[:200],
                "transport": e.transport,
                "trust": e.trust,
                "installed": e.id in installed,
            })
        return json.dumps({
            "ok": True,
            "count": len(results),
            "results": results,
            "hint": "Use mcp_install to install a server by id",
        }, ensure_ascii=False)

    return cast(str, tool_json_loud(_run))


def _tool_mcp_install(
    server_id: str,
    scope: str = "global",
    env: list[str] | None = None,
    **_: Any,
) -> str:
    """Install an MCP server from the catalog."""

    sid = (server_id or "").strip()
    if not sid:
        return json.dumps({"ok": False, "error": "server_id is required"})

    use_project = scope.strip().lower() in ("project", "项目")
    workspace = None
    if use_project:
        workspace = resolve_project_workspace_safe()
        if workspace is None:
            return json.dumps({"ok": False, "error": "No active project for project-scope install"})

    def _run() -> str:

        ok, msg = install_catalog_server(
            sid,
            env_assignments=env,
            workspace=workspace,
            use_project=use_project,
        )
        payload: dict[str, Any] = {"ok": ok, "message": msg}
        if ok:
            attach_post_install_verify_safe(payload, sid, workspace)
        return json.dumps(payload, ensure_ascii=False)

    return cast(str, tool_json_loud(_run))


def _tool_mcp_remove(server_id: str, **_: Any) -> str:
    """Remove an installed MCP server."""

    sid = (server_id or "").strip()
    if not sid:
        return json.dumps({"ok": False, "error": "server_id is required"})

    def _run() -> str:

        workspace = resolve_project_workspace_safe()
        ok, msg = remove_mcp_server(sid, workspace=workspace)
        return json.dumps({"ok": ok, "message": msg}, ensure_ascii=False)

    return cast(str, tool_json_loud(_run))


def _tool_mcp_list_installed(**_: Any) -> str:
    """List currently installed MCP servers."""

    def _run() -> str:

        svc = McpCatalogService()
        installed = svc.list_installed_ids()
        lock = svc.load_lock_summary()
        servers_info = lock.get("servers", {}) if isinstance(lock, dict) else {}

        items = []
        for sid in installed:
            info = servers_info.get(sid, {})
            items.append({
                "id": sid,
                "installed_at": info.get("installed_at", "unknown"),
                "probe_tool_count": (info.get("probe") or {}).get("tool_count", 0),
            })
        return json.dumps({
            "ok": True,
            "count": len(items),
            "servers": items,
        }, ensure_ascii=False)

    return cast(str, tool_json_loud(_run))


def register_mcp_self_service_tools(register: Callable[..., None]) -> None:
    if not _mcp_self_service_enabled():
        return

    register(
        name="mcp_catalog_search",
        description=(
            "【marketplace·discover】Search curated MCP catalog by keyword; "
            "returns server_id candidates from the index. Read-only; no local yaml scan."
        ),
        schema={
            "type": "object",
            "properties": {
                "query": {
                    "type": "string",
                    "description": "Search keyword (e.g. 'github', 'database', 'slack'). Empty returns all.",
                },
            },
        },
        handler=_tool_mcp_catalog_search,
        toolset="mcp_self_service",
    )
    register(
        name="mcp_install",
        description=(
            "【install·mutation】Add a catalog server_id into mcp.yaml (global or project scope). "
            "Requires server_id; performs config write."
        ),
        schema={
            "type": "object",
            "properties": {
                "server_id": {"type": "string", "description": "MCP server id from catalog"},
                "scope": {
                    "type": "string",
                    "enum": ["global", "project"],
                    "description": "Install scope: global or project-level",
                },
                "env": {
                    "type": "array",
                    "items": {"type": "string"},
                    "description": "Environment variable assignments (e.g. ['API_KEY=xxx'])",
                },
            },
            "required": ["server_id"],
        },
        handler=_tool_mcp_install,
        toolset="mcp_self_service",
    )
    register(
        name="mcp_remove",
        description="Remove an installed MCP server by server_id.",
        schema={
            "type": "object",
            "properties": {
                "server_id": {"type": "string", "description": "MCP server id to remove"},
            },
            "required": ["server_id"],
        },
        handler=_tool_mcp_remove,
        toolset="mcp_self_service",
    )
    register(
        name="mcp_list_installed",
        description=(
            "【local·inventory】List MCP servers already in mcp.yaml with runtime status. "
            "No marketplace keyword search."
        ),
        schema={"type": "object", "properties": {}},
        handler=_tool_mcp_list_installed,
        toolset="mcp_self_service",
    )