@yoda.digital/gitlab-mcp-server/dist/transport.js

GitLab MCP Server - A Model Context Protocol server for GitLab integration

import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import { SSEServerTransport } from "@modelcontextprotocol/sdk/server/sse.js";
import { createServer } from "http";
import { parse } from "url";
/**
 * Sets up the appropriate transport for the server based on the options
 *
 * @param server - The MCP server instance
 * @param options - Transport configuration options
 * @returns A promise that resolves when the transport is set up
 */
export async function setupTransport(server, options = {}) {
    const { port = 3000, useSSE = false } = options;
    if (useSSE) {
        // Create an object to store active SSE transports by session ID
        const transports = {};
        // Create raw HTTP server
        const httpServer = createServer(async (req, res) => {
            // Enable CORS
            res.setHeader('Access-Control-Allow-Origin', '*');
            res.setHeader('Access-Control-Allow-Methods', 'GET, POST');
            res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
            if (req.method === 'OPTIONS') {
                res.writeHead(204);
                res.end();
                return;
            }
            const { pathname, query } = parse(req.url || '', true);
            try {
                if (req.method === 'GET' && pathname === '/sse') {
                    // Create a new SSE transport
                    const transport = new SSEServerTransport("/messages", res);
                    // Store the transport by session ID
                    transports[transport.sessionId] = transport;
                    // Set up cleanup handler
                    req.on("close", () => {
                        delete transports[transport.sessionId];
                    });
                    // Connect the server to the transport
                    await server.connect(transport);
                }
                else if (req.method === 'POST' && pathname === '/messages') {
                    const sessionId = query.sessionId;
                    const transport = transports[sessionId];
                    if (!transport) {
                        res.writeHead(400);
                        res.end('No transport found for sessionId');
                        return;
                    }
                    // Pass the raw Node.js request to the transport
                    await transport.handlePostMessage(req, res);
                }
                else {
                    res.writeHead(404);
                    res.end();
                }
            }
            catch (error) {
                console.error('Server error:', error);
                if (!res.headersSent) {
                    res.writeHead(500);
                    res.end('Internal server error');
                }
            }
        });
        // Start the server
        httpServer.listen(port, () => {
            console.error(`SSE server listening on port ${port}`);
        });
    }
    else {
        // Set up stdio transport
        const transport = new StdioServerTransport();
        await server.connect(transport);
    }
}
//# sourceMappingURL=transport.js.map