def analyze_form(snapshot):
    """分析表單結(jié)構(gòu)"""
    form_info = {
        "textboxes": [],
        "checkboxes": [],
        "radios": [],
        "selects": [],
        "buttons": []
    }
    
    for element in snapshot["elements"]:
        element_type = element.get("type")
        
        if element_type == "textbox":
            form_info["textboxes"].append({
                "ref": element["ref"],
                "label": element.get("label", ""),
                "placeholder": element.get("placeholder", ""),
                "required": element.get("required", False)
            })
        elif element_type == "checkbox":
            form_info["checkboxes"].append(element)
        elif element_type == "radio":
            form_info["radios"].append(element)
        elif element_type == "combobox":
            form_info["selects"].append(element)
        elif element_type == "button":
            form_info["buttons"].append(element)
    
    return form_info

def map_form_fields(form_info, data):
    """將數(shù)據(jù)映射到表單字段"""
    mapping = {}
    
    for field in form_info["textboxes"]:
        label = field.get("label", "").lower()
        placeholder = field.get("placeholder", "").lower()
        
        # 智能匹配
        for key, value in data.items():
            key_lower = key.lower()
            if key_lower in label or key_lower in placeholder:
                mapping[field["ref"]] = value
                break
    
    return mapping

def auto_login(url, username, password):
    """自動登錄"""
    # 1. 打開登錄頁面
    browser(action="open", targetUrl=url)
    
    # 2. 等待頁面加載
    browser(action="act", kind="wait", timeMs=2000)
    
    # 3. 獲取頁面快照
    snapshot = browser(
        action="snapshot",
        interactive=True
    )
    
    # 4. 查找用戶名輸入框
    username_field = find_element_by_keywords(
        snapshot, 
        "textbox", 
        ["用戶名", "username", "賬號", "email"]
    )
    
    # 5. 輸入用戶名
    browser(
        action="act",
        kind="type",
        ref=username_field["ref"],
        text=username
    )
    
    # 6. 查找密碼輸入框
    password_field = find_element_by_keywords(
        snapshot,
        "textbox",
        ["密碼", "password", "pwd"]
    )
    
    # 7. 輸入密碼
    browser(
        action="act",
        kind="type",
        ref=password_field["ref"],
        text=password
    )
    
    # 8. 查找登錄按鈕
    login_btn = find_element_by_keywords(
        snapshot,
        "button",
        ["登錄", "login", "signin", "提交"]
    )
    
    # 9. 點擊登錄
    browser(
        action="act",
        kind="click",
        ref=login_btn["ref"]
    )
    
    # 10. 等待登錄完成
    browser(action="act", kind="wait", timeMs=3000)
    
    # 11. 驗證登錄結(jié)果
    new_snapshot = browser(action="snapshot")
    return check_login_success(new_snapshot)

def find_element_by_keywords(snapshot, element_type, keywords):
    """通過關(guān)鍵詞查找元素"""
    for element in snapshot["elements"]:
        if element.get("type") != element_type:
            continue
        
        text = element.get("text", "").lower()
        label = element.get("label", "").lower()
        placeholder = element.get("placeholder", "").lower()
        
        for keyword in keywords:
            if keyword.lower() in text or keyword.lower() in label or keyword.lower() in placeholder:
                return element
    
    return None

def safe_auto_login(url, username, password, max_retries=3):
    """帶錯誤處理的自動登錄"""
    for attempt in range(max_retries):
        try:
            result = auto_login(url, username, password)
            return result
        except Exception as e:
            print(f"登錄失敗 (嘗試 {attempt + 1}/{max_retries}): {e}")
            
            if attempt < max_retries - 1:
                # 刷新頁面重試
                browser(action="navigate", targetUrl=url)
                browser(action="act", kind="wait", timeMs=2000)
    
    raise Exception("登錄失敗，已達(dá)最大重試次數(shù)")

def auto_register(url, user_data):
    """自動注冊"""
    # 1. 打開注冊頁面
    browser(action="open", targetUrl=url)
    browser(action="act", kind="wait", timeMs=2000)
    
    # 2. 獲取表單快照
    snapshot = browser(action="snapshot", interactive=True)
    
    # 3. 分析表單結(jié)構(gòu)
    form_info = analyze_form(snapshot)
    
    # 4. 映射數(shù)據(jù)到字段
    field_mapping = map_form_fields(form_info, user_data)
    
    # 5. 填寫文本字段
    for ref, value in field_mapping.items():
        browser(
            action="act",
            kind="type",
            ref=ref,
            text=str(value)
        )
        # 短暫等待，模擬人工輸入
        browser(action="act", kind="wait", timeMs=300)
    
    # 6. 處理復(fù)選框（如同意條款）
    for checkbox in form_info["checkboxes"]:
        text = checkbox.get("text", "").lower()
        if "同意" in text or "條款" in text or "agree" in text.lower():
            browser(
                action="act",
                kind="click",
                ref=checkbox["ref"]
            )
    
    # 7. 處理下拉選擇框
    for select in form_info["selects"]:
        label = select.get("label", "").lower()
        if "城市" in label or "city" in label:
            browser(
                action="act",
                kind="select",
                ref=select["ref"],
                value=user_data.get("city", "北京")
            )
    
    # 8. 提交表單
    submit_btn = find_element_by_keywords(
        snapshot,
        "button",
        ["注冊", "register", "提交", "submit"]
    )
    
    if submit_btn:
        browser(
            action="act",
            kind="click",
            ref=submit_btn["ref"]
        )
    
    # 9. 等待響應(yīng)
    browser(action="act", kind="wait", timeMs=3000)
    
    # 10. 檢查結(jié)果
    result_snapshot = browser(action="snapshot")
    return check_register_result(result_snapshot)

def handle_captcha():
    """處理驗證碼"""
    # 獲取快照
    snapshot = browser(action="snapshot")
    
    # 查找驗證碼圖片
    captcha_img = find_element(snapshot, "image", "驗證碼")
    
    if captcha_img:
        # 截取驗證碼圖片
        captcha_screenshot = browser(
            action="screenshot",
            element=captcha_img["ref"]
        )
        
        # 方案1：OCR識別
        captcha_text = ocr_recognize(captcha_screenshot)
        
        # 方案2：請求人工輸入
        # captcha_text = request_human_input("請輸入驗證碼")
        
        # 填寫驗證碼
        captcha_input = find_element(snapshot, "textbox", "驗證碼")
        browser(
            action="act",
            kind="type",
            ref=captcha_input["ref"],
            text=captcha_text
        )

def multi_step_form(url, steps_data):
    """多步驟表單處理"""
    # 1. 打開頁面
    browser(action="open", targetUrl=url)
    browser(action="act", kind="wait", timeMs=2000)
    
    # 2. 逐步處理
    for step_index, step_data in enumerate(steps_data):
        print(f"處理步驟 {step_index + 1}/{len(steps_data)}")
        
        # 獲取當(dāng)前頁面快照
        snapshot = browser(action="snapshot", interactive=True)
        
        # 填寫當(dāng)前步驟的數(shù)據(jù)
        fill_current_step(snapshot, step_data)
        
        # 查找下一步按鈕
        next_btn = find_element_by_keywords(
            snapshot,
            "button",
            ["下一步", "next", "繼續(xù)", "continue"]
        )
        
        if next_btn:
            browser(
                action="act",
                kind="click",
                ref=next_btn["ref"]
            )
            browser(action="act", kind="wait", timeMs=2000)
        else:
            # 最后一步，查找提交按鈕
            submit_btn = find_element_by_keywords(
                snapshot,
                "button",
                ["提交", "submit", "完成", "finish"]
            )
            
            if submit_btn:
                browser(
                    action="act",
                    kind="click",
                    ref=submit_btn["ref"]
                )
    
    # 3. 等待最終結(jié)果
    browser(action="act", kind="wait", timeMs=3000)
    
    # 4. 驗證完成
    final_snapshot = browser(action="snapshot")
    return verify_completion(final_snapshot)

def fill_current_step(snapshot, step_data):
    """填寫當(dāng)前步驟的數(shù)據(jù)"""
    for field_name, value in step_data.items():
        # 查找對應(yīng)字段
        element = find_element_by_keywords(
            snapshot,
            "textbox",
            [field_name]
        )
        
        if element:
            browser(
                action="act",
                kind="type",
                ref=element["ref"],
                text=str(value)
            )

def verify_step_completion(expected_elements):
    """驗證步驟完成"""
    snapshot = browser(action="snapshot")
    
    for element_desc in expected_elements:
        found = find_element_by_keywords(
            snapshot,
            element_desc["type"],
            [element_desc["text"]]
        )
        
        if not found:
            return False
    
    return True

def smart_wait_for_element(element_type, keywords, timeout=10000):
    """智能等待元素出現(xiàn)"""
    start_time = time.time()
    
    while time.time() - start_time < timeout / 1000:
        snapshot = browser(action="snapshot", interactive=True)
        element = find_element_by_keywords(snapshot, element_type, keywords)
        
        if element:
            return element
        
        time.sleep(0.5)
    
    raise TimeoutError(f"等待元素超時: {element_type} - {keywords}")

def preprocess_form_data(data, form_info):
    """預(yù)處理表單數(shù)據(jù)"""
    processed = {}
    
    for field in form_info["textboxes"]:
        label = field.get("label", "").lower()
        placeholder = field.get("placeholder", "").lower()
        
        # 根據(jù)字段類型處理數(shù)據(jù)
        if "郵箱" in label or "email" in placeholder:
            # 郵箱格式驗證
            email = data.get("email", "")
            if "@" in email:
                processed[field["ref"]] = email
        
        elif "電話" in label or "phone" in placeholder:
            # 電話格式化
            phone = data.get("phone", "")
            processed[field["ref"]] = phone.replace("-", "").replace(" ", "")
        
        elif "日期" in label or "date" in placeholder:
            # 日期格式化
            date = data.get("date", "")
            processed[field["ref"]] = format_date(date)
        
        else:
            # 默認(rèn)處理
            for key, value in data.items():
                if key.lower() in label or key.lower() in placeholder:
                    processed[field["ref"]] = str(value)
                    break
    
    return processed

def form_fill_with_recovery(url, data, max_retries=3):
    """帶錯誤恢復(fù)的表單填寫"""
    saved_state = {}
    
    for attempt in range(max_retries):
        try:
            # 恢復(fù)之前的狀態(tài)
            if saved_state:
                restore_state(saved_state)
            
            # 填寫表單
            result = fill_form(url, data)
            return result
            
        except Exception as e:
            print(f"填寫失敗: {e}")
            
            # 保存當(dāng)前狀態(tài)
            saved_state = save_current_state()
            
            if attempt < max_retries - 1:
                # 刷新頁面
                browser(action="navigate", targetUrl=url)
                browser(action="act", kind="wait", timeMs=2000)
    
    raise Exception("表單填寫失敗，已達(dá)最大重試次數(shù)")

# 批量操作
def batch_fill_fields(field_values):
    """批量填寫字段"""
    snapshot = browser(action="snapshot", interactive=True)
    for ref, value in field_values.items():
        browser(
            action="act",
            kind="type",
            ref=ref,
            text=value
        )
        # 減少等待時間
        time.sleep(0.1)  # 最小等待