@aria-cli/tools 1.0.0

package/dist-cjs/mcp/client.js

@@ -0,0 +1,186 @@
+"use strict";
+/**
+ * MCP Multi-Server Client
+ *
+ * Manages connections to multiple MCP servers and aggregates their capabilities.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MCPClient = void 0;
+const events_1 = require("events");
+const connection_js_1 = require("./connection.js");
+class MCPClient extends events_1.EventEmitter {
+    servers = new Map();
+    serverConfigs = new Map();
+    reconnectTimers = new Map();
+    logger;
+    constructor(options) {
+        super();
+        this.logger = options?.logger ?? console;
+    }
+    /**
+     * Connect to an MCP server
+     */
+    async connect(config) {
+        if (this.servers.has(config.name)) {
+            throw new Error(`Server already connected: ${config.name}`);
+        }
+        const connection = new connection_js_1.MCPServerConnection(config);
+        await connection.initialize();
+        // Forward events
+        connection.on("tools/list_changed", () => this.emit("toolsChanged", config.name));
+        connection.on("resources/list_changed", () => this.emit("resourcesChanged", config.name));
+        connection.on("prompts/list_changed", () => this.emit("promptsChanged", config.name));
+        connection.on("exit", (code) => {
+            this.servers.delete(config.name);
+            this.emit("serverExit", config.name, code);
+            // Attempt auto-reconnect on unexpected exit
+            if (code !== 0 && this.serverConfigs.has(config.name)) {
+                this.reconnect(config.name, config).catch((err) => {
+                    this.logger.warn("[MCPClient] Reconnect failed:", err?.message);
+                });
+            }
+        });
+        this.servers.set(config.name, connection);
+        this.serverConfigs.set(config.name, config);
+        return connection;
+    }
+    /**
+     * Disconnect from a specific server
+     */
+    async disconnect(name) {
+        this.serverConfigs.delete(name);
+        // Cancel any pending reconnect timer for this server
+        const timer = this.reconnectTimers.get(name);
+        if (timer !== undefined) {
+            clearTimeout(timer);
+            this.reconnectTimers.delete(name);
+        }
+        const conn = this.servers.get(name);
+        if (conn) {
+            this.servers.delete(name);
+            conn.removeAllListeners();
+            await conn.shutdown();
+        }
+    }
+    /**
+     * Attempt to reconnect to a server with exponential backoff
+     */
+    async reconnect(serverName, config, attempt = 0) {
+        const maxAttempts = 3;
+        const delay = Math.min(1000 * Math.pow(2, attempt), 30000);
+        if (attempt >= maxAttempts)
+            return;
+        await new Promise((resolve) => {
+            const timer = setTimeout(resolve, delay);
+            this.reconnectTimers.set(serverName, timer);
+        });
+        this.reconnectTimers.delete(serverName);
+        // Bail out if server was disconnected while we were waiting
+        if (!this.serverConfigs.has(serverName))
+            return;
+        // Guard against double-reconnect race: if another reconnect already
+        // re-established the connection, skip this attempt.
+        if (this.servers.has(serverName))
+            return;
+        try {
+            await this.connect(config);
+        }
+        catch {
+            await this.reconnect(serverName, config, attempt + 1);
+        }
+    }
+    /**
+     * Disconnect from all servers
+     */
+    async disconnectAll() {
+        // Collect names from both active connections and pending reconnects
+        const names = new Set([...this.servers.keys(), ...this.reconnectTimers.keys()]);
+        await Promise.all([...names].map((name) => this.disconnect(name)));
+    }
+    /**
+     * Get connected server names
+     */
+    getConnectedServers() {
+        return [...this.servers.keys()];
+    }
+    // ========== Tools ==========
+    /**
+     * List all tools from all connected servers
+     */
+    async listAllTools() {
+        const results = await Promise.all([...this.servers.entries()].map(async ([name, conn]) => {
+            try {
+                const tools = await conn.listTools();
+                return tools.map((t) => ({ ...t, server: name }));
+            }
+            catch {
+                return [];
+            }
+        }));
+        return results.flat();
+    }
+    /**
+     * Call a tool on a specific server
+     */
+    async callTool(server, tool, args, abortSignal) {
+        const conn = this.servers.get(server);
+        if (!conn) {
+            return { success: false, message: `Server not connected: ${server}` };
+        }
+        return conn.callTool(tool, args, abortSignal);
+    }
+    // ========== Resources ==========
+    /**
+     * List all resources from all connected servers
+     */
+    async listAllResources() {
+        const results = await Promise.all([...this.servers.entries()].map(async ([name, conn]) => {
+            try {
+                const resources = await conn.listResources();
+                return resources.map((r) => ({ ...r, server: name }));
+            }
+            catch {
+                return [];
+            }
+        }));
+        return results.flat();
+    }
+    /**
+     * Read a resource from a specific server
+     */
+    async readResource(server, uri) {
+        const conn = this.servers.get(server);
+        if (!conn) {
+            throw new Error(`Server not connected: ${server}`);
+        }
+        return conn.readResource(uri);
+    }
+    // ========== Prompts ==========
+    /**
+     * List all prompts from all connected servers
+     */
+    async listAllPrompts() {
+        const results = await Promise.all([...this.servers.entries()].map(async ([name, conn]) => {
+            try {
+                const prompts = await conn.listPrompts();
+                return prompts.map((p) => ({ ...p, server: name }));
+            }
+            catch {
+                return [];
+            }
+        }));
+        return results.flat();
+    }
+    /**
+     * Get a prompt from a specific server
+     */
+    async getPrompt(server, name, args) {
+        const conn = this.servers.get(server);
+        if (!conn) {
+            throw new Error(`Server not connected: ${server}`);
+        }
+        return conn.getPrompt(name, args);
+    }
+}
+exports.MCPClient = MCPClient;
+//# sourceMappingURL=client.js.map

package/dist-cjs/mcp/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/mcp/client.ts"],"names":[],"mappings":";AAAA;;;;GAIG;;;AAEH,mCAAsC;AACtC,mDAAsD;AAWtD,MAAa,SAAU,SAAQ,qBAAY;IACjC,OAAO,GAAqC,IAAI,GAAG,EAAE,CAAC;IACtD,aAAa,GAAiC,IAAI,GAAG,EAAE,CAAC;IACxD,eAAe,GAA+C,IAAI,GAAG,EAAE,CAAC;IACxE,MAAM,CAAS;IAEvB,YAAY,OAA6B;QACvC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,OAAO,CAAC;IAC3C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,MAAuB;QACnC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAClC,MAAM,IAAI,KAAK,CAAC,6BAA6B,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,MAAM,UAAU,GAAG,IAAI,mCAAmB,CAAC,MAAM,CAAC,CAAC;QACnD,MAAM,UAAU,CAAC,UAAU,EAAE,CAAC;QAE9B,iBAAiB;QACjB,UAAU,CAAC,EAAE,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QAClF,UAAU,CAAC,EAAE,CAAC,wBAAwB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1F,UAAU,CAAC,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;QACtF,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YAC7B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACjC,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC3C,4CAA4C;YAC5C,IAAI,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;gBACtD,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;oBAChD,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,+BAA+B,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;gBAClE,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC1C,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC5C,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,IAAY;QAC3B,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAChC,qDAAqD;QACrD,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC7C,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,IAAI,EAAE,CAAC;YACT,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC1B,IAAI,CAAC,kBAAkB,EAAE,CAAC;YAC1B,MAAM,IAAI,CAAC,QAAQ,EAAE,CAAC;QACxB,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,SAAS,CAAC,UAAkB,EAAE,MAAuB,EAAE,OAAO,GAAG,CAAC;QAC9E,MAAM,WAAW,GAAG,CAAC,CAAC;QACtB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,OAAO,CAAC,EAAE,KAAK,CAAC,CAAC;QAC3D,IAAI,OAAO,IAAI,WAAW;YAAE,OAAO;QACnC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE;YAClC,MAAM,KAAK,GAAG,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;YACzC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACxC,4DAA4D;QAC5D,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC;YAAE,OAAO;QAChD,oEAAoE;QACpE,oDAAoD;QACpD,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC;YAAE,OAAO;QACzC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC7B,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,GAAG,CAAC,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa;QACjB,oEAAoE;QACpE,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAChF,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACrE,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,OAAO,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;IAClC,CAAC;IAED,8BAA8B;IAE9B;;OAEG;IACH,KAAK,CAAC,YAAY;QAChB,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACrD,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,SAAS,EAAE,CAAC;gBACrC,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YACpD,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC,CAAC,CACH,CAAC;QACF,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CACZ,MAAc,EACd,IAAY,EACZ,IAAa,EACb,WAAyB;QAEzB,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACtC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,yBAAyB,MAAM,EAAE,EAAE,CAAC;QACxE,CAAC;QACD,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;IAChD,CAAC;IAED,kCAAkC;IAElC;;OAEG;IACH,KAAK,CAAC,gBAAgB;QACpB,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACrD,IAAI,CAAC;gBACH,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;gBAC7C,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YACxD,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC,CAAC,CACH,CAAC;QACF,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,GAAW;QAC5C,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACtC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,yBAAyB,MAAM,EAAE,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IAChC,CAAC;IAED,gCAAgC;IAEhC;;OAEG;IACH,KAAK,CAAC,cAAc;QAClB,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE;YACrD,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC;gBACzC,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YACtD,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC,CAAC,CACH,CAAC;QACF,OAAO,OAAO,CAAC,IAAI,EAAE,CAAC;IACxB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,SAAS,CACb,MAAc,EACd,IAAY,EACZ,IAA6B;QAE7B,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACtC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,yBAAyB,MAAM,EAAE,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IACpC,CAAC;CACF;AAvMD,8BAuMC"}

package/dist-cjs/mcp/connection.d.ts

@@ -0,0 +1,54 @@
+/**
+ * MCP Server Connection
+ *
+ * Manages lifecycle and protocol methods for a single MCP server.
+ */
+import { EventEmitter } from "events";
+import type { MCPServerConfig, MCPCapabilities, MCPTool, MCPResource, MCPResourceContent, MCPPrompt, MCPPromptMessage } from "./types.js";
+import type { ToolResult } from "../types.js";
+export declare class MCPServerConnection extends EventEmitter {
+    private config;
+    private process?;
+    private transport;
+    private rpc;
+    private capabilities;
+    private _initialized;
+    private sseAbortController?;
+    private sseStreamPromise?;
+    private ssePostUrl?;
+    private sseSessionId?;
+    private sseOutput?;
+    private _disconnecting;
+    private _reconnecting;
+    private _reconnectAttempts;
+    private static readonly RECONNECT_BASE_DELAY_MS;
+    private static readonly RECONNECT_MAX_DELAY_MS;
+    private static readonly RECONNECT_MAX_ATTEMPTS;
+    constructor(config: MCPServerConfig);
+    get name(): string;
+    get initialized(): boolean;
+    /**
+     * Initialize connection to MCP server
+     */
+    initialize(): Promise<MCPCapabilities>;
+    /**
+     * Attempt to reconnect with exponential backoff.
+     * Called when the transport drops unexpectedly (not via explicit shutdown).
+     */
+    private attemptReconnect;
+    private initializeStdioTransport;
+    private initializeSSETransport;
+    private postSSEMessages;
+    private consumeSSE;
+    private cleanupAfterInitializeFailure;
+    /**
+     * Gracefully shutdown the connection
+     */
+    shutdown(): Promise<void>;
+    listTools(): Promise<MCPTool[]>;
+    callTool(name: string, args: unknown, abortSignal?: AbortSignal): Promise<ToolResult>;
+    listResources(): Promise<MCPResource[]>;
+    readResource(uri: string): Promise<MCPResourceContent[]>;
+    listPrompts(): Promise<MCPPrompt[]>;
+    getPrompt(name: string, args?: Record<string, string>): Promise<MCPPromptMessage[]>;
+}