imean-service-engine 1.7.2 → 1.8.0

package/dist/mod.cjs

@@ -14,10 +14,6 @@ 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 html = require('hono/html');
 var jsxRuntime = require('hono/jsx/jsx-runtime');
 var dayjs = require('dayjs');
@@ -92,8 +88,7 @@ function Action(options) {
         printError: options.printError ?? false,
         cache: options.cache,
         ttl: options.ttl,
-        stream: options.stream,
-        mcp: options.mcp
+        stream: options.stream
       };
       prototype[ACTION_METADATA] = existingMetadata;
     });
@@ -471,14 +466,14 @@ var ActionHandler = class {
     this.microservice = microservice;
     this.moduleName = moduleName;
   }
-  async handle(req) {
+  async handle(req, ctx) {
     return await tracer2.startActiveSpan(
       `handle ${this.moduleName}.${this.actionName}`,
       async (span) => {
         span.setAttribute("module", this.moduleName);
         span.setAttribute("action", this.actionName);
         try {
-          return await this._handle(req);
+          return await this._handle(req, ctx);
         } catch (error) {
           span.recordException(error);
           span.setStatus({
@@ -523,7 +518,7 @@ var ActionHandler = class {
       span.end();
     }
   }
-  async _handle(req) {
+  async _handle(req, ctx) {
     const span = api.trace.getActiveSpan();
     const { args, requestHash } = await this._validate(req);
     const cacheHash = typeof this.metadata.cache === "function" ? this.metadata.cache(...args) : requestHash;
@@ -541,7 +536,7 @@ var ActionHandler = class {
     try {
       const result = await this.moduleInstance[this.actionName].apply(
         this.moduleInstance,
-        args
+        args.concat(ctx)
       );
       if (this.metadata.stream) {
         span?.setAttribute("stream", true);
@@ -610,9 +605,10 @@ function Page(options) {
       existingMetadata[methodName] = {
         name: methodName,
         description: options.description || "",
-        method: options.method,
+        method: options.method || "get",
         path: normalizePath(options.path),
-        absolutePath: options.absolutePath ?? false
+        absolutePath: options.absolutePath ?? false,
+        middlewares: options.middlewares ?? []
       };
       prototype[PAGE_METADATA] = existingMetadata;
     });
@@ -648,6 +644,9 @@ var PageHandler = class {
             this.moduleInstance,
             [ctx]
           );
+          if (result instanceof Response) {
+            return result;
+          }
           return ctx.html(result);
         } catch (error) {
           span.recordException(error);
@@ -664,8 +663,9 @@ var PageHandler = class {
   }
 };
 var WebSocketHandler = class {
-  constructor(microservice, options) {
+  constructor(microservice, ctx, options) {
     this.microservice = microservice;
+    this.ctx = ctx;
     this.options = {
       timeout: options?.timeout || 3e4
     };
@@ -705,7 +705,7 @@ var WebSocketHandler = class {
         await this.sendMessage(ws, { type: "pong" });
         return;
       }
-      const response = await this.handleRequest(ws, message);
+      const response = await this.handleRequest(ws, message, this.ctx);
       if (response) {
         await this.sendMessage(ws, response);
       }
@@ -732,7 +732,7 @@ var WebSocketHandler = class {
   onError(_error, ws) {
     this.onClose(ws);
   }
-  async handleRequest(ws, message) {
+  async handleRequest(ws, message, ctx) {
     if (!message.id || !message.module || !message.action) {
       throw new Error("Invalid request message");
     }
@@ -768,7 +768,7 @@ var WebSocketHandler = class {
           message.action
         );
         const args = message.args ? Object.values(message.args) : [];
-        const result = await handler.handle(args);
+        const result = await handler.handle(args, ctx);
         if (handler.metadata.stream) {
           try {
             for await (const value of result) {
@@ -929,20 +929,20 @@ var Microservice = class {
         );
         this.actionHandlers.set(`${moduleName}.${actionName}`, handler);
         logger_default.info(
-          `[ \u6CE8\u518C\u52A8\u4F5C ] ${moduleName}.${actionName} ${actionMetadata.description} ${actionMetadata.mcp ? "MCP:" + actionMetadata.mcp?.type : ""}`
+          `[ \u6CE8\u518C\u52A8\u4F5C ] ${moduleName}.${actionName} ${actionMetadata.description}`
         );
       }
       for (const [_, page] of Object.entries(pages)) {
-        const handler = new PageHandler(
-          moduleInstance,
-          page,
-          moduleName
-        );
+        const handler = new PageHandler(moduleInstance, page, moduleName);
         this.pageHandlers.set(`${moduleName}.${page.name}`, handler);
         const paths = Array.isArray(page.path) ? page.path : [page.path];
         for (let path of paths) {
           path = page.absolutePath ? path : `${this.options.prefix}${path}`;
-          this.app[page.method](path, (ctx) => handler.handle(ctx));
+          this.app[page.method](
+            path,
+            ...page.middlewares ?? [],
+            (ctx) => handler.handle(ctx)
+          );
           logger_default.info(
             `[ \u6CE8\u518C\u9875\u9762 ] ${moduleName}.${page.name} ${page.method.toUpperCase()} ${page.path} ${page.description}`
           );
@@ -1004,11 +1004,10 @@ var Microservice = class {
     });
     this.app.post(`${prefix}/:moduleName/:actionName`, this.handleRequest);
     if (this.options.websocket?.enabled) {
-      this.wsHandler = new WebSocketHandler(this);
       this.app.get(
         `${prefix}/ws`,
-        this.nodeWebSocket.upgradeWebSocket((_ctx) => {
-          const wsHandler = new WebSocketHandler(this, {
+        this.nodeWebSocket.upgradeWebSocket((ctx) => {
+          const wsHandler = new WebSocketHandler(this, ctx, {
             timeout: this.options.websocket?.timeout
           });
           return {
@@ -1230,7 +1229,9 @@ Received SIGTERM signal`);
   async waitForActiveRequests() {
     const timeout = this.options.gracefulShutdown?.timeout || 3e4;
     const startTime = Date.now();
-    logger_default.info(`Waiting for ${this.activeRequests.size} active requests to complete...`);
+    logger_default.info(
+      `Waiting for ${this.activeRequests.size} active requests to complete...`
+    );
     return new Promise((resolve) => {
       const checkInterval = setInterval(() => {
         const elapsed = Date.now() - startTime;
@@ -1240,10 +1241,14 @@ Received SIGTERM signal`);
           resolve();
         } else if (elapsed >= timeout) {
           clearInterval(checkInterval);
-          logger_default.warn(`Timeout waiting for requests to complete. ${this.activeRequests.size} requests still active`);
+          logger_default.warn(
+            `Timeout waiting for requests to complete. ${this.activeRequests.size} requests still active`
+          );
           resolve();
         } else {
-          logger_default.info(`Still waiting for ${this.activeRequests.size} requests... (${elapsed}ms elapsed)`);
+          logger_default.info(
+            `Still waiting for ${this.activeRequests.size} requests... (${elapsed}ms elapsed)`
+          );
         }
       }, 1e3);
     });
@@ -1265,7 +1270,10 @@ Received SIGTERM signal`);
         await Promise.race([
           Promise.resolve(hook.cleanup()),
           new Promise(
-            (_, reject) => setTimeout(() => reject(new Error(`Cleanup hook ${hook.name} timeout`)), timeout)
+            (_, reject) => setTimeout(
+              () => reject(new Error(`Cleanup hook ${hook.name} timeout`)),
+              timeout
+            )
           )
         ]);
         logger_default.info(`Cleanup hook ${hook.name} completed successfully`);
@@ -1346,10 +1354,13 @@ Received SIGTERM signal`);
     const requestId = crypto.randomUUID();
     const startTime = Date.now();
     if (this.status === "shutting_down") {
-      return ctx.json({
-        success: false,
-        error: "Service is shutting down"
-      }, 503);
+      return ctx.json(
+        {
+          success: false,
+          error: "Service is shutting down"
+        },
+        503
+      );
     }
     try {
       const paramsText = await ctx.req.text();
@@ -1360,7 +1371,7 @@ Received SIGTERM signal`);
         startTime,
         params: paramsText
       });
-      const result = await handler.handle(paramsText);
+      const result = await handler.handle(paramsText, ctx);
       if (handler.metadata.stream) {
         const encoder = new TextEncoder();
         const microservice = this;
@@ -1419,121 +1430,6 @@ Received SIGTERM signal`);
     await this.waitingInitialization;
   }
 };
-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`
-    );
-  };
-};
 var DEFAULT_FAVICON = /* @__PURE__ */ jsxRuntime.jsx(
   "link",
   {
@@ -1746,7 +1642,6 @@ exports.CacheAdapter = CacheAdapter;
 exports.HtmxLayout = HtmxLayout;
 exports.MemoryCacheAdapter = MemoryCacheAdapter;
 exports.Microservice = Microservice;
-exports.ModelContextProtocolPlugin = ModelContextProtocolPlugin;
 exports.Module = Module;
 exports.Page = Page;
 exports.PageRenderPlugin = PageRenderPlugin;

package/dist/mod.d.cts

@@ -3,7 +3,7 @@ export * from 'zod';
 import { Redis, Cluster } from 'ioredis';
 import { LRUCache } from 'lru-cache';
 import { Lease, Etcd3 } from 'etcd3';
-import { Hono } from 'hono';
+import { MiddlewareHandler, Context, Hono } from 'hono';
 export { default as dayjs } from 'dayjs';
 import winston from 'winston';
 import * as hono_utils_html from 'hono/utils/html';
@@ -27,9 +27,6 @@ declare class MemoryCacheAdapter extends CacheAdapter {
 }
 
 type CacheFn = (...args: any[]) => any;
-type McpOptions = {
-    type: "tool";
-};
 interface ActionOptions {
     params?: z.ZodType<any>[];
     returns?: z.ZodType<any>;
@@ -39,16 +36,17 @@ interface ActionOptions {
     cache?: boolean | CacheFn;
     ttl?: number;
     stream?: boolean;
-    mcp?: McpOptions;
 }
 interface PageOptions {
-    method: "get" | "post" | "put" | "delete" | "patch" | "options";
+    method?: "get" | "post" | "put" | "delete" | "patch" | "options";
+    middlewares?: MiddlewareHandler[];
     path: string | string[];
     absolutePath?: boolean;
     description?: string;
 }
 interface PageMetadata extends PageOptions {
     name: string;
+    method: "get" | "post" | "put" | "delete" | "patch" | "options";
 }
 interface ActionMetadata extends ActionOptions {
     name: string;
@@ -213,7 +211,7 @@ declare class ActionHandler {
     private microservice;
     private moduleName;
     constructor(moduleInstance: any, actionName: string, metadata: ActionMetadata, microservice: Microservice, moduleName: string);
-    handle(req: string | any[]): Promise<any>;
+    handle(req: string | any[], ctx: Context): Promise<any>;
     private _validate;
     private _handle;
 }
@@ -316,13 +314,6 @@ declare class Microservice {
     init(): Promise<void>;
 }
 
-declare class ModelContextProtocolPlugin extends Plugin {
-    private mcpServer;
-    private transports;
-    private registerMcpTools;
-    initialize: (engine: Microservice) => Promise<void>;
-}
-
 declare const BaseLayout: (props?: {
     children?: any;
     title?: string;
@@ -381,4 +372,4 @@ declare function Schedule(options: ScheduleOptions): Function;
 
 declare const logger: winston.Logger;
 
-export { Action, type ActionErrorEvent, type ActionMetadata, type ActionOptions, BaseLayout, CacheAdapter, type CacheFn, type CleanupHook, type EtcdConfig, type EventServiceInfo, HtmxLayout, type McpOptions, MemoryCacheAdapter, Microservice, type MicroserviceOptions, ModelContextProtocolPlugin, Module, type ModuleInfo, type ModuleMetadata, type ModuleOptions, Page, type PageMetadata, type PageOptions, PageRenderPlugin, type PageRenderPluginOptions, Plugin, type PreStartChecker, RedisCacheAdapter, type RequestInfo, Schedule, type ScheduleMetadata, ScheduleMode, type ScheduleOptions, ServiceContext, type ServiceInfo, ServiceInfoCards, type ServiceStats, ServiceStatusPage, type StatisticsEvent, type StreamResponse, logger, startCheck };
+export { Action, type ActionErrorEvent, type ActionMetadata, type ActionOptions, BaseLayout, CacheAdapter, type CacheFn, type CleanupHook, type EtcdConfig, type EventServiceInfo, HtmxLayout, MemoryCacheAdapter, Microservice, type MicroserviceOptions, Module, type ModuleInfo, type ModuleMetadata, type ModuleOptions, Page, type PageMetadata, type PageOptions, PageRenderPlugin, type PageRenderPluginOptions, Plugin, type PreStartChecker, RedisCacheAdapter, type RequestInfo, Schedule, type ScheduleMetadata, ScheduleMode, type ScheduleOptions, ServiceContext, type ServiceInfo, ServiceInfoCards, type ServiceStats, ServiceStatusPage, type StatisticsEvent, type StreamResponse, logger, startCheck };

package/dist/mod.d.ts

@@ -3,7 +3,7 @@ export * from 'zod';
 import { Redis, Cluster } from 'ioredis';
 import { LRUCache } from 'lru-cache';
 import { Lease, Etcd3 } from 'etcd3';
-import { Hono } from 'hono';
+import { MiddlewareHandler, Context, Hono } from 'hono';
 export { default as dayjs } from 'dayjs';
 import winston from 'winston';
 import * as hono_utils_html from 'hono/utils/html';
@@ -27,9 +27,6 @@ declare class MemoryCacheAdapter extends CacheAdapter {
 }
 
 type CacheFn = (...args: any[]) => any;
-type McpOptions = {
-    type: "tool";
-};
 interface ActionOptions {
     params?: z.ZodType<any>[];
     returns?: z.ZodType<any>;
@@ -39,16 +36,17 @@ interface ActionOptions {
     cache?: boolean | CacheFn;
     ttl?: number;
     stream?: boolean;
-    mcp?: McpOptions;
 }
 interface PageOptions {
-    method: "get" | "post" | "put" | "delete" | "patch" | "options";
+    method?: "get" | "post" | "put" | "delete" | "patch" | "options";
+    middlewares?: MiddlewareHandler[];
     path: string | string[];
     absolutePath?: boolean;
     description?: string;
 }
 interface PageMetadata extends PageOptions {
     name: string;
+    method: "get" | "post" | "put" | "delete" | "patch" | "options";
 }
 interface ActionMetadata extends ActionOptions {
     name: string;
@@ -213,7 +211,7 @@ declare class ActionHandler {
     private microservice;
     private moduleName;
     constructor(moduleInstance: any, actionName: string, metadata: ActionMetadata, microservice: Microservice, moduleName: string);
-    handle(req: string | any[]): Promise<any>;
+    handle(req: string | any[], ctx: Context): Promise<any>;
     private _validate;
     private _handle;
 }
@@ -316,13 +314,6 @@ declare class Microservice {
     init(): Promise<void>;
 }
 
-declare class ModelContextProtocolPlugin extends Plugin {
-    private mcpServer;
-    private transports;
-    private registerMcpTools;
-    initialize: (engine: Microservice) => Promise<void>;
-}
-
 declare const BaseLayout: (props?: {
     children?: any;
     title?: string;
@@ -381,4 +372,4 @@ declare function Schedule(options: ScheduleOptions): Function;
 
 declare const logger: winston.Logger;
 
-export { Action, type ActionErrorEvent, type ActionMetadata, type ActionOptions, BaseLayout, CacheAdapter, type CacheFn, type CleanupHook, type EtcdConfig, type EventServiceInfo, HtmxLayout, type McpOptions, MemoryCacheAdapter, Microservice, type MicroserviceOptions, ModelContextProtocolPlugin, Module, type ModuleInfo, type ModuleMetadata, type ModuleOptions, Page, type PageMetadata, type PageOptions, PageRenderPlugin, type PageRenderPluginOptions, Plugin, type PreStartChecker, RedisCacheAdapter, type RequestInfo, Schedule, type ScheduleMetadata, ScheduleMode, type ScheduleOptions, ServiceContext, type ServiceInfo, ServiceInfoCards, type ServiceStats, ServiceStatusPage, type StatisticsEvent, type StreamResponse, logger, startCheck };
+export { Action, type ActionErrorEvent, type ActionMetadata, type ActionOptions, BaseLayout, CacheAdapter, type CacheFn, type CleanupHook, type EtcdConfig, type EventServiceInfo, HtmxLayout, MemoryCacheAdapter, Microservice, type MicroserviceOptions, Module, type ModuleInfo, type ModuleMetadata, type ModuleOptions, Page, type PageMetadata, type PageOptions, PageRenderPlugin, type PageRenderPluginOptions, Plugin, type PreStartChecker, RedisCacheAdapter, type RequestInfo, Schedule, type ScheduleMetadata, ScheduleMode, type ScheduleOptions, ServiceContext, type ServiceInfo, ServiceInfoCards, type ServiceStats, ServiceStatusPage, type StatisticsEvent, type StreamResponse, logger, startCheck };

package/dist/mod.js

@@ -14,10 +14,6 @@ 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';
 import { html } from 'hono/html';
 import { jsx, jsxs } from 'hono/jsx/jsx-runtime';
 export { default as dayjs } from 'dayjs';
@@ -83,8 +79,7 @@ function Action(options) {
         printError: options.printError ?? false,
         cache: options.cache,
         ttl: options.ttl,
-        stream: options.stream,
-        mcp: options.mcp
+        stream: options.stream
       };
       prototype[ACTION_METADATA] = existingMetadata;
     });
@@ -462,14 +457,14 @@ var ActionHandler = class {
     this.microservice = microservice;
     this.moduleName = moduleName;
   }
-  async handle(req) {
+  async handle(req, ctx) {
     return await tracer2.startActiveSpan(
       `handle ${this.moduleName}.${this.actionName}`,
       async (span) => {
         span.setAttribute("module", this.moduleName);
         span.setAttribute("action", this.actionName);
         try {
-          return await this._handle(req);
+          return await this._handle(req, ctx);
         } catch (error) {
           span.recordException(error);
           span.setStatus({
@@ -514,7 +509,7 @@ var ActionHandler = class {
       span.end();
     }
   }
-  async _handle(req) {
+  async _handle(req, ctx) {
     const span = trace.getActiveSpan();
     const { args, requestHash } = await this._validate(req);
     const cacheHash = typeof this.metadata.cache === "function" ? this.metadata.cache(...args) : requestHash;
@@ -532,7 +527,7 @@ var ActionHandler = class {
     try {
       const result = await this.moduleInstance[this.actionName].apply(
         this.moduleInstance,
-        args
+        args.concat(ctx)
       );
       if (this.metadata.stream) {
         span?.setAttribute("stream", true);
@@ -601,9 +596,10 @@ function Page(options) {
       existingMetadata[methodName] = {
         name: methodName,
         description: options.description || "",
-        method: options.method,
+        method: options.method || "get",
         path: normalizePath(options.path),
-        absolutePath: options.absolutePath ?? false
+        absolutePath: options.absolutePath ?? false,
+        middlewares: options.middlewares ?? []
       };
       prototype[PAGE_METADATA] = existingMetadata;
     });
@@ -639,6 +635,9 @@ var PageHandler = class {
             this.moduleInstance,
             [ctx]
           );
+          if (result instanceof Response) {
+            return result;
+          }
           return ctx.html(result);
         } catch (error) {
           span.recordException(error);
@@ -655,8 +654,9 @@ var PageHandler = class {
   }
 };
 var WebSocketHandler = class {
-  constructor(microservice, options) {
+  constructor(microservice, ctx, options) {
     this.microservice = microservice;
+    this.ctx = ctx;
     this.options = {
       timeout: options?.timeout || 3e4
     };
@@ -696,7 +696,7 @@ var WebSocketHandler = class {
         await this.sendMessage(ws, { type: "pong" });
         return;
       }
-      const response = await this.handleRequest(ws, message);
+      const response = await this.handleRequest(ws, message, this.ctx);
       if (response) {
         await this.sendMessage(ws, response);
       }
@@ -723,7 +723,7 @@ var WebSocketHandler = class {
   onError(_error, ws) {
     this.onClose(ws);
   }
-  async handleRequest(ws, message) {
+  async handleRequest(ws, message, ctx) {
     if (!message.id || !message.module || !message.action) {
       throw new Error("Invalid request message");
     }
@@ -759,7 +759,7 @@ var WebSocketHandler = class {
           message.action
         );
         const args = message.args ? Object.values(message.args) : [];
-        const result = await handler.handle(args);
+        const result = await handler.handle(args, ctx);
         if (handler.metadata.stream) {
           try {
             for await (const value of result) {
@@ -920,20 +920,20 @@ var Microservice = class {
         );
         this.actionHandlers.set(`${moduleName}.${actionName}`, handler);
         logger_default.info(
-          `[ \u6CE8\u518C\u52A8\u4F5C ] ${moduleName}.${actionName} ${actionMetadata.description} ${actionMetadata.mcp ? "MCP:" + actionMetadata.mcp?.type : ""}`
+          `[ \u6CE8\u518C\u52A8\u4F5C ] ${moduleName}.${actionName} ${actionMetadata.description}`
         );
       }
       for (const [_, page] of Object.entries(pages)) {
-        const handler = new PageHandler(
-          moduleInstance,
-          page,
-          moduleName
-        );
+        const handler = new PageHandler(moduleInstance, page, moduleName);
         this.pageHandlers.set(`${moduleName}.${page.name}`, handler);
         const paths = Array.isArray(page.path) ? page.path : [page.path];
         for (let path of paths) {
           path = page.absolutePath ? path : `${this.options.prefix}${path}`;
-          this.app[page.method](path, (ctx) => handler.handle(ctx));
+          this.app[page.method](
+            path,
+            ...page.middlewares ?? [],
+            (ctx) => handler.handle(ctx)
+          );
           logger_default.info(
             `[ \u6CE8\u518C\u9875\u9762 ] ${moduleName}.${page.name} ${page.method.toUpperCase()} ${page.path} ${page.description}`
           );
@@ -995,11 +995,10 @@ var Microservice = class {
     });
     this.app.post(`${prefix}/:moduleName/:actionName`, this.handleRequest);
     if (this.options.websocket?.enabled) {
-      this.wsHandler = new WebSocketHandler(this);
       this.app.get(
         `${prefix}/ws`,
-        this.nodeWebSocket.upgradeWebSocket((_ctx) => {
-          const wsHandler = new WebSocketHandler(this, {
+        this.nodeWebSocket.upgradeWebSocket((ctx) => {
+          const wsHandler = new WebSocketHandler(this, ctx, {
             timeout: this.options.websocket?.timeout
           });
           return {
@@ -1221,7 +1220,9 @@ Received SIGTERM signal`);
   async waitForActiveRequests() {
     const timeout = this.options.gracefulShutdown?.timeout || 3e4;
     const startTime = Date.now();
-    logger_default.info(`Waiting for ${this.activeRequests.size} active requests to complete...`);
+    logger_default.info(
+      `Waiting for ${this.activeRequests.size} active requests to complete...`
+    );
     return new Promise((resolve) => {
       const checkInterval = setInterval(() => {
         const elapsed = Date.now() - startTime;
@@ -1231,10 +1232,14 @@ Received SIGTERM signal`);
           resolve();
         } else if (elapsed >= timeout) {
           clearInterval(checkInterval);
-          logger_default.warn(`Timeout waiting for requests to complete. ${this.activeRequests.size} requests still active`);
+          logger_default.warn(
+            `Timeout waiting for requests to complete. ${this.activeRequests.size} requests still active`
+          );
           resolve();
         } else {
-          logger_default.info(`Still waiting for ${this.activeRequests.size} requests... (${elapsed}ms elapsed)`);
+          logger_default.info(
+            `Still waiting for ${this.activeRequests.size} requests... (${elapsed}ms elapsed)`
+          );
         }
       }, 1e3);
     });
@@ -1256,7 +1261,10 @@ Received SIGTERM signal`);
         await Promise.race([
           Promise.resolve(hook.cleanup()),
           new Promise(
-            (_, reject) => setTimeout(() => reject(new Error(`Cleanup hook ${hook.name} timeout`)), timeout)
+            (_, reject) => setTimeout(
+              () => reject(new Error(`Cleanup hook ${hook.name} timeout`)),
+              timeout
+            )
           )
         ]);
         logger_default.info(`Cleanup hook ${hook.name} completed successfully`);
@@ -1337,10 +1345,13 @@ Received SIGTERM signal`);
     const requestId = crypto.randomUUID();
     const startTime = Date.now();
     if (this.status === "shutting_down") {
-      return ctx.json({
-        success: false,
-        error: "Service is shutting down"
-      }, 503);
+      return ctx.json(
+        {
+          success: false,
+          error: "Service is shutting down"
+        },
+        503
+      );
     }
     try {
       const paramsText = await ctx.req.text();
@@ -1351,7 +1362,7 @@ Received SIGTERM signal`);
         startTime,
         params: paramsText
       });
-      const result = await handler.handle(paramsText);
+      const result = await handler.handle(paramsText, ctx);
       if (handler.metadata.stream) {
         const encoder = new TextEncoder();
         const microservice = this;
@@ -1410,121 +1421,6 @@ Received SIGTERM signal`);
     await this.waitingInitialization;
   }
 };
-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`
-    );
-  };
-};
 var DEFAULT_FAVICON = /* @__PURE__ */ jsx(
   "link",
   {
@@ -1727,4 +1623,4 @@ async function startCheck(checkers, pass) {
   if (pass) await pass();
 }
 
-export { Action, BaseLayout, CacheAdapter, HtmxLayout, MemoryCacheAdapter, Microservice, ModelContextProtocolPlugin, Module, Page, PageRenderPlugin, Plugin, RedisCacheAdapter, Schedule, ScheduleMode, ServiceContext, ServiceInfoCards, ServiceStatusPage, logger_default as logger, startCheck };
+export { Action, BaseLayout, CacheAdapter, HtmxLayout, MemoryCacheAdapter, Microservice, Module, Page, PageRenderPlugin, Plugin, RedisCacheAdapter, Schedule, ScheduleMode, ServiceContext, ServiceInfoCards, ServiceStatusPage, logger_default as logger, startCheck };

package/package.json

@@ -1,87 +1,86 @@
-{
-  "name": "imean-service-engine",
-  "version": "1.7.2",
-  "description": "microservice engine",
-  "keywords": [
-    "microservice",
-    "websocket",
-    "http",
-    "node"
-  ],
-  "author": "imean",
-  "type": "module",
-  "license": "MIT",
-  "repository": {
-    "type": "git",
-    "url": "git+https://git.imean.tech/imean/imean-microservice-framework.git"
-  },
-  "main": "dist/mod.js",
-  "module": "dist/mod.js",
-  "types": "dist/mod.d.ts",
-  "exports": {
-    ".": {
-      "types": "./dist/mod.d.ts",
-      "import": "./dist/mod.js",
-      "require": "./dist/mod.cjs"
-    }
-  },
-  "files": [
-    "dist",
-    "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",
-    "@modelcontextprotocol/sdk": "^1.8.0",
-    "dayjs": "^1.11.13",
-    "ejson": "^2.2.3",
-    "etcd3": "^1.1.2",
-    "fs-extra": "^11.3.0",
-    "hono": "^4.6.17",
-    "lru-cache": "^11.0.2",
-    "prettier": "^3.4.2",
-    "ulid": "^3.0.0",
-    "winston": "^3.17.0",
-    "zod": "^3.24.1"
-  },
-  "peerDependencies": {
-    "@opentelemetry/api": "^1.x",
-    "ioredis": "^5.6.0"
-  },
-  "devDependencies": {
-    "@opentelemetry/auto-instrumentations-node": "^0.55.3",
-    "@opentelemetry/exporter-logs-otlp-proto": "^0.57.1",
-    "@opentelemetry/exporter-metrics-otlp-proto": "^0.57.1",
-    "@opentelemetry/exporter-trace-otlp-proto": "^0.57.1",
-    "@opentelemetry/instrumentation-winston": "^0.44.0",
-    "@opentelemetry/sdk-logs": "^0.57.1",
-    "@opentelemetry/sdk-metrics": "^1.30.1",
-    "@opentelemetry/sdk-node": "^0.57.1",
-    "@opentelemetry/sdk-trace-node": "^1.30.1",
-    "@opentelemetry/winston-transport": "^0.10.0",
-    "@types/ejson": "^2.2.2",
-    "@types/fs-extra": "^11.0.4",
-    "@types/ioredis-mock": "^8.2.5",
-    "@types/node": "^20.0.0",
-    "@vitest/coverage-v8": "^3.0.4",
-    "imean-service-client": "^1.5.0",
-    "ioredis-mock": "^8.9.0",
-    "opentelemetry-instrumentation-fetch-node": "^1.2.3",
-    "tslib": "^2.8.1",
-    "tsup": "^8.0.1",
-    "tsx": "^4.19.2",
-    "typescript": "^5.3.3",
-    "vite-tsconfig-paths": "^5.1.4",
-    "vitest": "^3.0.3"
-  },
-  "engines": {
-    "node": ">=20"
-  }
-}
+{
+  "name": "imean-service-engine",
+  "version": "1.8.0",
+  "description": "microservice engine",
+  "keywords": [
+    "microservice",
+    "websocket",
+    "http",
+    "node"
+  ],
+  "author": "imean",
+  "type": "module",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://git.imean.tech/imean/imean-microservice-framework.git"
+  },
+  "main": "dist/mod.js",
+  "module": "dist/mod.js",
+  "types": "dist/mod.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/mod.d.ts",
+      "import": "./dist/mod.js",
+      "require": "./dist/mod.cjs"
+    }
+  },
+  "files": [
+    "dist",
+    "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",
+    "dayjs": "^1.11.13",
+    "ejson": "^2.2.3",
+    "etcd3": "^1.1.2",
+    "fs-extra": "^11.3.0",
+    "hono": "^4.6.17",
+    "lru-cache": "^11.0.2",
+    "prettier": "^3.4.2",
+    "ulid": "^3.0.0",
+    "winston": "^3.17.0",
+    "zod": "^3.24.1"
+  },
+  "peerDependencies": {
+    "@opentelemetry/api": "^1.x",
+    "ioredis": "^5.6.0"
+  },
+  "devDependencies": {
+    "@opentelemetry/auto-instrumentations-node": "^0.55.3",
+    "@opentelemetry/exporter-logs-otlp-proto": "^0.57.1",
+    "@opentelemetry/exporter-metrics-otlp-proto": "^0.57.1",
+    "@opentelemetry/exporter-trace-otlp-proto": "^0.57.1",
+    "@opentelemetry/instrumentation-winston": "^0.44.0",
+    "@opentelemetry/sdk-logs": "^0.57.1",
+    "@opentelemetry/sdk-metrics": "^1.30.1",
+    "@opentelemetry/sdk-node": "^0.57.1",
+    "@opentelemetry/sdk-trace-node": "^1.30.1",
+    "@opentelemetry/winston-transport": "^0.10.0",
+    "@types/ejson": "^2.2.2",
+    "@types/fs-extra": "^11.0.4",
+    "@types/ioredis-mock": "^8.2.5",
+    "@types/node": "^20.0.0",
+    "@vitest/coverage-v8": "^3.0.4",
+    "imean-service-client": "^1.5.0",
+    "ioredis-mock": "^8.9.0",
+    "opentelemetry-instrumentation-fetch-node": "^1.2.3",
+    "tslib": "^2.8.1",
+    "tsup": "^8.0.1",
+    "tsx": "^4.19.2",
+    "typescript": "^5.3.3",
+    "vite-tsconfig-paths": "^5.1.4",
+    "vitest": "^3.0.3"
+  },
+  "engines": {
+    "node": ">=20"
+  }
+}