Update index.js

index.js

@@ -18,7 +18,8 @@ const CONFIG = {
     BASE_URL: process.env.DENO_URL || "https://partyrock.aws/stream/getCompletion",//如果需要多号循环，需要设置你自己的denourl
     API_KEY: process.env.API_KEY || "sk-123456",//自定义你自己的认证密钥，记得修改
     RedisUrl: process.env.RedisUrl,
-    RedisToken: process.env.RedisToken
+    RedisToken: process.env.RedisToken,
+    CAPSOLVER_KEY: process.env.CAPSOLVER_KEY // 完全从环境变量读取
   },
   SERVER: {
     PORT: process.env.PORT || 3000,
@@ -149,18 +150,50 @@ class Utils {
       return type[Math.floor(Math.random() * type.length)]
     }
   }
+  static async solveAwsWaf(images, questionType = 'aws:grid') {
+    try {
+      console.log(`开始Capsolver验证 (类型: ${questionType}), 图片数量: ${images.length}`);
+      
+      const response = await fetch('https://api.capsolver.com/createTask', {
+        method: 'POST',
+        headers: {
+          'Content-Type': 'application/json',
+        },
+        body: JSON.stringify({
+          clientKey: CONFIG.API.CAPSOLVER_KEY,
+          task: {
+            type: 'AwsWafClassification',
+            websiteURL: Tokens[currentIndex].refreshUrl,
+            images: images,
+            question: questionType
+          }
+        })
+      });
+
+      const data = await response.json();
+      if (data.errorId === 0) {
+        if (data.status === 'ready') {
+          console.log(`✅ Capsolver验证成功 | 任务ID: ${data.taskId} | 使用图片: ${images.length}张`);
+          return data.solution;
+        }
+        console.log(`⏳ Capsolver处理中 | 任务ID: ${data.taskId} | 状态: ${data.status}`);
+      } else {
+        console.error(`❌ Capsolver错误 | 代码: ${data.errorCode} | 描述: ${data.errorDescription}`);
+      }
+      return null;
+    } catch (error) {
+      console.error('Capsolver请求失败:', error);
+      return null;
+    }
+  }
   static async extractWaf() {
     puppeteer.use(StealthPlugin())
     const 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'],
       executablePath: CONFIG.CHROME_PATH
     });
+
     try {
       const page = await browser.newPage();
       await page.setExtraHTTPHeaders({
@@ -169,32 +202,55 @@ class Utils {
       await page.setUserAgent(
         CONFIG.DEFAULT_HEADERS["User-Agent"]
       )
+
+      // 监听验证挑战
+      page.on('response', async (response) => {
+        if (response.url().includes('aws-waf-challenge')) {
+          const screenshots = [];
+          
+          // 截取验证图片
+          const challengeElements = await page.$$('.aws-waf-challenge');
+          for (const element of challengeElements) {
+            const screenshot = await element.screenshot({ encoding: 'base64' });
+            screenshots.push(`data:image/png;base64,${screenshot}`);
+          }
+
+          // 调用Capsolver
+          const solution = await this.solveAwsWaf(screenshots, 'aws:grid');
+          
+          if (solution?.objects) {
+            console.log(`🎯 验证答案位置: ${solution.objects.join(', ')}`);
+            // 自动点击验证答案
+            await page.evaluate((objects) => {
+              document.querySelectorAll('.grid-item').forEach((el, index) => {
+                if (objects.includes(index)) el.click();
+              });
+            }, solution.objects);
+          }
+        }
+      });
+
       await page.goto(Tokens[currentIndex].refreshUrl, {
         waitUntil: 'networkidle2',
         timeout: 30000
       });
-      await page.evaluate(() => {
-        // 随机滚动
-        window.scrollBy(0, Math.random() * 500)
-      })
-      await page.evaluate(() => {
-        return new Promise(resolve => setTimeout(resolve, 2000))
-      })
-      // 直接从页面 cookies 中提取 aws-waf-token
+
+      // 获取最终token
       const awsWafToken = (await page.cookies()).find(
         cookie => cookie.name.toLowerCase() === 'aws-waf-token'
       )?.value;
 
       if (awsWafToken) {
-        await browser.close();
-        return awsWafToken;
+        console.log(`🔑 成功获取aws-waf-token | 长度: ${awsWafToken.length}字符`);
       } else {
-        await browser.close();
-        return null;
+        console.error('❌ 获取aws-waf-token失败');
       }
 
+      await browser.close();
+      return awsWafToken;
+
     } catch (error) {
-      console.error('获取 aws-waf-token 出错:', error);
+      console.error('获取aws-waf-token出错:', error);
       await browser.close();
       return null;
     }