@cnbcool/mcp-server/dist/streamable.js

CNB MCP Server. A comprehensive MCP server that provides seamless integration to the CNB's API(https://cnb.cool), offering a wide range of tools for repository management, pipelines operations and collaboration features

#!/usr/bin/env node
"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const express_1 = __importDefault(require("express"));
const node_crypto_1 = require("node:crypto");
const streamableHttp_js_1 = require("@modelcontextprotocol/sdk/server/streamableHttp.js");
const sse_js_1 = require("@modelcontextprotocol/sdk/server/sse.js");
const types_js_1 = require("@modelcontextprotocol/sdk/types.js");
const dotenv_1 = __importDefault(require("dotenv"));
const createMcpServer_js_1 = require("./helpers/createMcpServer.js");
const sendResponse_js_1 = require("./helpers/sendResponse.js");
dotenv_1.default.config();
const DEFAULT_APP_PORT = 3000;
// Store transports for each session type
const transports = {
    streamable: {},
    sse: {}
};
const app = (0, express_1.default)();
app.use(express_1.default.json());
app.post('/mcp', async (req, res) => {
    const sessionId = req.headers['mcp-session-id'];
    let transport;
    // Reuse existing transport
    if (sessionId && transports.streamable[sessionId]) {
        transport = transports.streamable[sessionId];
        if (!(transport instanceof streamableHttp_js_1.StreamableHTTPServerTransport)) {
            (0, sendResponse_js_1.stopWithWrongTransport)(res);
            return;
        }
        await transport.handleRequest(req, res, req.body);
        return;
    }
    // New initialization request
    if (!sessionId && (0, types_js_1.isInitializeRequest)(req.body)) {
        transport = new streamableHttp_js_1.StreamableHTTPServerTransport({
            sessionIdGenerator: () => (0, node_crypto_1.randomUUID)(),
            onsessioninitialized: (sessionId) => {
                transports.streamable[sessionId] = transport;
            }
        });
        // Clean up transport when closed
        transport.onclose = () => {
            if (transport.sessionId) {
                delete transports.streamable[transport.sessionId];
            }
        };
        const mcpServer = (0, createMcpServer_js_1.createMcpServer)(req);
        await mcpServer.connect(transport);
        await transport.handleRequest(req, res, req.body);
        return;
    }
    // Invalid request
    res.status(400).json({
        jsonrpc: '2.0',
        error: {
            code: -32000,
            message: 'Bad Request: No valid session ID provided'
        },
        id: null
    });
});
const handleSessionRequest = async (req, res) => {
    const sessionId = req.headers['mcp-session-id'];
    if (!sessionId || !transports.streamable[sessionId]) {
        res.status(400).send('Invalid or missing session ID');
        return;
    }
    const transport = transports.streamable[sessionId];
    await transport.handleRequest(req, res, req.body);
};
app.get('/mcp', handleSessionRequest);
app.delete('/mcp', handleSessionRequest);
app.get('/sse', async (req, res) => {
    const transport = new sse_js_1.SSEServerTransport('/messages', res);
    transports.sse[transport.sessionId] = transport;
    res.on('close', () => {
        delete transports.sse[transport.sessionId];
    });
    const mcpServer = (0, createMcpServer_js_1.createMcpServer)(req);
    await mcpServer.connect(transport);
});
app.post('/messages', async (req, res) => {
    const sessionId = req.query.sessionId;
    const transport = transports.sse[sessionId];
    if (!transport) {
        res.status(400).send('No transport found for sessionId');
        return;
    }
    if (!(transport instanceof sse_js_1.SSEServerTransport)) {
        (0, sendResponse_js_1.stopWithWrongTransport)(res);
        return;
    }
    await transport.handlePostMessage(req, res, req.body);
});
let port = parseInt(process.env.APP_PORT ?? '', 10);
if (isNaN(port)) {
    port = DEFAULT_APP_PORT;
}
const server = app.listen(port, () => {
    console.log(`MCP Streamable HTTP Server listening on port ${port}`);
});
process.on('SIGTERM', () => {
    console.log('SIGTERM signal received: closing HTTP server');
    server.close(() => {
        console.log('HTTP server closed');
    });
});