|
| 1 | +import z from "zod" |
| 2 | +import { Tool } from "../../tool/tool" |
| 3 | +import { discoverExternalMcp } from "../../mcp/discover" |
| 4 | +import { resolveConfigPath, addMcpToConfig, findAllConfigPaths, listMcpInConfig } from "../../mcp/config" |
| 5 | +import { Instance } from "../../project/instance" |
| 6 | +import { Global } from "../../global" |
| 7 | + |
| 8 | +/** |
| 9 | + * Check which MCP server names are permanently configured on disk |
| 10 | + * (as opposed to ephemeral auto-discovered servers in memory). |
| 11 | + */ |
| 12 | +async function getPersistedMcpNames(): Promise<Set<string>> { |
| 13 | + const configPaths = await findAllConfigPaths(Instance.worktree, Global.Path.config) |
| 14 | + const names = new Set<string>() |
| 15 | + for (const p of configPaths) { |
| 16 | + for (const name of await listMcpInConfig(p)) { |
| 17 | + names.add(name) |
| 18 | + } |
| 19 | + } |
| 20 | + return names |
| 21 | +} |
| 22 | + |
| 23 | +/** Redact server details for safe display — show type and name only, not commands/URLs */ |
| 24 | +function safeDetail(server: { type: string } & Record<string, any>): string { |
| 25 | + if (server.type === "remote") return "(remote)" |
| 26 | + if (server.type === "local" && Array.isArray(server.command) && server.command.length > 0) { |
| 27 | + // Show only the executable name, not args (which may contain credentials) |
| 28 | + return `(local: ${server.command[0]})` |
| 29 | + } |
| 30 | + return `(${server.type})` |
| 31 | +} |
| 32 | + |
| 33 | +export const McpDiscoverTool = Tool.define("mcp_discover", { |
| 34 | + description: |
| 35 | + "Discover MCP servers from external AI tool configs (VS Code, Claude Code, Copilot, Gemini) and optionally add them to altimate-code config permanently.", |
| 36 | + parameters: z.object({ |
| 37 | + action: z |
| 38 | + .enum(["list", "add"]) |
| 39 | + .describe('"list" to show discovered servers, "add" to write them to config'), |
| 40 | + scope: z |
| 41 | + .enum(["project", "global"]) |
| 42 | + .optional() |
| 43 | + .default("project") |
| 44 | + .describe('Where to write when action is "add". "project" = .altimate-code/altimate-code.json, "global" = ~/.config/opencode/'), |
| 45 | + servers: z |
| 46 | + .array(z.string()) |
| 47 | + .optional() |
| 48 | + .describe('Server names to add. If omitted with action "add", adds all new servers.'), |
| 49 | + }), |
| 50 | + async execute(args, ctx) { |
| 51 | + const { servers: discovered } = await discoverExternalMcp(Instance.worktree) |
| 52 | + const discoveredNames = Object.keys(discovered) |
| 53 | + |
| 54 | + if (discoveredNames.length === 0) { |
| 55 | + return { |
| 56 | + title: "MCP Discover: none found", |
| 57 | + metadata: { discovered: 0, new: 0, existing: 0, added: 0 }, |
| 58 | + output: |
| 59 | + "No MCP servers found in external configs.\nChecked: .vscode/mcp.json, .github/copilot/mcp.json, .mcp.json, ~/.gemini/settings.json, ~/.claude.json", |
| 60 | + } |
| 61 | + } |
| 62 | + |
| 63 | + // Check what's actually persisted on disk, NOT the merged in-memory config |
| 64 | + const persistedNames = await getPersistedMcpNames() |
| 65 | + const newServers = discoveredNames.filter((n) => !persistedNames.has(n)) |
| 66 | + const alreadyAdded = discoveredNames.filter((n) => persistedNames.has(n)) |
| 67 | + |
| 68 | + // Build discovery report — redact details for security (no raw commands/URLs) |
| 69 | + const lines: string[] = [] |
| 70 | + if (newServers.length > 0) { |
| 71 | + lines.push(`New servers (not yet in config):`) |
| 72 | + for (const name of newServers) { |
| 73 | + lines.push(` - ${name} ${safeDetail(discovered[name])}`) |
| 74 | + } |
| 75 | + } |
| 76 | + if (alreadyAdded.length > 0) { |
| 77 | + lines.push(`\nAlready in config: ${alreadyAdded.join(", ")}`) |
| 78 | + } |
| 79 | + |
| 80 | + if (args.action === "list") { |
| 81 | + return { |
| 82 | + title: `MCP Discover: ${newServers.length} new, ${alreadyAdded.length} existing`, |
| 83 | + metadata: { discovered: discoveredNames.length, new: newServers.length, existing: alreadyAdded.length, added: 0 }, |
| 84 | + output: lines.join("\n"), |
| 85 | + } |
| 86 | + } |
| 87 | + |
| 88 | + // action === "add" |
| 89 | + const toAdd = args.servers |
| 90 | + ? args.servers.filter((n) => newServers.includes(n)) |
| 91 | + : newServers |
| 92 | + |
| 93 | + if (toAdd.length === 0) { |
| 94 | + return { |
| 95 | + title: "MCP Discover: nothing to add", |
| 96 | + metadata: { discovered: discoveredNames.length, new: newServers.length, existing: alreadyAdded.length, added: 0 }, |
| 97 | + output: lines.join("\n") + "\n\nNo matching servers to add.", |
| 98 | + } |
| 99 | + } |
| 100 | + |
| 101 | + const useGlobal = args.scope === "global" |
| 102 | + const configPath = await resolveConfigPath( |
| 103 | + useGlobal ? Global.Path.config : Instance.worktree, |
| 104 | + useGlobal, |
| 105 | + ) |
| 106 | + |
| 107 | + for (const name of toAdd) { |
| 108 | + await addMcpToConfig(name, discovered[name], configPath) |
| 109 | + } |
| 110 | + |
| 111 | + lines.push(`\nAdded ${toAdd.length} server(s) to ${configPath}: ${toAdd.join(", ")}`) |
| 112 | + lines.push("These servers are already active in the current session via auto-discovery.") |
| 113 | + |
| 114 | + return { |
| 115 | + title: `MCP Discover: added ${toAdd.length} server(s)`, |
| 116 | + metadata: { discovered: discoveredNames.length, new: newServers.length, existing: alreadyAdded.length, added: toAdd.length }, |
| 117 | + output: lines.join("\n"), |
| 118 | + } |
| 119 | + }, |
| 120 | +}) |
0 commit comments