Update worker.js

worker.js

@@ -31,7 +31,8 @@ async function initBrowser() {
         console.log('🚀 启动浏览器...');
         browser = await puppeteer.launch({
             headless: true,
-            args: ['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage', '--disable-gpu']
+            args: ['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage', '--disable-gpu'],
+            protocolTimeout: 60000  // 增加协议超时时间
         });
         console.log('✅ 浏览器已启动');
     }
@@ -40,17 +41,43 @@ async function initBrowser() {
 
 async function performAjaxRequest(page, prompt, model, sessionId = null) {
     console.log('🚀 尝试直接AJAX请求到API...');
+    console.log('📋 调试信息:');
+    console.log(`  - prompt: "${prompt.substring(0, 100)}${prompt.length > 100 ? '...' : ''}"`);
+    console.log(`  - model: ${model}`);
+    console.log(`  - sessionId: ${sessionId ? `"${sessionId.substring(0, 20)}..."` : 'null'}`);
     
     const response = await page.evaluate(async (promptText, modelText, sessId) => {
+        console.log('🔍 浏览器端调试开始');
+        console.log('  promptText:', promptText);
+        console.log('  modelText:', modelText);
+        console.log('  sessId:', sessId);
+        
         try {
             // 构造请求数据
             const formData = new FormData();
             formData.append('text', promptText);
             formData.append('model', modelText || 'gemini-2.5-flash');
+            
             if (sessId) {
                 formData.append('session_id', sessId);
+                console.log('✅ 添加了session_id:', sessId);
+            } else {
+                console.log('⚠️ 没有session_id，将发送无session请求');
+            }
+            
+            // 转换为URLSearchParams以确保正确的编码
+            const urlParams = new URLSearchParams();
+            urlParams.append('text', promptText);
+            urlParams.append('model', modelText || 'gemini-2.5-flash');
+            if (sessId) {
+                urlParams.append('session_id', sessId);
             }
             
+            console.log('📤 最终发送的URL参数:');
+            console.log('  - text:', promptText);
+            console.log('  - model:', modelText || 'gemini-2.5-flash');
+            console.log('  - session_id:', sessId || 'undefined');
+            
             // 构造标准headers
             const headers = {
                 'Accept': '*/*',
@@ -67,30 +94,48 @@ async function performAjaxRequest(page, prompt, model, sessionId = null) {
                 'Sec-Fetch-Site': 'same-site'
             };
             
+            console.log('🌐 发送fetch请求到 https://data.toolbaz.com/writing.php');
+            
             // 发送请求到 writing.php
             const response = await fetch('https://data.toolbaz.com/writing.php', {
                 method: 'POST',
                 headers: headers,
-                body: new URLSearchParams(formData).toString()
+                body: urlParams.toString()
             });
             
-            console.log('AJAX响应状态:', response.status);
+            console.log('📡 AJAX响应状态:', response.status);
+            console.log('📡 AJAX响应类型:', response.type);
+            console.log('📡 AJAX响应URL:', response.url);
+            
+            // 检查响应头
+            const responseHeaders = {};
+            for (const [key, value] of response.headers.entries()) {
+                responseHeaders[key] = value;
+            }
+            console.log('📡 AJAX响应头:', responseHeaders);
             
             if (!response.ok) {
-                throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+                const errorText = await response.text();
+                console.error('❌ HTTP错误响应内容:', errorText);
+                throw new Error(`HTTP ${response.status}: ${response.statusText} - ${errorText}`);
             }
             
             const text = await response.text();
-            console.log('AJAX响应长度:', text.length);
+            console.log('📡 AJAX响应长度:', text.length);
+            console.log('📡 AJAX响应前200字符:', text.substring(0, 200));
             
             return { success: true, data: text };
             
         } catch (error) {
-            console.error('AJAX请求失败:', error);
-            return { success: false, error: error.message };
+            console.error('❌ AJAX请求失败:', error);
+            console.error('❌ 错误堆栈:', error.stack);
+            return { success: false, error: error.message, stack: error.stack };
         }
     }, prompt, model, sessionId);
     
+    console.log('🏁 浏览器端调试结束');
+    console.log('📋 Node.js收到的响应:', response);
+    
     return response;
 }
 
@@ -142,74 +187,195 @@ async function chatWithBrowser(prompt, model) {
             console.log(`🧹 已过滤前缀: ${prompt.substring(0, 50)}... -> ${cleanedPrompt.substring(0, 50)}...`);
         }
         
-        // 首先访问页面获取必要的会话信息
+        // 访问页面获取必要的会话信息，但不设置请求拦截以避免超时
         console.log('🌐 访问页面获取会话信息...');
         await page.goto('https://toolbaz.com/writer/ai-writer', { waitUntil: 'networkidle2', timeout: 30000 });
         
-        // 设置请求拦截，移除前缀文本
-        await page.setRequestInterception(true);
-        page.on('request', request => {
-            if (request.url().includes('writing.php')) {
-                const postData = request.postData();
-                if (postData) {
-                    try {
-                        // 解析POST数据
-                        const params = new URLSearchParams(postData);
-                        const text = params.get('text');
-                        
-                        if (text && text.includes('Generate an original and engaging piece of writing on the following topic')) {
-                            // 移除前缀，只保留实际内容
-                            const cleanText = text.replace(/^Generate an original and engaging piece of writing on the following topic\s*:\s*/, '');
-                            params.set('text', cleanText);
-                            
-                            console.log('🔧 拦截请求并移除前缀');
-                            console.log(`📤 原始文本: ${text.substring(0, 100)}...`);
-                            console.log(`📤 清理后: ${cleanText.substring(0, 100)}...`);
-                            
-                            // 继续发送修改后的请求
-                            request.continue({
-                                method: 'POST',
-                                postData: params.toString(),
-                                headers: request.headers()
-                            });
-                            return;
-                        }
-                    } catch (error) {
-                        console.log('⚠️ 请求拦截处理失败:', error.message);
+        // 等待页面完全加载
+        await page.waitForSelector('#input', { timeout: 10000 });
+        
+        console.log('🔍 开始搜索session ID...');
+        
+        // 详细搜索session ID
+        const sessionInfo = await page.evaluate(() => {
+            console.log('🔍 浏览器端session ID搜索开始');
+            
+            const results = {
+                found: [],
+                allInputs: [],
+                allHiddens: [],
+                sessionStorageData: {},
+                cookies: document.cookie,
+                localStorage: {},
+                finalSessionId: null
+            };
+            
+            // 搜索所有包含session的元素
+            const allElements = document.querySelectorAll('*');
+            const sessionElements = [];
+            for (const elem of allElements) {
+                if (elem.id && elem.id.toLowerCase().includes('session')) {
+                    sessionElements.push({
+                        tag: elem.tagName,
+                        id: elem.id,
+                        name: elem.name,
+                        value: elem.value,
+                        type: elem.type
+                    });
+                }
+                if (elem.name && elem.name.toLowerCase().includes('session')) {
+                    sessionElements.push({
+                        tag: elem.tagName,
+                        id: elem.id,
+                        name: elem.name,
+                        value: elem.value,
+                        type: elem.type
+                    });
+                }
+            }
+            results.found = sessionElements;
+            
+            // 检查所有input元素
+            const inputs = document.querySelectorAll('input');
+            inputs.forEach(input => {
+                results.allInputs.push({
+                    id: input.id,
+                    name: input.name,
+                    type: input.type,
+                    value: input.value
+                });
+            });
+            
+            // 检查所有hidden input
+            const hiddens = document.querySelectorAll('input[type="hidden"]');
+            hiddens.forEach(hidden => {
+                results.allHiddens.push({
+                    id: hidden.id,
+                    name: hidden.name,
+                    value: hidden.value
+                });
+            });
+            
+            // 检查sessionStorage
+            for (let i = 0; i < sessionStorage.length; i++) {
+                const key = sessionStorage.key(i);
+                results.sessionStorageData[key] = sessionStorage.getItem(key);
+            }
+            
+            // 检查localStorage
+            for (let i = 0; i < localStorage.length; i++) {
+                const key = localStorage.key(i);
+                results.localStorage[key] = localStorage.getItem(key);
+            }
+            
+            // 尝试各种方法获取session ID
+            let sessionId = null;
+            
+            // 方法1: 直接查找session_id输入框
+            const sessionInput = document.querySelector('input[name="session_id"]');
+            if (sessionInput && sessionInput.value) {
+                sessionId = sessionInput.value;
+                console.log('✅ 方法1成功: input[name="session_id"]', sessionId);
+            }
+            
+            // 方法2: 查找id为session_id的元素
+            if (!sessionId) {
+                const hiddenSession = document.querySelector('#session_id');
+                if (hiddenSession && hiddenSession.value) {
+                    sessionId = hiddenSession.value;
+                    console.log('✅ 方法2成功: #session_id', sessionId);
+                }
+            }
+            
+            // 方法3: 检查sessionStorage
+            if (!sessionId) {
+                const sessionStorageSess = sessionStorage.getItem('session_id');
+                if (sessionStorageSess) {
+                    sessionId = sessionStorageSess;
+                    console.log('✅ 方法3成功: sessionStorage session_id', sessionId);
+                }
+            }
+            
+            // 方法4: 检查全局变量
+            if (!sessionId && typeof window.session_id !== 'undefined') {
+                sessionId = window.session_id;
+                console.log('✅ 方法4成功: window.session_id', sessionId);
+            }
+            
+            // 方法5: 从cookie中提取（如果有的话）
+            if (!sessionId) {
+                const cookies = document.cookie.split(';');
+                for (const cookie of cookies) {
+                    const [name, value] = cookie.trim().split('=');
+                    if (name === 'session_id' || name === 'PHPSESSID') {
+                        sessionId = value;
+                        console.log('✅ 方法5成功: cookie', name, sessionId);
+                        break;
                     }
                 }
             }
             
-            // 其他请求正常处理
-            request.continue();
+            // 方法6: 尝试从URL参数获取
+            if (!sessionId) {
+                const urlParams = new URLSearchParams(window.location.search);
+                const urlSession = urlParams.get('session_id');
+                if (urlSession) {
+                    sessionId = urlSession;
+                    console.log('✅ 方法6成功: URL参数', sessionId);
+                }
+            }
+            
+            // 方法7: 检查是否有隐藏的包含session信息的meta标签
+            if (!sessionId) {
+                const metaSession = document.querySelector('meta[name="session_id"]');
+                if (metaSession && metaSession.content) {
+                    sessionId = metaSession.content;
+                    console.log('✅ 方法7成功: meta标签', sessionId);
+                }
+            }
+            
+            results.finalSessionId = sessionId;
+            results.searchMethods = [
+                'input[name="session_id"]',
+                '#session_id',
+                'sessionStorage["session_id"]',
+                'window.session_id',
+                'cookies',
+                'URL参数',
+                'meta标签'
+            ];
+            
+            console.log('🔍 最终session ID结果:', sessionId);
+            console.log('🔍 搜索完成，返回详细结果');
+            
+            return results;
         });
         
-        // 等待页面完全加载并获取session ID
-        await page.waitForSelector('#input', { timeout: 10000 });
-        
-        // 获取页面生成的session ID
-        const sessionId = await page.evaluate(() => {
-            // 检查各种可能的session存储位置
-            const sessionInput = document.querySelector('input[name="session_id"]');
-            const hiddenSession = document.querySelector('#session_id');
-            const sessionStorage = window.sessionStorage.getItem('session_id');
-            const windowSession = window.session_id;
-            
-            return sessionInput?.value || 
-                   hiddenSession?.value || 
-                   sessionStorage || 
-                   windowSession || 
-                   null;
-        });
+        console.log('📋 Session搜索结果:');
+        console.log('  - 找到的session相关元素:', sessionInfo.found.length);
+        console.log('  - 所有input元素:', sessionInfo.allInputs.length);
+        console.log('  - 隐藏input元素:', sessionInfo.allHiddens.length);
+        console.log('  - sessionStorage数据:', Object.keys(sessionInfo.sessionStorageData).length);
+        console.log('  - localStorage数据:', Object.keys(sessionInfo.localStorage).length);
+        console.log('  - cookies:', sessionInfo.cookies);
         
-        if (sessionId) {
-            console.log('✅ 找到session ID:', sessionId.substring(0, 20) + '...');
+        if (sessionInfo.finalSessionId) {
+            console.log('✅ 找到session ID:', `"${sessionInfo.finalSessionId}"`);
+            console.log('✅ session ID长度:', sessionInfo.finalSessionId.length);
+            
+            // 详细显示找到的session ID
+            if (sessionInfo.finalSessionId.length > 20) {
+                console.log('✅ session ID完整值:', sessionInfo.finalSessionId);
+            }
         } else {
-            console.log('⚠️ 未找到session ID，将使用无session请求');
+            console.log('❌ 未找到session ID');
+            console.log('📋 详细搜索结果:', JSON.stringify(sessionInfo, null, 2));
         }
         
-        // 尝试直接AJAX请求
-        console.log('📡 尝试直接AJAX请求（绕过前端JS）...');
+        const sessionId = sessionInfo.finalSessionId;
+        
+        // 尝试直接AJAX请求（现在使用详细的调试信息）
+        console.log('📡 尝试直接AJAX请求（纯AJAX模式）...');
         const ajaxResponse = await performAjaxRequest(page, cleanedPrompt, model, sessionId);
         
         if (ajaxResponse.success) {
@@ -219,14 +385,22 @@ async function chatWithBrowser(prompt, model) {
             // 验证响应是否有效
             if (cleanedResponse.length > 0 && 
                 !cleanedResponse.includes('Session ID is invalid') && 
-                !cleanedResponse.includes('Session expired')) {
+                !cleanedResponse.includes('Session expired') &&
+                !cleanedResponse.includes('Error:') &&
+                !cleanedResponse.includes('Invalid session')) {
                 console.log(`✅ AJAX响应成功: ${cleanedResponse.substring(0, 100)}...`);
+                console.log(`✅ 响应长度: ${cleanedResponse.length}`);
                 return cleanedResponse;
             } else {
-                console.log('⚠️ AJAX响应无效，回退到UI方法');
+                console.log('⚠️ AJAX响应内容无效或包含错误:');
+                console.log('  响应内容:', cleanedResponse.substring(0, 200));
+                console.log('🔄 回退到UI方法...');
             }
         } else {
             console.log('❌ AJAX请求失败:', ajaxResponse.error);
+            if (ajaxResponse.stack) {
+                console.log('❌ 错误堆栈:', ajaxResponse.stack);
+            }
             console.log('🔄 回退到UI方法...');
         }
         
@@ -248,6 +422,7 @@ async function chatWithBrowser(prompt, model) {
         
     } catch (error) {
         console.error(`❌ 聊天失败: ${error.message}`);
+        console.error(`❌ 错误堆栈:`, error.stack);
         throw error;
     } finally {
         await page.close();