Update index.js

index.js

@@ -98,17 +98,29 @@ class CapsolverClient {
   }
 
   async createTask(websiteURL, websiteKey, extraParams = {}) {
+    // 首先获取验证码图片
+    const captchaResponse = await fetch(extraParams.captchaUrl, {
+      headers: {
+        'Referer': 'https://partyrock.aws/',
+        'Origin': 'https://partyrock.aws',
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36'
+      }
+    });
+
+    if (!captchaResponse.ok) {
+      throw new Error('获取验证码失败');
+    }
+
+    const captchaData = await captchaResponse.json();
+    console.log('验证码数据:', captchaData);
+
     const requestBody = JSON.stringify({
       clientKey: this.apiKey,
       task: {
-        type: "AwsWafToken",
+        type: "AwsWafClassification",
         websiteURL: websiteURL,
-        websiteKey: websiteKey,
-        metadata: {
-          gokuProps: extraParams.gokuProps,
-          captchaUrl: extraParams.captchaUrl,
-          challengeUrl: extraParams.challengeUrl
-        }
+        images: captchaData.images || [],
+        question: captchaData.question || "aws:grid:bed" // 默认值，应该从验证码响应中获取
       }
     });
 
@@ -130,40 +142,29 @@ class CapsolverClient {
         throw new Error(`Capsolver API 请求失败: ${response.status}`);
       }
 
-      return await response.json();
-    } catch (error) {
-      console.error('调用 Capsolver API 时出错:', error);
-      throw error;
-    }
-  }
+      const result = await response.json();
+      console.log('Capsolver API 响应:', result);
 
-  async getTaskResult(taskId) {
-    try {
-      const response = await fetch(`${CAPSOLVER_CONFIG.API_URL}/getTaskResult`, {
-        method: 'POST',
-        headers: {
-          'Content-Type': 'application/json',
-          'Accept': 'application/json'
-        },
-        body: JSON.stringify({
-          clientKey: this.apiKey,
-          taskId: taskId
-        })
-      });
+      // 如果成功识别，提交验证结果
+      if (result.status === 'ready') {
+        const verifyResponse = await fetch(extraParams.captchaUrl + '/verify', {
+          method: 'POST',
+          headers: {
+            'Content-Type': 'application/json',
+            'Referer': 'https://partyrock.aws/',
+            'Origin': 'https://partyrock.aws'
+          },
+          body: JSON.stringify({
+            solution: result.solution
+          })
+        });
 
-      if (!response.ok) {
-        console.error('获取任务结果失败:', response.status);
-        const errorText = await response.text();
-        console.error('错误响应内容:', errorText);
-        throw new Error(`获取任务结果失败: ${response.status}`);
+        return await verifyResponse.json();
       }
 
-      const result = await response.json();
-      console.log('获取任务结果响应:', result);
       return result;
-
     } catch (error) {
-      console.error('获取任务结果时出错:', error);
+      console.error('调用 Capsolver API 时出错:', error);
       throw error;
     }
   }