Upload 7 files

src/CookieManager.js

@@ -0,0 +1,416 @@
+import { JSDOM } from 'jsdom';
+import fetch from 'node-fetch';
+import fs from 'fs';
+import path from 'path';
+import { fileURLToPath } from 'url';
+import { dirname } from 'path';
+
+// 获取当前文件的目录路径
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+// 日志配置
+const logger = {
+  info: (message) => console.log(`\x1b[34m[info] ${message}\x1b[0m`),
+  error: (message) => console.error(`\x1b[31m[error] ${message}\x1b[0m`),
+  warning: (message) => console.warn(`\x1b[33m[warn] ${message}\x1b[0m`),
+  success: (message) => console.log(`\x1b[32m[success] ${message}\x1b[0m`),
+};
+
+class CookieManager {
+  constructor() {
+    this.cookieEntries = []; // 存储cookie及其对应的ID
+    this.currentIndex = 0;
+    this.initialized = false;
+    this.maxRetries = 3; // 最大重试次数
+    this.proxyUrl = process.env.PROXY_URL || "";
+  }
+
+  /**
+   * 从文件加载cookie
+   * @param {string} filePath - cookie文件路径
+   * @returns {Promise<boolean>} - 是否加载成功
+   */
+  async loadFromFile(filePath) {
+    try {
+      // 确保文件路径是绝对路径
+      const absolutePath = path.isAbsolute(filePath) 
+        ? filePath 
+        : path.join(dirname(__dirname), filePath);
+      
+      logger.info(`从文件加载cookie: ${absolutePath}`);
+      
+      // 检查文件是否存在
+      if (!fs.existsSync(absolutePath)) {
+        logger.error(`Cookie文件不存在: ${absolutePath}`);
+        return false;
+      }
+      
+      // 读取文件内容
+      const fileContent = fs.readFileSync(absolutePath, 'utf8');
+      
+      // 根据文件扩展名处理不同格式
+      const ext = path.extname(absolutePath).toLowerCase();
+      let cookieArray = [];
+      
+      if (ext === '.json') {
+        // JSON格式
+        try {
+          const jsonData = JSON.parse(fileContent);
+          if (Array.isArray(jsonData)) {
+            cookieArray = jsonData;
+          } else if (jsonData.cookies && Array.isArray(jsonData.cookies)) {
+            cookieArray = jsonData.cookies;
+          } else {
+            logger.error('JSON文件格式错误，应为cookie数组或包含cookies数组的对象');
+            return false;
+          }
+        } catch (error) {
+          logger.error(`解析JSON文件失败: ${error.message}`);
+          return false;
+        }
+      } else {
+        // 文本格式，每行一个cookie
+        cookieArray = fileContent
+          .split('\n')
+          .map(line => line.trim())
+          .filter(line => line && !line.startsWith('#'));
+      }
+      
+      logger.info(`从文件中读取了 ${cookieArray.length} 个cookie`);
+      
+      // 初始化cookie
+      return await this.initialize(cookieArray.join('|'));
+      
+    } catch (error) {
+      logger.error(`从文件加载cookie失败: ${error.message}`);
+      return false;
+    }
+  }
+
+  /**
+   * 初始化cookie管理器
+   * @param {string} cookiesString - 以"|"分隔的cookie字符串
+   * @returns {Promise<boolean>} - 是否初始化成功
+   */
+  async initialize(cookiesString) {
+    if (!cookiesString) {
+      logger.error('未提供cookie字符串');
+      return false;
+    }
+
+    // 分割cookie字符串
+    const cookieArray = cookiesString.split('|').map(c => c.trim()).filter(c => c);
+    
+    if (cookieArray.length === 0) {
+      logger.error('没有有效的cookie');
+      return false;
+    }
+
+    logger.info(`发现 ${cookieArray.length} 个cookie，开始获取对应的ID信息...`);
+
+    // 清空现有条目
+    this.cookieEntries = [];
+    
+    // 为每个cookie获取ID
+    for (let i = 0; i < cookieArray.length; i++) {
+      const cookie = cookieArray[i];
+      logger.info(`正在处理第 ${i+1}/${cookieArray.length} 个cookie...`);
+      
+      const result = await this.fetchNotionIds(cookie);
+      if (result.success) {
+        this.cookieEntries.push({
+          cookie,
+          spaceId: result.spaceId,
+          userId: result.userId,
+          valid: true,
+          lastUsed: 0 // 记录上次使用时间戳
+        });
+        logger.success(`第 ${i+1} 个cookie验证成功`);
+      } else {
+        if (result.status === 401) {
+          logger.error(`第 ${i+1} 个cookie无效（401未授权），已跳过`);
+        } else {
+          logger.warning(`第 ${i+1} 个cookie验证失败: ${result.error}，已跳过`);
+        }
+      }
+    }
+
+    // 检查是否有有效的cookie
+    if (this.cookieEntries.length === 0) {
+      logger.error('没有有效的cookie，初始化失败');
+      return false;
+    }
+
+    logger.success(`成功初始化 ${this.cookieEntries.length}/${cookieArray.length} 个cookie`);
+    this.initialized = true;
+    this.currentIndex = 0;
+    return true;
+  }
+
+  /**
+   * 保存cookie到文件
+   * @param {string} filePath - 保存路径
+   * @param {boolean} onlyValid - 是否只保存有效的cookie
+   * @returns {boolean} - 是否保存成功
+   */
+  saveToFile(filePath, onlyValid = true) {
+    try {
+      // 确保文件路径是绝对路径
+      const absolutePath = path.isAbsolute(filePath) 
+        ? filePath 
+        : path.join(dirname(__dirname), filePath);
+      
+      // 获取要保存的cookie
+      const cookiesToSave = onlyValid 
+        ? this.cookieEntries.filter(entry => entry.valid).map(entry => entry.cookie)
+        : this.cookieEntries.map(entry => entry.cookie);
+      
+      // 根据文件扩展名选择保存格式
+      const ext = path.extname(absolutePath).toLowerCase();
+      
+      if (ext === '.json') {
+        // 保存为JSON格式
+        const jsonData = {
+          cookies: cookiesToSave,
+          updatedAt: new Date().toISOString(),
+          count: cookiesToSave.length
+        };
+        fs.writeFileSync(absolutePath, JSON.stringify(jsonData, null, 2), 'utf8');
+      } else {
+        // 保存为文本格式，每行一个cookie
+        const content = cookiesToSave.join('\n');
+        fs.writeFileSync(absolutePath, content, 'utf8');
+      }
+      
+      logger.success(`已将 ${cookiesToSave.length} 个cookie保存到文件: ${absolutePath}`);
+      return true;
+    } catch (error) {
+      logger.error(`保存cookie到文件失败: ${error.message}`);
+      return false;
+    }
+  }
+
+  /**
+   * 获取Notion的空间ID和用户ID
+   * @param {string} cookie - Notion cookie
+   * @returns {Promise<Object>} - 包含ID信息的对象
+   */
+  async fetchNotionIds(cookie, retryCount = 0) {
+    if (!cookie) {
+      return { success: false, error: '未提供cookie' };
+    }
+
+    try {
+      // 创建JSDOM实例模拟浏览器环境
+      const dom = new JSDOM("", {
+        url: "https://www.notion.so",
+        referrer: "https://www.notion.so/",
+        contentType: "text/html",
+        includeNodeLocations: true,
+        storageQuota: 10000000,
+        pretendToBeVisual: true,
+        userAgent: "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/136.0.0.0 Safari/537.36"
+      });
+      
+      // 设置全局对象
+      const { window } = dom;
+      
+      // 安全地设置全局对象
+      if (!global.window) global.window = window;
+      if (!global.document) global.document = window.document;
+      
+      // 设置navigator
+      if (!global.navigator) {
+        try {
+          Object.defineProperty(global, 'navigator', {
+            value: window.navigator,
+            writable: true,
+            configurable: true
+          });
+        } catch (navError) {
+          logger.warning(`无法设置navigator: ${navError.message}，继续执行`);
+        }
+      }
+      
+      // 设置cookie
+      document.cookie = cookie;
+      
+      // 创建fetch选项
+      const fetchOptions = {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+          'accept': '*/*',
+          '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/',
+          'user-agent': window.navigator.userAgent,
+          'Cookie': cookie
+        },
+        body: JSON.stringify({}),
+      };
+      
+      // 发送请求
+      const response = await fetch("https://www.notion.so/api/v3/getSpaces", fetchOptions);
+      
+      // 检查响应状态
+      if (response.status === 401) {
+        return { success: false, status: 401, error: '未授权，cookie无效' };
+      }
+      
+      if (!response.ok) {
+        throw new Error(`HTTP error! status: ${response.status}`);
+      }
+      
+      const data = await response.json();
+      
+      // 提取用户ID
+      const userIdKey = Object.keys(data)[0];
+      if (!userIdKey) {
+        throw new Error('无法从响应中提取用户ID');
+      }
+      
+      const userId = userIdKey;
+      
+      // 提取空间ID
+      const userRoot = data[userIdKey]?.user_root?.[userIdKey];
+      const spaceViewPointers = userRoot?.value?.value?.space_view_pointers;
+      
+      if (!spaceViewPointers || !Array.isArray(spaceViewPointers) || spaceViewPointers.length === 0) {
+        throw new Error('在响应中找不到space_view_pointers或spaceId');
+      }
+      
+      const spaceId = spaceViewPointers[0].spaceId;
+      
+      if (!spaceId) {
+        throw new Error('无法从space_view_pointers中提取spaceId');
+      }
+      
+      // 清理全局对象
+      this.cleanupGlobalObjects();
+      
+      return {
+        success: true,
+        userId,
+        spaceId
+      };
+      
+    } catch (error) {
+      // 清理全局对象
+      this.cleanupGlobalObjects();
+      
+      // 重试逻辑
+      if (retryCount < this.maxRetries && error.message !== '未授权，cookie无效') {
+        logger.warning(`获取Notion ID失败，正在重试 (${retryCount + 1}/${this.maxRetries}): ${error.message}`);
+        return await this.fetchNotionIds(cookie, retryCount + 1);
+      }
+      
+      return {
+        success: false,
+        error: error.message
+      };
+    }
+  }
+
+  /**
+   * 清理全局对象
+   */
+  cleanupGlobalObjects() {
+    try {
+      if (global.window) delete global.window;
+      if (global.document) delete global.document;
+      
+      // 安全地删除navigator
+      if (global.navigator) {
+        try {
+          delete global.navigator;
+        } catch (navError) {
+          // 如果无法删除，尝试将其设置为undefined
+          try {
+            Object.defineProperty(global, 'navigator', {
+              value: undefined,
+              writable: true,
+              configurable: true
+            });
+          } catch (defineError) {
+            logger.warning(`无法清理navigator: ${defineError.message}`);
+          }
+        }
+      }
+    } catch (cleanupError) {
+      logger.warning(`清理全局对象时出错: ${cleanupError.message}`);
+    }
+  }
+
+  /**
+   * 获取下一个可用的cookie及其ID
+   * @returns {Object|null} - cookie及其对应的ID，如果没有可用cookie则返回null
+   */
+  getNext() {
+    if (!this.initialized || this.cookieEntries.length === 0) {
+      return null;
+    }
+
+    // 轮询选择下一个cookie
+    const entry = this.cookieEntries[this.currentIndex];
+    
+    // 更新索引，实现轮询
+    this.currentIndex = (this.currentIndex + 1) % this.cookieEntries.length;
+    
+    // 更新最后使用时间
+    entry.lastUsed = Date.now();
+    
+    return {
+      cookie: entry.cookie,
+      spaceId: entry.spaceId,
+      userId: entry.userId
+    };
+  }
+
+  /**
+   * 标记cookie为无效
+   * @param {string} userId - 用户ID
+   */
+  markAsInvalid(userId) {
+    const index = this.cookieEntries.findIndex(entry => entry.userId === userId);
+    if (index !== -1) {
+      this.cookieEntries[index].valid = false;
+      logger.warning(`已将用户ID为 ${userId} 的cookie标记为无效`);
+      
+      // 过滤掉所有无效的cookie
+      this.cookieEntries = this.cookieEntries.filter(entry => entry.valid);
+      
+      // 重置当前索引
+      if (this.cookieEntries.length > 0) {
+        this.currentIndex = 0;
+      }
+    }
+  }
+
+  /**
+   * 获取有效cookie的数量
+   * @returns {number} - 有效cookie的数量
+   */
+  getValidCount() {
+    return this.cookieEntries.filter(entry => entry.valid).length;
+  }
+
+  /**
+   * 获取所有cookie的状态信息
+   * @returns {Array} - cookie状态数组
+   */
+  getStatus() {
+    return this.cookieEntries.map((entry, index) => ({
+      index,
+      userId: entry.userId.substring(0, 8) + '...',
+      spaceId: entry.spaceId.substring(0, 8) + '...',
+      valid: entry.valid,
+      lastUsed: entry.lastUsed ? new Date(entry.lastUsed).toLocaleString() : 'never'
+    }));
+  }
+}
+
+export const cookieManager = new CookieManager(); 

src/ProxyPool.js

@@ -0,0 +1,787 @@
+import axios from 'axios';
+
+/**
+ * 代理池类，用于管理和提供HTTP代理
+ */
+class ProxyPool {
+  /**
+   * 创建代理池实例
+   * @param {Object} options - 配置选项
+   * @param {number} options.targetCount - 目标代理数量，默认20
+   * @param {number} options.batchSize - 每次获取的代理数量，默认20
+   * @param {number} options.testTimeout - 测试代理超时时间(毫秒)，默认3000
+   * @param {number} options.requestTimeout - 请求目标网站超时时间(毫秒)，默认5000
+   * @param {string} options.targetUrl - 目标网站URL，默认'https://www.notion.so'
+   * @param {number} options.concurrentRequests - 并发请求数量，默认15
+   * @param {number} options.minThreshold - 可用代理数量低于此阈值时自动补充，默认5
+   * @param {number} options.checkInterval - 检查代理池状态的时间间隔(毫秒)，默认30000
+   * @param {string} options.proxyProtocol - 代理协议，默认'http'
+   * @param {number} options.maxRefillAttempts - 最大补充尝试次数，默认50
+   * @param {number} options.retryDelay - 重试延迟(毫秒)，默认500
+   * @param {boolean} options.useCache - 是否使用缓存，默认true
+   * @param {number} options.cacheExpiry - 缓存过期时间(毫秒)，默认3600000 (1小时)
+   * @param {string} options.logLevel - 日志级别，可选值：'debug', 'info', 'warn', 'error', 'none'，默认'info'
+   * @param {boolean} options.showProgressBar - 是否显示进度条，默认false
+   */
+  constructor(options = {}) {
+    // 配置参数
+    this.targetCount = options.targetCount || 20;
+    this.batchSize = options.batchSize || 20;
+    this.testTimeout = options.testTimeout || 3000; // 减少测试超时时间
+    this.requestTimeout = options.requestTimeout || 5000; // 减少请求超时时间
+    this.targetUrl = options.targetUrl || 'https://www.notion.so';
+    this.concurrentRequests = options.concurrentRequests || 15; // 增加默认并发请求数
+    this.minThreshold = options.minThreshold || 5;
+    this.checkInterval = options.checkInterval || 30000; // 默认30秒检查一次
+    this.proxyProtocol = options.proxyProtocol || 'http';
+    this.maxRefillAttempts = options.maxRefillAttempts || 50; // 增加最大尝试次数
+    this.retryDelay = options.retryDelay || 500; // 减少重试延迟
+    this.useCache = options.useCache !== undefined ? options.useCache : true;
+    this.cacheExpiry = options.cacheExpiry || 3600000; // 默认1小时
+    this.logLevel = options.logLevel || 'info'; // 默认日志级别为info
+    this.showProgressBar = options.showProgressBar !== undefined ? options.showProgressBar : false; // 是否显示进度条
+    
+    // 内部状态
+    this.availableProxies = [];
+    this.currentIndex = 0;
+    this.isInitialized = false;
+    this.isRefilling = false;
+    this.checkTimer = null;
+    this.proxyCache = new Map(); // 缓存验证过的代理
+    this.proxyCountry = 'us';
+    
+    // 进度条状态
+    this.progressTotal = 0;
+    this.progressCurrent = 0;
+    this.progressBarLength = 30; // 进度条长度
+    this.lastProgressString = ''; // 上次显示的进度字符串
+    
+    // 日志级别权重
+    this.logLevels = {
+      'debug': 0,
+      'info': 1,
+      'warn': 2,
+      'error': 3,
+      'none': 4
+    };
+    
+    // 绑定方法
+    this.getProxy = this.getProxy.bind(this);
+    this.removeProxy = this.removeProxy.bind(this);
+    this.checkAndRefill = this.checkAndRefill.bind(this);
+  }
+
+  /**
+   * 设置代理国家
+   * @param {string} country - 代理国家 支持us uk jp de fr ca
+   */
+  setCountry(country) {
+    this.proxyCountry = country;
+  }
+  
+  /**
+   * 根据设置的日志级别输出日志
+   * @param {string} level - 日志级别
+   * @param {...any} args - 日志参数
+   */
+  log(level, ...args) {
+    if (this.logLevels[level] >= this.logLevels[this.logLevel]) {
+      if (level === 'error') {
+        console.error(...args);
+      } else if (level === 'warn') {
+        console.warn(...args);
+      } else if (level === 'info' || level === 'debug') {
+        console.log(...args);
+      }
+    }
+  }
+  
+  /**
+   * 显示进度条
+   * @param {string} message - 进度条前的消息
+   * @param {number} current - 当前进度
+   * @param {number} total - 总进度
+   * @param {boolean} [complete=false] - 是否完成
+   */
+  showProgress(message, current, total) {
+    if (!this.showProgressBar) return;
+    
+    // 确保进度不超过总数
+    current = Math.min(current, total);
+    
+    // 计算进度百分比
+    const percent = total > 0 ? Math.floor((current / total) * 100) : 0;
+    
+    // 创建终端风格的箭头进度条
+    let bar = '';
+    let ProcessBarLength = 15;
+    let CurrentProcessBarLength = Math.floor(ProcessBarLength*(percent/100));
+    let ProcessBarRemainLength = ProcessBarLength-CurrentProcessBarLength;
+    let ProcessBar = '-'.repeat(CurrentProcessBarLength)+'>';
+    let ProcessBarRemain = '·'.repeat(ProcessBarRemainLength);
+    if (percent === 100) {
+      // 完成状态显示完整箭头
+      bar = '['+'-'.repeat(ProcessBarLength)+'>'+']';
+    } else {
+      bar = '['+ProcessBar+ProcessBarRemain+']';
+    }
+    
+    // 构建进度字符串，添加百分比
+    const progressString = `${message} ${bar} ${current}/${total} (${percent}%)`;
+    
+    // 如果进度字符串与上次相同，则不重复显示
+    if (progressString === this.lastProgressString) {
+      return;
+    }
+    
+    // 清除上一行
+    if (this.lastProgressString) {
+      process.stdout.clearLine(0);
+      process.stdout.cursorTo(0);
+    }
+    
+    // 显示进度条
+    process.stdout.write(progressString);
+    
+    // 如果完成，换行
+    if (percent === 100) {
+      process.stdout.write('\n');
+    }
+    
+    // 保存当前进度字符串
+    this.lastProgressString = progressString;
+  }
+  
+  /**
+   * 初始化代理池
+   * @returns {Promise<void>}
+   */
+  async initialize() {
+    if (this.isInitialized) return;
+    
+    this.log('info', `初始化代理池，目标数量: ${this.targetCount}`);
+    
+    // 重置进度条状态
+    this.progressTotal = this.targetCount;
+    this.progressCurrent = 0;
+    //this.showProgress('初始化代理池进度', this.progressCurrent, this.progressTotal);
+    
+    await this.refillProxies();
+    
+    // 设置定时检查
+    this.checkTimer = setInterval(this.checkAndRefill, this.checkInterval);
+    
+    this.isInitialized = true;
+    
+    // 完成进度条
+    this.showProgress('补充代理进度', this.availableProxies.length, this.progressTotal);
+    this.log('info', `代理池初始化完成，当前可用代理数量: ${this.availableProxies.length}`);
+  }
+  
+  /**
+   * 停止代理池服务
+   */
+  stop() {
+    if (this.checkTimer) {
+      clearInterval(this.checkTimer);
+      this.checkTimer = null;
+    }
+    this.log('info', '代理池服务已停止');
+  }
+  
+  /**
+   * 检查并补充代理
+   */
+  async checkAndRefill() {
+    if (this.availableProxies.length <= this.minThreshold && !this.isRefilling) {
+      this.log('info', `可用代理数量(${this.availableProxies.length})低于阈值(${this.minThreshold})，开始补充代理`);
+      await this.refillProxies();
+    }
+  }
+  
+  /**
+   * 补充代理到目标数量
+   * @returns {Promise<void>}
+   */
+  async refillProxies() {
+    if (this.isRefilling) return;
+    
+    this.isRefilling = true;
+    this.log('info', `开始补充代理，当前数量: ${this.availableProxies.length}，目标数量: ${this.targetCount}`);
+    
+    // 设置进度条状态
+    this.progressTotal = this.targetCount;
+    this.progressCurrent = this.availableProxies.length;
+    this.showProgress('补充代理进度', this.progressCurrent, this.progressTotal);
+    
+    let attempts = 0;
+    
+    try {
+      // 计算需要补充的代理数量
+      const neededProxies = this.targetCount - this.availableProxies.length;
+      
+      // 优先检查缓存中的代理
+      if (this.useCache && this.proxyCache.size > 0) {
+        await this.tryUsingCachedProxies(neededProxies);
+        
+        // 更新进度条
+        this.progressCurrent = this.availableProxies.length;
+        this.showProgress('补充代理进度', this.progressCurrent, this.progressTotal);
+      }
+      
+      // 如果缓存中的代理不足，继续获取新代理
+      while (this.availableProxies.length < this.targetCount && attempts < this.maxRefillAttempts) {
+        attempts++;
+        
+        this.log('debug', `补充尝试 #${attempts}，当前可用代理: ${this.availableProxies.length}/${this.targetCount}`);
+        
+        // 计算本次需要获取的批次大小
+        const remainingNeeded = this.targetCount - this.availableProxies.length;
+        const batchSizeNeeded = Math.max(remainingNeeded * 2, 10); // 获取更多代理以提高成功率，至少10个
+        
+        // 获取代理
+        const proxies = await this.getProxiesFromProvider(batchSizeNeeded);
+        
+        if (proxies.length === 0) {
+          this.log('debug', `没有获取到代理，等待${this.retryDelay/1000}秒后重试...`);
+          await new Promise(resolve => setTimeout(resolve, this.retryDelay));
+          continue;
+        }
+        
+        // 过滤掉已有的代理
+        const newProxies = this.filterExistingProxies(proxies);
+        
+        if (newProxies.length === 0) {
+          this.log('debug', '所有获取的代理都已存在，继续获取新代理...');
+          // 减少等待时间，立即继续尝试
+          continue;
+        }
+        
+        // 保存当前进度，测试代理过程中会临时更新进度条
+        const previousProgress = this.progressCurrent;
+        
+        // 测试代理
+        const results = await this.testProxiesConcurrently(newProxies);
+        
+        // 添加可用代理
+        this.addValidProxies(results);
+        
+        // 更新实际进度
+        this.progressCurrent = this.availableProxies.length;
+        this.showProgress('补充代理进度', this.progressCurrent, this.progressTotal);
+        
+        // 如果已经获取到足够的代理，提前结束
+        if (this.availableProxies.length >= this.targetCount) {
+          break;
+        }
+        
+        // 如果还没补充到足够的代理，减少等待时间再继续
+        if (this.availableProxies.length < this.targetCount) {
+          await new Promise(resolve => setTimeout(resolve, 500)); // 减少等待时间到500毫秒
+        }
+      }
+    } catch (error) {
+      this.log('error', '补充代理过程中出错:', error);
+    } finally {
+      this.isRefilling = false;      
+      if (this.availableProxies.length >= this.targetCount) {
+        this.log('info', `代理补充完成，当前可用代理: ${this.availableProxies.length}/${this.targetCount}`);
+      } else {
+        this.log('info', `已达到最大尝试次数 ${this.maxRefillAttempts}，当前可用代理: ${this.availableProxies.length}/${this.targetCount}`);
+      }
+    }
+  }
+  
+  /**
+   * 尝试使用缓存中的代理
+   * @param {number} neededProxies - 需要的代理数量
+   */
+  async tryUsingCachedProxies(neededProxies) {
+    const now = Date.now();
+    const cachedProxies = [];
+    
+    // 筛选未过期的缓存代理
+    for (const [proxyKey, data] of this.proxyCache.entries()) {
+      if (now - data.timestamp < this.cacheExpiry && data.valid) {
+        cachedProxies.push(proxyKey);
+        
+        if (cachedProxies.length >= neededProxies) {
+          break;
+        }
+      }
+    }
+    
+    if (cachedProxies.length > 0) {
+      this.log('debug', `从缓存中找到 ${cachedProxies.length} 个可能可用的代理`);
+      
+      // 验证缓存的代理是否仍然可用
+      const results = await this.testProxiesConcurrently(cachedProxies);
+      this.addValidProxies(results);
+    }
+  }
+  
+  /**
+   * 过滤掉已存在的代理
+   * @param {Array<string>} proxies - 代理列表
+   * @returns {Array<string>} - 新代理列表
+   */
+  filterExistingProxies(proxies) {
+    return proxies.filter(proxy => {
+      const proxyParts = proxy.split(':');
+      const ip = proxyParts[0];
+      const port = proxyParts[1];
+      
+      // 如果有用户名和密码，也需要比较这些信息
+      if (proxyParts.length >= 4) {
+        const username = proxyParts[2];
+        const password = proxyParts[3];
+        return !this.availableProxies.some(p => 
+          p.ip === ip && 
+          p.port === port && 
+          p.username === username && 
+          p.password === password
+        );
+      }
+      
+      // 没有用户名和密码时，只比较IP和端口
+      return !this.availableProxies.some(p => p.ip === ip && p.port === port);
+    });
+  }
+  
+  /**
+   * 添加有效的代理到代理池
+   * @param {Array<{proxy: string, result: boolean}>} results - 测试结果
+   */
+  addValidProxies(results) {
+    for (const { proxy, result } of results) {
+      if (result) {
+        const proxyParts = proxy.split(':');
+        const ip = proxyParts[0];
+        const port = proxyParts[1];
+        
+        // 检查是否已存在
+        let exists = false;
+        
+        if (proxyParts.length >= 4) {
+          // 有用户名和密码时，同时比较IP、端口、用户名和密码
+          const username = proxyParts[2];
+          const password = proxyParts[3];
+          exists = this.availableProxies.some(p => 
+            p.ip === ip && 
+            p.port === port && 
+            p.username === username && 
+            p.password === password
+          );
+        } else {
+          // 没有用户名和密码时，只比较IP和端口
+          exists = this.availableProxies.some(p => p.ip === ip && p.port === port);
+        }
+        
+        if (!exists) {
+          const proxyObj = {
+            ip,
+            port,
+            protocol: this.proxyProtocol,
+            full: `${this.proxyProtocol}://${proxy}`,
+            addedAt: new Date().toISOString()
+          };
+          
+          // 如果有用户名和密码，添加到代理对象
+          if (proxyParts.length >= 4) {
+            proxyObj.username = proxyParts[2];
+            proxyObj.password = proxyParts[3];
+            proxyObj.full = `${this.proxyProtocol}://${proxyObj.username}:${proxyObj.password}@${ip}:${port}`;
+          }
+          
+          this.availableProxies.push(proxyObj);
+          
+          // 添加到缓存
+          if (this.useCache) {
+            this.proxyCache.set(proxy, { 
+              valid: true, 
+              timestamp: Date.now() 
+            });
+          }
+          
+          this.log('debug', `成功添加代理: ${proxyObj.full}，当前可用代理: ${this.availableProxies.length}/${this.targetCount}`);
+          
+          if (this.availableProxies.length >= this.targetCount) {
+            break;
+          }
+        }
+      } else if (this.useCache) {
+        // 记录无效代理��缓存
+        this.proxyCache.set(proxy, { 
+          valid: false, 
+          timestamp: Date.now() 
+        });
+      }
+    }
+  }
+  
+  /**
+   * 从代理服务获取代理URL
+   * @param {number} count - 请求的代理数量
+   * @returns {Promise<Array<string>>} - 代理URL列表
+   */
+  async getProxiesFromProvider(count = null) {
+    try {
+      const requestCount = count || this.batchSize;
+      // 限制单次请求数量最大为10
+      const actualCount = Math.min(requestCount, 10);
+      
+      // 计算需要发送的请求数量，以获取足够的代理
+      const requestsNeeded = Math.ceil(requestCount / 1); // 假设每次请求只返回1个代理
+      const maxParallelRequests = 5; // 最大并行请求数
+      const actualRequests = Math.min(requestsNeeded, maxParallelRequests);
+      
+      this.log('debug', `需要 ${requestCount} 个代理，将发送 ${actualRequests} 个并行请求`);
+      
+      // 并行发送多个请求
+      const requestPromises = [];
+      for (let i = 0; i < actualRequests; i++) {
+        const url = `https://proxy.doudouzi.me/random/${this.proxyCountry}?number=${actualCount}&protocol=${this.proxyProtocol}&type=json`;
+        requestPromises.push(
+          axios.get(url, { 
+            timeout: 10000,
+            validateStatus: status => true
+          })
+        );
+      }
+      
+      // 等待所有请求完成
+      const responses = await Promise.all(requestPromises);
+      
+      // 处理所有响应，合并代理列表
+      let proxies = [];
+      
+      for (const response of responses) {
+        if (!response.data) continue;
+        
+        // 处理不同的返回格式
+        if (typeof response.data === 'string') {
+          // 如果返回的是字符串，尝试按行解析
+          const proxyDataArray = response.data.trim().split('\n').filter(line => line.trim() !== '');
+          
+          for (const line of proxyDataArray) {
+            try {
+              const proxyData = JSON.parse(line);
+              if (proxyData.ip && proxyData.port) {
+                // 如果有用户名和密码，则使用认证格式
+                if (proxyData.username && proxyData.password) {
+                  proxies.push(`${proxyData.ip}:${proxyData.port}:${proxyData.username}:${proxyData.password}`);
+                } else {
+                  proxies.push(`${proxyData.ip}:${proxyData.port}`);
+                }
+              }
+            } catch (err) {
+              this.log('error', '解析代理数据出错:', err.message);
+            }
+          }
+        } else if (typeof response.data === 'object') {
+          // 如果返回的是对象
+          if (Array.isArray(response.data)) {
+            // 如果是数组，遍历处理
+            for (const proxyData of response.data) {
+              if (proxyData.ip && proxyData.port) {
+                if (proxyData.username && proxyData.password) {
+                  proxies.push(`${proxyData.ip}:${proxyData.port}:${proxyData.username}:${proxyData.password}`);
+                } else {
+                  proxies.push(`${proxyData.ip}:${proxyData.port}`);
+                }
+              }
+            }
+          } else {
+            // 如果是单个对象
+            if (response.data.ip && response.data.port) {
+              if (response.data.username && response.data.password) {
+                proxies.push(`${response.data.ip}:${response.data.port}:${response.data.username}:${response.data.password}`);
+              } else {
+                proxies.push(`${response.data.ip}:${response.data.port}`);
+              }
+            }
+          }
+        }
+      }
+      
+      this.log('debug', `成功获取 ${proxies.length} 个代理`);
+      return proxies;
+    } catch (error) {
+      this.log('error', '获取代理出错:', error.message);
+      return [];
+    }
+  }
+  
+  /**
+   * 并发测试多个代理
+   * @param {Array<string>} proxies - 代理列表
+   * @returns {Promise<Array<{proxy: string, result: boolean}>>} - 测试结果
+   */
+  async testProxiesConcurrently(proxies) {
+    const results = [];
+    const remainingNeeded = this.targetCount - this.availableProxies.length;
+    
+    // 增加并发数以加快处理速度
+    const concurrentRequests = Math.min(this.concurrentRequests * 3, 30); // 增加到最多30个并发请求
+    
+    // 分批处理代理
+    for (let i = 0; i < proxies.length; i += concurrentRequests) {
+      const batch = proxies.slice(i, i + concurrentRequests);
+      
+      // 不再显示测试代理的单独进度条，而是更新补充代理的总体进度
+      if (this.showProgressBar) {
+        // 计算当前测试进度对应的总体进度
+        const testedPercent = i / proxies.length;
+        const progressStep = Math.min(1, testedPercent) * 0.2; // 测试部分占总体进度的20%
+        const currentProgress = this.progressCurrent + progressStep * (this.targetCount - this.progressCurrent);
+        this.showProgress('补充代理进度', Math.floor(currentProgress), this.progressTotal);
+      }
+      
+      const promises = batch.map(proxy => {
+        // 检查缓存中是否有近期验证过的结果
+        if (this.useCache && this.proxyCache.has(proxy)) {
+          const cachedResult = this.proxyCache.get(proxy);
+          const isFresh = (Date.now() - cachedResult.timestamp) < this.cacheExpiry;
+          
+          if (isFresh) {
+            // 使用缓存结果，避免重复测试
+            return Promise.resolve({ proxy, result: cachedResult.valid });
+          }
+        }
+        
+        return this.testProxy(proxy)
+          .then(result => ({ proxy, result }))
+          .catch(() => ({ proxy, result: false }));
+      });
+      
+      const batchResults = await Promise.all(promises);
+      results.push(...batchResults);
+      
+      // 如果已经找到足够的代理，提前结束测试
+      const successCount = results.filter(item => item.result).length;
+      if (successCount >= remainingNeeded) {
+        break;
+      }
+    }
+    
+    // 不再显示测试完成的进度条
+    
+    return results;
+  }
+  
+  /**
+   * 测试代理是否可用
+   * @param {string} proxyUrl - 代理URL
+   * @returns {Promise<boolean>} - 代理是否可用
+   */
+  async testProxy(proxyUrl) {
+    try {
+      // 创建代理配置
+      const proxyParts = proxyUrl.split(':');
+      const proxyConfig = {
+        host: proxyParts[0],
+        port: parseInt(proxyParts[1]),
+        protocol: this.proxyProtocol
+      };
+      
+      // 如果有用户名和密码，添加认证信息
+      if (proxyParts.length >= 4) {
+        proxyConfig.auth = {
+          username: proxyParts[2],
+          password: proxyParts[3]
+        };
+      }
+      
+      // 发送请求到目标网站
+      const response = await axios.get(this.targetUrl, {
+        proxy: proxyConfig,
+        headers: {
+          'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
+          'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8',
+          'Accept-Language': 'en-US,en;q=0.5',
+          'Connection': 'keep-alive',
+          'Upgrade-Insecure-Requests': '1'
+        },
+        timeout: Math.min(this.requestTimeout, 5000), // 减少超时时间，最多5秒
+        validateStatus: status => true,
+        maxRedirects: 5, // 减少最大重定向次数
+        followRedirect: true
+      });
+      
+      // 检查响应是否包含目标网站特有的内容
+      const isTargetContent = response.data && 
+                             (typeof response.data === 'string') && 
+                             (response.data.includes('notion') || 
+                              response.data.includes('Notion'));
+      
+      const isValid = response.status === 200 && isTargetContent;
+      
+      if (isValid) {
+        this.log('debug', `代理 ${proxyUrl} 请求目标网站成功，状态码: ${response.status}`);
+      } else {
+        this.log('debug', `代理 ${proxyUrl} 请求目标网站失败，状态码: ${response.status}`);
+      }
+      
+      return isValid;
+    } catch (error) {
+      this.log('debug', `代理 ${proxyUrl} 请求出错: ${error.message}`);
+      return false;
+    }
+  }
+  
+  /**
+   * 获取一个可用代理
+   * @returns {Object|null} - 代理对象，如果没有可用代理则返回null
+   */
+  getProxy() {
+    if (this.availableProxies.length === 0) {
+      this.log('warn', '没有可用代理');
+      return null;
+    }
+    
+    // 轮询方式获取代理
+    const proxy = this.availableProxies[this.currentIndex];
+    this.currentIndex = (this.currentIndex + 1) % this.availableProxies.length;
+    
+    return proxy;
+  }
+  
+  /**
+   * 移除指定代理
+   * @param {string} ip - 代理IP
+   * @param {string|number} port - 代理端口
+   * @returns {boolean} - 是否成功移除
+   */
+  removeProxy(ip, port) {
+    const portStr = port.toString();
+    const initialLength = this.availableProxies.length;
+    
+    // 找到要移除的代理
+    const proxyToRemove = this.availableProxies.find(
+      proxy => proxy.ip === ip && proxy.port === portStr
+    );
+    
+    if (proxyToRemove) {
+      // 更新缓存，标记为无效
+      if (this.useCache) {
+        const proxyKey = `${ip}:${portStr}`;
+        this.proxyCache.set(proxyKey, { valid: false, timestamp: Date.now() });
+      }
+    }
+    
+    this.availableProxies = this.availableProxies.filter(
+      proxy => !(proxy.ip === ip && proxy.port === portStr)
+    );
+    
+    // 重置当前索引，确保不会越界
+    if (this.currentIndex >= this.availableProxies.length && this.availableProxies.length > 0) {
+      this.currentIndex = 0;
+    }
+    
+    const removed = initialLength > this.availableProxies.length;
+    
+    if (removed) {
+      this.log('debug', `已移除代理 ${ip}:${port}，当前可用代理: ${this.availableProxies.length}`);
+    } else {
+      this.log('debug', `未找到要移除的代理 ${ip}:${port}`);
+    }
+    
+    // 如果移除后代理数量低于阈值，触发补充
+    this.checkAndRefill();
+    
+    return removed;
+  }
+  
+  /**
+   * 获取所有可用代理
+   * @returns {Array<Object>} - 代理对象数组
+   */
+  getAllProxies() {
+    return [...this.availableProxies];
+  }
+  
+  /**
+   * 获取可用代理数量
+   * @returns {number} - 代理数量
+   */
+  getCount() {
+    return this.availableProxies.length;
+  }
+  
+  /**
+   * 清理过期的缓存条目
+   */
+  cleanupCache() {
+    if (!this.useCache) return;
+    
+    const now = Date.now();
+    let cleanupCount = 0;
+    
+    for (const [key, data] of this.proxyCache.entries()) {
+      if (now - data.timestamp > this.cacheExpiry) {
+        this.proxyCache.delete(key);
+        cleanupCount++;
+      }
+    }
+    
+    if (cleanupCount > 0) {
+      this.log('debug', `清理了 ${cleanupCount} 个过期的缓存代理`);
+    }
+  }
+}
+
+// 使用示例
+async function example() {
+  // 创建代理池实例
+  const proxyPool = new ProxyPool({
+    targetCount: 10,           // 目标保持10个代理
+    minThreshold: 3,           // 当可用代理少于3个时，自动补充
+    checkInterval: 60000,      // 每60秒检查一次
+    targetUrl: 'https://www.notion.so',
+    concurrentRequests: 20,    // 增加并发请求数
+    useCache: true,            // 启用缓存
+    maxRefillAttempts: 50,     // 增加最大尝试次数
+    retryDelay: 500,           // 减少重试延迟
+    testTimeout: 3000,         // 减少测试超时时间
+    requestTimeout: 5000,      // 减少请求超时时间
+    logLevel: 'info',          // 设置日志级别
+    showProgressBar: true      // 启用进度条
+  });
+  
+  // 初始化代理池
+  await proxyPool.initialize();
+  
+  // 获取一个代理
+  const proxy = proxyPool.getProxy();
+  console.log('获取到代理:', proxy);
+  
+  // 模拟使用一段时间后，移除一个代理
+  setTimeout(() => {
+    if (proxy) {
+      proxyPool.removeProxy(proxy.ip, proxy.port);
+    }
+    
+    // 获取所有代理
+    const allProxies = proxyPool.getAllProxies();
+    console.log(`当前所有代理(${allProxies.length}):`, allProxies);
+    
+    // 使用完毕后停止服务
+    setTimeout(() => {
+      proxyPool.stop();
+      console.log('代理池示例运行完毕');
+    }, 5000);
+  }, 5000);
+}
+
+// 如果直接运行此文件，则执行示例
+if (typeof require !== 'undefined' && require.main === module) {
+  example().catch(err => console.error('示例运行出错:', err));
+}
+
+// 导出 ProxyPool 类和实例
+export default ProxyPool;
+export const proxyPool = new ProxyPool({
+  logLevel: 'info', // 默认导出的实例使用info级别的日志
+  showProgressBar: true // 启用进度条
+});
+

src/ProxyServer.js

@@ -0,0 +1,204 @@
+import { spawn } from 'child_process';
+import { fileURLToPath } from 'url';
+import { dirname, join } from 'path';
+import fs from 'fs';
+import os from 'os';
+import dotenv from 'dotenv';
+import chalk from 'chalk';
+
+// 获取当前文件的目录路径
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+// 加载环境变量
+dotenv.config({ path: join(dirname(__dirname), '.env') });
+
+// 日志配置
+const logger = {
+  info: (message) => console.log(chalk.blue(`[ProxyServer] ${message}`)),
+  error: (message) => console.error(chalk.red(`[ProxyServer] ${message}`)),
+  warning: (message) => console.warn(chalk.yellow(`[ProxyServer] ${message}`)),
+  success: (message) => console.log(chalk.green(`[ProxyServer] ${message}`)),
+};
+
+class ProxyServer {
+  constructor() {
+    this.proxyProcess = null;
+    this.platform = process.env.PROXY_SERVER_PLATFORM || 'auto';
+    this.port = process.env.PROXY_SERVER_PORT || 10655;
+    this.logPath = process.env.PROXY_SERVER_LOG_PATH || './proxy_server.log';
+    this.enabled = process.env.ENABLE_PROXY_SERVER === 'true';
+    this.proxyAuthToken = process.env.PROXY_AUTH_TOKEN || 'default_token';
+    this.logStream = null;
+  }
+
+  // 获取当前系统平台
+  detectPlatform() {
+    if (this.platform !== 'auto') {
+      return this.platform;
+    }
+
+    const platform = os.platform();
+    const arch = os.arch();
+
+    if (platform === 'win32') {
+      return 'windows';
+    } else if (platform === 'linux') {
+      if (arch === 'arm64') {
+        return 'android';
+      } else {
+        return 'linux';
+      }
+    } else if (platform === 'android') {
+      return 'android';
+    } else {
+      logger.warning(`未知平台: ${platform}, ${arch}, 默认使用linux版本`);
+      return 'linux';
+    }
+  }
+
+  // 获取代理服务器可执行文件路径
+  getProxyServerPath() {
+    const platform = this.detectPlatform();
+    const proxyDir = join(__dirname, 'proxy');
+
+    switch (platform) {
+      case 'windows':
+        return join(proxyDir, 'chrome_proxy_server_windows_amd64.exe');
+      case 'linux':
+        return join(proxyDir, 'chrome_proxy_server_linux_amd64');
+      case 'android':
+        return join(proxyDir, 'chrome_proxy_server_android_arm64');
+      default:
+        logger.error(`不支持的平台: ${platform}`);
+        return null;
+    }
+  }
+
+  // 启动代理服务器
+  async start() {
+    if (!this.enabled) {
+      logger.info('代理服务器未启用，跳过启动');
+      return;
+    }
+
+    if (this.proxyProcess) {
+      logger.warning('代理服务器已经在运行中');
+      return;
+    }
+
+    const proxyServerPath = this.getProxyServerPath();
+    if (!proxyServerPath) {
+      logger.error('无法获取代理服务器路径');
+      return;
+    }
+
+    try {
+      // 确保可执行文件有执行权限（在Linux/Android上）
+      if (this.detectPlatform() !== 'windows') {
+        try {
+          fs.chmodSync(proxyServerPath, 0o755);
+        } catch (err) {
+          logger.warning(`无法设置执行权限: ${err.message}`);
+        }
+      }
+
+      // 创建日志文件
+      this.logStream = fs.createWriteStream(this.logPath, { flags: 'a' });
+      
+      // 修复 stdio 参数问题
+      // 启动代理服务器进程
+      this.proxyProcess = spawn(proxyServerPath, [
+        '--port', this.port.toString(),
+        '--token', this.proxyAuthToken
+      ], {
+        stdio: ['ignore', 'pipe', 'pipe'], // 使用pipe而不是直接传递流
+        detached: false
+      });
+
+      // 将进程的输出重定向到日志文件
+      if (this.proxyProcess.stdout) {
+        this.proxyProcess.stdout.pipe(this.logStream);
+      }
+      
+      if (this.proxyProcess.stderr) {
+        this.proxyProcess.stderr.pipe(this.logStream);
+      }
+
+      // 设置进程事件处理
+      this.proxyProcess.on('error', (err) => {
+        logger.error(`代理服务器启动失败: ${err.message}`);
+        this.proxyProcess = null;
+        if (this.logStream) {
+          this.logStream.end();
+          this.logStream = null;
+        }
+      });
+
+      this.proxyProcess.on('exit', (code, signal) => {
+        logger.info(`代理服务器已退出，退出码: ${code}, 信号: ${signal}`);
+        this.proxyProcess = null;
+        if (this.logStream) {
+          this.logStream.end();
+          this.logStream = null;
+        }
+      });
+
+      // 等待一段时间，确保服务器启动
+      await new Promise(resolve => setTimeout(resolve, 1000));
+      
+      if (this.proxyProcess && this.proxyProcess.exitCode === null) {
+        logger.success(`代理服务器已启动，端口: ${this.port}, 日志文件: ${this.logPath}`);
+        return true;
+      } else {
+        logger.error('代理服务器启动失败');
+        if (this.logStream) {
+          this.logStream.end();
+          this.logStream = null;
+        }
+        return false;
+      }
+    } catch (error) {
+      logger.error(`启动代理服务器时出错: ${error.message}`);
+      if (this.logStream) {
+        this.logStream.end();
+        this.logStream = null;
+      }
+      return false;
+    }
+  }
+
+  // 停止代理服务器
+  stop() {
+    if (!this.proxyProcess) {
+      //logger.info('代理服务器已关闭');
+      return;
+    }
+
+    try {
+      // 在Windows上使用taskkill确保子进程也被终止
+      if (this.detectPlatform() === 'windows' && this.proxyProcess.pid) {
+        spawn('taskkill', ['/pid', this.proxyProcess.pid, '/f', '/t']);
+      } else {
+        // 在Linux/Android上使用kill信号
+        this.proxyProcess.kill('SIGTERM');
+      }
+
+      logger.success('代理服务器已停止');
+    } catch (error) {
+      logger.error(`停止代理服务器时出错: ${error.message}`);
+    } finally {
+      this.proxyProcess = null;
+      if (this.logStream) {
+        this.logStream.end();
+        this.logStream = null;
+      }
+    }
+  }
+}
+
+// 创建单例
+const proxyServer = new ProxyServer();
+
+// 导出
+export { proxyServer }; 

src/cookie-cli.js

@@ -0,0 +1,301 @@
+#!/usr/bin/env node
+
+import { cookieManager } from './CookieManager.js';
+import dotenv from 'dotenv';
+import { fileURLToPath } from 'url';
+import { dirname, join } from 'path';
+import fs from 'fs';
+import readline from 'readline';
+import chalk from 'chalk';
+
+// 获取当前文件的目录路径
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+// 加载环境变量
+dotenv.config({ path: join(dirname(__dirname), '.env') });
+
+// 日志配置
+const logger = {
+  info: (message) => console.log(chalk.blue(`[信息] ${message}`)),
+  error: (message) => console.error(chalk.red(`[错误] ${message}`)),
+  warning: (message) => console.warn(chalk.yellow(`[警告] ${message}`)),
+  success: (message) => console.log(chalk.green(`[成功] ${message}`)),
+};
+
+// 创建readline接口
+const rl = readline.createInterface({
+  input: process.stdin,
+  output: process.stdout
+});
+
+// 默认cookie文件路径
+const DEFAULT_COOKIE_FILE = 'cookies.txt';
+
+let isSaveCookie = false;
+let isAddCookie = false;
+
+// 显示帮助信息
+function showHelp() {
+  console.log(chalk.cyan('Notion Cookie 管理工具'));
+  console.log(chalk.cyan('===================='));
+  console.log('');
+  console.log('可用命令:');
+  console.log('  help      - 显示此帮助信息');
+  console.log('  list      - 列出所有cookie');
+  console.log('  add       - 添加新的cookie');
+  console.log('  validate  - 验证所有cookie');
+  console.log('  remove    - 删除指定cookie');
+  console.log('  save      - 保存cookie到文件');
+  console.log('  load      - 从文件加载cookie');
+  console.log('  exit      - 退出程序');
+  console.log('');
+}
+
+// 列出所有cookie
+async function listCookies() {
+  if (!cookieManager.initialized || cookieManager.getValidCount() === 0) {
+    logger.warning('没有可用的cookie，请先加载或添加cookie');
+    return;
+  }
+
+  const status = cookieManager.getStatus();
+  console.log(chalk.cyan('\nCookie 列表:'));
+  console.log(chalk.cyan('==========='));
+  
+  status.forEach((entry, idx) => {
+    const validMark = entry.valid ? chalk.green('✓') : chalk.red('✗');
+    console.log(`${idx + 1}. ${validMark} 用户ID: ${entry.userId}, 空间ID: ${entry.spaceId}, 上次使用: ${entry.lastUsed}`);
+  });
+  
+  console.log(`\n共有 ${status.length} 个cookie，${cookieManager.getValidCount()} 个有效\n`);
+}
+
+// 添加新cookie
+async function addCookie() {
+  return new Promise((resolve) => {
+    rl.question(chalk.yellow('请输入Notion cookie: '), async (cookie) => {
+      if (!cookie || cookie.trim() === '') {
+        logger.error('Cookie不能为空');
+        resolve();
+        return;
+      }
+      
+      logger.info('正在验证cookie...');
+      const result = await cookieManager.fetchNotionIds(cookie.trim());
+      
+      if (result.success) {
+        // 如果cookie管理器尚未初始化，先初始化
+        if (!cookieManager.initialized) {
+          await cookieManager.initialize(cookie.trim());
+        } else {
+          // 已初始化，直接添加到现有条目
+          cookieManager.cookieEntries.push({
+            cookie: cookie.trim(),
+            spaceId: result.spaceId,
+            userId: result.userId,
+            valid: true,
+            lastUsed: 0
+          });
+        }
+        isAddCookie = true;
+        logger.success(`Cookie添加成功! 用户ID: ${result.userId}, 空间ID: ${result.spaceId}`);
+
+      } else {
+        logger.error(`Cookie验证失败: ${result.error}`);
+      }
+      
+      resolve();
+    });
+  });
+}
+
+// 验证所有cookie
+async function validateCookies() {
+  if (!cookieManager.initialized || cookieManager.cookieEntries.length === 0) {
+    logger.warning('没有可用的cookie，请先加载或添加cookie');
+    return;
+  }
+  
+  logger.info('开始验证所有cookie...');
+  
+  const originalEntries = [...cookieManager.cookieEntries];
+  cookieManager.cookieEntries = [];
+  
+  for (let i = 0; i < originalEntries.length; i++) {
+    const entry = originalEntries[i];
+    logger.info(`正在验证第 ${i+1}/${originalEntries.length} 个cookie...`);
+    
+    const result = await cookieManager.fetchNotionIds(entry.cookie);
+    if (result.success) {
+      cookieManager.cookieEntries.push({
+        cookie: entry.cookie,
+        spaceId: result.spaceId,
+        userId: result.userId,
+        valid: true,
+        lastUsed: entry.lastUsed || 0
+      });
+      logger.success(`第 ${i+1} 个cookie验证成功`);
+    } else {
+      logger.error(`第 ${i+1} 个cookie验证失败: ${result.error}`);
+    }
+  }
+  
+  logger.info(`验证完成，共 ${originalEntries.length} 个cookie，${cookieManager.cookieEntries.length} 个有效`);
+}
+
+// 删除指定cookie
+async function removeCookie() {
+  if (!cookieManager.initialized || cookieManager.cookieEntries.length === 0) {
+    logger.warning('没有可用的cookie，请先加载或添加cookie');
+    return;
+  }
+  
+  // 先列出所有cookie
+  await listCookies();
+  
+  return new Promise((resolve) => {
+    rl.question(chalk.yellow('请输入要删除的cookie编号: '), (input) => {
+      const index = parseInt(input) - 1;
+      
+      if (isNaN(index) || index < 0 || index >= cookieManager.cookieEntries.length) {
+        logger.error('无效的编号');
+        resolve();
+        return;
+      }
+      
+      const removed = cookieManager.cookieEntries.splice(index, 1)[0];
+      logger.success(`已删除编号 ${index + 1} 的cookie (用户ID: ${removed.userId.substring(0, 8)}...)`);
+      
+      // 重置当前索引
+      if (cookieManager.cookieEntries.length > 0) {
+        cookieManager.currentIndex = 0;
+      }
+      
+      resolve();
+    });
+  });
+}
+
+// 保存cookie到文件
+async function saveCookies() {
+  if (!cookieManager.initialized || cookieManager.cookieEntries.length === 0) {
+    logger.warning('没有可用的cookie，请先加载或添加cookie');
+    return;
+  }
+  
+  return new Promise((resolve) => {
+    rl.question(chalk.yellow(`请输入保存文件路径 (默认: ${DEFAULT_COOKIE_FILE}): `), (filePath) => {
+      const path = filePath.trim() || DEFAULT_COOKIE_FILE;
+      
+      rl.question(chalk.yellow('是否只保存有效的cookie? (y/n, 默认: y): '), (onlyValidInput) => {
+        const onlyValid = onlyValidInput.toLowerCase() !== 'n';
+        
+        const success = cookieManager.saveToFile(path, onlyValid);
+        if (success) {
+          logger.success(`Cookie已保存到文件: ${path}`);
+          isSaveCookie = true;
+        }
+        
+        resolve();
+      });
+    });
+  });
+}
+
+// 从文件加载cookie
+async function loadCookies() {
+  return new Promise((resolve) => {
+    rl.question(chalk.yellow(`请输入cookie文件路径 (默认: ${DEFAULT_COOKIE_FILE}): `), async (filePath) => {
+      const path = filePath.trim() || DEFAULT_COOKIE_FILE;
+      
+      logger.info(`正在从文件加载cookie: ${path}`);
+      const success = await cookieManager.loadFromFile(path);
+      
+      if (success) {
+        logger.success(`成功从文件加载cookie，共 ${cookieManager.getValidCount()} 个有效cookie`);
+      } else {
+        logger.error(`从文件加载cookie失败`);
+      }
+      
+      resolve();
+    });
+  });
+}
+
+// 主函数
+async function main() {
+  // 显示欢迎信息
+  console.log(chalk.cyan('\nNotion Cookie 管理工具'));
+  console.log(chalk.cyan('====================\n'));
+  
+  // 检查是否有环境变量中的cookie
+  const envCookie = process.env.NOTION_COOKIE;
+  if (envCookie) {
+    logger.info('检测到环境变量中的NOTION_COOKIE，正在初始化...');
+    await cookieManager.initialize(envCookie);
+  }
+  
+  // 检查是否有环境变量中的cookie文件
+  const envCookieFile = process.env.COOKIE_FILE;
+  if (envCookieFile && !cookieManager.initialized) {
+    logger.info(`检测到环境变量中的COOKIE_FILE: ${envCookieFile}，正在加载...`);
+    await cookieManager.loadFromFile(envCookieFile);
+  }
+  
+  // 如果没有cookie，检查默认文件
+  if (!cookieManager.initialized && fs.existsSync(DEFAULT_COOKIE_FILE)) {
+    logger.info(`检测到默认cookie文件: ${DEFAULT_COOKIE_FILE}，正在加载...`);
+    await cookieManager.loadFromFile(DEFAULT_COOKIE_FILE);
+  }
+  
+  showHelp();
+  
+  // 命令循环
+  while (true) {
+    const command = await new Promise((resolve) => {
+      rl.question(chalk.green('> '), (cmd) => {
+        resolve(cmd.trim().toLowerCase());
+      });
+    });
+    
+    switch (command) {
+      case 'help':
+        showHelp();
+        break;
+      case 'list':
+        await listCookies();
+        break;
+      case 'add':
+        await addCookie();
+        break;
+      case 'validate':
+        await validateCookies();
+        break;
+      case 'remove':
+        await removeCookie();
+        break;
+      case 'save':
+        await saveCookies();
+        break;
+      case 'load':
+        await loadCookies();
+        break;
+      case 'exit':
+      case 'quit':
+      case 'q':
+        logger.info('感谢使用，再见!');
+        rl.close();
+        process.exit(0);
+      default:
+        logger.error(`未知命令: ${command}`);
+        logger.info('输入 "help" 查看可用命令');
+    }
+  }
+}
+
+// 启动程序
+main().catch((error) => {
+  logger.error(`程序出错: ${error.message}`);
+  process.exit(1);
+}); 

src/lightweight-client-express.js

@@ -0,0 +1,343 @@
+import express from 'express';
+import dotenv from 'dotenv';
+import { randomUUID } from 'crypto';
+import { fileURLToPath } from 'url';
+import { dirname, join } from 'path';
+import chalk from 'chalk';
+import {
+  ChatMessage, ChatCompletionRequest, Choice, ChoiceDelta, ChatCompletionChunk
+} from './models.js';
+import {
+  initialize,
+  streamNotionResponse,
+  buildNotionRequest,
+  INITIALIZED_SUCCESSFULLY
+} from './lightweight-client.js';
+import { proxyPool } from './ProxyPool.js';
+import { cookieManager } from './CookieManager.js';
+
+// 获取当前文件的目录路径
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+// 加载环境变量
+dotenv.config({ path: join(dirname(__dirname), '.env') });
+
+// 活跃流管理器 - 用于跟踪和管理当前活跃的请求流
+const activeStreams = new Map();
+
+// 日志配置
+const logger = {
+  info: (message) => console.log(chalk.blue(`[info] ${message}`)),
+  error: (message) => console.error(chalk.red(`[error] ${message}`)),
+  warning: (message) => console.warn(chalk.yellow(`[warn] ${message}`)),
+  success: (message) => console.log(chalk.green(`[success] ${message}`)),
+  request: (method, path, status, time) => {
+    const statusColor = status >= 500 ? chalk.red : 
+                        status >= 400 ? chalk.yellow : 
+                        status >= 300 ? chalk.cyan : 
+                        status >= 200 ? chalk.green : chalk.white;
+    console.log(`${chalk.magenta(`[${method}]`)} - ${path} ${statusColor(status)} ${chalk.gray(`${time}ms`)}`);
+  }
+};
+
+// 认证配置
+const EXPECTED_TOKEN = process.env.PROXY_AUTH_TOKEN || "default_token";
+
+// 创建Express应用
+const app = express();
+app.use(express.json({ limit: '50mb' }));
+app.use(express.urlencoded({ extended: true, limit: '50mb' }));
+
+// 请求日志中间件
+app.use((req, res, next) => {
+  const start = Date.now();
+  
+  // 保存原始的 end 方法
+  const originalEnd = res.end;
+  
+  // 重写 end 方法以记录请求完成时间
+  res.end = function(...args) {
+    const duration = Date.now() - start;
+    logger.request(req.method, req.path, res.statusCode, duration);
+    return originalEnd.apply(this, args);
+  };
+  
+  next();
+});
+
+// 认证中间件
+function authenticate(req, res, next) {
+  const authHeader = req.headers.authorization;
+  
+  if (!authHeader || !authHeader.startsWith('Bearer ')) {
+    return res.status(401).json({
+      error: {
+        message: "Authentication required. Please provide a valid Bearer token.",
+        type: "authentication_error"
+      }
+    });
+  }
+  
+  const token = authHeader.split(' ')[1];
+  
+  if (token !== EXPECTED_TOKEN) {
+    return res.status(401).json({
+      error: {
+        message: "Invalid authentication credentials",
+        type: "authentication_error"
+      }
+    });
+  }
+  
+  next();
+}
+
+// 流管理函数
+function manageStream(clientId, stream) {
+  // 如果该客户端已有活跃流，先关闭它
+  if (activeStreams.has(clientId)) {
+    try {
+      const oldStream = activeStreams.get(clientId);
+      logger.info(`关闭客户端 ${clientId} 的旧流`);
+      oldStream.end();
+    } catch (error) {
+      logger.error(`关闭旧流时出错: ${error.message}`);
+    }
+  }
+  
+  // 注册新流
+  activeStreams.set(clientId, stream);
+  
+  // 当流结束时从管理器中移除
+  stream.on('end', () => {
+    if (activeStreams.get(clientId) === stream) {
+      activeStreams.delete(clientId);
+      //logger.info(`客户端 ${clientId} 的流已结束并移除`);
+    }
+  });
+  
+  stream.on('error', (error) => {
+    logger.error(`流错误: ${error.message}`);
+    if (activeStreams.get(clientId) === stream) {
+      activeStreams.delete(clientId);
+    }
+  });
+  
+  return stream;
+}
+
+// API路由
+
+// 获取模型列表
+app.get('/v1/models', authenticate, (req, res) => {
+  // 返回可用模型列表
+  const modelList = {
+    data: [
+      { id: "openai-gpt-4.1" },
+      { id: "anthropic-opus-4" },
+      { id: "anthropic-sonnet-4" },
+      { id: "anthropic-sonnet-3.x-stable" },
+      { id: "google-gemini-2.5-pro"}, //vertex-gemini-2.5-pro
+      { id: "google-gemini-2.5-flash"}, //vertex-gemini-2.5-flash
+    ]
+  };
+  
+  res.json(modelList);
+});
+
+// 聊天完成端点
+app.post('/v1/chat/completions', authenticate, async (req, res) => {
+  try {
+    // 生成或获取客户端ID
+    const clientId = req.headers['x-client-id'] || randomUUID();
+    
+    // 检查是否成功初始化
+    if (!INITIALIZED_SUCCESSFULLY) {
+      return res.status(500).json({
+        error: {
+          message: "系统未成功初始化。请检查您的NOTION_COOKIE是否有效。",
+          type: "server_error"
+        }
+      });
+    }
+    
+    // 检查是否有可用的cookie
+    if (cookieManager.getValidCount() === 0) {
+      return res.status(500).json({
+        error: {
+          message: "没有可用的有效cookie。请检查您的NOTION_COOKIE配置。",
+          type: "server_error"
+        }
+      });
+    }
+    
+    // 验证请求数据
+    const requestData = req.body;
+    
+    if (!requestData.messages || !Array.isArray(requestData.messages) || requestData.messages.length === 0) {
+      return res.status(400).json({
+        error: {
+          message: "Invalid request: 'messages' field must be a non-empty array.",
+          type: "invalid_request_error"
+        }
+      });
+    }
+    
+    // 构建Notion请求
+    const notionRequestBody = buildNotionRequest(requestData);
+    
+    // 处理流式响应
+    if (requestData.stream) {
+      res.setHeader('Content-Type', 'text/event-stream');
+      res.setHeader('Cache-Control', 'no-cache');
+      res.setHeader('Connection', 'keep-alive');
+      
+      logger.info(`开始流式响应`);
+      const stream = await streamNotionResponse(notionRequestBody);
+      
+      // 管理流
+      manageStream(clientId, stream);
+      
+      stream.pipe(res);
+      
+      // 处理客户端断开连接
+      req.on('close', () => {
+        logger.info(`客户端 ${clientId} 断开连接`);
+        if (activeStreams.has(clientId)) {
+          try {
+            activeStreams.get(clientId).end();
+            activeStreams.delete(clientId);
+          } catch (error) {
+            logger.error(`关闭流时出错: ${error.message}`);
+          }
+        }
+      });
+    } else {
+      // 非流式响应
+      // 创建一个内部流来收集完整响应
+      logger.info(`开始非流式响应`);
+      const chunks = [];
+      const stream = await streamNotionResponse(notionRequestBody);
+      
+      // 管理流
+      manageStream(clientId, stream);
+      
+      return new Promise((resolve, reject) => {
+        stream.on('data', (chunk) => {
+          const chunkStr = chunk.toString();
+          if (chunkStr.startsWith('data: ') && !chunkStr.includes('[DONE]')) {
+            try {
+              const dataJson = chunkStr.substring(6).trim();
+              if (dataJson) {
+                const chunkData = JSON.parse(dataJson);
+                if (chunkData.choices && chunkData.choices[0].delta && chunkData.choices[0].delta.content) {
+                  chunks.push(chunkData.choices[0].delta.content);
+                }
+              }
+            } catch (error) {
+              logger.error(`解析非流式响应块时出错: ${error}`);
+            }
+          }
+        });
+        
+        stream.on('end', () => {
+          const fullResponse = {
+            id: `chatcmpl-${randomUUID()}`,
+            object: "chat.completion",
+            created: Math.floor(Date.now() / 1000),
+            model: requestData.model,
+            choices: [
+              {
+                index: 0,
+                message: {
+                  role: "assistant",
+                  content: chunks.join('')
+                },
+                finish_reason: "stop"
+              }
+            ],
+            usage: {
+              prompt_tokens: null,
+              completion_tokens: null,
+              total_tokens: null
+            }
+          };
+          
+          res.json(fullResponse);
+          resolve();
+        });
+        
+        stream.on('error', (error) => {
+          logger.error(`非流式响应出错: ${error}`);
+          reject(error);
+        });
+        
+        // 处理客户端断开连接
+        req.on('close', () => {
+          logger.info(`客户端 ${clientId} 断开连接（非流式）`);
+          if (activeStreams.has(clientId)) {
+            try {
+              activeStreams.get(clientId).end();
+              activeStreams.delete(clientId);
+            } catch (error) {
+              logger.error(`关闭流时出错: ${error.message}`);
+            }
+          }
+        });
+      });
+    }
+  } catch (error) {
+    logger.error(`聊天完成端点错误: ${error}`);
+    res.status(500).json({
+      error: {
+        message: `Internal server error: ${error.message}`,
+        type: "server_error"
+      }
+    });
+  }
+});
+
+// 健康检查端点
+app.get('/health', (req, res) => {
+  res.json({
+    status: 'ok',
+    timestamp: new Date().toISOString(),
+    initialized: INITIALIZED_SUCCESSFULLY,
+    valid_cookies: cookieManager.getValidCount(),
+    active_streams: activeStreams.size
+  });
+});
+
+// Cookie状态查询端点
+app.get('/cookies/status', authenticate, (req, res) => {
+  res.json({
+    total_cookies: cookieManager.getValidCount(),
+    cookies: cookieManager.getStatus()
+  });
+});
+
+// 启动服务器
+const PORT = process.env.PORT || 7860;
+
+// 设置代理池日志级别为warn，减少详细日志输出
+proxyPool.logLevel = 'info';
+
+// 初始化并启动服务器
+initialize().then(() => {
+  app.listen(PORT, () => {
+    logger.info(`服务已启动 - 端口: ${PORT}`);
+    logger.info(`访问地址: http://localhost:${PORT}`);
+    
+    if (INITIALIZED_SUCCESSFULLY) {
+      logger.success(`系统初始化状态: ✅`);
+      logger.success(`可用cookie数量: ${cookieManager.getValidCount()}`);
+    } else {
+      logger.warning(`系统初始化状态: ❌`);
+      logger.warning(`警告: 系统未成功初始化，API调用将无法正常工作`);
+      logger.warning(`请检查NOTION_COOKIE配置是否有效`);
+    }
+  });
+}).catch((error) => {
+  logger.error(`初始化失败: ${error}`);
+}); 

src/lightweight-client.js

@@ -0,0 +1,882 @@
+import fetch from 'node-fetch';
+import { JSDOM } from 'jsdom';
+import dotenv from 'dotenv';
+import { randomUUID } from 'crypto';
+import { fileURLToPath } from 'url';
+import { dirname, join } from 'path';
+import { PassThrough } from 'stream';
+import chalk from 'chalk';
+import {
+  NotionTranscriptConfigValue,
+  NotionTranscriptContextValue, NotionTranscriptItem, NotionDebugOverrides,
+  NotionRequestBody, ChoiceDelta, Choice, ChatCompletionChunk, NotionTranscriptItemByuser
+} from './models.js';
+import { proxyPool } from './ProxyPool.js';
+import { proxyServer } from './ProxyServer.js';
+import { cookieManager } from './CookieManager.js';
+
+// 获取当前文件的目录路径
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+// 加载环境变量
+dotenv.config({ path: join(dirname(__dirname), '.env') });
+
+// 日志配置
+const logger = {
+  info: (message) => console.log(chalk.blue(`[info] ${message}`)),
+  error: (message) => console.error(chalk.red(`[error] ${message}`)),
+  warning: (message) => console.warn(chalk.yellow(`[warn] ${message}`)),
+  success: (message) => console.log(chalk.green(`[success] ${message}`)),
+};
+
+// 配置
+const NOTION_API_URL = "https://www.notion.so/api/v3/runInferenceTranscript";
+// 这些变量将由cookieManager动态提供
+let currentCookieData = null;
+const USE_NATIVE_PROXY_POOL = process.env.USE_NATIVE_PROXY_POOL === 'true';
+const ENABLE_PROXY_SERVER = process.env.ENABLE_PROXY_SERVER === 'true';
+let proxy = null;
+
+// 代理配置
+const PROXY_URL = process.env.PROXY_URL || "";
+
+// 标记是否成功初始化
+let INITIALIZED_SUCCESSFULLY = false;
+
+// 注册进程退出事件，确保代理服务器在程序退出时关闭
+process.on('exit', () => {
+  try {
+    if (proxyServer) {
+      proxyServer.stop();
+    }
+  } catch (error) {
+    logger.error(`程序退出时关闭代理服务器出错: ${error.message}`);
+  }
+});
+
+// 捕获意外退出信号
+['SIGINT', 'SIGTERM', 'SIGQUIT'].forEach(signal => {
+  process.on(signal, () => {
+    logger.info(`收到${signal}信号，正在关闭代理服务器...`);
+    try {
+      if (proxyServer) {
+        proxyServer.stop();
+      }
+    } catch (error) {
+      logger.error(`关闭代理服务器出错: ${error.message}`);
+    }
+    process.exit(0);
+  });
+});
+
+// 构建Notion请求
+function buildNotionRequest(requestData) {
+  // 确保我们有当前的cookie数据
+  if (!currentCookieData) {
+    currentCookieData = cookieManager.getNext();
+    if (!currentCookieData) {
+      throw new Error('没有可用的cookie');
+    }
+  }
+
+  // 当前时间
+  const now = new Date();
+  // 格式化为ISO字符串，确保包含毫秒和时区
+  const isoString = now.toISOString();
+  
+  // 生成随机名称，类似于Python版本
+  const randomWords = ["Project", "Workspace", "Team", "Studio", "Lab", "Hub", "Zone", "Space"];
+  const userName = `User${Math.floor(Math.random() * 900) + 100}`; // 生成100-999之间的随机数
+  const spaceName = `${randomWords[Math.floor(Math.random() * randomWords.length)]} ${Math.floor(Math.random() * 99) + 1}`;
+  
+  // 创建transcript数组
+  const transcript = [];
+  
+  // 添加配置项
+  if(requestData.model === 'anthropic-sonnet-3.x-stable'){
+    transcript.push(new NotionTranscriptItem({
+      type: "config",
+      value: new NotionTranscriptConfigValue({
+      })
+    }));
+  } else if(requestData.model === 'google-gemini-2.5-pro'){
+    transcript.push(new NotionTranscriptItem({
+      type: "config",
+      value: new NotionTranscriptConfigValue({
+        model: 'vertex-gemini-2.5-pro'
+      })
+    }));
+  } else if (requestData.model === 'google-gemini-2.5-flash'){
+    transcript.push(new NotionTranscriptItem({
+      type: "config",
+      value: new NotionTranscriptConfigValue({
+        model: 'vertex-gemini-2.5-flash'
+      })
+    }));
+  }
+  else{
+      transcript.push(new NotionTranscriptItem({
+    type: "config",
+    value: new NotionTranscriptConfigValue({
+      model: requestData.model
+    })
+  }));
+  }
+
+  
+  // 添加上下文项
+  transcript.push(new NotionTranscriptItem({
+    type: "context",
+    value: new NotionTranscriptContextValue({
+      userId: currentCookieData.userId,
+      spaceId: currentCookieData.spaceId,
+      surface: "home_module",
+      timezone: "America/Los_Angeles",
+      userName: userName,
+      spaceName: spaceName,
+      spaceViewId: randomUUID(),
+      currentDatetime: isoString
+    })
+  }));
+  
+  // 添加agent-integration项
+  transcript.push(new NotionTranscriptItem({
+    type: "agent-integration"
+  }));
+  
+  // 添加消息
+  for (const message of requestData.messages) {
+    // 处理消息内容，确保格式一致
+    let content = message.content;
+    
+    // 处理内容为数组的情况
+    if (Array.isArray(content)) {
+      let textContent = "";
+      for (const part of content) {
+        if (part && typeof part === 'object' && part.type === 'text') {
+          if (typeof part.text === 'string') {
+            textContent += part.text;
+          }
+        }
+      }
+      content = textContent || ""; // 使用提取的文本或空字符串
+    } else if (typeof content !== 'string') {
+      content = ""; // 如果不是字符串或数组，则默认为空字符串
+    }
+    
+    if (message.role === "system") {
+      // 系统消息作为用户消息添加
+      transcript.push(new NotionTranscriptItemByuser({
+        type: "user",
+        value: [[content]],
+        userId: currentCookieData.userId,
+        createdAt: message.createdAt || isoString
+      }));
+    } else if (message.role === "user") {
+      // 用户消息
+      transcript.push(new NotionTranscriptItemByuser({
+        type: "user",
+        value: [[content]],
+        userId: currentCookieData.userId,
+        createdAt: message.createdAt || isoString
+      }));
+    } else if (message.role === "assistant") {
+      // 助手消息
+      transcript.push(new NotionTranscriptItem({
+        type: "markdown-chat",
+        value: content,
+        traceId: message.traceId || randomUUID(),
+        createdAt: message.createdAt || isoString
+      }));
+    }
+  }
+  
+  // 创建请求体
+  return new NotionRequestBody({
+    spaceId: currentCookieData.spaceId,
+    transcript: transcript,
+    createThread: true,
+    traceId: randomUUID(),
+    debugOverrides: new NotionDebugOverrides({
+      cachedInferences: {},
+      annotationInferences: {},
+      emitInferences: false
+    }),
+    generateTitle: false,
+    saveAllThreadOperations: false
+  });
+}
+
+// 流式处理Notion响应
+async function streamNotionResponse(notionRequestBody) {
+  // 确保我们有当前的cookie数据
+  if (!currentCookieData) {
+    currentCookieData = cookieManager.getNext();
+    if (!currentCookieData) {
+      throw new Error('没有可用的cookie');
+    }
+  }
+
+  // 创建流
+  const stream = new PassThrough();
+  
+  // 标记流状态
+  let streamClosed = false;
+  
+  // 重写stream.end方法，确保安全关闭
+  const originalEnd = stream.end;
+  stream.end = function(...args) {
+    if (streamClosed) return; // 避免重复关闭
+    streamClosed = true;
+    return originalEnd.apply(this, args);
+  };
+  
+  // 添加初始数据，确保连接建立
+  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': currentCookieData.userId,
+    'x-notion-space-id': currentCookieData.spaceId
+  };
+  
+  // 设置超时处理，确保流不会无限等待
+  const timeoutId = setTimeout(() => {
+    if (streamClosed) return;
+    
+    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}`);
+      if (!streamClosed) stream.end();
+    }
+  }, 30000); // 30秒超时
+  
+  // 启动fetch处理
+  fetchNotionResponse(
+    stream,
+    notionRequestBody,
+    headers,
+    NOTION_API_URL,
+    currentCookieData.cookie,
+    timeoutId
+  ).catch((error) => {
+    if (streamClosed) return;
+    
+    logger.error(`流处理出错: ${error}`);
+    clearTimeout(timeoutId);  // 清除超时计时器
+    
+    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 {
+      if (!streamClosed) stream.end();
+    }
+  });
+  
+  return stream;
+}
+
+// 使用fetch调用Notion API并处理流式响应
+async function fetchNotionResponse(chunkQueue, notionRequestBody, headers, notionApiUrl, notionCookie, timeoutId) {
+  let responseReceived = false;
+  let dom = null;
+  
+  // 检查流是否已关闭的辅助函数
+  const isStreamClosed = () => {
+    return chunkQueue.destroyed || (typeof chunkQueue.closed === 'boolean' && chunkQueue.closed);
+  };
+  
+  // 安���写入函数，确保只向开启的流写入数据
+  const safeWrite = (data) => {
+    if (!isStreamClosed()) {
+      try {
+        return chunkQueue.write(data);
+      } catch (error) {
+        logger.error(`流写入错误: ${error.message}`);
+        return false;
+      }
+    }
+    return false;
+  };
+  
+  try {
+    // 创建JSDOM实例模拟浏览器环境
+    dom = new JSDOM("", {
+      url: "https://www.notion.so",
+      referrer: "https://www.notion.so/chat",
+      contentType: "text/html",
+      includeNodeLocations: true,
+      storageQuota: 10000000,
+      pretendToBeVisual: true,
+      userAgent: "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/136.0.0.0 Safari/537.36"
+    });
+    
+    // 设置全局对象
+    const { window } = dom;
+    
+    // 使用更安全的方式设置全局对象
+    try {
+      if (!global.window) {
+        global.window = window;
+      }
+      
+      if (!global.document) {
+        global.document = window.document;
+      }
+      
+      // 安全地设置navigator
+      if (!global.navigator) {
+        try {
+          Object.defineProperty(global, 'navigator', {
+            value: window.navigator,
+            writable: true,
+            configurable: true
+          });
+        } catch (navError) {
+          logger.warning(`无法设置navigator: ${navError.message}，继续执行`);
+          // 继续执行，不会中断流程
+        }
+      }
+    } catch (globalError) {
+      logger.warning(`设置全局对象时出错: ${globalError.message}`);
+    }
+    
+    // 设置cookie
+    document.cookie = notionCookie;
+    
+    // 创建fetch选项
+    const fetchOptions = {
+      method: 'POST',
+      headers: {
+        ...headers,
+        'user-agent': window.navigator.userAgent,
+        'Cookie': notionCookie
+      },
+      body: JSON.stringify(notionRequestBody),
+    };
+    
+    // 添加代理配置（如果有）
+    if (USE_NATIVE_PROXY_POOL && ENABLE_PROXY_SERVER && !PROXY_URL) {
+      proxy = proxyPool.getProxy();
+      if (proxy !== null)
+      {
+        logger.info(`使用代理: ${proxy.full}`);
+      }
+      else{
+        logger.warning(`没有可用代理`);
+      }
+    } else if(USE_NATIVE_PROXY_POOL&&!PROXY_URL&&!ENABLE_PROXY_SERVER) {
+      const { HttpsProxyAgent } = await import('https-proxy-agent');
+      proxy = proxyPool.getProxy();
+      fetchOptions.agent = new HttpsProxyAgent(proxy.full);
+      logger.info(`使用代理: ${proxy.full}`);
+    }else if(PROXY_URL){
+      const { HttpsProxyAgent } = await import('https-proxy-agent');
+      fetchOptions.agent = new HttpsProxyAgent(PROXY_URL);
+      logger.info(`使用代理: ${PROXY_URL}`);
+    }
+    let response = null;
+    // 发送请求
+    if (ENABLE_PROXY_SERVER && USE_NATIVE_PROXY_POOL){
+      response = await fetch('http://127.0.0.1:10655/proxy', {
+        method: 'POST',
+        body: JSON.stringify({
+          method: 'POST',
+          url: notionApiUrl,
+          headers: fetchOptions.headers,
+          body: fetchOptions.body,
+          stream:true,
+          proxy:proxy.full
+        }),
+      });
+    }
+    else if (ENABLE_PROXY_SERVER && !USE_NATIVE_PROXY_POOL && PROXY_URL){
+      response = await fetch('http://127.0.0.1:10655/proxy', {
+        method: 'POST',
+        body: JSON.stringify({
+          method: 'POST',
+          url: notionApiUrl,
+          headers: fetchOptions.headers,
+          body: fetchOptions.body,
+          proxy: PROXY_URL,
+          stream:true,
+        }),
+      });
+    }
+    else if(ENABLE_PROXY_SERVER && !USE_NATIVE_PROXY_POOL){
+      response = await fetch('http://127.0.0.1:10655/proxy', {
+        method: 'POST',
+        body: JSON.stringify({
+          method: 'POST',
+          url: notionApiUrl,
+          headers: fetchOptions.headers,
+          body: fetchOptions.body,
+          stream:true,
+        }),
+      });
+    }
+    else{
+      response = await fetch(notionApiUrl, fetchOptions);
+    }
+
+    // 检查是否收到401错误（未授权）
+    if (response.status === 401) {
+      logger.error(`收到401未授权错误，cookie可能已失效`);
+      // 标记当前cookie为无效
+      cookieManager.markAsInvalid(currentCookieData.userId);
+      // 尝试获取下一个cookie
+      currentCookieData = cookieManager.getNext();
+      
+      if (!currentCookieData) {
+        throw new Error('所有cookie均已失效，无法继续请求');
+      }
+      
+      // 使用新cookie重新构建请求体
+      const newRequestBody = buildNotionRequest({
+        model: notionRequestBody.transcript[0]?.value?.model || '',
+        messages: [] // 这里应该根据实际情况重构消息
+      });
+      
+      // 使用新cookie重试请求
+      return fetchNotionResponse(
+        chunkQueue,
+        newRequestBody,
+        {
+          ...headers,
+          'x-notion-active-user-header': currentCookieData.userId,
+          'x-notion-space-id': currentCookieData.spaceId
+        },
+        notionApiUrl,
+        currentCookieData.cookie,
+        timeoutId
+      );
+    }
+    
+    if (!response.ok) {
+      throw new Error(`HTTP error! status: ${response.status}`);
+    }
+    
+    // 处理流式响应
+    if (!response.body) {
+      throw new Error("Response body is null");
+    }
+    
+    // 创建流读取器
+    const reader = response.body;
+    let buffer = '';
+    
+    // 处理数据块
+    reader.on('data', (chunk) => {
+      // 检查流是否已关闭
+      if (isStreamClosed()) {
+        try {
+          reader.destroy();
+        } catch (error) {
+          logger.error(`销毁reader时出错: ${error.message}`);
+        }
+        return;
+      }
+      
+      try {
+        // 标记已收到响应
+        if (!responseReceived) {
+          responseReceived = true;
+          logger.info(`已连接Notion API`);
+          clearTimeout(timeoutId);  // 清除超时计时器
+        }
+        
+        // 解码数据
+        const text = chunk.toString('utf8');
+        buffer += text;
+        
+        // 按行分割并处理完整的JSON对象
+        const lines = buffer.split('\n');
+        buffer = lines.pop() || ''; // 保留最后一行（可能不完整）
+        
+        for (const line of lines) {
+          if (!line.trim()) continue;
+          
+          try {
+            const jsonData = JSON.parse(line);
+            
+            // 提取内容
+            if (jsonData?.type === "markdown-chat" && typeof jsonData?.value === "string") {
+              const content = jsonData.value;
+              if (!content) continue;
+              
+              // 创建OpenAI格式的块
+              const chunk = new ChatCompletionChunk({
+                choices: [
+                  new Choice({
+                    delta: new ChoiceDelta({ content }),
+                    finish_reason: null
+                  })
+                ]
+              });
+              
+              // 添加到队列
+              const dataStr = `data: ${JSON.stringify(chunk)}\n\n`;
+              if (!safeWrite(dataStr)) {
+                // 如果写入失败，结束处理
+                try {
+                  reader.destroy();
+                } catch (error) {
+                  logger.error(`写入失败后销毁reader时出错: ${error.message}`);
+                }
+                return;
+              }
+            } else if (jsonData?.recordMap) {
+              // 忽略recordMap响应
+            } else {
+              // 忽略其他类型响应
+            }
+          } catch (jsonError) {
+            logger.error(`解析JSON出错: ${jsonError}`);
+          }
+        }
+      } catch (error) {
+        logger.error(`处理数据块出错: ${error}`);
+      }
+    });
+    
+    // 处理流结束
+    reader.on('end', () => {
+      try {
+        logger.info(`响应完成`);
+        if (cookieManager.getValidCount() > 1){
+          // 尝试切换到下一个cookie
+          currentCookieData = cookieManager.getNext();
+          logger.info(`切换到下一个cookie: ${currentCookieData.userId}`);
+        }
+        
+        // 如果没有收到任何响应，发送一个提示消息
+        if (!responseReceived) {
+          if (!ENABLE_PROXY_SERVER){
+            logger.warning(`未从Notion收到内容响应，请尝试启用tls代理服务`)
+          }else if (USE_NATIVE_PROXY_POOL){
+            logger.warning(`未从Notion收到内容响应，请重roll，或者切换cookie`)
+          }else{
+            logger.warning(`未从Notion收到内容响应,请更换ip重试`);
+          }
+          if (USE_NATIVE_PROXY_POOL) {
+            proxyPool.removeProxy(proxy.ip, proxy.port);
+          }
+          
+          const noContentChunk = new ChatCompletionChunk({
+            choices: [
+              new Choice({
+                delta: new ChoiceDelta({ content: "未从Notion收到内容响应,请更换ip重试。" }),
+                finish_reason: "no_content"
+              })
+            ]
+          });
+          safeWrite(`data: ${JSON.stringify(noContentChunk)}\n\n`);
+        }
+        
+        // 创建结束块
+        const endChunk = new ChatCompletionChunk({
+          choices: [
+            new Choice({
+              delta: new ChoiceDelta({ content: null }),
+              finish_reason: "stop"
+            })
+          ]
+        });
+        
+        // 添加到队列
+        safeWrite(`data: ${JSON.stringify(endChunk)}\n\n`);
+        safeWrite('data: [DONE]\n\n');
+        
+        // 清除超时计时器（如果尚未清除）
+        if (timeoutId) clearTimeout(timeoutId);
+        
+        // 清理全局对象
+        try {
+          if (global.window) delete global.window;
+          if (global.document) delete global.document;
+          
+          // 安全地删除navigator
+          if (global.navigator) {
+            try {
+              delete global.navigator;
+            } catch (navError) {
+              // 如果无法删除，尝试将其设置为undefined
+              try {
+                Object.defineProperty(global, 'navigator', {
+                  value: undefined,
+                  writable: true,
+                  configurable: true
+                });
+              } catch (defineError) {
+                logger.warning(`无法清理navigator: ${defineError.message}`);
+              }
+            }
+          }
+        } catch (cleanupError) {
+          logger.warning(`清理全局对象时出错: ${cleanupError.message}`);
+        }
+        
+        // 结束流
+        if (!isStreamClosed()) {
+          chunkQueue.end();
+        }
+      } catch (error) {
+        logger.error(`Error in stream end handler: ${error}`);
+        if (timeoutId) clearTimeout(timeoutId);
+        
+        // 清理全局对象
+        try {
+          if (global.window) delete global.window;
+          if (global.document) delete global.document;
+          
+          // 安全地删除navigator
+          if (global.navigator) {
+            try {
+              delete global.navigator;
+            } catch (navError) {
+              // 如果无法删除，尝试将其设置为undefined
+              try {
+                Object.defineProperty(global, 'navigator', {
+                  value: undefined,
+                  writable: true,
+                  configurable: true
+                });
+              } catch (defineError) {
+                logger.warning(`无法清理navigator: ${defineError.message}`);
+              }
+            }
+          }
+        } catch (cleanupError) {
+          logger.warning(`清理全局对象时出错: ${cleanupError.message}`);
+        }
+        
+        if (!isStreamClosed()) {
+          chunkQueue.end();
+        }
+      }
+    });
+    
+    // 处理错误
+    reader.on('error', (error) => {
+      logger.error(`Stream error: ${error}`);
+      if (timeoutId) clearTimeout(timeoutId);
+      
+      // 清理全局对象
+      try {
+        if (global.window) delete global.window;
+        if (global.document) delete global.document;
+        
+        // 安全地删除navigator
+        if (global.navigator) {
+          try {
+            delete global.navigator;
+          } catch (navError) {
+            // 如果无法删除，尝试将其设置为undefined
+            try {
+              Object.defineProperty(global, 'navigator', {
+                value: undefined,
+                writable: true,
+                configurable: true
+              });
+            } catch (defineError) {
+              logger.warning(`无法清理navigator: ${defineError.message}`);
+            }
+          }
+        }
+      } catch (cleanupError) {
+        logger.warning(`清理全局对象时出错: ${cleanupError.message}`);
+      }
+      
+      try {
+        const errorChunk = new ChatCompletionChunk({
+          choices: [
+            new Choice({
+              delta: new ChoiceDelta({ content: `流读取错误: ${error.message}` }),
+              finish_reason: "error"
+            })
+          ]
+        });
+        safeWrite(`data: ${JSON.stringify(errorChunk)}\n\n`);
+        safeWrite('data: [DONE]\n\n');
+      } catch (e) {
+        logger.error(`Error sending error message: ${e}`);
+      } finally {
+        if (!isStreamClosed()) {
+          chunkQueue.end();
+        }
+      }
+    });
+  } catch (error) {
+    logger.error(`Notion API请求失败: ${error}`);
+    // 清理全局对象
+    try {
+      if (global.window) delete global.window;
+      if (global.document) delete global.document;
+      
+      // 安全地删除navigator
+      if (global.navigator) {
+        try {
+          delete global.navigator;
+        } catch (navError) {
+          // 如果无法删除，尝试将其设置为undefined
+          try {
+            Object.defineProperty(global, 'navigator', {
+              value: undefined,
+              writable: true,
+              configurable: true
+            });
+          } catch (defineError) {
+            logger.warning(`无法清理navigator: ${defineError.message}`);
+          }
+        }
+      }
+    } catch (cleanupError) {
+      logger.warning(`清理全局对象时出错: ${cleanupError.message}`);
+    }
+    
+    if (timeoutId) clearTimeout(timeoutId);
+    
+    // 确保在错误情况下也触发流结束
+    try {
+      if (!responseReceived && !isStreamClosed()) {
+        const errorChunk = new ChatCompletionChunk({
+          choices: [
+            new Choice({
+              delta: new ChoiceDelta({ content: `Notion API请求失败: ${error.message}` }),
+              finish_reason: "error"
+            })
+          ]
+        });
+        safeWrite(`data: ${JSON.stringify(errorChunk)}\n\n`);
+        safeWrite('data: [DONE]\n\n');
+      }
+    } catch (e) {
+      logger.error(`发送错误消息时出错: ${e}`);
+    }
+    
+    if (!isStreamClosed()) {
+      chunkQueue.end();
+    }
+    
+    throw error; // 重新抛出错误以便上层捕获
+  }
+}
+
+// 应用初始化
+async function initialize() {
+  logger.info(`初始化Notion配置...`);
+  
+  // 启动代理服务器
+  try {
+    await proxyServer.start();
+  } catch (error) {
+    logger.error(`启动代理服务器失败: ${error.message}`);
+  }
+  
+  // 初始化cookie管理器
+  let initResult = false;
+  
+  // 检查是否配置了cookie文件
+  const cookieFilePath = process.env.COOKIE_FILE;
+  if (cookieFilePath) {
+    logger.info(`检测到COOKIE_FILE配置: ${cookieFilePath}`);
+    initResult = await cookieManager.loadFromFile(cookieFilePath);
+    
+    if (!initResult) {
+      logger.error(`从文件加载cookie失败，尝试使用环境变量中的NOTION_COOKIE`);
+    }
+  }
+  
+  // 如果文件加载失败或未配置文件，尝试从环境变量加载
+  if (!initResult) {
+    const cookiesString = process.env.NOTION_COOKIE;
+    if (!cookiesString) {
+      logger.error(`错误: 未设置NOTION_COOKIE环境变量或COOKIE_FILE路径，应用无法正常工作`);
+      logger.error(`请在.env文件中设置有效的NOTION_COOKIE值或COOKIE_FILE路径`);
+      INITIALIZED_SUCCESSFULLY = false;
+      return;
+    }
+    
+    logger.info(`正在从环境变量初始化cookie管理器...`);
+    initResult = await cookieManager.initialize(cookiesString);
+    
+    if (!initResult) {
+      logger.error(`初始化cookie管理器失败，应用无法正常工作`);
+      INITIALIZED_SUCCESSFULLY = false;
+      return;
+    }
+  }
+  
+  // 获取第一个可用的cookie数据
+  currentCookieData = cookieManager.getNext();
+  if (!currentCookieData) {
+    logger.error(`没有可用的cookie，应用无法正常工作`);
+    INITIALIZED_SUCCESSFULLY = false;
+    return;
+  }
+  
+  logger.success(`成功初始化cookie管理器，共有 ${cookieManager.getValidCount()} 个有效cookie`);
+  logger.info(`当前使用的cookie对应的用户ID: ${currentCookieData.userId}`);
+  logger.info(`当前使用的cookie对应的空间ID: ${currentCookieData.spaceId}`);
+  
+  if (process.env.USE_NATIVE_PROXY_POOL === 'true') {
+    logger.info(`正在初始化本地代理池...`);
+    // 设置代理池的日志级别为warn，减少详细日志输出
+    proxyPool.logLevel = 'info';
+    // 启用进度条显示
+    proxyPool.showProgressBar = true;
+    
+    if (['us', 'uk', 'jp', 'de', 'fr', 'ca'].includes(process.env.PROXY_COUNTRY)) {
+      proxyPool.setCountry(process.env.PROXY_COUNTRY);
+    } else {
+      logger.warning(`未设置正确PROXY_COUNTRY，使用默认代理国家: us`);
+      proxyPool.setCountry('us');
+    }
+    await proxyPool.initialize();
+    await new Promise(resolve => setTimeout(resolve, 1000));
+    logger.success(`代理池初始化完成，当前代理国家: ${proxyPool.proxyCountry}`);
+  }
+  
+  INITIALIZED_SUCCESSFULLY = true;
+}
+
+// 导出函数
+export {
+  initialize,
+  streamNotionResponse,
+  buildNotionRequest,
+  INITIALIZED_SUCCESSFULLY
+}; 

src/models.js

@@ -0,0 +1,213 @@
+import { randomUUID } from 'crypto';
+
+// 输入模型 (OpenAI-like)
+export class ChatMessage {
+  constructor({
+    id = generateCustomId(),
+    role,
+    content,
+    userId = null,
+    createdAt = null,
+    traceId = null
+  }) {
+    this.id = id;
+    this.role = role; // "system", "user", "assistant"
+    this.content = content;
+    this.userId = userId;
+    this.createdAt = createdAt;
+    this.traceId = traceId;
+  }
+}
+
+export class ChatCompletionRequest {
+  constructor({
+    messages,
+    model = "notion-proxy",
+    stream = false,
+    notion_model = "anthropic-opus-4"
+  }) {
+    this.messages = messages;
+    this.model = model;
+    this.stream = stream;
+    this.notion_model = notion_model;
+  }
+}
+
+// Notion 模型
+export class NotionTranscriptConfigValue {
+  constructor({
+    type = "markdown-chat",
+    model
+  }) {
+    this.type = type;
+    this.model = model;
+  }
+}
+
+
+export class NotionTranscriptContextValue {
+  constructor({
+    userId,
+    spaceId,
+    surface = "home_module",
+    timezone = "America/Los_Angeles",
+    userName,
+    spaceName,
+    spaceViewId,
+    currentDatetime
+  }) {
+    this.userId = userId;
+    this.spaceId = spaceId;
+    this.surface = surface;
+    this.timezone = timezone;
+    this.userName = userName;
+    this.spaceName = spaceName;
+    this.spaceViewId = spaceViewId;
+    this.currentDatetime = currentDatetime;
+  }
+}
+
+export class NotionTranscriptItem {
+  constructor({
+    id = generateCustomId(),
+    type,
+    value = null,
+
+  }) {
+    this.id = id;
+    this.type = type; // "markdown-chat", "agent-integration", "context"
+    this.value = value;
+  }
+}
+
+export class NotionTranscriptItemByuser {
+  constructor({
+    id = generateCustomId(),
+    type,
+    value = null,
+    userId,
+    createdAt
+
+  }) {
+    this.id = id;
+    this.type = type; // "config", "user"
+    this.value = value;
+    this.userId = userId;
+    this.createdAt = createdAt;
+  }
+}
+
+export class NotionDebugOverrides {
+  constructor({
+    cachedInferences = {},
+    annotationInferences = {},
+    emitInferences = false
+  }) {
+    this.cachedInferences = cachedInferences;
+    this.annotationInferences = annotationInferences;
+    this.emitInferences = emitInferences;
+  }
+}
+
+export function generateCustomId() {
+  // 创建固定部分
+  const prefix1 = '2036702a';
+  const prefix2 = '4d19';
+  const prefix5 = '00aa';
+  
+  // 生成随机十六进制字符
+  function randomHex(length) {
+    return Array(length).fill(0).map(() => 
+      Math.floor(Math.random() * 16).toString(16)
+    ).join('');
+  }
+  
+  // 组合所有部分
+  const part3 = '80' + randomHex(2);  // 8xxx
+  const part4 = randomHex(4);        // xxxx
+  const part5 = prefix5 + randomHex(8); // 00aaxxxxxxxx
+  
+  return `${prefix1}-${prefix2}-${part3}-${part4}-${part5}`;
+}
+
+export class NotionRequestBody {
+  constructor({
+    traceId = randomUUID(),
+    spaceId,
+    transcript,
+    createThread = false,
+    debugOverrides = new NotionDebugOverrides({}),
+    generateTitle = true,
+    saveAllThreadOperations = true,
+  }) {
+    this.traceId = traceId;
+    this.spaceId = spaceId;
+    this.transcript = transcript;
+    this.createThread = createThread;
+    this.debugOverrides = debugOverrides;
+    this.generateTitle = generateTitle;
+    this.saveAllThreadOperations = saveAllThreadOperations;
+  }
+}
+
+// 输出模型 (OpenAI SSE)
+export class ChoiceDelta {
+  constructor({
+    content = null
+  }) {
+    this.content = content;
+  }
+}
+
+export class Choice {
+  constructor({
+    index = 0,
+    delta,
+    finish_reason = null
+  }) {
+    this.index = index;
+    this.delta = delta;
+    this.finish_reason = finish_reason;
+  }
+}
+
+export class ChatCompletionChunk {
+  constructor({
+    id = `chatcmpl-${randomUUID()}`,
+    object = "chat.completion.chunk",
+    created = Math.floor(Date.now() / 1000),
+    model = "notion-proxy",
+    choices
+  }) {
+    this.id = id;
+    this.object = object;
+    this.created = created;
+    this.model = model;
+    this.choices = choices;
+  }
+}
+
+// 模型列表端点 /v1/models
+export class Model {
+  constructor({
+    id,
+    object = "model",
+    created = Math.floor(Date.now() / 1000),
+    owned_by = "notion"
+  }) {
+    this.id = id;
+    this.object = object;
+    this.created = created;
+    this.owned_by = owned_by;
+  }
+}
+
+export class ModelList {
+  constructor({
+    object = "list",
+    data
+  }) {
+    this.object = object;
+    this.data = data;
+  }
+}