fix: response error write

Author: akitaSummer

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

@@ -35,7 +35,7 @@ export interface MCPControllerHook {
   onStreamSessionInitialized?: (ctx: Context, transport: StreamableHTTPServerTransport, register: MCPControllerRegister) => Promise<void>
 
   // COMMON
-  preProxy?: (ctx: Context, proxyReq: Parameters<typeof http.createServer>['0'], proxyResp: Parameters<typeof http.createServer>['1']) => Promise<void>
+  preProxy?: (ctx: Context, proxyReq: http.IncomingMessage, proxyResp: http.ServerResponse) => Promise<void>
   schemaLoader?: (controllerMeta: MCPControllerMeta, meta: MCPPromptMeta | MCPToolMeta) => Promise<Parameters<McpServer['tool']>['2'] | Parameters<McpServer['prompt']>['2']>
   checkAndRunProxy?: (ctx: Context, type: MCPProtocols, sessionId: string) => Promise<boolean>;
 }
@@ -195,7 +195,6 @@ export class MCPControllerRegister implements ControllerRegister {
     // }
     const self = this;
     const initHandler = async (ctx: Context) => {
-      ctx.respond = false;
       const transport = new SSEServerTransport(self.app.config.mcp.sseMessagePath, ctx.res);
       const id = transport.sessionId;
       if (MCPControllerRegister.hooks.length > 0) {
@@ -218,6 +217,7 @@ export class MCPControllerRegister implements ControllerRegister {
         'content-type': 'text/event-stream',
         'transfer-encoding': 'chunked',
       });
+      ctx.respond = false;
       await self.mcpServer.connect(transport);
       return self.sseCtxStorageRun.bind(self)(ctx, transport);
     };

plugin/mcp-proxy/index.ts

@@ -142,7 +142,7 @@ export class MCPProxyApiClient extends APIClientBase {
       async onStreamSessionInitialized(_ctx, transport, self) {
         const sessionId = transport.sessionId!;
         self.streamTransports[sessionId] = transport;
-        apiClient.setProxyHandler(MCPProtocols.STREAM, async (req, res) => {
+        apiClient.setProxyHandler(MCPProtocols.STREAM, async (req: http.IncomingMessage, res: http.ServerResponse) => {
           // eslint-disable-next-line @typescript-eslint/ban-ts-comment
           // @ts-ignore
           self.app.RequestClass = EggRequest;
@@ -157,43 +157,46 @@ export class MCPProxyApiClient extends APIClientBase {
           }
           const sessionId = req.headers['mcp-session-id'] as string | undefined;
           if (!sessionId) {
-            res.status(500).json({
+            res.writeHead(500, { 'content-type': 'application/json' });
+            res.end(JSON.stringify({
               jsonrpc: '2.0',
               error: {
                 code: -32603,
                 message: 'session id not have and run in proxy',
               },
               id: null,
-            });
+            }));
           } else {
             let transport: StreamableHTTPServerTransport;
             const existingTransport = self.streamTransports[sessionId];
             if (existingTransport instanceof StreamableHTTPServerTransport) {
               transport = existingTransport;
             } else {
-              res.status(400).json({
+              res.writeHead(400, { 'content-type': 'application/json' });
+              res.end(JSON.stringify({
                 jsonrpc: '2.0',
                 error: {
                   code: -32000,
                   message: 'Bad Request: Session exists but uses a different transport protocol',
                 },
                 id: null,
-              });
+              }));
               return;
             }
             if (transport) {
               await self.app.ctxStorage.run(ctx, async () => {
                 await transport.handleRequest(ctx.req, ctx.res);
               });
             } else {
-              res.status(404).send({
+              res.writeHead(400, { 'content-type': 'application/json' });
+              res.end(JSON.stringify({
                 jsonrpc: '2.0',
                 error: {
                   code: -32602,
                   message: 'Bad Request: No transport found for sessionId',
                 },
                 id: null,
-              });
+              }));
             }
           }
         });