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

@@ -11,6 +11,7 @@ import {
   PromptArgsSchema,
   Logger,
   Inject,
+  ContextProto,
   ToolArgsSchema,
 } from '@eggjs/tegg';
 import z from 'zod';
@@ -25,13 +26,26 @@ export const ToolType = {
   }),
 };
 
+@ContextProto()
+export class CommonService {
+  @Inject()
+  logger: Logger;
+
+  async sayHello(): Promise<string> {
+    this.logger.info('hello world');
+    return 'hello world';
+  }
+}
 
 @MCPController()
 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 +64,7 @@ export class McpController {
 
   @MCPTool()
   async bar(@ToolArgsSchema(ToolType) args: ToolArgs<typeof ToolType>): Promise<MCPToolResponse> {
+    await this.commonService.sayHello();
     return {
       content: [
         {

plugin/controller/test/mcp/mcp.test.ts

@@ -93,7 +93,7 @@ describe('plugin/controller/test/mcp/mcp.test.ts', () => {
       return app.close();
     });
 
-    it('sse should work', async () => {
+    it.only('sse should work', async () => {
       const sseClient = new Client({
         name: 'sse-demo-client',
         version: '1.0.0',