import requests
import base64
import time
import mimetypes

# --- 配置 ---
# 替换成你的 Worker 地址
WORKER_URL = "https://your-worker-name.your-subdomain.workers.dev" 
# 替换成你在 Cloudflare 中设置的 AUTH_TOKEN
AUTH_TOKEN = "your_auth_token"  
# 替换成你在 Cloudflare 中设置的 API_KEY
API_KEY = "your_api_key"        
# 替换成你的本地验证码图片路径
IMAGE_PATH = "path/to/your/captcha.png" 

def solve_captcha(image_path):
    """
    调用API来识别验证码。
    """
    # 1. 读取图片并进行Base64编码
    try:
        with open(image_path, "rb") as image_file:
            image_base64 = base64.b64encode(image_file.read()).decode('utf-8')
        # 自动猜测文件的MIME类型
        mime_type = mimetypes.guess_type(image_path)[0]
        if not mime_type:
            raise ValueError("无法确定图片的MIME类型")
    except FileNotFoundError:
        print(f"错误: 文件未找到 at {image_path}")
        return None
    except Exception as e:
        print(f"读取文件时出错: {e}")
        return None

    # 2. 提交任务
    submit_url = f"{WORKER_URL}/submit"
    submit_headers = {
        "Authorization": f"Bearer {AUTH_TOKEN}",
        "Content-Type": "application/json"
    }
    submit_payload = {
        "image": image_base64,
        "mimeType": mime_type
    }

    try:
        print("正在提交任务...")
        submit_res = requests.post(submit_url, headers=submit_headers, json=submit_payload, timeout=15)
        submit_res.raise_for_status() # 如果请求失败 (非2xx状态码), 抛出异常
        
        task_id = submit_res.json().get("taskId")
        if not task_id:
            print("错误: 未能从响应中获取 taskId")
            print("响应内容:", submit_res.text)
            return None
        print(f"任务提交成功, Task ID: {task_id}")
        
    except requests.exceptions.RequestException as e:
        print(f"提交任务时出错: {e}")
        return None

    # 3. 轮询结果
    result_url = f"{WORKER_URL}/result?taskId={task_id}&apiKey={API_KEY}"
    max_polls = 30  # 最多轮询30次 (30 * 2秒 = 60秒超时)
    poll_interval = 2 # 每次轮询间隔2秒

    print("开始轮询结果...")
    for i in range(max_polls):
        try:
            result_res = requests.get(result_url, timeout=10)
            
            if result_res.status_code == 404:
                print("错误: 任务未找到或已过期。")
                return None
            result_res.raise_for_status()
            
            data = result_res.json()
            status = data.get("status")

            if status == "completed":
                solution = data.get("solution")
                print(f"识别成功! 结果: {solution}")
                return solution
            elif status == "error":
                message = data.get("message", "未知错误")
                print(f"识别失败: {message}")
                return None
            elif status == "pending":
                print(f"任务仍在处理中... ({i+1}/{max_polls})")
                time.sleep(poll_interval) # 等待后继续轮询
            else:
                print(f"收到未知的状态: {status}")
                return None

        except requests.exceptions.RequestException as e:
            print(f"查询结果时出错: {e}")
            return None
            
    print("轮询超时，未能获取结果。")
    return None

if __name__ == "__main__":
    captcha_solution = solve_captcha(IMAGE_PATH)
    if captcha_solution:
        print(f"\n[+] 最终识别结果是: {captcha_solution}")