@spences10/pi-mcp 0.0.1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Scott Spence
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,97 @@
+# @spences10/pi-mcp
+
+[![built with vite+](https://img.shields.io/badge/built%20with-Vite+-646CFF?logo=vite&logoColor=white)](https://viteplus.dev)
+[![tested with vitest](https://img.shields.io/badge/tested%20with-Vitest-6E9F18?logo=vitest&logoColor=white)](https://vitest.dev)
+
+Pi extension for connecting MCP servers and exposing their tools
+inside Pi.
+
+## Install
+
+```bash
+pi install npm:@spences10/pi-mcp
+```
+
+Local development from this monorepo:
+
+```bash
+pnpm --filter @spences10/pi-mcp run build
+pi install ./packages/pi-mcp
+# or for one run only
+pi -e ./packages/pi-mcp
+```
+
+## Configuration
+
+The extension loads MCP server definitions from `mcp.json` files in
+global and project locations.
+
+A typical project `mcp.json` looks like:
+
+```json
+{
+	"mcpServers": {
+		"sqlite": {
+			"command": "npx",
+			"args": ["-y", "mcp-sqlite-tools", "./data.db"]
+		}
+	}
+}
+```
+
+Server tools are registered as Pi tools using this naming format:
+
+```text
+mcp__<server>__<tool>
+```
+
+For example, a `sqlite` server tool named `execute_read_query`
+becomes:
+
+```text
+mcp__sqlite__execute_read_query
+```
+
+## Commands
+
+```text
+/mcp list
+/mcp enable <server>
+/mcp disable <server>
+```
+
+Use `/mcp list` to inspect connection state and `/mcp enable` or
+`/mcp disable` to toggle a server's registered tools during a session.
+
+## What it does
+
+- reads MCP server config
+- connects to stdio or HTTP MCP servers
+- performs the MCP `initialize` handshake
+- discovers tools via `tools/list`
+- registers each discovered MCP tool with Pi
+- forwards model tool calls to the MCP server
+- cleans up server processes on session shutdown
+
+## Using from a custom harness
+
+```ts
+import mcp from '@spences10/pi-mcp';
+
+// pass `mcp` as an ExtensionFactory to your Pi runtime
+```
+
+`my-pi` imports this package directly and enables it as the built-in
+MCP extension.
+
+## Development
+
+```bash
+pnpm --filter @spences10/pi-mcp run check
+pnpm --filter @spences10/pi-mcp run test
+pnpm --filter @spences10/pi-mcp run build
+```
+
+## License
+
+MIT

package/dist/client.d.ts

@@ -0,0 +1,27 @@
+export interface McpStdioServerConfig {
+    name: string;
+    transport: 'stdio';
+    command: string;
+    args?: string[];
+    env?: Record<string, string>;
+}
+export interface McpHttpServerConfig {
+    name: string;
+    transport: 'http';
+    url: string;
+    headers?: Record<string, string>;
+}
+export type McpServerConfig = McpStdioServerConfig | McpHttpServerConfig;
+export interface McpToolInfo {
+    name: string;
+    description?: string;
+    inputSchema?: Record<string, unknown>;
+}
+export declare class McpClient {
+    #private;
+    constructor(config: McpServerConfig);
+    connect(): Promise<void>;
+    listTools(): Promise<McpToolInfo[]>;
+    callTool(name: string, args: Record<string, unknown>): Promise<unknown>;
+    disconnect(): Promise<void>;
+}

package/dist/client.js

@@ -0,0 +1,227 @@
+import { spawn } from 'node:child_process';
+export class McpClient {
+    #proc = null;
+    #config;
+    #nextId = 1;
+    #pending = new Map();
+    #buffer = '';
+    #sessionId;
+    constructor(config) {
+        this.#config = config;
+    }
+    async connect() {
+        if (this.#config.transport === 'stdio') {
+            await this.#connect_stdio();
+        }
+        await this.#request('initialize', {
+            protocolVersion: '2024-11-05',
+            capabilities: {},
+            clientInfo: { name: 'my-pi', version: '0.0.1' },
+        });
+        await this.#send({
+            jsonrpc: '2.0',
+            method: 'notifications/initialized',
+        });
+    }
+    async listTools() {
+        const result = (await this.#request('tools/list', {}));
+        return result.tools;
+    }
+    async callTool(name, args) {
+        return this.#request('tools/call', {
+            name,
+            arguments: args,
+        });
+    }
+    async disconnect() {
+        if (this.#config.transport === 'http') {
+            await this.#disconnect_http();
+        }
+        if (this.#proc) {
+            this.#proc.kill();
+            this.#proc = null;
+        }
+        this.#pending.clear();
+    }
+    async #connect_stdio() {
+        const { command, args = [], env, } = this.#config;
+        this.#proc = spawn(command, args, {
+            stdio: ['pipe', 'pipe', 'pipe'],
+            env: { ...process.env, ...env },
+        });
+        this.#proc.stdout.setEncoding('utf8');
+        this.#proc.stdout.on('data', (chunk) => {
+            this.#buffer += chunk;
+            const lines = this.#buffer.split('\n');
+            this.#buffer = lines.pop() || '';
+            for (const line of lines) {
+                if (!line.trim())
+                    continue;
+                try {
+                    this.#handle_message(JSON.parse(line));
+                }
+                catch {
+                    // ignore non-JSON lines
+                }
+            }
+        });
+    }
+    #request(method, params) {
+        return new Promise((resolve, reject) => {
+            const id = this.#nextId++;
+            this.#pending.set(id, { resolve, reject });
+            this.#send({ jsonrpc: '2.0', id, method, params }).catch((error) => {
+                if (this.#pending.has(id)) {
+                    this.#pending.delete(id);
+                    reject(error);
+                }
+            });
+            setTimeout(() => {
+                if (this.#pending.has(id)) {
+                    this.#pending.delete(id);
+                    reject(new Error(`MCP request ${method} timed out`));
+                }
+            }, 30_000);
+        });
+    }
+    async #send(msg) {
+        if (this.#config.transport === 'http') {
+            await this.#send_http(msg);
+            return;
+        }
+        if (!this.#proc?.stdin?.writable) {
+            throw new Error('MCP server not connected');
+        }
+        this.#proc.stdin.write(JSON.stringify(msg) + '\n');
+    }
+    async #send_http(msg) {
+        const config = this.#config;
+        const headers = new Headers(config.headers ?? {});
+        headers.set('content-type', 'application/json');
+        headers.set('accept', 'application/json, text/event-stream');
+        if (this.#sessionId) {
+            headers.set('mcp-session-id', this.#sessionId);
+        }
+        const response = await fetch(config.url, {
+            method: 'POST',
+            headers,
+            body: JSON.stringify(msg),
+        });
+        const sessionId = response.headers.get('mcp-session-id');
+        if (sessionId) {
+            this.#sessionId = sessionId;
+        }
+        if (!response.ok) {
+            const body = await response.text().catch(() => '');
+            throw new Error(`MCP HTTP ${response.status}${body ? `: ${body}` : ''}`);
+        }
+        if (response.status === 204)
+            return;
+        const contentType = response.headers.get('content-type') ?? '';
+        if (contentType.includes('text/event-stream')) {
+            await this.#consume_sse_response(response, config.name);
+            return;
+        }
+        const body = await response.text();
+        if (!body.trim())
+            return;
+        let parsed;
+        try {
+            parsed = JSON.parse(body);
+        }
+        catch {
+            throw new Error(`Invalid MCP HTTP response from ${config.name}: ${body.slice(0, 200)}`);
+        }
+        this.#dispatch_message(parsed);
+    }
+    async #disconnect_http() {
+        const config = this.#config;
+        if (!this.#sessionId)
+            return;
+        const headers = new Headers(config.headers ?? {});
+        headers.set('mcp-session-id', this.#sessionId);
+        const response = await fetch(config.url, {
+            method: 'DELETE',
+            headers,
+        });
+        if (response.status !== 405 && !response.ok) {
+            const body = await response.text().catch(() => '');
+            throw new Error(`MCP HTTP disconnect ${response.status}${body ? `: ${body}` : ''}`);
+        }
+        this.#sessionId = undefined;
+    }
+    async #consume_sse_response(response, server_name) {
+        if (!response.body)
+            return;
+        const reader = response.body.getReader();
+        const decoder = new TextDecoder();
+        let buffer = '';
+        let event_lines = [];
+        const flush_event = () => {
+            if (event_lines.length === 0)
+                return;
+            const data_lines = event_lines
+                .filter((line) => line.startsWith('data:'))
+                .map((line) => line.slice(5).trimStart());
+            event_lines = [];
+            if (data_lines.length === 0)
+                return;
+            const payload = data_lines.join('\n').trim();
+            if (!payload)
+                return;
+            try {
+                this.#dispatch_message(JSON.parse(payload));
+            }
+            catch {
+                throw new Error(`Invalid MCP SSE payload from ${server_name}: ${payload.slice(0, 200)}`);
+            }
+        };
+        while (true) {
+            const { done, value } = await reader.read();
+            buffer += decoder.decode(value ?? new Uint8Array(), {
+                stream: !done,
+            });
+            const normalized = buffer.replace(/\r\n/g, '\n');
+            const lines = normalized.split('\n');
+            buffer = lines.pop() ?? '';
+            for (const line of lines) {
+                if (line === '') {
+                    flush_event();
+                    continue;
+                }
+                if (line.startsWith(':'))
+                    continue;
+                event_lines.push(line);
+            }
+            if (done)
+                break;
+        }
+        if (buffer.trim()) {
+            event_lines.push(buffer.trim());
+        }
+        flush_event();
+    }
+    #dispatch_message(message) {
+        if (Array.isArray(message)) {
+            for (const item of message) {
+                this.#dispatch_message(item);
+            }
+            return;
+        }
+        if (!message || typeof message !== 'object')
+            return;
+        this.#handle_message(message);
+    }
+    #handle_message(msg) {
+        if (msg.id == null || !this.#pending.has(msg.id))
+            return;
+        const pending = this.#pending.get(msg.id);
+        this.#pending.delete(msg.id);
+        if (msg.error) {
+            pending.reject(new Error(`MCP error ${msg.error.code}: ${msg.error.message}`));
+            return;
+        }
+        pending.resolve(msg.result);
+    }
+}
+//# sourceMappingURL=client.js.map

package/dist/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAqB,MAAM,oBAAoB,CAAC;AAyC9D,MAAM,OAAO,SAAS;IACrB,KAAK,GAAwB,IAAI,CAAC;IAClC,OAAO,CAAkB;IACzB,OAAO,GAAG,CAAC,CAAC;IACZ,QAAQ,GAAG,IAAI,GAAG,EAMf,CAAC;IACJ,OAAO,GAAG,EAAE,CAAC;IACb,UAAU,CAAU;IAEpB,YAAY,MAAuB;QAClC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,OAAO;QACZ,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,KAAK,OAAO,EAAE,CAAC;YACxC,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC7B,CAAC;QAED,MAAM,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE;YACjC,eAAe,EAAE,YAAY;YAC7B,YAAY,EAAE,EAAE;YAChB,UAAU,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE;SAC/C,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,KAAK,CAAC;YAChB,OAAO,EAAE,KAAK;YACd,MAAM,EAAE,2BAA2B;SACnC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,SAAS;QACd,MAAM,MAAM,GAAG,CAAC,MAAM,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE,EAAE,CAAC,CAEpD,CAAC;QACF,OAAO,MAAM,CAAC,KAAK,CAAC;IACrB,CAAC;IAED,KAAK,CAAC,QAAQ,CACb,IAAY,EACZ,IAA6B;QAE7B,OAAO,IAAI,CAAC,QAAQ,CAAC,YAAY,EAAE;YAClC,IAAI;YACJ,SAAS,EAAE,IAAI;SACf,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,UAAU;QACf,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,KAAK,MAAM,EAAE,CAAC;YACvC,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAC/B,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YAClB,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;QACnB,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,cAAc;QACnB,MAAM,EACL,OAAO,EACP,IAAI,GAAG,EAAE,EACT,GAAG,GACH,GAAG,IAAI,CAAC,OAA+B,CAAC;QAEzC,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE;YACjC,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;YAC/B,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,GAAG,EAAE;SAC/B,CAAC,CAAC;QAEH,IAAI,CAAC,KAAK,CAAC,MAAO,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QACvC,IAAI,CAAC,KAAK,CAAC,MAAO,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;YAC/C,IAAI,CAAC,OAAO,IAAI,KAAK,CAAC;YACtB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACvC,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YAEjC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBAC1B,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;oBAAE,SAAS;gBAC3B,IAAI,CAAC;oBACJ,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAoB,CAAC,CAAC;gBAC3D,CAAC;gBAAC,MAAM,CAAC;oBACR,wBAAwB;gBACzB,CAAC;YACF,CAAC;QACF,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,QAAQ,CAAC,MAAc,EAAE,MAAe;QACvC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACtC,MAAM,EAAE,GAAG,IAAI,CAAC,OAAO,EAAE,CAAC;YAC1B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;YAC3C,IAAI,CAAC,KAAK,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,KAAK,CACvD,CAAC,KAAK,EAAE,EAAE;gBACT,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;oBAC3B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBACzB,MAAM,CAAC,KAAc,CAAC,CAAC;gBACxB,CAAC;YACF,CAAC,CACD,CAAC;YAEF,UAAU,CAAC,GAAG,EAAE;gBACf,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;oBAC3B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBACzB,MAAM,CAAC,IAAI,KAAK,CAAC,eAAe,MAAM,YAAY,CAAC,CAAC,CAAC;gBACtD,CAAC;YACF,CAAC,EAAE,MAAM,CAAC,CAAC;QACZ,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,GAAmB;QAC9B,IAAI,IAAI,CAAC,OAAO,CAAC,SAAS,KAAK,MAAM,EAAE,CAAC;YACvC,MAAM,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YAC3B,OAAO;QACR,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC7C,CAAC;QACD,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,CAAC;IACpD,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,GAAmB;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,OAA8B,CAAC;QACnD,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;QAClD,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,kBAAkB,CAAC,CAAC;QAChD,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,qCAAqC,CAAC,CAAC;QAC7D,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACrB,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE;YACxC,MAAM,EAAE,MAAM;YACd,OAAO;YACP,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;SACzB,CAAC,CAAC;QAEH,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACzD,IAAI,SAAS,EAAE,CAAC;YACf,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;QAC7B,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAClB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,IAAI,KAAK,CACd,YAAY,QAAQ,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CACvD,CAAC;QACH,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG;YAAE,OAAO;QAEpC,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;QAC/D,IAAI,WAAW,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;YAC/C,MAAM,IAAI,CAAC,qBAAqB,CAAC,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;YACxD,OAAO;QACR,CAAC;QAED,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;YAAE,OAAO;QAEzB,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACJ,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC;QAAC,MAAM,CAAC;YACR,MAAM,IAAI,KAAK,CACd,kCAAkC,MAAM,CAAC,IAAI,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CACtE,CAAC;QACH,CAAC;QACD,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,gBAAgB;QACrB,MAAM,MAAM,GAAG,IAAI,CAAC,OAA8B,CAAC;QACnD,IAAI,CAAC,IAAI,CAAC,UAAU;YAAE,OAAO;QAE7B,MAAM,OAAO,GAAG,IAAI,OAAO,CAAC,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;QAClD,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAC/C,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE;YACxC,MAAM,EAAE,QAAQ;YAChB,OAAO;SACP,CAAC,CAAC;QACH,IAAI,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAC7C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;YACnD,MAAM,IAAI,KAAK,CACd,uBAAuB,QAAQ,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAClE,CAAC;QACH,CAAC;QACD,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,qBAAqB,CAC1B,QAAkB,EAClB,WAAmB;QAEnB,IAAI,CAAC,QAAQ,CAAC,IAAI;YAAE,OAAO;QAE3B,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;QAClC,IAAI,MAAM,GAAG,EAAE,CAAC;QAChB,IAAI,WAAW,GAAa,EAAE,CAAC;QAE/B,MAAM,WAAW,GAAG,GAAG,EAAE;YACxB,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO;YACrC,MAAM,UAAU,GAAG,WAAW;iBAC5B,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;iBAC1C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC,CAAC;YAC3C,WAAW,GAAG,EAAE,CAAC;YACjB,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO;YACpC,MAAM,OAAO,GAAG,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;YAC7C,IAAI,CAAC,OAAO;gBAAE,OAAO;YAErB,IAAI,CAAC;gBACJ,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;YAC7C,CAAC;YAAC,MAAM,CAAC;gBACR,MAAM,IAAI,KAAK,CACd,gCAAgC,WAAW,KAAK,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CACvE,CAAC;YACH,CAAC;QACF,CAAC,CAAC;QAEF,OAAO,IAAI,EAAE,CAAC;YACb,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,IAAI,IAAI,UAAU,EAAE,EAAE;gBACnD,MAAM,EAAE,CAAC,IAAI;aACb,CAAC,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YACjD,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACrC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;YAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBAC1B,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;oBACjB,WAAW,EAAE,CAAC;oBACd,SAAS;gBACV,CAAC;gBACD,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;oBAAE,SAAS;gBACnC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACxB,CAAC;YAED,IAAI,IAAI;gBAAE,MAAM;QACjB,CAAC;QAED,IAAI,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;YACnB,WAAW,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QACjC,CAAC;QACD,WAAW,EAAE,CAAC;IACf,CAAC;IAED,iBAAiB,CAAC,OAAgB;QACjC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC5B,KAAK,MAAM,IAAI,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC9B,CAAC;YACD,OAAO;QACR,CAAC;QACD,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;YAAE,OAAO;QACpD,IAAI,CAAC,eAAe,CAAC,OAA0B,CAAC,CAAC;IAClD,CAAC;IAED,eAAe,CAAC,GAAoB;QACnC,IAAI,GAAG,CAAC,EAAE,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAAE,OAAO;QACzD,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAE,CAAC;QAC3C,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC7B,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,MAAM,CACb,IAAI,KAAK,CACR,aAAa,GAAG,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,KAAK,CAAC,OAAO,EAAE,CACnD,CACD,CAAC;YACF,OAAO;QACR,CAAC;QACD,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC7B,CAAC;CACD"}

package/dist/config.d.ts

@@ -0,0 +1,2 @@
+import type { McpServerConfig } from './client.js';
+export declare function load_mcp_config(cwd: string): McpServerConfig[];

package/dist/config.js

@@ -0,0 +1,73 @@
+import { existsSync, readFileSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { join } from 'node:path';
+function is_string_record(value, label, name) {
+    if (value === undefined)
+        return true;
+    if (!value || typeof value !== 'object' || Array.isArray(value)) {
+        throw new Error(`Invalid MCP server "${name}": ${label} must be an object of string values`);
+    }
+    for (const [key, entry] of Object.entries(value)) {
+        if (typeof entry !== 'string') {
+            throw new Error(`Invalid MCP server "${name}": ${label}.${key} must be a string`);
+        }
+    }
+    return true;
+}
+function parse_server(name, entry) {
+    const type = typeof entry.type === 'string'
+        ? entry.type.trim().toLowerCase()
+        : '';
+    if (type && !['stdio', 'http', 'streamable-http'].includes(type)) {
+        throw new Error(`Invalid MCP server "${name}": unsupported transport type "${type}"`);
+    }
+    if (type === 'http' ||
+        type === 'streamable-http' ||
+        entry.url !== undefined) {
+        if (typeof entry.url !== 'string' || !entry.url.trim()) {
+            throw new Error(`Invalid MCP server "${name}": http transport requires a url`);
+        }
+        is_string_record(entry.headers, 'headers', name);
+        const headers = entry.headers;
+        const config = {
+            name,
+            transport: 'http',
+            url: entry.url.trim(),
+            ...(headers ? { headers } : {}),
+        };
+        return config;
+    }
+    if (typeof entry.command !== 'string' || !entry.command.trim()) {
+        throw new Error(`Invalid MCP server "${name}": stdio transport requires a command`);
+    }
+    if (entry.args !== undefined &&
+        (!Array.isArray(entry.args) ||
+            entry.args.some((value) => typeof value !== 'string'))) {
+        throw new Error(`Invalid MCP server "${name}": args must be an array of strings`);
+    }
+    is_string_record(entry.env, 'env', name);
+    const args = entry.args;
+    const env = entry.env;
+    const config = {
+        name,
+        transport: 'stdio',
+        command: entry.command.trim(),
+        ...(args ? { args } : {}),
+        ...(env ? { env } : {}),
+    };
+    return config;
+}
+function read_config(path) {
+    if (!existsSync(path))
+        return {};
+    const raw = readFileSync(path, 'utf-8');
+    const config = JSON.parse(raw);
+    return config.mcpServers || {};
+}
+export function load_mcp_config(cwd) {
+    const global_servers = read_config(join(homedir(), '.pi', 'agent', 'mcp.json'));
+    const project_servers = read_config(join(cwd, 'mcp.json'));
+    const merged = { ...global_servers, ...project_servers };
+    return Object.entries(merged).map(([name, server]) => parse_server(name, server));
+}
+//# sourceMappingURL=config.js.map

package/dist/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAClC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAoBjC,SAAS,gBAAgB,CACxB,KAAc,EACd,KAAa,EACb,IAAY;IAEZ,IAAI,KAAK,KAAK,SAAS;QAAE,OAAO,IAAI,CAAC;IACrC,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACjE,MAAM,IAAI,KAAK,CACd,uBAAuB,IAAI,MAAM,KAAK,qCAAqC,CAC3E,CAAC;IACH,CAAC;IAED,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAClD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CACd,uBAAuB,IAAI,MAAM,KAAK,IAAI,GAAG,mBAAmB,CAChE,CAAC;QACH,CAAC;IACF,CAAC;IAED,OAAO,IAAI,CAAC;AACb,CAAC;AAED,SAAS,YAAY,CACpB,IAAY,EACZ,KAAwB;IAExB,MAAM,IAAI,GACT,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;QAC7B,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE;QACjC,CAAC,CAAC,EAAE,CAAC;IAEP,IAAI,IAAI,IAAI,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,iBAAiB,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QAClE,MAAM,IAAI,KAAK,CACd,uBAAuB,IAAI,kCAAkC,IAAI,GAAG,CACpE,CAAC;IACH,CAAC;IAED,IACC,IAAI,KAAK,MAAM;QACf,IAAI,KAAK,iBAAiB;QAC1B,KAAK,CAAC,GAAG,KAAK,SAAS,EACtB,CAAC;QACF,IAAI,OAAO,KAAK,CAAC,GAAG,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC;YACxD,MAAM,IAAI,KAAK,CACd,uBAAuB,IAAI,kCAAkC,CAC7D,CAAC;QACH,CAAC;QACD,gBAAgB,CAAC,KAAK,CAAC,OAAO,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,KAAK,CAAC,OAEV,CAAC;QACb,MAAM,MAAM,GAAwB;YACnC,IAAI;YACJ,SAAS,EAAE,MAAM;YACjB,GAAG,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE;YACrB,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC/B,CAAC;QACF,OAAO,MAAM,CAAC;IACf,CAAC;IAED,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;QAChE,MAAM,IAAI,KAAK,CACd,uBAAuB,IAAI,uCAAuC,CAClE,CAAC;IACH,CAAC;IACD,IACC,KAAK,CAAC,IAAI,KAAK,SAAS;QACxB,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC;YAC1B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,EACtD,CAAC;QACF,MAAM,IAAI,KAAK,CACd,uBAAuB,IAAI,qCAAqC,CAChE,CAAC;IACH,CAAC;IACD,gBAAgB,CAAC,KAAK,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;IACzC,MAAM,IAAI,GAAG,KAAK,CAAC,IAA4B,CAAC;IAChD,MAAM,GAAG,GAAG,KAAK,CAAC,GAAyC,CAAC;IAE5D,MAAM,MAAM,GAAyB;QACpC,IAAI;QACJ,SAAS,EAAE,OAAO;QAClB,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE;QAC7B,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACzB,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACvB,CAAC;IACF,OAAO,MAAM,CAAC;AACf,CAAC;AAED,SAAS,WAAW,CAAC,IAAY;IAChC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC;QAAE,OAAO,EAAE,CAAC;IACjC,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACxC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAqB,CAAC;IACnD,OAAO,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,GAAW;IAC1C,MAAM,cAAc,GAAG,WAAW,CACjC,IAAI,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,CAAC,CAC3C,CAAC;IACF,MAAM,eAAe,GAAG,WAAW,CAAC,IAAI,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC,CAAC;IAC3D,MAAM,MAAM,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,eAAe,EAAE,CAAC;IAEzD,OAAO,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,EAAE,CACpD,YAAY,CAAC,IAAI,EAAE,MAAM,CAAC,CAC1B,CAAC;AACH,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+import { type BeforeAgentStartEvent, type ExtensionAPI } from '@mariozechner/pi-coding-agent';
+export declare function should_wait_for_mcp_connections(event: Pick<BeforeAgentStartEvent, 'systemPromptOptions'>): boolean;
+export default function mcp(pi: ExtensionAPI): Promise<void>;

package/dist/index.js

@@ -0,0 +1,301 @@
+import { defineTool, } from '@mariozechner/pi-coding-agent';
+import { McpClient } from './client.js';
+import { load_mcp_config } from './config.js';
+function create_server_states(configs) {
+    return new Map(configs.map((config) => [
+        config.name,
+        {
+            config,
+            tool_names: [],
+            enabled: true,
+            status: 'disconnected',
+        },
+    ]));
+}
+function remove_server_tools_from_active(pi, tool_names) {
+    const tool_set = new Set(tool_names);
+    pi.setActiveTools(pi.getActiveTools().filter((tool) => !tool_set.has(tool)));
+}
+function format_server_status(state) {
+    switch (state.status) {
+        case 'connected':
+            return state.enabled ? 'enabled' : 'disabled';
+        case 'connecting':
+            return state.enabled ? 'connecting' : 'connecting, disabled';
+        case 'failed':
+            return state.enabled ? 'failed' : 'failed, disabled';
+        default:
+            return state.enabled ? 'not connected yet' : 'disabled';
+    }
+}
+function count_pending_enabled_servers(servers) {
+    return Array.from(servers.values()).filter((state) => state.enabled && state.status !== 'connected').length;
+}
+function report_mcp_failure(state, ctx) {
+    const message = `MCP server failed (${state.config.name}): ${state.error}`;
+    if (ctx?.hasUI) {
+        ctx.ui.notify(message, 'warning');
+        return;
+    }
+    console.error(message);
+}
+function update_mcp_status(ctx, servers) {
+    if (!ctx.hasUI)
+        return;
+    if (servers.size === 0) {
+        ctx.ui.setStatus('mcp', undefined);
+        return;
+    }
+    const states = Array.from(servers.values());
+    const enabled = states.filter((state) => state.enabled).length;
+    const connected = states.filter((state) => state.enabled && state.status === 'connected').length;
+    const connecting = states.filter((state) => state.enabled && state.status === 'connecting').length;
+    const failed = states.filter((state) => state.enabled && state.status === 'failed').length;
+    const fragments = [`MCP ${connected}/${enabled} connected`];
+    if (connecting > 0)
+        fragments.push(`${connecting} connecting`);
+    if (failed > 0)
+        fragments.push(`${failed} failed`);
+    ctx.ui.setStatus('mcp', ctx.ui.theme.fg('dim', fragments.join(' · ')));
+}
+function set_connect_feedback(ctx, pending_server_count) {
+    if (!ctx.hasUI) {
+        return () => { };
+    }
+    const label = pending_server_count === 1
+        ? 'Connecting 1 MCP server...'
+        : `Connecting ${pending_server_count} MCP servers...`;
+    ctx.ui.setWorkingMessage(label);
+    ctx.ui.setWorkingIndicator({
+        frames: [
+            ctx.ui.theme.fg('dim', '·'),
+            ctx.ui.theme.fg('muted', '•'),
+            ctx.ui.theme.fg('accent', '●'),
+            ctx.ui.theme.fg('muted', '•'),
+        ],
+        intervalMs: 120,
+    });
+    ctx.ui.setStatus('mcp', ctx.ui.theme.fg('dim', label));
+    return () => {
+        ctx.ui.setWorkingMessage();
+        ctx.ui.setWorkingIndicator();
+    };
+}
+export function should_wait_for_mcp_connections(event) {
+    const selected_tools = event.systemPromptOptions?.selectedTools;
+    return (!selected_tools ||
+        selected_tools.some((tool) => tool.startsWith('mcp__')));
+}
+// Default export for Pi Package / additionalExtensionPaths loading
+export default async function mcp(pi) {
+    let initialized_cwd = null;
+    let servers = new Map();
+    const registered_tool_names = new Set();
+    const ensure_servers = (cwd) => {
+        if (initialized_cwd !== null)
+            return;
+        servers = create_server_states(load_mcp_config(cwd));
+        initialized_cwd = cwd;
+    };
+    const connect_server = async (state, ctx) => {
+        if (state.status === 'connected')
+            return;
+        if (state.connect_promise) {
+            await state.connect_promise;
+            return;
+        }
+        state.connect_promise = (async () => {
+            state.status = 'connecting';
+            state.error = undefined;
+            if (ctx)
+                update_mcp_status(ctx, servers);
+            const client = new McpClient(state.config);
+            try {
+                await client.connect();
+                state.client = client;
+                const mcp_tools = await client.listTools();
+                const tool_names = [];
+                for (const mcp_tool of mcp_tools) {
+                    const tool_name = `mcp__${state.config.name}__${mcp_tool.name}`;
+                    tool_names.push(tool_name);
+                    if (registered_tool_names.has(tool_name))
+                        continue;
+                    registered_tool_names.add(tool_name);
+                    pi.registerTool(defineTool({
+                        name: tool_name,
+                        label: `${state.config.name}: ${mcp_tool.name}`,
+                        description: mcp_tool.description || mcp_tool.name,
+                        parameters: (mcp_tool.inputSchema || {
+                            type: 'object',
+                            properties: {},
+                        }),
+                        execute: async (_id, params) => {
+                            const result = (await state.client.callTool(mcp_tool.name, params));
+                            const text = result?.content
+                                ?.map((c) => c.text || '')
+                                .join('\n') || JSON.stringify(result);
+                            return {
+                                content: [{ type: 'text', text }],
+                                details: {},
+                            };
+                        },
+                    }));
+                }
+                state.tool_names = tool_names;
+                state.status = 'connected';
+                if (!state.enabled) {
+                    remove_server_tools_from_active(pi, state.tool_names);
+                }
+            }
+            catch (error) {
+                state.status = 'failed';
+                state.error =
+                    error instanceof Error ? error.message : String(error);
+                state.client = undefined;
+                await client.disconnect().catch(() => { });
+                report_mcp_failure(state, ctx);
+                throw error;
+            }
+            finally {
+                state.connect_promise = undefined;
+                if (ctx)
+                    update_mcp_status(ctx, servers);
+            }
+        })();
+        await state.connect_promise;
+    };
+    const connect_all_servers = async (options = {}) => {
+        await Promise.allSettled(Array.from(servers.values())
+            .filter((state) => state.enabled)
+            .filter((state) => options.include_failed || state.status !== 'failed')
+            .map((state) => connect_server(state, options.ctx)));
+        if (options.ctx)
+            update_mcp_status(options.ctx, servers);
+    };
+    pi.on('session_start', async (_event, ctx) => {
+        ensure_servers(ctx.cwd);
+        update_mcp_status(ctx, servers);
+        void connect_all_servers({ ctx });
+    });
+    pi.on('before_agent_start', async (event, ctx) => {
+        ensure_servers(ctx.cwd);
+        if (!should_wait_for_mcp_connections(event)) {
+            void connect_all_servers({ ctx });
+            return event;
+        }
+        const pending_server_count = count_pending_enabled_servers(servers);
+        if (pending_server_count === 0) {
+            update_mcp_status(ctx, servers);
+            return event;
+        }
+        const restore_feedback = set_connect_feedback(ctx, pending_server_count);
+        try {
+            await connect_all_servers({ ctx });
+            return event;
+        }
+        finally {
+            restore_feedback();
+            update_mcp_status(ctx, servers);
+        }
+    });
+    pi.registerCommand('mcp', {
+        description: 'Manage MCP servers (list, enable, disable)',
+        getArgumentCompletions: (prefix) => {
+            const parts = prefix.split(' ');
+            if (parts.length <= 1) {
+                return ['list', 'enable', 'disable']
+                    .filter((s) => s.startsWith(prefix))
+                    .map((s) => ({ value: s, label: s }));
+            }
+            if (parts[0] === 'enable' || parts[0] === 'disable') {
+                const name_prefix = parts[1] || '';
+                return Array.from(servers.keys())
+                    .filter((n) => n.startsWith(name_prefix))
+                    .map((n) => ({
+                    value: `${parts[0]} ${n}`,
+                    label: n,
+                }));
+            }
+            return null;
+        },
+        handler: async (args, ctx) => {
+            ensure_servers(ctx.cwd);
+            const [sub, ...rest] = args.trim().split(/\s+/);
+            const name = rest.join(' ');
+            switch (sub || 'list') {
+                case 'list': {
+                    if (servers.size === 0) {
+                        ctx.ui.notify('No MCP servers configured');
+                        return;
+                    }
+                    const lines = [];
+                    for (const [sname, state] of servers.entries()) {
+                        lines.push(`${sname} (${format_server_status(state)}) — ${state.tool_names.length} tools${state.error ? ` — ${state.error}` : ''}`);
+                    }
+                    update_mcp_status(ctx, servers);
+                    ctx.ui.notify(lines.join('\n'));
+                    break;
+                }
+                case 'enable': {
+                    const server = servers.get(name);
+                    if (!server) {
+                        ctx.ui.notify(`Unknown server: ${name}`, 'warning');
+                        return;
+                    }
+                    if (server.enabled && server.status !== 'failed') {
+                        ctx.ui.notify(`${name} already enabled`);
+                        return;
+                    }
+                    server.enabled = true;
+                    if (server.status === 'connected') {
+                        const active = pi.getActiveTools();
+                        pi.setActiveTools([
+                            ...new Set([...active, ...server.tool_names]),
+                        ]);
+                        update_mcp_status(ctx, servers);
+                        ctx.ui.notify(`Enabled ${name}`);
+                        return;
+                    }
+                    if (server.status === 'failed') {
+                        server.status = 'disconnected';
+                        server.error = undefined;
+                    }
+                    update_mcp_status(ctx, servers);
+                    void connect_server(server, ctx);
+                    ctx.ui.notify(`Enabling ${name} and connecting in background`);
+                    break;
+                }
+                case 'disable': {
+                    const server = servers.get(name);
+                    if (!server) {
+                        ctx.ui.notify(`Unknown server: ${name}`, 'warning');
+                        return;
+                    }
+                    if (!server.enabled) {
+                        ctx.ui.notify(`${name} already disabled`);
+                        return;
+                    }
+                    server.enabled = false;
+                    remove_server_tools_from_active(pi, server.tool_names);
+                    update_mcp_status(ctx, servers);
+                    ctx.ui.notify(`Disabled ${name}`);
+                    break;
+                }
+                default:
+                    ctx.ui.notify(`Unknown subcommand: ${sub}. Use list, enable, or disable.`, 'warning');
+            }
+        },
+    });
+    pi.on('session_shutdown', async (_event, ctx) => {
+        await Promise.allSettled(Array.from(servers.values()).map(async (server) => {
+            await server.connect_promise?.catch(() => { });
+            await server.client?.disconnect();
+            server.client = undefined;
+            if (server.status !== 'failed') {
+                server.status = 'disconnected';
+            }
+        }));
+        ctx.ui.setStatus('mcp', undefined);
+    });
+}
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAIN,UAAU,GACV,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,SAAS,EAAwB,MAAM,aAAa,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAY9C,SAAS,oBAAoB,CAC5B,OAA0B;IAE1B,OAAO,IAAI,GAAG,CACb,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;QACvB,MAAM,CAAC,IAAI;QACX;YACC,MAAM;YACN,UAAU,EAAE,EAAE;YACd,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,cAAuB;SAC/B;KACD,CAAC,CACF,CAAC;AACH,CAAC;AAED,SAAS,+BAA+B,CACvC,EAAgB,EAChB,UAAoB;IAEpB,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,CAAC;IACrC,EAAE,CAAC,cAAc,CAChB,EAAE,CAAC,cAAc,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CACzD,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAkB;IAC/C,QAAQ,KAAK,CAAC,MAAM,EAAE,CAAC;QACtB,KAAK,WAAW;YACf,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC;QAC/C,KAAK,YAAY;YAChB,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,sBAAsB,CAAC;QAC9D,KAAK,QAAQ;YACZ,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,kBAAkB,CAAC;QACtD;YACC,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,UAAU,CAAC;IAC1D,CAAC;AACF,CAAC;AAED,SAAS,6BAA6B,CACrC,OAAyC;IAEzC,OAAO,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CACzC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,MAAM,KAAK,WAAW,CACxD,CAAC,MAAM,CAAC;AACV,CAAC;AAED,SAAS,kBAAkB,CAC1B,KAAkB,EAClB,GAAsB;IAEtB,MAAM,OAAO,GAAG,sBAAsB,KAAK,CAAC,MAAM,CAAC,IAAI,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;IAC3E,IAAI,GAAG,EAAE,KAAK,EAAE,CAAC;QAChB,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;QAClC,OAAO;IACR,CAAC;IACD,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;AACxB,CAAC;AAED,SAAS,iBAAiB,CACzB,GAAqB,EACrB,OAAyC;IAEzC,IAAI,CAAC,GAAG,CAAC,KAAK;QAAE,OAAO;IACvB,IAAI,OAAO,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QACxB,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACnC,OAAO;IACR,CAAC;IAED,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC5C,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;IAC/D,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAC9B,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,MAAM,KAAK,WAAW,CACxD,CAAC,MAAM,CAAC;IACT,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAC/B,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,MAAM,KAAK,YAAY,CACzD,CAAC,MAAM,CAAC;IACT,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAC3B,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,CACrD,CAAC,MAAM,CAAC;IAET,MAAM,SAAS,GAAG,CAAC,OAAO,SAAS,IAAI,OAAO,YAAY,CAAC,CAAC;IAC5D,IAAI,UAAU,GAAG,CAAC;QAAE,SAAS,CAAC,IAAI,CAAC,GAAG,UAAU,aAAa,CAAC,CAAC;IAC/D,IAAI,MAAM,GAAG,CAAC;QAAE,SAAS,CAAC,IAAI,CAAC,GAAG,MAAM,SAAS,CAAC,CAAC;IAEnD,GAAG,CAAC,EAAE,CAAC,SAAS,CACf,KAAK,EACL,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAC7C,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAC5B,GAAqB,EACrB,oBAA4B;IAE5B,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;QAChB,OAAO,GAAG,EAAE,GAAE,CAAC,CAAC;IACjB,CAAC;IAED,MAAM,KAAK,GACV,oBAAoB,KAAK,CAAC;QACzB,CAAC,CAAC,4BAA4B;QAC9B,CAAC,CAAC,cAAc,oBAAoB,iBAAiB,CAAC;IAExD,GAAG,CAAC,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAChC,GAAG,CAAC,EAAE,CAAC,mBAAmB,CAAC;QAC1B,MAAM,EAAE;YACP,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,CAAC;YAC3B,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,CAAC;YAC7B,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,CAAC;YAC9B,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,CAAC;SAC7B;QACD,UAAU,EAAE,GAAG;KACf,CAAC,CAAC;IACH,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC;IAEvD,OAAO,GAAG,EAAE;QACX,GAAG,CAAC,EAAE,CAAC,iBAAiB,EAAE,CAAC;QAC3B,GAAG,CAAC,EAAE,CAAC,mBAAmB,EAAE,CAAC;IAC9B,CAAC,CAAC;AACH,CAAC;AAED,MAAM,UAAU,+BAA+B,CAC9C,KAAyD;IAEzD,MAAM,cAAc,GAAG,KAAK,CAAC,mBAAmB,EAAE,aAAa,CAAC;IAChE,OAAO,CACN,CAAC,cAAc;QACf,cAAc,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CACvD,CAAC;AACH,CAAC;AAED,mEAAmE;AACnE,MAAM,CAAC,OAAO,CAAC,KAAK,UAAU,GAAG,CAAC,EAAgB;IACjD,IAAI,eAAe,GAAkB,IAAI,CAAC;IAC1C,IAAI,OAAO,GAAG,IAAI,GAAG,EAAuB,CAAC;IAC7C,MAAM,qBAAqB,GAAG,IAAI,GAAG,EAAU,CAAC;IAEhD,MAAM,cAAc,GAAG,CAAC,GAAW,EAAQ,EAAE;QAC5C,IAAI,eAAe,KAAK,IAAI;YAAE,OAAO;QACrC,OAAO,GAAG,oBAAoB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC;QACrD,eAAe,GAAG,GAAG,CAAC;IACvB,CAAC,CAAC;IAEF,MAAM,cAAc,GAAG,KAAK,EAC3B,KAAkB,EAClB,GAAsB,EACN,EAAE;QAClB,IAAI,KAAK,CAAC,MAAM,KAAK,WAAW;YAAE,OAAO;QACzC,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;YAC3B,MAAM,KAAK,CAAC,eAAe,CAAC;YAC5B,OAAO;QACR,CAAC;QAED,KAAK,CAAC,eAAe,GAAG,CAAC,KAAK,IAAI,EAAE;YACnC,KAAK,CAAC,MAAM,GAAG,YAAY,CAAC;YAC5B,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC;YACxB,IAAI,GAAG;gBAAE,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YAEzC,MAAM,MAAM,GAAG,IAAI,SAAS,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YAC3C,IAAI,CAAC;gBACJ,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;gBACvB,KAAK,CAAC,MAAM,GAAG,MAAM,CAAC;gBAEtB,MAAM,SAAS,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;gBAC3C,MAAM,UAAU,GAAa,EAAE,CAAC;gBAEhC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;oBAClC,MAAM,SAAS,GAAG,QAAQ,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,EAAE,CAAC;oBAChE,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBAE3B,IAAI,qBAAqB,CAAC,GAAG,CAAC,SAAS,CAAC;wBAAE,SAAS;oBACnD,qBAAqB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;oBAErC,EAAE,CAAC,YAAY,CACd,UAAU,CAAC;wBACV,IAAI,EAAE,SAAS;wBACf,KAAK,EAAE,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,IAAI,EAAE;wBAC/C,WAAW,EAAE,QAAQ,CAAC,WAAW,IAAI,QAAQ,CAAC,IAAI;wBAClD,UAAU,EAAE,CAAC,QAAQ,CAAC,WAAW,IAAI;4BACpC,IAAI,EAAE,QAAQ;4BACd,UAAU,EAAE,EAAE;yBACd,CAAmD;wBACpD,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE;4BAC9B,MAAM,MAAM,GAAG,CAAC,MAAM,KAAK,CAAC,MAAO,CAAC,QAAQ,CAC3C,QAAQ,CAAC,IAAI,EACb,MAAiC,CACjC,CAKA,CAAC;4BAEF,MAAM,IAAI,GACT,MAAM,EAAE,OAAO;gCACd,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;iCACzB,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;4BAExC,OAAO;gCACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,CAAC;gCAC1C,OAAO,EAAE,EAAE;6BACX,CAAC;wBACH,CAAC;qBACD,CAAC,CACF,CAAC;gBACH,CAAC;gBAED,KAAK,CAAC,UAAU,GAAG,UAAU,CAAC;gBAC9B,KAAK,CAAC,MAAM,GAAG,WAAW,CAAC;gBAC3B,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;oBACpB,+BAA+B,CAAC,EAAE,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;gBACvD,CAAC;YACF,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,KAAK,CAAC,MAAM,GAAG,QAAQ,CAAC;gBACxB,KAAK,CAAC,KAAK;oBACV,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACxD,KAAK,CAAC,MAAM,GAAG,SAAS,CAAC;gBACzB,MAAM,MAAM,CAAC,UAAU,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;gBAC1C,kBAAkB,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;gBAC/B,MAAM,KAAK,CAAC;YACb,CAAC;oBAAS,CAAC;gBACV,KAAK,CAAC,eAAe,GAAG,SAAS,CAAC;gBAClC,IAAI,GAAG;oBAAE,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YAC1C,CAAC;QACF,CAAC,CAAC,EAAE,CAAC;QAEL,MAAM,KAAK,CAAC,eAAe,CAAC;IAC7B,CAAC,CAAC;IAEF,MAAM,mBAAmB,GAAG,KAAK,EAChC,UAGI,EAAE,EACU,EAAE;QAClB,MAAM,OAAO,CAAC,UAAU,CACvB,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;aAC1B,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;aAChC,MAAM,CACN,CAAC,KAAK,EAAE,EAAE,CACT,OAAO,CAAC,cAAc,IAAI,KAAK,CAAC,MAAM,KAAK,QAAQ,CACpD;aACA,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,cAAc,CAAC,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC,CACpD,CAAC;QACF,IAAI,OAAO,CAAC,GAAG;YAAE,iBAAiB,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;IAC1D,CAAC,CAAC;IAEF,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;QAC5C,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACxB,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAChC,KAAK,mBAAmB,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,oBAAoB,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;QAChD,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACxB,IAAI,CAAC,+BAA+B,CAAC,KAAK,CAAC,EAAE,CAAC;YAC7C,KAAK,mBAAmB,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;YAClC,OAAO,KAAK,CAAC;QACd,CAAC;QAED,MAAM,oBAAoB,GACzB,6BAA6B,CAAC,OAAO,CAAC,CAAC;QACxC,IAAI,oBAAoB,KAAK,CAAC,EAAE,CAAC;YAChC,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;YAChC,OAAO,KAAK,CAAC;QACd,CAAC;QAED,MAAM,gBAAgB,GAAG,oBAAoB,CAC5C,GAAG,EACH,oBAAoB,CACpB,CAAC;QACF,IAAI,CAAC;YACJ,MAAM,mBAAmB,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;YACnC,OAAO,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACV,gBAAgB,EAAE,CAAC;YACnB,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACjC,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,eAAe,CAAC,KAAK,EAAE;QACzB,WAAW,EAAE,4CAA4C;QACzD,sBAAsB,EAAE,CAAC,MAAM,EAAE,EAAE;YAClC,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAChC,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;gBACvB,OAAO,CAAC,MAAM,EAAE,QAAQ,EAAE,SAAS,CAAC;qBAClC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;qBACnC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YACxC,CAAC;YACD,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,SAAS,EAAE,CAAC;gBACrD,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBACnC,OAAO,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;qBAC/B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;qBACxC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;oBACZ,KAAK,EAAE,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE;oBACzB,KAAK,EAAE,CAAC;iBACR,CAAC,CAAC,CAAC;YACN,CAAC;YACD,OAAO,IAAI,CAAC;QACb,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE;YAC5B,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACxB,MAAM,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAChD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAE5B,QAAQ,GAAG,IAAI,MAAM,EAAE,CAAC;gBACvB,KAAK,MAAM,CAAC,CAAC,CAAC;oBACb,IAAI,OAAO,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;wBACxB,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,2BAA2B,CAAC,CAAC;wBAC3C,OAAO;oBACR,CAAC;oBACD,MAAM,KAAK,GAAa,EAAE,CAAC;oBAC3B,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;wBAChD,KAAK,CAAC,IAAI,CACT,GAAG,KAAK,KAAK,oBAAoB,CAAC,KAAK,CAAC,OAAO,KAAK,CAAC,UAAU,CAAC,MAAM,SAAS,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CACvH,CAAC;oBACH,CAAC;oBACD,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;oBAChC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;oBAChC,MAAM;gBACP,CAAC;gBACD,KAAK,QAAQ,CAAC,CAAC,CAAC;oBACf,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;oBACjC,IAAI,CAAC,MAAM,EAAE,CAAC;wBACb,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,mBAAmB,IAAI,EAAE,EAAE,SAAS,CAAC,CAAC;wBACpD,OAAO;oBACR,CAAC;oBACD,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;wBAClD,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,IAAI,kBAAkB,CAAC,CAAC;wBACzC,OAAO;oBACR,CAAC;oBACD,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC;oBACtB,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;wBACnC,MAAM,MAAM,GAAG,EAAE,CAAC,cAAc,EAAE,CAAC;wBACnC,EAAE,CAAC,cAAc,CAAC;4BACjB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,MAAM,EAAE,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC;yBAC7C,CAAC,CAAC;wBACH,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;wBAChC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;wBACjC,OAAO;oBACR,CAAC;oBACD,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;wBAChC,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC;wBAC/B,MAAM,CAAC,KAAK,GAAG,SAAS,CAAC;oBAC1B,CAAC;oBACD,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;oBAChC,KAAK,cAAc,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;oBACjC,GAAG,CAAC,EAAE,CAAC,MAAM,CACZ,YAAY,IAAI,+BAA+B,CAC/C,CAAC;oBACF,MAAM;gBACP,CAAC;gBACD,KAAK,SAAS,CAAC,CAAC,CAAC;oBAChB,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;oBACjC,IAAI,CAAC,MAAM,EAAE,CAAC;wBACb,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,mBAAmB,IAAI,EAAE,EAAE,SAAS,CAAC,CAAC;wBACpD,OAAO;oBACR,CAAC;oBACD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;wBACrB,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,IAAI,mBAAmB,CAAC,CAAC;wBAC1C,OAAO;oBACR,CAAC;oBACD,MAAM,CAAC,OAAO,GAAG,KAAK,CAAC;oBACvB,+BAA+B,CAAC,EAAE,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;oBACvD,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;oBAChC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;oBAClC,MAAM;gBACP,CAAC;gBACD;oBACC,GAAG,CAAC,EAAE,CAAC,MAAM,CACZ,uBAAuB,GAAG,iCAAiC,EAC3D,SAAS,CACT,CAAC;YACJ,CAAC;QACF,CAAC;KACD,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,kBAAkB,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;QAC/C,MAAM,OAAO,CAAC,UAAU,CACvB,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;YACjD,MAAM,MAAM,CAAC,eAAe,EAAE,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAC9C,MAAM,MAAM,CAAC,MAAM,EAAE,UAAU,EAAE,CAAC;YAClC,MAAM,CAAC,MAAM,GAAG,SAAS,CAAC;YAC1B,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAChC,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC;YAChC,CAAC;QACF,CAAC,CAAC,CACF,CAAC;QACF,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IACpC,CAAC,CAAC,CAAC;AACJ,CAAC"}

package/package.json

@@ -0,0 +1,55 @@
+{
+  "name": "@spences10/pi-mcp",
+  "version": "0.0.1",
+  "description": "Pi extension for connecting MCP servers and exposing their tools",
+  "keywords": [
+    "mcp",
+    "model-context-protocol",
+    "pi",
+    "pi-package"
+  ],
+  "license": "MIT",
+  "author": "Scott Spence <me@scottspence.com>",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/spences10/my-pi.git",
+    "directory": "packages/pi-mcp"
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "@mariozechner/pi-coding-agent": "^0.70.2"
+  },
+  "devDependencies": {
+    "@types/node": "^25.6.0",
+    "typescript": "^5.9.3",
+    "vitest": "^4.1.5"
+  },
+  "engines": {
+    "node": ">=22.0.0"
+  },
+  "pi": {
+    "extensions": [
+      "./dist/index.js"
+    ]
+  },
+  "scripts": {
+    "build": "tsc -p tsconfig.build.json",
+    "check": "tsc --noEmit -p tsconfig.json",
+    "test": "vitest run"
+  }
+}