Update index.js

index.js

@@ -85,6 +85,56 @@ var RedisClient = class {
     }
   }
 };
+
+// 添加 Capsolver 相关配置
+const CAPSOLVER_CONFIG = {
+  API_KEY: process.env.CAPSOLVER_KEY || "YOUR_CAPSOLVER_KEY",
+  API_URL: "https://api.capsolver.com"
+};
+
+class CapsolverClient {
+  constructor(apiKey) {
+    this.apiKey = apiKey;
+  }
+
+  async createTask(websiteURL, websiteKey) {
+    const response = await fetch(`${CAPSOLVER_CONFIG.API_URL}/createTask`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json'
+      },
+      body: JSON.stringify({
+        clientKey: this.apiKey,
+        task: {
+          type: "AwsWafCaptcha",
+          websiteURL: websiteURL,
+          websiteKey: websiteKey,
+          proxy: {
+            // 如果需要代理可以在这里配置
+          }
+        }
+      })
+    });
+
+    return await response.json();
+  }
+
+  async getTaskResult(taskId) {
+    const response = await fetch(`${CAPSOLVER_CONFIG.API_URL}/getTaskResult`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json'
+      },
+      body: JSON.stringify({
+        clientKey: this.apiKey,
+        taskId: taskId
+      })
+    });
+
+    return await response.json();
+  }
+}
+
 class TokenManager {
   async updateRedisTokens() {
     await redisClient.set(`tokens_${currentIndex}`, JSON.stringify(Tokens[currentIndex]));
@@ -102,20 +152,78 @@ class TokenManager {
     CONFIG.DEFAULT_HEADERS["request-id"] = `request-id-${Utils.uuidv4()}`;
   }
 
+  async solveCaptcha(url) {
+    const capsolverClient = new CapsolverClient(CAPSOLVER_CONFIG.API_KEY);
+    
+    try {
+      // 创建验证码任务
+      const createTaskResponse = await capsolverClient.createTask(
+        "partyrock.aws",
+        // 从URL中提取websiteKey
+        url.split('/')[2].split('.')[0]
+      );
+
+      if (createTaskResponse.errorId > 0) {
+        throw new Error(`创建验证码任务失败: ${createTaskResponse.errorDescription}`);
+      }
+
+      // 轮询获取结果
+      let taskResult;
+      for (let i = 0; i < 30; i++) {
+        await new Promise(resolve => setTimeout(resolve, 2000)); // 等待2秒
+        taskResult = await capsolverClient.getTaskResult(createTaskResponse.taskId);
+        
+        if (taskResult.status === 'ready') {
+          break;
+        }
+      }
+
+      if (!taskResult || taskResult.status !== 'ready') {
+        throw new Error('验证码解决超时');
+      }
+
+      return taskResult.solution;
+      
+    } catch (error) {
+      console.error('验证码解决失败:', error);
+      throw error;
+    }
+  }
+
   async updateTokens(response, isWaf = false) {
     if (isWaf) {
-      var wafToken = await Utils.extractWaf();
-      if (wafToken) {
-        Tokens[currentIndex].aws_waf_token = wafToken;
-        await this.updateCacheTokens();
-        this.updateRedisTokens();
-        currentIndex = (currentIndex + 1) % Tokens.length;
-        console.log("成功提取 aws-waf-token");
-      } else {
-        currentIndex = (currentIndex + 1) % Tokens.length;
-        await this.updateCacheTokens();
-        console.log("提取aws-waf-token失败");
+      try {
+        // 获取验证码URL
+        const captchaUrl = response.headers.get('x-amz-waf-challenge-url');
+        if (captchaUrl) {
+          const solution = await this.solveCaptcha(captchaUrl);
+          
+          // 使用解决方案更新token
+          const verifyResponse = await fetch(captchaUrl, {
+            method: 'POST',
+            headers: {
+              'Content-Type': 'application/json',
+              ...CONFIG.DEFAULT_HEADERS
+            },
+            body: JSON.stringify(solution)
+          });
+
+          if (verifyResponse.ok) {
+            const wafToken = await Utils.extractWaf();
+            if (wafToken) {
+              Tokens[currentIndex].aws_waf_token = wafToken;
+              await this.updateCacheTokens();
+              this.updateRedisTokens();
+              console.log("成功更新 aws-waf-token");
+            }
+          }
+        }
+      } catch (error) {
+        console.error('处理验证码失败:', error);
       }
+      
+      currentIndex = (currentIndex + 1) % Tokens.length;
+      
     } else {
       const newCsrfToken = response.headers.get('anti-csrftoken-a2z');
       const cookies = response.headers.get('set-cookie');
@@ -134,7 +242,6 @@ class TokenManager {
   }
 }
 
-
 class Utils {
   static async getRandomUserAgent() {
     try {