const { McpServer } = require('@modelcontextprotocol/sdk/server/mcp.js');
const { StreamableHTTPServerTransport } = require('@modelcontextprotocol/sdk/server/streamableHttp.js');

const mcp = async (ctx, next) => {
  const server = new McpServer({
    name: 'mcp-server',
    version: '0.1.0',
  });
  const transport = new StreamableHTTPServerTransport({
    sessionIdGenerator: undefined,
    enableJsonResponse: true,
  });
  await server.connect(transport);
  await transport.handleRequest(ctx.req, ctx.res);
  await next();
};

module.exports = app => {
  const { router } = app;
  router.all('/mcp', mcp);
};