agentpool 2.1.9__py3-none-any.whl → 2.5.0__py3-none-any.whl

agentpool_server/opencode_server/routes/agent_routes.py

@@ -0,0 +1,473 @@
+"""Agent, command, MCP, LSP, formatter, and logging routes."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from fastapi import APIRouter, HTTPException
+import httpx
+from llmling_models.auth.anthropic_auth import (
+    AnthropicTokenStore,
+    build_authorization_url,
+    exchange_code_for_token,
+    generate_pkce,
+)
+from pydantic import BaseModel, HttpUrl
+
+from agentpool.mcp_server.manager import MCPManager
+from agentpool.resource_providers import AggregatingResourceProvider
+from agentpool_config.mcp_server import (
+    SSEMCPServerConfig,
+    StdioMCPServerConfig,
+    StreamableHTTPMCPServerConfig,
+)
+from agentpool_server.opencode_server.dependencies import StateDep
+from agentpool_server.opencode_server.models import (
+    Agent,
+    Command,
+    LogRequest,
+    McpResource,
+    MCPStatus,
+)
+
+
+router = APIRouter(tags=["agent"])
+
+
+@router.get("/agent")
+async def list_agents(state: StateDep) -> list[Agent]:
+    """List available agents from the AgentPool.
+
+    Returns all agents with their configurations, suitable for the agent
+    switcher UI. Agents are marked as primary (visible in switcher) or
+    subagent (hidden, used internally).
+    """
+    if state.agent.agent_pool is None:
+        return [
+            Agent(
+                name="default",
+                description="Default AgentPool agent",
+                mode="primary",
+                default=True,
+            )
+        ]
+
+    pool = state.agent.agent_pool
+    agents: list[Agent] = []
+    first_agent_name = next(iter(pool.all_agents.keys()), None)
+
+    for name, agent in pool.all_agents.items():
+        # Get description from agent
+        agents.append(
+            Agent(
+                name=name,
+                description=agent.description or f"Agent: {name}",
+                # model=AgentModel(model_id=agent.model_name or "unknown", provider_id=""),
+                mode="primary",  # All agents visible for now; add hidden config later
+                default=(name == first_agent_name),  # First agent is default
+            )
+        )
+
+    return (
+        agents
+        if agents
+        else [Agent(name="default", description="Default agent", mode="primary", default=True)]
+    )
+
+
+@router.get("/command")
+async def list_commands(state: StateDep) -> list[Command]:
+    """List available slash commands.
+
+    Commands are derived from MCP prompts available to the agent.
+    """
+    try:
+        prompts = await state.agent.tools.list_prompts()
+        return [Command(name=p.name, description=p.description or "") for p in prompts]
+    except Exception:  # noqa: BLE001
+        return []
+
+
+@router.get("/mcp")
+async def get_mcp_status(state: StateDep) -> dict[str, MCPStatus]:
+    """Get MCP server status.
+
+    Returns status for each connected MCP server.
+    """
+    # Use agent's get_mcp_server_info method which handles different agent types
+    server_info = state.agent.get_mcp_server_info()
+
+    # Convert MCPServerStatus dataclass to MCPStatus response model
+    return {
+        name: MCPStatus(
+            name=status.name,
+            status=status.status,
+            error=status.error,
+        )
+        for name, status in server_info.items()
+    }
+
+
+class AddMCPServerRequest(BaseModel):
+    """Request to add an MCP server dynamically."""
+
+    command: str | None = None
+    """Command to run (for stdio servers)."""
+
+    args: list[str] | None = None
+    """Arguments for the command."""
+
+    url: str | None = None
+    """URL for HTTP/SSE servers."""
+
+    env: dict[str, str] | None = None
+    """Environment variables for the server."""
+
+
+@router.post("/mcp")
+async def add_mcp_server(request: AddMCPServerRequest, state: StateDep) -> MCPStatus:
+    """Add an MCP server dynamically.
+
+    Supports stdio servers (command + args) or HTTP/SSE servers (url).
+    """
+    # Build the config based on request
+    # Note: client_id is auto-generated from command/url, custom names not supported
+    config: SSEMCPServerConfig | StdioMCPServerConfig | StreamableHTTPMCPServerConfig
+    if request.url:
+        # HTTP-based server
+        if request.url.endswith("/sse"):
+            config = SSEMCPServerConfig(url=HttpUrl(request.url))
+        else:
+            config = StreamableHTTPMCPServerConfig(url=HttpUrl(request.url))
+    elif request.command:  # Stdio server
+        args = request.args or []
+        config = StdioMCPServerConfig(command=request.command, args=args, env=request.env)
+    else:
+        detail = "Must provide either 'command' (for stdio) or 'url' (for HTTP/SSE)"
+        raise HTTPException(status_code=400, detail=detail)
+
+    # Find the MCPManager and add the server
+    manager: MCPManager | None = None
+    for provider in state.agent.tools.external_providers:
+        if isinstance(provider, MCPManager):
+            manager = provider
+            break
+        if isinstance(provider, AggregatingResourceProvider):
+            for nested in provider.providers:
+                if isinstance(nested, MCPManager):
+                    manager = nested
+                    break
+
+    if manager is None:
+        raise HTTPException(status_code=400, detail="No MCP manager available")
+
+    try:
+        await manager.setup_server(config, add_to_config=True)
+        return MCPStatus(name=config.client_id, status="connected")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Failed to add MCP server: {e}") from e
+
+
+@router.post("/log")
+async def log(request: LogRequest, state: StateDep) -> bool:
+    """Write a log entry.
+
+    TODO: Integrate with proper logging.
+    """
+    _ = state  # unused for now
+    print(f"[{request.level}] {request.service}: {request.message}")
+    return True
+
+
+@router.get("/experimental/resource")
+async def list_mcp_resources(state: StateDep) -> dict[str, McpResource]:
+    """Get all available MCP resources from connected servers.
+
+    Returns a dictionary mapping resource keys to McpResource objects.
+    Keys are formatted as "{client}:{resource_name}" for uniqueness.
+    """
+    try:
+        resources = await state.agent.tools.list_resources()
+        result: dict[str, McpResource] = {}
+
+        for resource in resources:
+            # Create unique key: sanitize client and resource names
+            client_name = (resource.client or "unknown").replace("/", "_")
+            resource_name = resource.name.replace("/", "_")
+            key = f"{client_name}:{resource_name}"
+
+            result[key] = McpResource(
+                name=resource.name,
+                uri=resource.uri,
+                description=resource.description,
+                mime_type=resource.mime_type,
+                client=resource.client or "unknown",
+            )
+    except Exception:  # noqa: BLE001
+        return {}
+    else:
+        return result
+
+
+@router.get("/experimental/tool/ids")
+async def list_tool_ids(state: StateDep) -> list[str]:
+    """List all available tool IDs.
+
+    Returns a list of tool names that are available to the agent.
+    OpenCode expects: Array<string>
+    """
+    try:
+        tools = await state.agent.tools.get_tools()
+        return [tool.name for tool in tools]
+    except Exception:  # noqa: BLE001
+        return []
+
+
+class ToolListItem(BaseModel):
+    """Tool info matching OpenCode SDK ToolListItem type."""
+
+    id: str
+    description: str
+    parameters: dict[str, Any]
+
+
+@router.get("/experimental/tool")
+async def list_tools_with_schemas(  # noqa: D417
+    state: StateDep,
+    provider: str | None = None,
+    model: str | None = None,
+) -> list[ToolListItem]:
+    """List tools with their JSON schemas.
+
+    Args:
+        provider: Optional provider filter (not used currently)
+        model: Optional model filter (not used currently)
+
+    Returns list of tools matching OpenCode's ToolListItem format:
+    - id: string
+    - description: string
+    - parameters: unknown (JSON schema)
+    """
+    _ = provider, model  # Currently unused, for future filtering
+
+    try:
+        tools = await state.agent.tools.get_tools()
+        result = []
+        for tool in tools:
+            # Extract parameters schema from the OpenAI function schema
+            schema = tool.schema
+            params = schema.get("function", {}).get("parameters", {})
+            item = ToolListItem(id=tool.name, description=tool.description or "", parameters=params)
+            result.append(item)
+    except Exception:  # noqa: BLE001
+        return []
+    else:
+        return result
+
+
+@router.get("/lsp")
+async def get_lsp_status(state: StateDep) -> list[dict[str, Any]]:
+    """Get LSP server status.
+
+    Returns status of all running LSP servers.
+    """
+    try:
+        lsp_manager = state.get_or_create_lsp_manager()
+        servers = []
+        for server_id, server_state in lsp_manager._servers.items():
+            # OpenCode TUI expects "connected" or "error" for status colors
+            status = "connected" if server_state.initialized else "error"
+            servers.append({
+                "id": server_id,
+                "name": server_id,
+                "status": status,
+                "language": server_state.language,
+                "root": server_state.root_uri,  # TUI uses "root" not "rootUri"
+            })
+    except Exception:  # noqa: BLE001
+        return []
+    else:
+        return servers
+
+
+@router.get("/formatter")
+async def get_formatter_status(state: StateDep) -> list[dict[str, Any]]:
+    """Get formatter status.
+
+    Returns empty list - formatters not supported yet.
+    """
+    _ = state
+    return []
+
+
+@router.get("/provider/auth")
+async def get_provider_auth(state: StateDep) -> dict[str, list[dict[str, Any]]]:
+    """Get provider authentication methods.
+
+    Returns available OAuth providers with their auth methods.
+    """
+    _ = state
+    return {
+        "anthropic": [
+            {
+                "type": "oauth",
+                "label": "Connect Claude Max/Pro",
+                "method": "code",
+            }
+        ],
+        "copilot": [
+            {
+                "type": "oauth",
+                "label": "Connect GitHub Copilot",
+                "method": "device_code",
+            }
+        ],
+    }
+
+
+# Store for active OAuth flows (in production, use Redis or similar)
+_oauth_flows: dict[str, dict[str, Any]] = {}
+
+
+@router.post("/provider/{provider_id}/oauth/authorize")
+async def oauth_authorize(provider_id: str, state: StateDep) -> dict[str, Any]:
+    """Start OAuth authorization flow for a provider.
+
+    Returns URL and instructions for the user to complete authorization.
+    """
+    _ = state
+
+    if provider_id == "anthropic":
+        verifier, challenge = generate_pkce()
+        auth_url = build_authorization_url(verifier, challenge)
+
+        # Store verifier for callback
+        _oauth_flows[f"anthropic:{verifier}"] = {"verifier": verifier}
+
+        return {
+            "url": auth_url,
+            "instructions": "Sign in with your Anthropic account and copy the authorization code",
+            "method": "code",
+            "state": verifier,
+        }
+
+    if provider_id == "copilot":
+        async with httpx.AsyncClient(timeout=30.0) as client:
+            resp = await client.post(
+                "https://github.com/login/device/code",
+                headers={
+                    "accept": "application/json",
+                    "editor-version": "Neovim/0.6.1",
+                    "editor-plugin-version": "copilot.vim/1.16.0",
+                    "content-type": "application/json",
+                    "user-agent": "GithubCopilot/1.155.0",
+                },
+                json={"client_id": "Iv1.b507a08c87ecfe98", "scope": "read:user"},
+            )
+            resp.raise_for_status()
+            data = resp.json()
+
+        device_code = data["device_code"]
+        user_code = data["user_code"]
+        verification_uri = data["verification_uri"]
+
+        # Store device_code for callback
+        _oauth_flows[f"copilot:{device_code}"] = {"device_code": device_code}
+
+        return {
+            "url": verification_uri,
+            "instructions": f"Enter code: {user_code}",
+            "method": "device_code",
+            "user_code": user_code,
+            "device_code": device_code,
+        }
+
+    raise HTTPException(status_code=404, detail=f"Unknown provider: {provider_id}")
+
+
+@router.post("/provider/{provider_id}/oauth/callback")
+async def oauth_callback(
+    provider_id: str,
+    state: StateDep,
+    code: str | None = None,
+    device_code: str | None = None,
+    verifier: str | None = None,
+) -> dict[str, Any]:
+    """Handle OAuth callback/code exchange.
+
+    For Anthropic: exchanges authorization code for tokens.
+    For Copilot: polls for token using device code.
+    """
+    _ = state
+
+    if provider_id == "anthropic":
+        if not code or not verifier:
+            raise HTTPException(
+                status_code=400, detail="Missing code or verifier for Anthropic OAuth"
+            )
+
+        try:
+            token = exchange_code_for_token(code, verifier)
+            # Save token
+            store = AnthropicTokenStore()
+            store.save(token)
+
+            # Clean up flow state
+            _oauth_flows.pop(f"anthropic:{verifier}", None)
+        except Exception as e:
+            raise HTTPException(status_code=400, detail=str(e)) from e
+        else:
+            return {
+                "type": "success",
+                "access": token.access_token,
+                "refresh": token.refresh_token,
+                "expires": token.expires_at,
+            }
+
+    if provider_id == "copilot":
+        if not device_code:
+            raise HTTPException(
+                status_code=400,
+                detail="Missing device_code for Copilot OAuth",
+            )
+
+        async with httpx.AsyncClient(timeout=30.0) as client:
+            resp = await client.post(
+                "https://github.com/login/oauth/access_token",
+                headers={
+                    "accept": "application/json",
+                    "editor-version": "Neovim/0.6.1",
+                    "editor-plugin-version": "copilot.vim/1.16.0",
+                    "content-type": "application/json",
+                    "user-agent": "GithubCopilot/1.155.0",
+                },
+                json={
+                    "client_id": "Iv1.b507a08c87ecfe98",
+                    "device_code": device_code,
+                    "grant_type": "urn:ietf:params:oauth:grant-type:device_code",
+                },
+            )
+            data = resp.json()
+
+        if "error" in data:
+            if data["error"] == "authorization_pending":
+                return {"type": "pending", "message": "Waiting for user authorization"}
+            raise HTTPException(
+                status_code=400,
+                detail=data.get("error_description", data["error"]),
+            )
+
+        access_token = data.get("access_token")
+        if access_token:
+            # Clean up flow state
+            _oauth_flows.pop(f"copilot:{device_code}", None)
+
+            return {
+                "type": "success",
+                "access": access_token,
+                "refresh": data.get("refresh_token"),
+                "expires": None,  # Copilot tokens don't expire the same way
+            }
+
+        return {"type": "pending", "message": "No token received yet"}
+    raise HTTPException(status_code=404, detail=f"Unknown provider: {provider_id}")

agentpool_server/opencode_server/routes/app_routes.py

@@ -0,0 +1,202 @@
+"""App, project, path, and VCS routes."""
+
+from __future__ import annotations
+
+from pathlib import Path
+import subprocess
+from typing import TYPE_CHECKING
+
+from fastapi import APIRouter, HTTPException
+
+from agentpool_server.opencode_server.dependencies import StateDep
+from agentpool_server.opencode_server.models import (
+    App,
+    AppTimeInfo,
+    PathInfo,
+    Project,
+    ProjectTime,
+    ProjectUpdatedEvent,
+    ProjectUpdateRequest,
+    VcsInfo,
+)
+
+
+if TYPE_CHECKING:
+    from agentpool.sessions.models import ProjectData
+
+
+router = APIRouter(tags=["app"])
+
+
+@router.get("/app")
+async def get_app(state: StateDep) -> App:
+    """Get app information."""
+    working_path = Path(state.working_dir)
+    return App(
+        git=(working_path / ".git").is_dir(),
+        hostname="localhost",
+        path=PathInfo(
+            config="",
+            cwd=state.working_dir,
+            data="",
+            root=state.working_dir,
+            state="",
+        ),
+        time=AppTimeInfo(initialized=state.start_time),
+    )
+
+
+def _project_data_to_response(data: ProjectData) -> Project:
+    """Convert ProjectData to OpenCode Project response."""
+    working_path = Path(data.worktree)
+    vcs_dir: str | None = None
+    if data.vcs == "git":
+        vcs_dir = str(working_path / ".git")
+    elif data.vcs == "hg":
+        vcs_dir = str(working_path / ".hg")
+
+    return Project(
+        id=data.project_id,
+        worktree=data.worktree,
+        vcs_dir=vcs_dir,
+        vcs=data.vcs,
+        time=ProjectTime(created=int(data.created_at.timestamp() * 1000)),
+    )
+
+
+async def _get_current_project(state: StateDep) -> ProjectData:
+    """Get or create the current project from storage."""
+    from agentpool_storage.project_store import ProjectStore
+
+    storage = state.pool.storage
+    project_store = ProjectStore(storage)
+    return await project_store.get_or_create(state.working_dir)
+
+
+@router.get("/project")
+async def list_projects(state: StateDep) -> list[Project]:
+    """List all projects."""
+    from agentpool_storage.project_store import ProjectStore
+
+    storage = state.pool.storage
+    project_store = ProjectStore(storage)
+    projects = await project_store.list_recent(limit=50)
+    return [_project_data_to_response(p) for p in projects]
+
+
+@router.get("/project/current")
+async def get_project_current(state: StateDep) -> Project:
+    """Get current project."""
+    project = await _get_current_project(state)
+    return _project_data_to_response(project)
+
+
+@router.patch("/project/{project_id}")
+async def update_project(
+    project_id: str,
+    update: ProjectUpdateRequest,
+    state: StateDep,
+) -> Project:
+    """Update project metadata (name, settings).
+
+    Emits a project.updated event when successful.
+
+    Args:
+        project_id: Project identifier
+        update: Fields to update (name and/or settings)
+        state: Server state
+
+    Returns:
+        Updated project data
+
+    Raises:
+        HTTPException: If project not found
+    """
+    from agentpool_storage.project_store import ProjectStore
+
+    store = ProjectStore(state.pool.storage)
+    project_data = None
+
+    # Update name if provided
+    if update.name is not None:
+        project_data = await store.set_name(project_id, update.name)
+        if not project_data:
+            raise HTTPException(status_code=404, detail="Project not found")
+
+    # Update settings if provided
+    if update.settings:
+        if project_data:
+            # Already fetched from set_name, update with settings
+            project_data = await store.update_settings(project_id, **update.settings)
+        else:
+            project_data = await store.update_settings(project_id, **update.settings)
+
+        if not project_data:
+            raise HTTPException(status_code=404, detail="Project not found")
+
+    # If neither name nor settings provided, just fetch the project
+    if not project_data:
+        project_data = await store.get_by_id(project_id)
+        if not project_data:
+            raise HTTPException(status_code=404, detail="Project not found")
+
+    # Convert to OpenCode Project model
+    project = _project_data_to_response(project_data)
+
+    # Broadcast event
+    await state.broadcast_event(ProjectUpdatedEvent.create(project))
+
+    return project
+
+
+@router.get("/path")
+async def get_path(state: StateDep) -> PathInfo:
+    """Get current path info."""
+    return PathInfo(
+        config="",
+        cwd=state.working_dir,
+        data="",
+        root=state.working_dir,
+        state="",
+    )
+
+
+@router.get("/vcs")
+async def get_vcs(state: StateDep) -> VcsInfo:
+    """Get VCS info.
+
+    TODO: For remote/ACP support, these git commands should run through
+    state.env.execute_command() instead of subprocess.run() so they
+    execute on the client side where the repository lives.
+    """
+    git_dir = Path(state.working_dir) / ".git"
+    if not git_dir.is_dir():
+        return VcsInfo(branch=None, dirty=False, commit=None)
+
+    try:
+        branch = subprocess.run(
+            ["git", "rev-parse", "--abbrev-ref", "HEAD"],
+            cwd=state.working_dir,
+            capture_output=True,
+            text=True,
+            check=True,
+        ).stdout.strip()
+        commit = subprocess.run(
+            ["git", "rev-parse", "HEAD"],
+            cwd=state.working_dir,
+            capture_output=True,
+            text=True,
+            check=True,
+        ).stdout.strip()
+        dirty = bool(
+            subprocess.run(
+                ["git", "status", "--porcelain"],
+                cwd=state.working_dir,
+                capture_output=True,
+                text=True,
+                check=True,
+            ).stdout.strip()
+        )
+        return VcsInfo(branch=branch, dirty=dirty, commit=commit)
+    except subprocess.CalledProcessError:
+        return VcsInfo(branch=None, dirty=False, commit=None)