Update index.js

index.js

@@ -107,15 +107,18 @@ class TokenManager {
     if (isWaf) {
       var wafToken = await Utils.extractWaf();
       if (wafToken) {
+        // 更新当前token的aws_waf_token
         Tokens[currentIndex].aws_waf_token = wafToken;
+        // 立即更新缓存headers
         await this.updateCacheTokens();
-        this.updateRedisTokens();
+        // 更新Redis存储
+        await this.updateRedisTokens();
+        console.log("✅ 成功更新aws-waf-token到缓存");
         currentIndex = (currentIndex + 1) % Tokens.length;
-        console.log("成功提取 aws-waf-token");
       } else {
+        console.log("❌ 提取aws-waf-token失败，切换账号");
         currentIndex = (currentIndex + 1) % Tokens.length;
         await this.updateCacheTokens();
-        console.log("提取aws-waf-token失败");
       }
     } else {
       const newCsrfToken = response.headers.get('anti-csrftoken-a2z');
@@ -235,6 +238,10 @@ class Utils {
         timeout: 30000
       });
 
+      // 新增：等待验证完成
+      await page.waitForTimeout(1000); // 等待1秒确保验证完成
+      await page.waitForSelector('.aws-waf-challenge', { hidden: true, timeout: 15000 });
+
       // 获取最终token
       const awsWafToken = (await page.cookies()).find(
         cookie => cookie.name.toLowerCase() === 'aws-waf-token'