next
Version:
The React Framework
39 lines (38 loc) • 1.49 kB
JavaScript
import { getOrCreateMcpServer } from './get-or-create-mcp-server';
import { parseBody } from '../api-utils/node/parse-body';
import { StreamableHTTPServerTransport } from 'next/dist/compiled/@modelcontextprotocol/sdk/server/streamableHttp';
export function getMcpMiddleware(options) {
return async function(req, res, next) {
const { pathname } = new URL(req.url || '', 'http://n');
if (!pathname.startsWith('/_next/mcp')) {
return next();
}
const mcpServer = getOrCreateMcpServer(options);
const transport = new StreamableHTTPServerTransport({
sessionIdGenerator: undefined
});
try {
res.on('close', ()=>{
transport.close();
});
await mcpServer.connect(transport);
const parsedBody = await parseBody(req, 1024 * 1024) // 1MB limit
;
await transport.handleRequest(req, res, parsedBody);
} catch (error) {
if (!res.headersSent) {
res.statusCode = 500;
res.setHeader('Content-Type', 'application/json; charset=utf-8');
res.end(JSON.stringify({
jsonrpc: '2.0',
error: {
code: -32000,
message: 'Internal server error'
},
id: null
}));
}
}
};
}
//# sourceMappingURL=get-mcp-middleware.js.map