pi-mcp-extension 1.5.4 → 1.6.0

package/README.md

@@ -159,6 +159,8 @@ Config files are loaded from two locations. **Project config overrides global co
 
 ## Commands
 
+Server-name arguments provide autocomplete suggestions from the configured MCP servers. `/mcp:auth` suggests only servers configured for OAuth.
+
 | Command | Description |
 |---|---|
 | `/mcp` | Show status summary of all configured servers |

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-mcp-extension",
-  "version": "1.5.4",
+  "version": "1.6.0",
   "description": "MCP (Model Context Protocol) client extension for the Pi coding agent — connect Pi to any MCP server",
   "type": "module",
   "keywords": [

package/src/index.ts

@@ -157,6 +157,42 @@ export function parseAuthArguments(args: string, serverNames: string[]): { name:
   return { name: reset ? name.replace(/\s+--reset$/, "").trimEnd() : name, reset };
 }
 
+interface ServerCompletionEntry {
+  name: string;
+  state: string;
+  config: {
+    transport: string;
+    lifecycle: string;
+    auth?: unknown;
+  };
+}
+
+export function serverNameCompletions(
+  argumentPrefix: string,
+  servers: ServerCompletionEntry[],
+  options: { authOnly?: boolean } = {},
+) {
+  const prefix = argumentPrefix.trimStart();
+  const completions = servers
+    .filter((server) => !options.authOnly || server.config.auth)
+    .filter((server) => server.name.startsWith(prefix))
+    .sort((a, b) => a.name.localeCompare(b.name))
+    .map((server) => ({
+      value: server.name,
+      label: server.name,
+      description: [
+        server.state,
+        server.config.transport,
+        server.config.lifecycle,
+        server.config.auth ? "OAuth" : undefined,
+      ]
+        .filter(Boolean)
+        .join(" · "),
+    }));
+
+  return completions.length > 0 ? completions : null;
+}
+
 export default async function (pi: ExtensionAPI, paths: { bootstrapCwd?: string; globalConfigPath?: string; authStorageDir?: string; browserOpen?: (url: string) => void } = {}): Promise<void> {
   // ── 1. Load and validate config ──────────────────────────────────────────
   // cwd is available on the ExtensionContext passed to event handlers.
@@ -277,6 +313,7 @@ export default async function (pi: ExtensionAPI, paths: { bootstrapCwd?: string;
         ctx.ui.notify(manager.getStatusSummary(), "info");
       }
     },
+    getArgumentCompletions: (prefix) => serverNameCompletions(prefix, manager.getAllServers()),
   });
 
   // ── 5. /mcp:stop — stop a server ─────────────────────────────────────────
@@ -297,6 +334,7 @@ export default async function (pi: ExtensionAPI, paths: { bootstrapCwd?: string;
       await manager.stopServer(serverName);
       ctx.ui.notify(`pi-mcp: Stopped ${serverName}`, "info");
     },
+    getArgumentCompletions: (prefix) => serverNameCompletions(prefix, manager.getAllServers()),
   });
 
   // ── 6. /mcp:start — manually start a lazy server ─────────────────────────
@@ -320,6 +358,7 @@ export default async function (pi: ExtensionAPI, paths: { bootstrapCwd?: string;
         ctx.ui.notify(`pi-mcp: Failed to start ${serverName} — ${msg}`, "error");
       }
     },
+    getArgumentCompletions: (prefix) => serverNameCompletions(prefix, manager.getAllServers()),
   });
 
   async function authorize(serverName: string, ctx: ExtensionContext, signal: AbortSignal, reset: boolean): Promise<void> {
@@ -565,5 +604,6 @@ export default async function (pi: ExtensionAPI, paths: { bootstrapCwd?: string;
         ctx.ui.notify(`pi-mcp: ${err instanceof Error ? err.message : String(err)}`, err instanceof AuthCancelledError ? "info" : "error");
       }
     },
+    getArgumentCompletions: (prefix) => serverNameCompletions(prefix, manager.getAllServers(), { authOnly: true }),
   });
 }