imean-service-engine 1.4.0 → 1.4.2

package/dist/mod.cjs

@@ -13,6 +13,10 @@ var prettier = require('prettier');
 var crypto2 = require('crypto');
 var zlib = require('zlib');
 var nodeWs = require('@hono/node-ws');
+var mcp_js = require('@modelcontextprotocol/sdk/server/mcp.js');
+var types_js = require('@modelcontextprotocol/sdk/types.js');
+var streaming = require('hono/streaming');
+var ulid = require('ulid');
 var dayjs = require('dayjs');
 
 function _interopDefault (e) { return e && e.__esModule ? e : { default: e }; }
@@ -523,6 +527,7 @@ var ActionHandler = class {
       const now = Date.now();
       if (cached !== null && (!this.metadata.ttl || cached?.expireAt > now)) {
         span?.setAttribute("cacheHit", true);
+        span?.setAttribute("cacheKey", cacheKey);
         return cached.data;
       }
     }
@@ -1236,6 +1241,121 @@ async function startCheck(checkers, pass) {
   logger_default.info("[ \u9884\u68C0\u5B8C\u6210 ]");
   if (pass) await pass();
 }
+var HonoTransport = class {
+  constructor(url, stream, closeStream) {
+    this.url = url;
+    this.stream = stream;
+    this.closeStream = closeStream;
+    this.sessionId = ulid.ulid();
+  }
+  sessionId;
+  onclose;
+  onerror;
+  onmessage;
+  async start() {
+    await this.stream.writeSSE({
+      event: "endpoint",
+      data: `${encodeURI(this.url)}?sessionId=${this.sessionId}`
+    });
+    this.stream.onAbort(() => {
+      this.close();
+    });
+  }
+  async handleMessage(message) {
+    let parsedMessage;
+    try {
+      parsedMessage = types_js.JSONRPCMessageSchema.parse(message);
+    } catch (error) {
+      this.onerror?.(error);
+      throw error;
+    }
+    this.onmessage?.(parsedMessage);
+  }
+  async send(message) {
+    await this.stream.writeln(
+      `event: message
+data: ${JSON.stringify(message)}
+
+`
+    );
+  }
+  async close() {
+    this.onclose?.();
+    this.closeStream();
+  }
+};
+var ModelContextProtocolPlugin = class extends Plugin {
+  mcpServer;
+  transports = {};
+  registerMcpTools(engine) {
+    const modules = engine.getModules(true);
+    for (const module of Object.values(modules)) {
+      for (const action of Object.values(module.actions)) {
+        if (action.mcp?.type !== "tool") {
+          continue;
+        }
+        const args = {};
+        const argsIndex = {};
+        for (const [index, param] of (action.params ?? []).entries()) {
+          args[param.description] = param;
+          argsIndex[param.description] = index;
+        }
+        this.mcpServer.tool(
+          `${module.name}.${action.name}`,
+          action.description ?? "",
+          args,
+          async (params) => {
+            const argsList = [];
+            for (const [key, value] of Object.entries(params)) {
+              argsList[argsIndex[key]] = value;
+            }
+            const result = await engine.getActionHandler(module.name, action.name).handle(argsList);
+            return {
+              content: [{ type: "text", text: JSON.stringify(result) }]
+            };
+          }
+        );
+      }
+    }
+  }
+  initialize = async (engine) => {
+    const app = engine.getApp();
+    this.mcpServer = new mcp_js.McpServer({
+      name: engine.options.name,
+      version: engine.options.version
+    });
+    this.registerMcpTools(engine);
+    app.get(`${engine.options.prefix}/mcp_sse`, async (ctx) => {
+      return streaming.streamSSE(ctx, async (stream) => {
+        return new Promise(async (resolve) => {
+          const transport = new HonoTransport(
+            `${engine.options.prefix}/mcp_messages`,
+            stream,
+            () => {
+              delete this.transports[transport.sessionId];
+              resolve();
+            }
+          );
+          this.transports[transport.sessionId] = transport;
+          await this.mcpServer.connect(transport);
+        });
+      });
+    });
+    app.post(`${engine.options.prefix}/mcp_messages`, async (ctx) => {
+      const sessionId = ctx.req.query("sessionId");
+      if (!sessionId) {
+        return ctx.text("No transport found for sessionId", 400);
+      }
+      const transport = this.transports[sessionId];
+      const message = await ctx.req.json();
+      await transport.handleMessage(message);
+      return ctx.text("Accepted", 202);
+    });
+    logger_default.info(
+      `ModelContextProtocolPlugin endpoint: ${engine.options.prefix}/mcp_sse`
+    );
+  };
+};
 
 Object.defineProperty(exports, "dayjs", {
   enumerable: true,
@@ -1245,6 +1365,7 @@ exports.Action = Action;
 exports.CacheAdapter = CacheAdapter;
 exports.MemoryCacheAdapter = MemoryCacheAdapter;
 exports.Microservice = Microservice;
+exports.ModelContextProtocolPlugin = ModelContextProtocolPlugin;
 exports.Module = Module;
 exports.Plugin = Plugin;
 exports.RedisCacheAdapter = RedisCacheAdapter;

package/dist/mod.d.cts

@@ -260,6 +260,13 @@ interface PreStartChecker {
 }
 declare function startCheck(checkers: PreStartChecker[], pass?: () => void | Promise<void>): Promise<void>;
 
+declare class ModelContextProtocolPlugin extends Plugin {
+    private mcpServer;
+    private transports;
+    private registerMcpTools;
+    initialize: (engine: Microservice) => Promise<void>;
+}
+
 /**
  * 用于给微服务模块方法添加的注解
  */
@@ -279,4 +286,4 @@ declare function Schedule(options: ScheduleOptions): Function;
 
 declare const logger: winston.Logger;
 
-export { Action, type ActionErrorEvent, type ActionMetadata, type ActionOptions, CacheAdapter, type CacheFn, type EtcdConfig, type EventServiceInfo, type McpOptions, MemoryCacheAdapter, Microservice, type MicroserviceOptions, Module, type ModuleInfo, type ModuleMetadata, type ModuleOptions, Plugin, type PreStartChecker, RedisCacheAdapter, Schedule, type ScheduleMetadata, ScheduleMode, type ScheduleOptions, ServiceContext, type ServiceInfo, type ServiceStats, type StatisticsEvent, type StreamResponse, logger, startCheck };
+export { Action, type ActionErrorEvent, type ActionMetadata, type ActionOptions, CacheAdapter, type CacheFn, type EtcdConfig, type EventServiceInfo, type McpOptions, MemoryCacheAdapter, Microservice, type MicroserviceOptions, ModelContextProtocolPlugin, Module, type ModuleInfo, type ModuleMetadata, type ModuleOptions, Plugin, type PreStartChecker, RedisCacheAdapter, Schedule, type ScheduleMetadata, ScheduleMode, type ScheduleOptions, ServiceContext, type ServiceInfo, type ServiceStats, type StatisticsEvent, type StreamResponse, logger, startCheck };

package/dist/mod.d.ts

@@ -260,6 +260,13 @@ interface PreStartChecker {
 }
 declare function startCheck(checkers: PreStartChecker[], pass?: () => void | Promise<void>): Promise<void>;
 
+declare class ModelContextProtocolPlugin extends Plugin {
+    private mcpServer;
+    private transports;
+    private registerMcpTools;
+    initialize: (engine: Microservice) => Promise<void>;
+}
+
 /**
  * 用于给微服务模块方法添加的注解
  */
@@ -279,4 +286,4 @@ declare function Schedule(options: ScheduleOptions): Function;
 
 declare const logger: winston.Logger;
 
-export { Action, type ActionErrorEvent, type ActionMetadata, type ActionOptions, CacheAdapter, type CacheFn, type EtcdConfig, type EventServiceInfo, type McpOptions, MemoryCacheAdapter, Microservice, type MicroserviceOptions, Module, type ModuleInfo, type ModuleMetadata, type ModuleOptions, Plugin, type PreStartChecker, RedisCacheAdapter, Schedule, type ScheduleMetadata, ScheduleMode, type ScheduleOptions, ServiceContext, type ServiceInfo, type ServiceStats, type StatisticsEvent, type StreamResponse, logger, startCheck };
+export { Action, type ActionErrorEvent, type ActionMetadata, type ActionOptions, CacheAdapter, type CacheFn, type EtcdConfig, type EventServiceInfo, type McpOptions, MemoryCacheAdapter, Microservice, type MicroserviceOptions, ModelContextProtocolPlugin, Module, type ModuleInfo, type ModuleMetadata, type ModuleOptions, Plugin, type PreStartChecker, RedisCacheAdapter, Schedule, type ScheduleMetadata, ScheduleMode, type ScheduleOptions, ServiceContext, type ServiceInfo, type ServiceStats, type StatisticsEvent, type StreamResponse, logger, startCheck };

package/dist/mod.js

@@ -13,6 +13,10 @@ import crypto2 from 'node:crypto';
 import { brotliDecompress } from 'node:zlib';
 import { brotliCompress, constants } from 'zlib';
 import { createNodeWebSocket } from '@hono/node-ws';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { JSONRPCMessageSchema } from '@modelcontextprotocol/sdk/types.js';
+import { streamSSE } from 'hono/streaming';
+import { ulid } from 'ulid';
 export { default as dayjs } from 'dayjs';
 
 // mod.ts
@@ -514,6 +518,7 @@ var ActionHandler = class {
       const now = Date.now();
       if (cached !== null && (!this.metadata.ttl || cached?.expireAt > now)) {
         span?.setAttribute("cacheHit", true);
+        span?.setAttribute("cacheKey", cacheKey);
         return cached.data;
       }
     }
@@ -1227,5 +1232,120 @@ async function startCheck(checkers, pass) {
   logger_default.info("[ \u9884\u68C0\u5B8C\u6210 ]");
   if (pass) await pass();
 }
+var HonoTransport = class {
+  constructor(url, stream, closeStream) {
+    this.url = url;
+    this.stream = stream;
+    this.closeStream = closeStream;
+    this.sessionId = ulid();
+  }
+  sessionId;
+  onclose;
+  onerror;
+  onmessage;
+  async start() {
+    await this.stream.writeSSE({
+      event: "endpoint",
+      data: `${encodeURI(this.url)}?sessionId=${this.sessionId}`
+    });
+    this.stream.onAbort(() => {
+      this.close();
+    });
+  }
+  async handleMessage(message) {
+    let parsedMessage;
+    try {
+      parsedMessage = JSONRPCMessageSchema.parse(message);
+    } catch (error) {
+      this.onerror?.(error);
+      throw error;
+    }
+    this.onmessage?.(parsedMessage);
+  }
+  async send(message) {
+    await this.stream.writeln(
+      `event: message
+data: ${JSON.stringify(message)}
+
+`
+    );
+  }
+  async close() {
+    this.onclose?.();
+    this.closeStream();
+  }
+};
+var ModelContextProtocolPlugin = class extends Plugin {
+  mcpServer;
+  transports = {};
+  registerMcpTools(engine) {
+    const modules = engine.getModules(true);
+    for (const module of Object.values(modules)) {
+      for (const action of Object.values(module.actions)) {
+        if (action.mcp?.type !== "tool") {
+          continue;
+        }
+        const args = {};
+        const argsIndex = {};
+        for (const [index, param] of (action.params ?? []).entries()) {
+          args[param.description] = param;
+          argsIndex[param.description] = index;
+        }
+        this.mcpServer.tool(
+          `${module.name}.${action.name}`,
+          action.description ?? "",
+          args,
+          async (params) => {
+            const argsList = [];
+            for (const [key, value] of Object.entries(params)) {
+              argsList[argsIndex[key]] = value;
+            }
+            const result = await engine.getActionHandler(module.name, action.name).handle(argsList);
+            return {
+              content: [{ type: "text", text: JSON.stringify(result) }]
+            };
+          }
+        );
+      }
+    }
+  }
+  initialize = async (engine) => {
+    const app = engine.getApp();
+    this.mcpServer = new McpServer({
+      name: engine.options.name,
+      version: engine.options.version
+    });
+    this.registerMcpTools(engine);
+    app.get(`${engine.options.prefix}/mcp_sse`, async (ctx) => {
+      return streamSSE(ctx, async (stream) => {
+        return new Promise(async (resolve) => {
+          const transport = new HonoTransport(
+            `${engine.options.prefix}/mcp_messages`,
+            stream,
+            () => {
+              delete this.transports[transport.sessionId];
+              resolve();
+            }
+          );
+          this.transports[transport.sessionId] = transport;
+          await this.mcpServer.connect(transport);
+        });
+      });
+    });
+    app.post(`${engine.options.prefix}/mcp_messages`, async (ctx) => {
+      const sessionId = ctx.req.query("sessionId");
+      if (!sessionId) {
+        return ctx.text("No transport found for sessionId", 400);
+      }
+      const transport = this.transports[sessionId];
+      const message = await ctx.req.json();
+      await transport.handleMessage(message);
+      return ctx.text("Accepted", 202);
+    });
+    logger_default.info(
+      `ModelContextProtocolPlugin endpoint: ${engine.options.prefix}/mcp_sse`
+    );
+  };
+};
 
-export { Action, CacheAdapter, MemoryCacheAdapter, Microservice, Module, Plugin, RedisCacheAdapter, Schedule, ScheduleMode, ServiceContext, logger_default as logger, startCheck };
+export { Action, CacheAdapter, MemoryCacheAdapter, Microservice, ModelContextProtocolPlugin, Module, Plugin, RedisCacheAdapter, Schedule, ScheduleMode, ServiceContext, logger_default as logger, startCheck };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "imean-service-engine",
-  "version": "1.4.0",
+  "version": "1.4.2",
   "description": "microservice engine",
   "keywords": [
     "microservice",
@@ -30,12 +30,6 @@
     "README.md",
     "LICENSE"
   ],
-  "scripts": {
-    "dev": "tsx watch dev/index.ts",
-    "build": "tsup",
-    "test": "vitest run",
-    "prepublishOnly": "npm run build && npm run test"
-  },
   "dependencies": {
     "@hono/node-server": "^1.13.7",
     "@hono/node-ws": "^1.0.6",
@@ -83,5 +77,10 @@
   },
   "engines": {
     "node": ">=20"
+  },
+  "scripts": {
+    "dev": "tsx watch dev/index.ts",
+    "build": "tsup",
+    "test": "vitest run"
   }
-}
+}