flex-mcp 1.0.0

package/dist/mcp/post-sse-transport.js

@@ -0,0 +1,375 @@
+/**
+ * POST-based SSE Transport
+ * 实现 MCP SDK Transport 接口，支持需要 POST 方法的 SSE 端点（streamable-http）
+ */
+import axios from 'axios';
+/**
+ * POST-based SSE Transport
+ * 兼容 MCP SDK Transport 接口
+ * 通过 POST 请求建立 SSE 连接，使用流式响应处理 SSE 消息
+ */
+export class PostSSETransport {
+    url;
+    headers;
+    stream = null;
+    abortController = null;
+    endpoint = null;
+    buffer = '';
+    isStarted = false;
+    startPromise = null;
+    endpointReceived = false;
+    initializeResponseReceived = false;
+    // Transport 接口要求的回调
+    onclose;
+    onerror;
+    onmessage;
+    constructor(options) {
+        this.url = options.url;
+        this.headers = {
+            'Content-Type': 'application/json',
+            'Accept': 'application/json, text/event-stream',
+            ...options.headers,
+        };
+    }
+    /**
+     * 启动传输（实现 Transport 接口）
+     */
+    async start() {
+        if (this.isStarted) {
+            throw new Error('PostSSETransport already started!');
+        }
+        this.abortController = new AbortController();
+        this.isStarted = true;
+        this.endpointReceived = false;
+        this.initializeResponseReceived = false;
+        return new Promise((resolve, reject) => {
+            this.startPromise = { resolve, reject };
+            // 发送 POST 请求建立 SSE 连接
+            // 根据 curl 测试，端点需要 POST 方法，且需要 Accept: application/json, text/event-stream
+            axios.post(this.url.toString(), {
+                jsonrpc: '2.0',
+                method: 'initialize',
+                id: 1,
+                params: {
+                    protocolVersion: '2024-11-05',
+                    capabilities: {},
+                    clientInfo: {
+                        name: 'flex-mcp-client',
+                        version: '1.0.0',
+                    },
+                },
+            }, {
+                headers: this.headers,
+                responseType: 'stream',
+                timeout: 0, // 无超时，保持连接
+                signal: this.abortController.signal,
+            })
+                .then((response) => {
+                console.log(`[MCP] POST-based SSE connection established, status: ${response.status}`);
+                // 处理流式响应
+                this.stream = response.data;
+                this.handleStream();
+                // 设置超时，如果 10 秒内没有收到响应，reject
+                setTimeout(() => {
+                    if (this.startPromise && !this.endpointReceived && !this.initializeResponseReceived) {
+                        const err = new Error('Timeout waiting for endpoint or initialize response');
+                        this.startPromise.reject(err);
+                        this.startPromise = null;
+                    }
+                }, 10000);
+            })
+                .catch((error) => {
+                this.isStarted = false;
+                const err = new Error(`Failed to connect to POST-based SSE endpoint: ${error.message}`);
+                this.onerror?.(err);
+                if (this.startPromise) {
+                    this.startPromise.reject(err);
+                    this.startPromise = null;
+                }
+            });
+        });
+    }
+    /**
+     * 处理 SSE 流式响应
+     */
+    handleStream() {
+        if (!this.stream) {
+            return;
+        }
+        this.stream.on('data', (chunk) => {
+            this.buffer += chunk.toString();
+            // 处理 SSE 格式的数据
+            const lines = this.buffer.split('\n');
+            this.buffer = lines.pop() || ''; // 保留最后不完整的行
+            let currentEvent = 'message';
+            let currentData = '';
+            for (const line of lines) {
+                const trimmedLine = line.trim();
+                if (trimmedLine.startsWith('event:')) {
+                    currentEvent = trimmedLine.substring(6).trim();
+                }
+                else if (trimmedLine.startsWith('data:')) {
+                    // 处理多行 data（SSE 规范支持多行 data）
+                    const dataContent = trimmedLine.substring(5);
+                    if (currentData) {
+                        currentData += '\n' + dataContent;
+                    }
+                    else {
+                        currentData = dataContent;
+                    }
+                }
+                else if (trimmedLine === '') {
+                    // 空行表示一个完整的 SSE 消息
+                    if (currentData) {
+                        this.processSSEMessage(currentEvent, currentData);
+                        currentEvent = 'message';
+                        currentData = '';
+                    }
+                }
+            }
+        });
+        this.stream.on('end', () => {
+            console.log(`[MCP] POST-based SSE stream ended`);
+            // 对于 streamable-http，流可能在发送响应后结束
+            // 如果已经收到 initialize 响应，这是正常的，不要立即关闭连接
+            // 只有在明确关闭时才调用 onclose
+            if (!this.initializeResponseReceived && !this.endpointReceived) {
+                // 如果流结束但还没有收到响应，可能是错误
+                console.warn(`[MCP] Stream ended before receiving initialize response or endpoint`);
+                this.isStarted = false;
+                this.onclose?.();
+            }
+            else {
+                // 已经收到响应，流结束是正常的，保持连接状态
+                console.log(`[MCP] Stream ended after receiving response, keeping connection alive`);
+            }
+        });
+        this.stream.on('error', (error) => {
+            console.error(`[MCP] POST-based SSE stream error:`, error);
+            this.isStarted = false;
+            this.onerror?.(error);
+            if (this.startPromise) {
+                this.startPromise.reject(error);
+                this.startPromise = null;
+            }
+        });
+    }
+    /**
+     * 处理 SSE 消息
+     */
+    processSSEMessage(event, data) {
+        try {
+            const trimmedData = data.trim();
+            // 处理 endpoint 事件（MCP SSE 协议）
+            if (event === 'endpoint') {
+                try {
+                    // endpoint 事件的 data 是直接的 URL 字符串
+                    this.endpoint = new URL(trimmedData, this.url);
+                    if (this.endpoint.origin !== this.url.origin) {
+                        throw new Error(`Endpoint origin does not match connection origin: ${this.endpoint.origin}`);
+                    }
+                    console.log(`[MCP] Received endpoint: ${this.endpoint.toString()}`);
+                    // 如果 start() 还在等待，resolve 它
+                    if (this.startPromise && !this.endpointReceived) {
+                        this.endpointReceived = true;
+                        this.startPromise.resolve();
+                        this.startPromise = null;
+                    }
+                }
+                catch (error) {
+                    const err = new Error(`Invalid endpoint URL: ${error.message}`);
+                    this.onerror?.(err);
+                    if (this.startPromise) {
+                        this.startPromise.reject(err);
+                        this.startPromise = null;
+                    }
+                    return;
+                }
+                return;
+            }
+            // 处理普通消息（message 事件或默认事件）
+            if (event === 'message' || !event || event === '') {
+                try {
+                    const message = JSON.parse(trimmedData);
+                    // 检查是否是 initialize 响应
+                    // 类型守卫：确保 message 有 id 和 result 属性
+                    if (message.jsonrpc === '2.0' && 'id' in message && message.id === 1 && 'result' in message && message.result && !this.initializeResponseReceived) {
+                        this.initializeResponseReceived = true;
+                        // 对于 streamable-http，如果没有 endpoint，使用原始 URL
+                        if (!this.endpoint) {
+                            this.endpoint = this.url;
+                            console.log(`[MCP] Initialize response received, using original URL as endpoint`);
+                        }
+                        // 如果 start() 还在等待，resolve 它
+                        if (this.startPromise && !this.endpointReceived) {
+                            this.startPromise.resolve();
+                            this.startPromise = null;
+                        }
+                    }
+                    // 调用消息回调
+                    this.onmessage?.(message);
+                }
+                catch (parseError) {
+                    // 如果不是有效的 JSON-RPC 消息，记录警告但不中断连接
+                    console.warn(`[MCP] Failed to parse SSE message as JSON-RPC:`, parseError.message);
+                    console.warn(`[MCP] Raw data:`, trimmedData.substring(0, 200));
+                }
+            }
+            else {
+                // 其他事件类型（如 ping），记录但不处理
+                console.log(`[MCP] Received SSE event '${event}':`, trimmedData.substring(0, 100));
+            }
+        }
+        catch (error) {
+            console.error(`[MCP] Error processing SSE message:`, error);
+            this.onerror?.(error);
+            if (this.startPromise) {
+                this.startPromise.reject(error);
+                this.startPromise = null;
+            }
+        }
+    }
+    /**
+     * 发送消息（实现 Transport 接口）
+     */
+    async send(message) {
+        if (!this.isStarted) {
+            throw new Error('Transport not started');
+        }
+        // 对于 POST-based SSE，使用 endpoint（如果已收到）或原始 URL
+        const targetUrl = this.endpoint || this.url;
+        try {
+            // 对于 streamable-http，每个请求都是独立的 POST
+            // 如果响应是流式的，需要处理 SSE 流
+            const response = await axios.post(targetUrl.toString(), message, {
+                headers: {
+                    ...this.headers,
+                    'Content-Type': 'application/json',
+                    'Accept': 'application/json, text/event-stream',
+                },
+                responseType: 'stream',
+                signal: this.abortController?.signal,
+            });
+            // 检查响应内容类型
+            const contentType = response.headers['content-type'] || '';
+            // 如果响应是流式的（SSE），处理 SSE 流
+            if (response.data && typeof response.data.on === 'function' && contentType.includes('text/event-stream')) {
+                const stream = response.data;
+                let buffer = '';
+                // Remove unused 'currentEvent'
+                let currentData = '';
+                stream.on('data', (chunk) => {
+                    buffer += chunk.toString();
+                    const lines = buffer.split('\n');
+                    buffer = lines.pop() || '';
+                    for (const line of lines) {
+                        const trimmedLine = line.trim();
+                        // Remove assignment and use of 'currentEvent' since it's unused
+                        // if (trimmedLine.startsWith('event:')) {
+                        //   currentEvent = trimmedLine.substring(6).trim();
+                        // } else 
+                        if (trimmedLine.startsWith('data:')) {
+                            const dataContent = trimmedLine.substring(5);
+                            if (currentData) {
+                                currentData += '\n' + dataContent;
+                            }
+                            else {
+                                currentData = dataContent;
+                            }
+                        }
+                        else if (trimmedLine === '') {
+                            if (currentData) {
+                                try {
+                                    const responseMessage = JSON.parse(currentData.trim());
+                                    this.onmessage?.(responseMessage);
+                                }
+                                catch (e) {
+                                    console.warn(`[MCP] Failed to parse response message:`, e);
+                                }
+                                currentData = '';
+                            }
+                        }
+                    }
+                });
+                stream.on('end', () => {
+                    // 流结束是正常的
+                });
+                stream.on('error', (error) => {
+                    this.onerror?.(error);
+                });
+            }
+            else {
+                // 如果响应是 JSON（不是流），直接处理
+                // 某些端点可能返回 JSON 而不是 SSE 流
+                let responseData = response.data;
+                // 如果响应是流但内容类型是 JSON，读取整个流
+                if (response.data && typeof response.data.on === 'function' && contentType.includes('application/json')) {
+                    const stream = response.data;
+                    let buffer = '';
+                    stream.on('data', (chunk) => {
+                        buffer += chunk.toString();
+                    });
+                    stream.on('end', () => {
+                        try {
+                            responseData = JSON.parse(buffer);
+                            this.onmessage?.(responseData);
+                        }
+                        catch (e) {
+                            console.warn(`[MCP] Failed to parse JSON response:`, e);
+                            this.onerror?.(new Error(`Failed to parse JSON response: ${e}`));
+                        }
+                    });
+                    stream.on('error', (error) => {
+                        this.onerror?.(error);
+                    });
+                }
+                else if (typeof responseData === 'object') {
+                    // 如果响应已经是对象，直接处理
+                    this.onmessage?.(responseData);
+                }
+                else if (typeof responseData === 'string') {
+                    // 如果响应是字符串，尝试解析为 JSON
+                    try {
+                        const parsed = JSON.parse(responseData);
+                        this.onmessage?.(parsed);
+                    }
+                    catch (e) {
+                        console.warn(`[MCP] Failed to parse string response as JSON:`, e);
+                        this.onerror?.(new Error(`Failed to parse response: ${e}`));
+                    }
+                }
+            }
+        }
+        catch (error) {
+            const err = error.response?.data
+                ? new Error(`Error POSTing to endpoint (HTTP ${error.response.status}): ${JSON.stringify(error.response.data)}`)
+                : error;
+            this.onerror?.(err);
+            throw err;
+        }
+    }
+    /**
+     * 关闭连接（实现 Transport 接口）
+     */
+    async close() {
+        this.abortController?.abort();
+        if (this.stream) {
+            // 移除所有事件监听器，避免内存泄漏
+            this.stream.removeAllListeners();
+            this.stream.destroy();
+            this.stream = null;
+        }
+        this.isStarted = false;
+        this.endpoint = null;
+        this.buffer = '';
+        this.endpointReceived = false;
+        this.initializeResponseReceived = false;
+        if (this.startPromise) {
+            // 如果 start() 还在等待，reject 它
+            this.startPromise.reject(new Error('Transport closed before connection established'));
+            this.startPromise = null;
+        }
+        this.onclose?.();
+    }
+}
+//# sourceMappingURL=post-sse-transport.js.map

package/dist/mcp/post-sse-transport.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"post-sse-transport.js","sourceRoot":"","sources":["../../src/mcp/post-sse-transport.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH,OAAO,KAAK,MAAM,OAAO,CAAC;AAQ1B;;;;GAIG;AACH,MAAM,OAAO,gBAAgB;IACnB,GAAG,CAAM;IACT,OAAO,CAAyB;IAChC,MAAM,GAAoB,IAAI,CAAC;IAC/B,eAAe,GAA2B,IAAI,CAAC;IAC/C,QAAQ,GAAe,IAAI,CAAC;IAC5B,MAAM,GAAW,EAAE,CAAC;IACpB,SAAS,GAAG,KAAK,CAAC;IAClB,YAAY,GAAmE,IAAI,CAAC;IACpF,gBAAgB,GAAG,KAAK,CAAC;IACzB,0BAA0B,GAAG,KAAK,CAAC;IAE3C,oBAAoB;IACpB,OAAO,CAAc;IACrB,OAAO,CAA0B;IACjC,SAAS,CAAqC;IAE9C,YAAY,OAAgC;QAC1C,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG;YACb,cAAc,EAAE,kBAAkB;YAClC,QAAQ,EAAE,qCAAqC;YAC/C,GAAG,OAAO,CAAC,OAAO;SACnB,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC7C,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;QAC9B,IAAI,CAAC,0BAA0B,GAAG,KAAK,CAAC;QAExC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,YAAY,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAExC,sBAAsB;YACtB,0EAA0E;YAC1E,KAAK,CAAC,IAAI,CACR,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,EACnB;gBACE,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,YAAY;gBACpB,EAAE,EAAE,CAAC;gBACL,MAAM,EAAE;oBACN,eAAe,EAAE,YAAY;oBAC7B,YAAY,EAAE,EAAE;oBAChB,UAAU,EAAE;wBACV,IAAI,EAAE,iBAAiB;wBACvB,OAAO,EAAE,OAAO;qBACjB;iBACF;aACF,EACD;gBACE,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,YAAY,EAAE,QAAQ;gBACtB,OAAO,EAAE,CAAC,EAAE,WAAW;gBACvB,MAAM,EAAE,IAAI,CAAC,eAAe,CAAC,MAAM;aACpC,CACF;iBACA,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;gBACjB,OAAO,CAAC,GAAG,CAAC,wDAAwD,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;gBAEvF,SAAS;gBACT,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC,IAAgB,CAAC;gBACxC,IAAI,CAAC,YAAY,EAAE,CAAC;gBAEpB,6BAA6B;gBAC7B,UAAU,CAAC,GAAG,EAAE;oBACd,IAAI,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,gBAAgB,IAAI,CAAC,IAAI,CAAC,0BAA0B,EAAE,CAAC;wBACpF,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;wBAC7E,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;wBAC9B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;oBAC3B,CAAC;gBACH,CAAC,EAAE,KAAK,CAAC,CAAC;YACZ,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,KAAU,EAAE,EAAE;gBACpB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;gBACvB,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,iDAAiD,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBACxF,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG,CAAC,CAAC;gBACpB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBACtB,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;oBAC9B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;gBAC3B,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,YAAY;QAClB,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;YACvC,IAAI,CAAC,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;YAEhC,eAAe;YACf,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACtC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,CAAC,YAAY;YAE7C,IAAI,YAAY,GAAG,SAAS,CAAC;YAC7B,IAAI,WAAW,GAAG,EAAE,CAAC;YAErB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;gBAEhC,IAAI,WAAW,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACrC,YAAY,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBACjD,CAAC;qBAAM,IAAI,WAAW,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC3C,6BAA6B;oBAC7B,MAAM,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;oBAC7C,IAAI,WAAW,EAAE,CAAC;wBAChB,WAAW,IAAI,IAAI,GAAG,WAAW,CAAC;oBACpC,CAAC;yBAAM,CAAC;wBACN,WAAW,GAAG,WAAW,CAAC;oBAC5B,CAAC;gBACH,CAAC;qBAAM,IAAI,WAAW,KAAK,EAAE,EAAE,CAAC;oBAC9B,mBAAmB;oBACnB,IAAI,WAAW,EAAE,CAAC;wBAChB,IAAI,CAAC,iBAAiB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC;wBAClD,YAAY,GAAG,SAAS,CAAC;wBACzB,WAAW,GAAG,EAAE,CAAC;oBACnB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;YACzB,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;YACjD,iCAAiC;YACjC,sCAAsC;YACtC,sBAAsB;YACtB,IAAI,CAAC,IAAI,CAAC,0BAA0B,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBAC/D,sBAAsB;gBACtB,OAAO,CAAC,IAAI,CAAC,qEAAqE,CAAC,CAAC;gBACpF,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;gBACvB,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YACnB,CAAC;iBAAM,CAAC;gBACN,wBAAwB;gBACxB,OAAO,CAAC,GAAG,CAAC,uEAAuE,CAAC,CAAC;YACvF,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAY,EAAE,EAAE;YACvC,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,KAAK,CAAC,CAAC;YAC3D,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;YACvB,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC;YACtB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YAC3B,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,KAAa,EAAE,IAAY;QACnD,IAAI,CAAC;YACH,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YAEhC,6BAA6B;YAC7B,IAAI,KAAK,KAAK,UAAU,EAAE,CAAC;gBACzB,IAAI,CAAC;oBACH,iCAAiC;oBACjC,IAAI,CAAC,QAAQ,GAAG,IAAI,GAAG,CAAC,WAAW,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;oBAC/C,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,KAAK,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,CAAC;wBAC7C,MAAM,IAAI,KAAK,CAAC,qDAAqD,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;oBAC/F,CAAC;oBACD,OAAO,CAAC,GAAG,CAAC,4BAA4B,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;oBAEpE,4BAA4B;oBAC5B,IAAI,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;wBAChD,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;wBAC7B,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;wBAC5B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;oBAC3B,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAU,EAAE,CAAC;oBACpB,MAAM,GAAG,GAAG,IAAI,KAAK,CAAC,yBAAyB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;oBAChE,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG,CAAC,CAAC;oBACpB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;wBACtB,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;wBAC9B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;oBAC3B,CAAC;oBACD,OAAO;gBACT,CAAC;gBACD,OAAO;YACT,CAAC;YAED,0BAA0B;YAC1B,IAAI,KAAK,KAAK,SAAS,IAAI,CAAC,KAAK,IAAI,KAAK,KAAK,EAAE,EAAE,CAAC;gBAClD,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAmB,CAAC;oBAE1D,sBAAsB;oBACtB,mCAAmC;oBACnC,IAAI,OAAO,CAAC,OAAO,KAAK,KAAK,IAAI,IAAI,IAAI,OAAO,IAAI,OAAO,CAAC,EAAE,KAAK,CAAC,IAAI,QAAQ,IAAI,OAAO,IAAI,OAAO,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,0BAA0B,EAAE,CAAC;wBAClJ,IAAI,CAAC,0BAA0B,GAAG,IAAI,CAAC;wBACvC,4CAA4C;wBAC5C,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC;4BACnB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC;4BACzB,OAAO,CAAC,GAAG,CAAC,oEAAoE,CAAC,CAAC;wBACpF,CAAC;wBAED,4BAA4B;wBAC5B,IAAI,IAAI,CAAC,YAAY,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;4BAChD,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;4BAC5B,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;wBAC3B,CAAC;oBACH,CAAC;oBAED,SAAS;oBACT,IAAI,CAAC,SAAS,EAAE,CAAC,OAAO,CAAC,CAAC;gBAC5B,CAAC;gBAAC,OAAO,UAAe,EAAE,CAAC;oBACzB,iCAAiC;oBACjC,OAAO,CAAC,IAAI,CAAC,gDAAgD,EAAE,UAAU,CAAC,OAAO,CAAC,CAAC;oBACnF,OAAO,CAAC,IAAI,CAAC,iBAAiB,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,wBAAwB;gBACxB,OAAO,CAAC,GAAG,CAAC,6BAA6B,KAAK,IAAI,EAAE,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;YACrF,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,KAAK,CAAC,CAAC;YAC5D,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC;YACtB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBAChC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,OAAuB;QAChC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;QAC3C,CAAC;QAED,8CAA8C;QAC9C,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC;QAE5C,IAAI,CAAC;YACH,oCAAoC;YACpC,sBAAsB;YACtB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAC/B,SAAS,CAAC,QAAQ,EAAE,EACpB,OAAO,EACP;gBACE,OAAO,EAAE;oBACP,GAAG,IAAI,CAAC,OAAO;oBACf,cAAc,EAAE,kBAAkB;oBAClC,QAAQ,EAAE,qCAAqC;iBAChD;gBACD,YAAY,EAAE,QAAQ;gBACtB,MAAM,EAAE,IAAI,CAAC,eAAe,EAAE,MAAM;aACrC,CACF,CAAC;YAEF,WAAW;YACX,MAAM,WAAW,GAAG,QAAQ,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;YAE3D,yBAAyB;YACzB,IAAI,QAAQ,CAAC,IAAI,IAAI,OAAO,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,UAAU,IAAI,WAAW,CAAC,QAAQ,CAAC,mBAAmB,CAAC,EAAE,CAAC;gBACzG,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAgB,CAAC;gBACzC,IAAI,MAAM,GAAG,EAAE,CAAC;gBAChB,+BAA+B;gBAC/B,IAAI,WAAW,GAAG,EAAE,CAAC;gBAErB,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;oBAClC,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;oBAC3B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;oBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;oBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;wBACzB,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;wBAChC,gEAAgE;wBAChE,0CAA0C;wBAC1C,oDAAoD;wBACpD,UAAU;wBACV,IAAI,WAAW,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;4BACpC,MAAM,WAAW,GAAG,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;4BAC7C,IAAI,WAAW,EAAE,CAAC;gCAChB,WAAW,IAAI,IAAI,GAAG,WAAW,CAAC;4BACpC,CAAC;iCAAM,CAAC;gCACN,WAAW,GAAG,WAAW,CAAC;4BAC5B,CAAC;wBACH,CAAC;6BAAM,IAAI,WAAW,KAAK,EAAE,EAAE,CAAC;4BAC9B,IAAI,WAAW,EAAE,CAAC;gCAChB,IAAI,CAAC;oCACH,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,EAAE,CAAmB,CAAC;oCACzE,IAAI,CAAC,SAAS,EAAE,CAAC,eAAe,CAAC,CAAC;gCACpC,CAAC;gCAAC,OAAO,CAAC,EAAE,CAAC;oCACX,OAAO,CAAC,IAAI,CAAC,yCAAyC,EAAE,CAAC,CAAC,CAAC;gCAC7D,CAAC;gCACD,WAAW,GAAG,EAAE,CAAC;4BACnB,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC,CAAC,CAAC;gBAEH,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;oBACpB,UAAU;gBACZ,CAAC,CAAC,CAAC;gBAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAY,EAAE,EAAE;oBAClC,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC;gBACxB,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,uBAAuB;gBACvB,0BAA0B;gBAC1B,IAAI,YAAY,GAAG,QAAQ,CAAC,IAAI,CAAC;gBAEjC,0BAA0B;gBAC1B,IAAI,QAAQ,CAAC,IAAI,IAAI,OAAO,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,UAAU,IAAI,WAAW,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC;oBACxG,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAgB,CAAC;oBACzC,IAAI,MAAM,GAAG,EAAE,CAAC;oBAEhB,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;wBAClC,MAAM,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;oBAC7B,CAAC,CAAC,CAAC;oBAEH,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;wBACpB,IAAI,CAAC;4BACH,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;4BAClC,IAAI,CAAC,SAAS,EAAE,CAAC,YAA8B,CAAC,CAAC;wBACnD,CAAC;wBAAC,OAAO,CAAC,EAAE,CAAC;4BACX,OAAO,CAAC,IAAI,CAAC,sCAAsC,EAAE,CAAC,CAAC,CAAC;4BACxD,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,KAAK,CAAC,kCAAkC,CAAC,EAAE,CAAC,CAAC,CAAC;wBACnE,CAAC;oBACH,CAAC,CAAC,CAAC;oBAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAY,EAAE,EAAE;wBAClC,IAAI,CAAC,OAAO,EAAE,CAAC,KAAK,CAAC,CAAC;oBACxB,CAAC,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;oBAC5C,iBAAiB;oBACjB,IAAI,CAAC,SAAS,EAAE,CAAC,YAA8B,CAAC,CAAC;gBACnD,CAAC;qBAAM,IAAI,OAAO,YAAY,KAAK,QAAQ,EAAE,CAAC;oBAC5C,sBAAsB;oBACtB,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;wBACxC,IAAI,CAAC,SAAS,EAAE,CAAC,MAAwB,CAAC,CAAC;oBAC7C,CAAC;oBAAC,OAAO,CAAC,EAAE,CAAC;wBACX,OAAO,CAAC,IAAI,CAAC,gDAAgD,EAAE,CAAC,CAAC,CAAC;wBAClE,IAAI,CAAC,OAAO,EAAE,CAAC,IAAI,KAAK,CAAC,6BAA6B,CAAC,EAAE,CAAC,CAAC,CAAC;oBAC9D,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,MAAM,GAAG,GAAG,KAAK,CAAC,QAAQ,EAAE,IAAI;gBAC9B,CAAC,CAAC,IAAI,KAAK,CAAC,mCAAmC,KAAK,CAAC,QAAQ,CAAC,MAAM,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;gBAChH,CAAC,CAAC,KAAK,CAAC;YACV,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG,CAAC,CAAC;YACpB,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,KAAK;QACT,IAAI,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;QAE9B,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,mBAAmB;YACnB,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACtB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACrB,CAAC;QAED,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;QACjB,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC;QAC9B,IAAI,CAAC,0BAA0B,GAAG,KAAK,CAAC;QAExC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,2BAA2B;YAC3B,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC,CAAC;YACtF,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;QAC3B,CAAC;QAED,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;IACnB,CAAC;CACF"}

package/dist/mcp/service.d.ts

@@ -0,0 +1,49 @@
+/**
+ * MCP 服务核心逻辑
+ */
+import type { User } from '../types/index.js';
+import type { MCPConfig } from '../types/index.js';
+import { MCPAggregator } from './aggregator.js';
+export declare class MCPService {
+    private aggregators;
+    private aggregatorPromises;
+    private sseStreams;
+    private mcpClientConfigs;
+    /**
+     * 验证用户凭据并返回用户信息
+     */
+    validateAndGetUser(email: string, apiKey: string): Promise<User | null>;
+    /**
+     * 获取用户的 MCP 配置
+     */
+    getUserConfig(userId: string): Promise<MCPConfig | null>;
+    /**
+     * 获取或创建用户的 MCP 聚合器
+     * 使用 Promise 缓存防止并发初始化
+     * 如果配置已更新，会重新创建聚合器
+     */
+    getAggregator(userId: string): Promise<MCPAggregator>;
+    /**
+     * 获取用户的 MCP 客户端配置（headers 和 config）
+     */
+    getMCPClientConfig(userId: string): {
+        headers: Record<string, string>;
+        config: Record<string, any>;
+    } | null;
+    /**
+     * 处理 SSE 连接（服务端推送）
+     */
+    handleSSEConnection(userId: string, stream: any, endpointUrl: string, mcpHeaders?: Record<string, string>, mcpConfig?: Record<string, any>): Promise<void>;
+    private removeStream;
+    /**
+     * 处理 HTTP 请求（MCP 协议请求）
+     * 添加超时机制，默认 60 秒
+     */
+    handleRequest(userId: string, request: any, timeout?: number): Promise<void>;
+    /**
+     * 清理用户的聚合器（当配置更新时）
+     */
+    clearAggregator(userId: string): Promise<void>;
+}
+export declare const mcpService: MCPService;
+//# sourceMappingURL=service.d.ts.map

package/dist/mcp/service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.d.ts","sourceRoot":"","sources":["../../src/mcp/service.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AAI9C,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAEhD,qBAAa,UAAU;IACrB,OAAO,CAAC,WAAW,CAAyC;IAC5D,OAAO,CAAC,kBAAkB,CAAkD;IAC5E,OAAO,CAAC,UAAU,CAAoC;IAEtD,OAAO,CAAC,gBAAgB,CAA4F;IAEpH;;OAEG;IACG,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,GAAG,IAAI,CAAC;IAS7E;;OAEG;IACG,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,SAAS,GAAG,IAAI,CAAC;IAyB9D;;;;OAIG;IACG,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAwD3D;;OAEG;IACH,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG;QAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,GAAG,IAAI;IAI3G;;OAEG;IACG,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,WAAW,EAAE,MAAM,EAAE,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IA+HhK,OAAO,CAAC,YAAY;IAiBpB;;;OAGG;IACG,aAAa,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,EAAE,OAAO,GAAE,MAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAoFzF;;OAEG;IACG,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;CAkBrD;AAED,eAAO,MAAM,UAAU,YAAmB,CAAC"}