feat: add MCPServer

Author: QuantGeekDev

src/core/MCPServer.ts

@@ -0,0 +1,75 @@
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import {
+  CallToolRequestSchema,
+  ListToolsRequestSchema,
+} from "@modelcontextprotocol/sdk/types.js";
+import { ToolLoader, ToolLoaderOptions } from "./toolLoader.js";
+import { BaseTool } from "../tools/BaseTool.js";
+
+export interface MCPServerConfig {
+  name: string;
+  version: string;
+  toolsDir?: string;
+  excludeTools?: string[];
+}
+
+export class MCPServer {
+  private server: Server;
+  private toolsMap: Map<string, BaseTool> = new Map();
+  private toolLoader: ToolLoader;
+
+  constructor(config: MCPServerConfig) {
+    this.server = new Server(
+      {
+        name: config.name,
+        version: config.version,
+      },
+      {
+        capabilities: {
+          tools: {},
+        },
+      }
+    );
+
+    this.toolLoader = new ToolLoader({
+      toolsDir: config.toolsDir,
+      exclude: config.excludeTools,
+    });
+
+    this.setupHandlers();
+  }
+
+  private setupHandlers() {
+    this.server.setRequestHandler(ListToolsRequestSchema, async () => ({
+      tools: Array.from(this.toolsMap.values()).map(
+        (tool) => tool.toolDefinition
+      ),
+    }));
+
+    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+      const tool = this.toolsMap.get(request.params.name);
+      if (!tool) {
+        throw new Error(
+          `Unknown tool: ${request.params.name}. Available tools: ${Array.from(
+            this.toolsMap.keys()
+          ).join(", ")}`
+        );
+      }
+      return tool.toolCall(request);
+    });
+  }
+
+  async start() {
+    try {
+      const tools = await this.toolLoader.loadTools();
+      this.toolsMap = new Map(tools.map((tool: BaseTool) => [tool.name, tool]));
+
+      const transport = new StdioServerTransport();
+      await this.server.connect(transport);
+    } catch (error) {
+      console.error("Server initialization error:", error);
+      throw error;
+    }
+  }
+}