Upload 6 files

src/CookieManager.js

@@ -0,0 +1,423 @@
+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({}),
+      };
+      
+      // 添加代理配置（如果有）
+      if (this.proxyUrl) {
+        const { HttpsProxyAgent } = await import('https-proxy-agent');
+        fetchOptions.agent = new HttpsProxyAgent(this.proxyUrl);
+        logger.info(`使用代理: ${this.proxyUrl}`);
+      }
+      
+      // 发送请求
+      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,526 @@
+import axios from 'axios';
+
+/**
+ * 代理池类，用于管理和提供HTTP代理
+ */
+class ProxyPool {
+  /**
+   * 创建代理池实例
+   * @param {Object} options - 配置选项
+   * @param {number} options.targetCount - 目标代理数量，默认20
+   * @param {number} options.batchSize - 每次获取的代理数量，默认20
+   * @param {number} options.testTimeout - 测试代理超时时间(毫秒)，默认5000
+   * @param {number} options.requestTimeout - 请求目标网站超时时间(毫秒)，默认10000
+   * @param {string} options.targetUrl - 目标网站URL，默认'https://www.notion.so'
+   * @param {number} options.concurrentRequests - 并发请求数量，默认10
+   * @param {number} options.minThreshold - 可用代理数量低于此阈值时自动补充，默认5
+   * @param {number} options.checkInterval - 检查代理池状态的时间间隔(毫秒)，默认30000
+   * @param {string} options.proxyProtocol - 代理协议，默认'http'
+   * @param {number} options.maxRefillAttempts - 最大补充尝试次数，默认20
+   * @param {number} options.retryDelay - 重试延迟(毫秒)，默认1000
+   * @param {boolean} options.useCache - 是否使用缓存，默认true
+   * @param {number} options.cacheExpiry - 缓存过期时间(毫秒)，默认3600000 (1小时)
+   */
+  constructor(options = {}) {
+    // 配置参数
+    this.targetCount = options.targetCount || 20;
+    this.batchSize = options.batchSize || 20;
+    this.testTimeout = options.testTimeout || 5000;
+    this.requestTimeout = options.requestTimeout || 10000;
+    this.targetUrl = options.targetUrl || 'https://www.notion.so';
+    this.concurrentRequests = options.concurrentRequests || 10;
+    this.minThreshold = options.minThreshold || 5;
+    this.checkInterval = options.checkInterval || 30000; // 默认30秒检查一次
+    this.proxyProtocol = options.proxyProtocol || 'http';
+    this.maxRefillAttempts = options.maxRefillAttempts || 20; // 减少最大尝试次数
+    this.retryDelay = options.retryDelay || 1000; // 减少重试延迟
+    this.useCache = options.useCache !== undefined ? options.useCache : true;
+    this.cacheExpiry = options.cacheExpiry || 3600000; // 默认1小时
+    
+    // 内部状态
+    this.availableProxies = [];
+    this.currentIndex = 0;
+    this.isInitialized = false;
+    this.isRefilling = false;
+    this.checkTimer = null;
+    this.proxyCache = new Map(); // 缓存验证过的代理
+    
+    // 绑定方法
+    this.getProxy = this.getProxy.bind(this);
+    this.removeProxy = this.removeProxy.bind(this);
+    this.checkAndRefill = this.checkAndRefill.bind(this);
+  }
+  
+  /**
+   * 初始化代理池
+   * @returns {Promise<void>}
+   */
+  async initialize() {
+    if (this.isInitialized) return;
+    
+    console.log(`初始化代理池，目标数量: ${this.targetCount}`);
+    await this.refillProxies();
+    
+    // 设置定时检查
+    this.checkTimer = setInterval(this.checkAndRefill, this.checkInterval);
+    
+    this.isInitialized = true;
+    console.log(`代理池初始化完成，当前可用代理数量: ${this.availableProxies.length}`);
+  }
+  
+  /**
+   * 停止代理池服务
+   */
+  stop() {
+    if (this.checkTimer) {
+      clearInterval(this.checkTimer);
+      this.checkTimer = null;
+    }
+    console.log('代理池服务已停止');
+  }
+  
+  /**
+   * 检查并补充代理
+   */
+  async checkAndRefill() {
+    if (this.availableProxies.length <= this.minThreshold && !this.isRefilling) {
+      console.log(`可用代理数量(${this.availableProxies.length})低于阈值(${this.minThreshold})，开始补充代理`);
+      await this.refillProxies();
+    }
+  }
+  
+  /**
+   * 补充代理到目标数量
+   * @returns {Promise<void>}
+   */
+  async refillProxies() {
+    if (this.isRefilling) return;
+    
+    this.isRefilling = true;
+    console.log(`开始补充代理，当前数量: ${this.availableProxies.length}，目标数量: ${this.targetCount}`);
+    
+    let attempts = 0;
+    
+    try {
+      // 计算需要补充的代理数量
+      const neededProxies = this.targetCount - this.availableProxies.length;
+      
+      // 优先检查缓存中的代理
+      if (this.useCache && this.proxyCache.size > 0) {
+        await this.tryUsingCachedProxies(neededProxies);
+      }
+      
+      // 如果缓存中的代理不足，继续获取新代理
+      while (this.availableProxies.length < this.targetCount && attempts < this.maxRefillAttempts) {
+        attempts++;
+        
+        console.log(`补充尝试 #${attempts}，当前可用代理: ${this.availableProxies.length}/${this.targetCount}`);
+        
+        // 计算本次需要获取的批次大小
+        const remainingNeeded = this.targetCount - this.availableProxies.length;
+        const batchSizeNeeded = Math.max(this.batchSize, remainingNeeded * 2); // 获取更多代理以提高成功率
+        
+        // 获取代理
+        const proxies = await this.getProxiesFromProvider(batchSizeNeeded);
+        
+        if (proxies.length === 0) {
+          console.log(`没有获取到代理，等待${this.retryDelay/1000}秒后重试...`);
+          await new Promise(resolve => setTimeout(resolve, this.retryDelay));
+          continue;
+        }
+        
+        // 过滤掉已有的代理
+        const newProxies = this.filterExistingProxies(proxies);
+        
+        if (newProxies.length === 0) {
+          console.log('所有获取的代理都已存在，继续获取新代理...');
+          continue;
+        }
+        
+        // 测试代理
+        const results = await this.testProxiesConcurrently(newProxies);
+        
+        // 添加可用代理
+        this.addValidProxies(results);
+        
+        // 如果已经获取到足够的代理，提前结束
+        if (this.availableProxies.length >= this.targetCount) {
+          break;
+        }
+        
+        // 如果还没补充到足够的代理，等待一段时间再继续
+        if (this.availableProxies.length < this.targetCount) {
+          await new Promise(resolve => setTimeout(resolve, this.retryDelay));
+        }
+      }
+    } catch (error) {
+      console.error('补充代理过程中出错:', error);
+    } finally {
+      this.isRefilling = false;
+      
+      if (this.availableProxies.length >= this.targetCount) {
+        console.log(`代理补充完成，当前可用代理: ${this.availableProxies.length}/${this.targetCount}`);
+      } else {
+        console.log(`已达到最大尝试次数 ${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) {
+      console.log(`从缓存中找到 ${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 [ip, port] = proxy.split(':');
+      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 [ip, port] = proxy.split(':');
+        
+        // 检查是否已存在
+        if (!this.availableProxies.some(p => p.ip === ip && p.port === port)) {
+          const proxyObj = {
+            ip,
+            port,
+            protocol: this.proxyProtocol,
+            full: `${this.proxyProtocol}://${proxy}`,
+            addedAt: new Date().toISOString()
+          };
+          
+          this.availableProxies.push(proxyObj);
+          
+          // 添加到缓存
+          if (this.useCache) {
+            this.proxyCache.set(proxy, { 
+              valid: true, 
+              timestamp: Date.now() 
+            });
+          }
+          
+          console.log(`成功添加代理: ${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;
+      const url = `https://proxy.scdn.io/api/get_proxy.php?protocol=${this.proxyProtocol}&count=${requestCount}`;
+      console.log(`正在获取代理，URL: ${url}`);
+      
+      const response = await axios.get(url, { 
+        timeout: 10000,
+        validateStatus: status => true
+      });
+      
+      if (response.data && response.data.code === 200) {
+        console.log(`成功获取 ${response.data.data.count} 个代理`);
+        return response.data.data.proxies;
+      } else {
+        console.error('获取代理失败:', response.data ? response.data.message : '未知错误');
+        return [];
+      }
+    } catch (error) {
+      console.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 * 2, 20);
+    
+    // 分批处理代理
+    for (let i = 0; i < proxies.length; i += concurrentRequests) {
+      const batch = proxies.slice(i, i + concurrentRequests);
+      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 proxyConfig = {
+        host: proxyUrl.split(':')[0],
+        port: parseInt(proxyUrl.split(':')[1]),
+        protocol: this.proxyProtocol
+      };
+      
+      // 发送请求到目标网站
+      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: this.requestTimeout,
+        validateStatus: status => true,
+        maxRedirects: 10,
+        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) {
+        console.log(`代理 ${proxyUrl} 请求目标网站成功，状态码: ${response.status}`);
+      } else {
+        console.log(`代理 ${proxyUrl} 请求目标网站失败，状态码: ${response.status}`);
+      }
+      
+      return isValid;
+    } catch (error) {
+      console.log(`代理 ${proxyUrl} 请求出错: ${error.message}`);
+      return false;
+    }
+  }
+  
+  /**
+   * 获取一个可用代理
+   * @returns {Object|null} - 代理对象，如果没有可用代理则返回null
+   */
+  getProxy() {
+    if (this.availableProxies.length === 0) {
+      console.log('没有可用代理');
+      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) {
+      console.log(`已移除代理 ${ip}:${port}，当前可用代理: ${this.availableProxies.length}`);
+    } else {
+      console.log(`未找到要移除的代理 ${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) {
+      console.log(`清理了 ${cleanupCount} 个过期的缓存代理`);
+    }
+  }
+}
+
+// 使用示例
+async function example() {
+  // 创建代理池实例
+  const proxyPool = new ProxyPool({
+    targetCount: 10,           // 目标保持10个代理
+    minThreshold: 3,           // 当可用代理少于3个时，自动补充
+    checkInterval: 60000,      // 每60秒检查一次
+    targetUrl: 'https://www.notion.so',
+    concurrentRequests: 15,    // 增加并发请求数
+    useCache: true,            // 启用缓存
+    maxRefillAttempts: 15,     // 减少最大尝试次数
+    retryDelay: 1000           // 减少重试延迟
+  });
+  
+  // 初始化代理池
+  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();

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,269 @@
+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 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();
+}
+
+// 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" }
+    ]
+  };
+  
+  res.json(modelList);
+});
+
+// 聊天完成端点
+app.post('/v1/chat/completions', authenticate, async (req, res) => {
+  try {
+    // 检查是否成功初始化
+    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);
+      stream.pipe(res);
+      
+      // 处理客户端断开连接
+      req.on('close', () => {
+        stream.end();
+      });
+    } else {
+      // 非流式响应
+      // 创建一个内部流来收集完整响应
+      logger.info(`开始非流式响应`);
+      const chunks = [];
+      const stream = await streamNotionResponse(notionRequestBody);
+      
+      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);
+        });
+      });
+    }
+  } 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()
+  });
+});
+
+// Cookie状态查询端点
+app.get('/cookies/status', authenticate, (req, res) => {
+  res.json({
+    total_cookies: cookieManager.getValidCount(),
+    cookies: cookieManager.getStatus()
+  });
+});
+
+// 启动服务器
+const PORT = process.env.PORT || 7860;
+
+// 初始化并启动服务器
+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/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;
+  }
+}