@theia/ai-mcp 1.57.0-next.112

package/src/node/mcp-server-manager-impl.ts

@@ -0,0 +1,93 @@
+// *****************************************************************************
+// Copyright (C) 2024 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+import { injectable } from '@theia/core/shared/inversify';
+import { MCPServerDescription, MCPServerManager } from '../common/mcp-server-manager';
+import { MCPServer } from './mcp-server';
+
+@injectable()
+export class MCPServerManagerImpl implements MCPServerManager {
+
+    protected servers: Map<string, MCPServer> = new Map();
+
+    async stopServer(serverName: string): Promise<void> {
+        const server = this.servers.get(serverName);
+        if (!server) {
+            throw new Error(`MCP server "${serverName}" not found.`);
+        }
+        server.stop();
+        console.log(`MCP server "${serverName}" stopped.`);
+    }
+
+    async getStartedServers(): Promise<string[]> {
+        const startedServers: string[] = [];
+        for (const [name, server] of this.servers.entries()) {
+            if (server.isStarted()) {
+                startedServers.push(name);
+            }
+        }
+        return startedServers;
+    }
+
+    callTool(serverName: string, toolName: string, arg_string: string): ReturnType<MCPServer['callTool']> {
+        const server = this.servers.get(serverName);
+        if (!server) {
+            throw new Error(`MCP server "${toolName}" not found.`);
+        }
+        return server.callTool(toolName, arg_string);
+    }
+
+    async startServer(serverName: string): Promise<void> {
+        const server = this.servers.get(serverName);
+        if (!server) {
+            throw new Error(`MCP server "${serverName}" not found.`);
+        }
+        await server.start();
+    }
+    async getServerNames(): Promise<string[]> {
+        return Array.from(this.servers.keys());
+    }
+
+    public async getTools(serverName: string): ReturnType<MCPServer['getTools']> {
+        const server = this.servers.get(serverName);
+        if (!server) {
+            throw new Error(`MCP server "${serverName}" not found.`);
+        }
+        return server.getTools();
+
+    }
+
+    addOrUpdateServer(description: MCPServerDescription): void {
+        const { name, command, args, env } = description;
+        const existingServer = this.servers.get(name);
+
+        if (existingServer) {
+            existingServer.update(command, args, env);
+        } else {
+            const newServer = new MCPServer(name, command, args, env);
+            this.servers.set(name, newServer);
+        }
+    }
+
+    removeServer(name: string): void {
+        const server = this.servers.get(name);
+        if (server) {
+            server.stop();
+            this.servers.delete(name);
+        } else {
+            console.warn(`MCP server "${name}" not found.`);
+        }
+    }
+}

package/src/node/mcp-server.ts

@@ -0,0 +1,111 @@
+// *****************************************************************************
+// Copyright (C) 2024 EclipseSource GmbH.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio';
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+
+export class MCPServer {
+    private name: string;
+    private command: string;
+    private args?: string[];
+    private client: Client;
+    private env?: { [key: string]: string };
+    private started: boolean = false;
+
+    constructor(name: string, command: string, args?: string[], env?: Record<string, string>) {
+        this.name = name;
+        this.command = command;
+        this.args = args;
+        this.env = env;
+    }
+
+    isStarted(): boolean {
+        return this.started;
+    }
+
+    async start(): Promise<void> {
+        if (this.started) {
+            return;
+        }
+        console.log(`Starting server "${this.name}" with command: ${this.command} and args: ${this.args?.join(' ')} and env: ${JSON.stringify(this.env)}`);
+        // Filter process.env to exclude undefined values
+        const sanitizedEnv: Record<string, string> = Object.fromEntries(
+            Object.entries(process.env).filter((entry): entry is [string, string] => entry[1] !== undefined)
+        );
+
+        const mergedEnv: Record<string, string> = {
+            ...sanitizedEnv,
+            ...(this.env || {})
+        };
+        const transport = new StdioClientTransport({
+            command: this.command,
+            args: this.args,
+            env: mergedEnv,
+        });
+        transport.onerror = error => {
+            console.error('Error: ' + error);
+        };
+
+        this.client = new Client({
+            name: 'theia-client',
+            version: '1.0.0',
+        }, {
+            capabilities: {}
+        });
+        this.client.onerror = error => {
+            console.error('Error in MCP client: ' + error);
+        };
+
+        await this.client.connect(transport);
+        this.started = true;
+    }
+
+    async callTool(toolName: string, arg_string: string): ReturnType<Client['callTool']> {
+        let args;
+        try {
+            args = JSON.parse(arg_string);
+        } catch (error) {
+            console.error(
+                `Failed to parse arguments for calling tool "${toolName}" in MCP server "${this.name}" with command "${this.command}".
+                Invalid JSON: ${arg_string}`,
+                error
+            );
+        }
+        const params = {
+            name: toolName,
+            arguments: args,
+        };
+        return this.client.callTool(params);
+    }
+
+    async getTools(): ReturnType<Client['listTools']> {
+        return this.client.listTools();
+    }
+
+    update(command: string, args?: string[], env?: { [key: string]: string }): void {
+        this.command = command;
+        this.args = args;
+        this.env = env;
+    }
+
+    stop(): void {
+        if (!this.started || !this.client) {
+            return;
+        }
+        console.log(`Stopping MCP server "${this.name}"`);
+        this.client.close();
+        this.started = false;
+    }
+}

package/src/package.spec.ts

@@ -0,0 +1,28 @@
+// *****************************************************************************
+// Copyright (C) 2024 EclipseSource GmbH and others.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Eclipse Public License v. 2.0 which is available at
+// http://www.eclipse.org/legal/epl-2.0.
+//
+// This Source Code may also be made available under the following Secondary
+// Licenses when the conditions for such availability set forth in the Eclipse
+// Public License v. 2.0 are satisfied: GNU General Public License, version 2
+// with the GNU Classpath Exception which is available at
+// https://www.gnu.org/software/classpath/license.html.
+//
+// SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0
+// *****************************************************************************
+
+/* note: this bogus test file is required so that
+   we are able to run mocha unit tests on this
+   package, without having any actual unit tests in it.
+   This way a coverage report will be generated,
+   showing 0% coverage, instead of no report.
+   This file can be removed once we have real unit
+   tests in place. */
+
+describe('ai-mcp package', () => {
+
+    it('support code coverage statistics', () => true);
+});