fix: mcp context proto

Author: akitaSummer

plugin/controller/lib/impl/mcp/MCPControllerRegister.ts

@@ -344,9 +344,21 @@ export class MCPControllerRegister implements ControllerRegister {
       this.app.logger.error('session %s error %o', transport.sessionId, error);
     };
     const messageFunc = transport.onmessage;
+    const self = this;
     transport.onmessage = async (...args: [ JSONRPCMessage ]) => {
-      await ctx.app.ctxStorage.run(ctx, async () => {
-        await mw(ctx, async () => {
+      // 这里需要 new 一个新的 ctx，否则 ContextProto 会未被初始化
+      const socket = new Socket();
+      const req = new IncomingMessage(socket);
+      const res = new ServerResponse(req);
+      req.method = 'POST';
+      req.url = self.mcpConfig.sseInitPath;
+      req.headers = {
+        accept: 'application/json, text/event-stream',
+        'content-type': 'application/json',
+      };
+      const newCtx = self.app.createContext(req, res) as unknown as Context;
+      await ctx.app.ctxStorage.run(newCtx, async () => {
+        await mw(newCtx, async () => {
           await messageFunc!(...args);
         });
       });

plugin/controller/test/fixtures/apps/mcp-app/app/controller/McpController.ts

@@ -14,6 +14,7 @@ import {
   ToolArgsSchema,
 } from '@eggjs/tegg';
 import z from 'zod';
+import { CommonService } from '../services/common';
 
 export const PromptType = {
   name: z.string(),
@@ -32,6 +33,9 @@ export class McpController {
   @Inject()
   logger: Logger;
 
+  @Inject()
+  commonService: CommonService;
+
   @MCPPrompt()
   async foo(@PromptArgsSchema(PromptType) args: PromptArgs<typeof PromptType>): Promise<MCPPromptResponse> {
     this.logger.info('hello world');
@@ -50,6 +54,7 @@ export class McpController {
 
   @MCPTool()
   async bar(@ToolArgsSchema(ToolType) args: ToolArgs<typeof ToolType>): Promise<MCPToolResponse> {
+    await this.commonService.sayHello();
     return {
       content: [
         {

plugin/controller/test/fixtures/apps/mcp-app/app/services/common.ts

@@ -0,0 +1,17 @@
+import {
+  Logger,
+  ContextProto,
+  Inject,
+} from '@eggjs/tegg';
+
+@ContextProto()
+export class CommonService {
+
+  @Inject()
+  logger: Logger;
+
+  async sayHello(): Promise<string> {
+    this.logger.info('hello world');
+    return 'hello world';
+  }
+}