Upload 14 files

src/lightweight-client.js

@@ -30,6 +30,11 @@ const logger = {
 
 // 配置
 const NOTION_API_URL = "https://www.notion.so/api/v3/runInferenceTranscript";
+let NOTION_COOKIES = [];
+let NOTION_SPACE_IDS = {};
+let NOTION_USER_IDS = {};
+let currentCookieIndex = 0;
+
 let NOTION_COOKIE = '';
 let NOTION_SPACE_ID = process.env.NOTION_SPACE_ID;
 let FETCHED_NOTION_USER_ID = process.env.NOTION_ACTIVE_USER_HEADER;
@@ -73,20 +78,12 @@ process.on('exit', () => {
 });
 
 function getCookie() {
-  const cookies = process.env.NOTION_COOKIE;
-  if (cookies) {
-      let cookiePairs = cookies.split('|');
-      if (cookiePairs.length > 1){
-        isMoreThanOne=true;
-      }
-      let currentIndex = index;
-      index++;
-      if (index >= cookiePairs.length) {
-        index = 0;
-      }
-      return cookiePairs[currentIndex];
+  if (NOTION_COOKIES.length === 0) {
+    return '';
   }
-  return '';
+  const cookie = NOTION_COOKIES[currentCookieIndex];
+  currentCookieIndex = (currentCookieIndex + 1) % NOTION_COOKIES.length;
+  return cookie;
 }
 
 // 添加一个通用的cookie解析函数
@@ -136,8 +133,8 @@ function parseCookiesFromString(cookieString, logPrefix = "") {
 }
 
 // 获取Notion空间ID和用户ID
-async function fetchAndSetNotionIds() {
-  if (!NOTION_COOKIE) {
+async function fetchAndSetNotionIds(cookie) {
+  if (!cookie) {
     logger.error(`无法获取Notion ID: 未设置NOTION_COOKIE`);
     FETCHED_IDS_SUCCESSFULLY = false;
     return;
@@ -188,7 +185,7 @@ async function fetchAndSetNotionIds() {
     }
     
     // 设置cookie
-    document.cookie = NOTION_COOKIE;
+    document.cookie = cookie;
     
     // 创建fetch选项
     const fetchOptions = {
@@ -202,7 +199,7 @@ async function fetchAndSetNotionIds() {
         'origin': 'https://www.notion.so',
         'referer': 'https://www.notion.so/',
         'user-agent': window.navigator.userAgent,
-        'Cookie': NOTION_COOKIE
+        'Cookie': cookie
       },
       body: JSON.stringify({}),
     };
@@ -280,6 +277,7 @@ async function fetchAndSetNotionIds() {
       logger.warning(`清理全局对象时出错: ${cleanupError.message}`);
     }
     
+    return { spaceId: NOTION_SPACE_ID, userId: userIdKey };
   } catch (error) {
     logger.error(`获取Notion ID时出错: ${error}`);
     FETCHED_IDS_SUCCESSFULLY = false;
@@ -429,80 +427,44 @@ function buildNotionRequest(requestData) {
 
 // 流式处理Notion响应
 async function streamNotionResponse(notionRequestBody) {
-  // 创建流
-  const stream = new PassThrough();
-  
-  // 添加初始数据，确保连接建立
-  stream.write(':\n\n');  // 发送一个空注释行，保持连接活跃
-  
-  // 设置HTTP头模板
-  const headers = {
-    'Content-Type': 'application/json',
-    'accept': 'application/x-ndjson',
-    'accept-language': 'en-US,en;q=0.9',
-    'notion-audit-log-platform': 'web',
-    'notion-client-version': '23.13.0.3686',
-    'origin': 'https://www.notion.so',
-    'referer': 'https://www.notion.so/chat',
-    'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/136.0.0.0 Safari/537.36',
-    'x-notion-active-user-header': FETCHED_NOTION_USER_ID,
-    'x-notion-space-id': NOTION_SPACE_ID
-  };
-  
-  // 设置超时处理，确保流不会无限等待
-  const timeoutId = setTimeout(() => {
-    logger.warning(`请求超时，30秒内未收到响应`);
-    try {
-      // 发送结束消息
-      const endChunk = new ChatCompletionChunk({
-        choices: [
-          new Choice({
-            delta: new ChoiceDelta({ content: "请求超时，未收到Notion响应。" }),
-            finish_reason: "timeout"
-          })
-        ]
-      });
-      stream.write(`data: ${JSON.stringify(endChunk)}\n\n`);
-      stream.write('data: [DONE]\n\n');
-      stream.end();
-    } catch (error) {
-      logger.error(`发送超时消息时出错: ${error}`);
-      stream.end();
+  const chunkQueue = new PassThrough();
+  let timeoutId = null;
+
+  try {
+    // 动态获取cookie
+    const currentCookie = getCookie();
+    if (!currentCookie) {
+      throw new Error("无法获取 Notion cookie");
     }
-  }, 30000); // 30秒超时
-  
-  // 启动fetch处理
-  fetchNotionResponse(
-    stream,
-    notionRequestBody,
-    headers,
-    NOTION_API_URL,
-    NOTION_COOKIE,
-    timeoutId
-  ).catch((error) => {
-    logger.error(`流处理出错: ${error}`);
-    clearTimeout(timeoutId);  // 清除超时计时器
+    NOTION_COOKIE = currentCookie; // 更新当前使用的cookie
+    
+    // 获取对应的 space_id 和 user_id
+    notionRequestBody.spaceId = NOTION_SPACE_IDS[currentCookie];
+    const headers = {
+      'Content-Type': 'application/json',
+      'Cookie': currentCookie,
+      'x-notion-active-user-header': NOTION_USER_IDS[currentCookie],
+      "accept": "application/json",
+      "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/136.0.0.0 Safari/537.36"
+    };
     
-    try {
-      // 发送错误消息
-      const errorChunk = new ChatCompletionChunk({
-        choices: [
-          new Choice({
-            delta: new ChoiceDelta({ content: `处理请求时出错: ${error.message}` }),
-            finish_reason: "error"
-          })
-        ]
-      });
-      stream.write(`data: ${JSON.stringify(errorChunk)}\n\n`);
-      stream.write('data: [DONE]\n\n');
-    } catch (e) {
-      logger.error(`发送错误消息时出错: ${e}`);
-    } finally {
-      stream.end();
-    }
-  });
-  
-  return stream;
+    fetchNotionResponse(chunkQueue, notionRequestBody, headers, NOTION_API_URL, currentCookie, timeoutId);
+  } catch (error) {
+    logger.error(`streamNotionResponse 错误: ${error}`);
+    const errorChunk = new ChatCompletionChunk({
+      choices: [
+        new Choice({
+          delta: new ChoiceDelta({ content: `处理流式响应时出错: ${error.message}` }),
+          finish_reason: "error"
+        })
+      ]
+    });
+    chunkQueue.write(`data: ${JSON.stringify(errorChunk)}\n\n`);
+    chunkQueue.write('data: [DONE]\n\n');
+    chunkQueue.end();
+  }
+
+  return chunkQueue;
 }
 
 // 使用fetch调用Notion API并处理流式响应
@@ -891,41 +853,43 @@ async function initialize() {
     logger.error(`启动代理服务器失败: ${error.message}`);
   }
   
-  // 检查是否已从环境变量中获取ID
-  if (NOTION_SPACE_ID && FETCHED_NOTION_USER_ID) {
-    logger.info(`使用环境变量中的Notion ID配置
-      - NOTION_SPACE_ID: ${NOTION_SPACE_ID}
-      - NOTION_ACTIVE_USER_HEADER: ${FETCHED_NOTION_USER_ID}`);
-    // logger.info(`- NOTION_SPACE_ID: ${NOTION_SPACE_ID}`);
-    // logger.info(`- NOTION_ACTIVE_USER_HEADER: ${FETCHED_NOTION_USER_ID}`);
+  // 从环境变量加载所有 cookies
+  const cookiesEnv = process.env.NOTION_COOKIE;
+  if (!cookiesEnv) {
+    logger.error(`错误: 未设置NOTION_COOKIE环境变量，应用无法正常工作`);
+    return;
+  }
+  NOTION_COOKIES = cookiesEnv.split('|').map(c => c.trim()).filter(c => c);
+  
+  if (NOTION_COOKIES.length === 0) {
+    logger.error(`错误: NOTION_COOKIE环境变量中未找到有效cookie`);
+    return;
+  }
+
+  logger.info(`共找到 ${NOTION_COOKIES.length} 个 Notion cookie，正在逐个获取 ID...`);
+
+  let successfulCookies = 0;
+  for (const cookie of NOTION_COOKIES) {
+    try {
+      const ids = await fetchAndSetNotionIds(cookie);
+      if (ids) {
+        NOTION_SPACE_IDS[cookie] = ids.spaceId;
+        NOTION_USER_IDS[cookie] = ids.userId;
+        logger.success(`成功获取 cookie 的 ID: ...${cookie.slice(-10)}`);
+        successfulCookies++;
+      }
+    } catch (error) {
+      logger.error(`获取 cookie 的 ID 失败: ...${cookie.slice(-10)} - ${error.message}`);
+    }
+  }
+
+  if (successfulCookies > 0) {
     FETCHED_IDS_SUCCESSFULLY = true;
+    logger.success(`成功初始化 ${successfulCookies} 个 Notion 账号`);
   } else {
-    // 尝试从Notion API获取ID
-    NOTION_COOKIE = getCookie(); // 获取第一个cookie
-    if (!NOTION_COOKIE) {
-      logger.error(`错误: 未设置NOTION_COOKIE环境变量，应用无法正常工作`);
-      logger.error(`请在.env文件中设置有效的NOTION_COOKIE值`);
-      return;
-    }
-    
-    logger.info(`未检测到ID配置，正在通过API自动获取Notion ID...`);
-    await fetchAndSetNotionIds();
-    
-    if (!FETCHED_IDS_SUCCESSFULLY) {
-      logger.error(`获取Notion ID失败，应用无法正常工作`);
-      logger.error(`请检查以下可能的问题:`);
-      logger.error(`1. NOTION_COOKIE是否有效`);
-      logger.error(`2. 网络连接是否正常`);
-      logger.error(`3. Notion服务是否可访问`);
-      logger.error(`或者手动在.env文件中设置NOTION_SPACE_ID和NOTION_ACTIVE_USER_HEADER`);
-    } else {
-      logger.info(`成功通过API获取Notion ID
-        - 用户ID: ${FETCHED_NOTION_USER_ID}
-        - 空间ID: ${NOTION_SPACE_ID}`);
-      // logger.info(`- 用户ID: ${FETCHED_NOTION_USER_ID}`);
-      // logger.info(`- 空间ID: ${NOTION_SPACE_ID}`);
-    }
+    logger.error(`所有 Notion cookie 均无法成功初始化，应用可能无法正常工作`);
   }
+
   if (process.env.USE_NATIVE_PROXY_POOL === 'true') {
     logger.info(`正在初始化本地代理池...`);
     await proxyPool.initialize();