如何使用Puppeteer [JavaScript] 通过CapSolver扩展解决AWS验证码

const puppeteer = require('puppeteer');
const path = require('path');

// 定义您提取的CapSolver扩展文件夹的路径
const extensionPath = path.join(__dirname, 'capsolver-extension');

async function launchBrowserWithExtension() {
    const browser = await puppeteer.launch({
        headless: false, // 必须为非无头模式以确保扩展可靠运行
        args: [
            `--disable-extensions-except=${extensionPath}`,
            `--load-extension=${extensionPath}`,
            '--no-sandbox', // 某些环境推荐使用
        ],
    });
    return browser;
}

// 示例用法:
// const browser = await launchBrowserWithExtension();
// const page = await browser.newPage();
// await page.goto('https://your-aws-waf-protected-site.com');

const puppeteer = require('puppeteer');
const path = require('path');

// ...（来自步骤2的launchBrowserWithExtension函数）...

async function runAwsWafAutomation() {
    const browser = await launchBrowserWithExtension();
    const page = await browser.newPage();

    // 1. 导航到AWS WAF保护的URL
    const targetUrl = 'https://efw47fpad9.execute-api.us-east-1.amazonaws.com/latest'; // 示例URL
    console.log(`正在导航到 ${targetUrl}...`);
    await page.goto(targetUrl, { waitUntil: 'domcontentloaded' });

    // 2. 等待验证码被解决
    // CapSolver扩展将自动检测并解决AWS WAF验证码。
    
    // 最佳实践：等待验证码元素消失或目标内容出现。
    try {
        // 假设验证码有一个特定的选择器，例如 '#aws-waf-captcha-container'
        // 我们等待该元素消失（即验证码被解决，页面继续）
        console.log("等待CapSolver扩展解决AWS WAF验证码...");
        await page.waitForSelector('#aws-waf-captcha-container', { hidden: true, timeout: 60000 });
        console.log("验证码已解决！继续AI搜索自动化。");
        
        // 3. 继续您的AI搜索自动化逻辑
        // 示例：从现在可访问的页面中提取数据
        const pageTitle = await page.title();
        console.log(`页面标题（验证码后）: ${pageTitle}`);
        
    } catch (error) {
        console.error("验证码解决超时或失败:", error.message);
    }

    await browser.close();
}

// runAwsWafAutomation();

const fetch = require('node-fetch'); // 旧版Node.js使用'node-fetch'，新版使用原生fetch

const CAPSOLVER_API_KEY = 'YOUR_CAPSOLVER_API_KEY';
const API_URL = 'https://api.capsolver.com';

/**
 * 使用CapSolver API解决AWS WAF验证码。
 * @param {string} websiteURL 验证码页面的URL。
 * @param {object} awsParams 从被阻止的页面中提取的参数（awsKey、awsIv、awsContext等）。
 * @returns {Promise<string>} aws-waf-token cookie值。
 */
async function solveAwsWafCaptcha(websiteURL, awsParams) {
    // 1. 创建任务
    const createTaskPayload = {
        clientKey: CAPSOLVER_API_KEY,
        task: {
            type: "AntiAwsWafTaskProxyLess", // 如果需要指定代理，请使用AntiAwsWafTask
            websiteURL: websiteURL,
            ...awsParams // 传递提取的参数
        }
    };

    let response = await fetch(`${API_URL}/createTask`, {
        method: 'POST',
        headers: { 'Content-Type': 'application/json' },
        body: JSON.stringify(createTaskPayload)
    });
    let result = await response.json();

    if (result.errorId !== 0) {
        throw new Error(`CapSolver API错误（createTask）: ${result.errorDescription}`);
    }

    const taskId = result.taskId;
    console.log(`任务创建成功，ID: ${taskId}。正在等待结果...`);

    // 2. 轮询结果
    const getResultPayload = {
        clientKey: CAPSOLVER_API_KEY,
        taskId: taskId
    };

    let solution = null;
    for (let i = 0; i < 15; i++) { // 最多轮询15次（最大30秒）
        await new Promise(resolve => setTimeout(resolve, 2000)); // 等待2秒

        response = await fetch(`${API_URL}/getTaskResult`, {
            method: 'POST',
            headers: { 'Content-Type': 'application/json' },
            body: JSON.stringify(getResultPayload)
        });
        result = await response.json();

        if (result.errorId !== 0) {
            throw new Error(`CapSolver API错误（getTaskResult）: ${result.errorDescription}`);
        }

        if (result.status === 'ready') {
            solution = result.solution;
            break;
        }
        console.log(`状态: ${result.status}。重试中...`);
    }

    if (!solution || !solution.cookie) {
        throw new Error("CapSolver在超时内未能返回有效令牌。");
    }

    // solution.cookie包含aws-waf-token
    return solution.cookie;
}

// 在无头Puppeteer脚本中的示例用法:
/*
async function runHeadlessApiAutomation() {
    // 1. 使用Puppeteer导航并提取AWS WAF参数（例如，从脚本标签或网络响应中）
    // 此步骤需要高级的Puppeteer技能来拦截网络请求或解析HTML源代码。
    const websiteURL = 'https://efw47fpad9.execute-api.us-east-1.amazonaws.com/latest';
    const extractedAwsParams = {
        awsKey: "AQIDAHjcYu/GjX+QlghicBg......", 
        awsIv: "CgAAFDIlckAAAAid",
        awsContext: "7DhQfG5CmoY90ZdxdHCi8WtJ3z......",
        // ...其他可选参数
    };

    try {
        const awsWafToken = await solveAwsWafCaptcha(websiteURL, extractedAwsParams);
        console.log(`成功获取AWS WAF令牌: ${awsWafToken.substring(0, 30)}...`);

        // 2. 将令牌注入Puppeteer会话或在后续请求的头/cookie中使用
        // 示例：将令牌作为cookie注入下一次请求
        // await page.setCookie({
        //     name: 'aws-waf-token',
        //     value: awsWafToken,
        //     domain: new URL(websiteURL).hostname,
        //     path: '/',
        // });
        // await page.reload({ waitUntil: 'networkidle0' });

    } catch (error) {
        console.error("API自动化失败:", error.message);
    }
}
*/