from DrissionPage import ChromiumOptions, ChromiumPage
import time
import os
import signal
import random

# 在文件开头添加全局变量
_translator = None

def cleanup_chrome_processes(translator=None):
    """清理所有Chrome相关进程"""
    print("\n正在清理Chrome进程...")
    try:
        if os.name == 'nt':
            os.system('taskkill /F /IM chrome.exe /T 2>nul')
            os.system('taskkill /F /IM chromedriver.exe /T 2>nul')
        else:
            os.system('pkill -f chrome')
            os.system('pkill -f chromedriver')
    except Exception as e:
        if translator:
            print(f"{translator.get('register.cleanup_error', error=str(e))}")
        else:
            print(f"清理进程时出错: {e}")

def signal_handler(signum, frame):
    """处理Ctrl+C信号"""
    global _translator
    if _translator:
        print(f"{_translator.get('register.exit_signal')}")
    else:
        print("\n接收到退出信号，正在关闭...")
    cleanup_chrome_processes(_translator)
    os._exit(0)

def simulate_human_input(page, url, translator=None):
    """访问网址"""
    if translator:
        print(f"{translator.get('register.visiting_url')}: {url}")
    else:
        print("正在访问网址...")
    
    # 先访问空白页面
    page.get('about:blank')
    time.sleep(random.uniform(1.0, 2.0))
    
    # 访问目标页面
    page.get(url)
    time.sleep(random.uniform(2.0, 3.0))  # 等待页面加载

def fill_signup_form(page, first_name, last_name, email, translator=None):
    """填写注册表单"""
    try:
        if translator:
            print(f"{translator.get('register.filling_form')}")
        else:
            print("\n正在填写注册表单...")
        
        # 填写名字
        first_name_input = page.ele("@name=first_name")
        if first_name_input:
            first_name_input.input(first_name)
            time.sleep(random.uniform(0.5, 1.0))
        
        # 填写姓氏
        last_name_input = page.ele("@name=last_name")
        if last_name_input:
            last_name_input.input(last_name)
            time.sleep(random.uniform(0.5, 1.0))
        
        # 填写邮箱
        email_input = page.ele("@name=email")
        if email_input:
            email_input.input(email)
            time.sleep(random.uniform(0.5, 1.0))
        
        # 点击提交按钮
        submit_button = page.ele("@type=submit")
        if submit_button:
            submit_button.click()
            time.sleep(random.uniform(2.0, 3.0))
            
        if translator:
            print(f"{translator.get('register.form_success')}")
        else:
            print("表单填写完成")
        return True
        
    except Exception as e:
        if translator:
            print(f"{translator.get('register.form_error', error=str(e))}")
        else:
            print(f"填写表单时出错: {e}")
        return False

def setup_driver(translator=None):
    """设置浏览器驱动"""
    co = ChromiumOptions()
    
    # 使用无痕模式
    co.set_argument("--incognito")
    
    # 设置随机端口
    co.auto_port()
    
    # 使用有头模式
    co.headless(False)
    
    try:
        # 加载插件
        extension_path = os.path.join(os.getcwd(), "turnstilePatch")
        if os.path.exists(extension_path):
            co.set_argument("--allow-extensions-in-incognito")
            co.add_extension(extension_path)
    except Exception as e:
        if translator:
            print(f"{translator.get('register.extension_load_error', error=str(e))}")
        else:
            print(f"加载插件失败: {e}")
    
    if translator:
        print(f"{translator.get('register.starting_browser')}")
    else:
        print("正在启动浏览器...")
    page = ChromiumPage(co)
    
    return page

def handle_turnstile(page, translator=None):
    """处理 Turnstile 验证"""
    try:
        if translator:
            print(f"{translator.get('register.handling_turnstile')}")
        else:
            print("\n正在处理 Turnstile 验证...")
        
        max_retries = 2
        retry_count = 0

        while retry_count < max_retries:
            retry_count += 1
            if translator:
                print(f"{translator.get('register.retry_verification', attempt=retry_count)}")
            else:
                print(f"第 {retry_count} 次尝试验证...")

            try:
                # 尝试重置 turnstile
                page.run_js("try { turnstile.reset() } catch(e) { }")
                time.sleep(2)

                # 定位验证框元素
                challenge_check = (
                    page.ele("@id=cf-turnstile", timeout=2)
                    .child()
                    .shadow_root.ele("tag:iframe")
                    .ele("tag:body")
                    .sr("tag:input")
                )

                if challenge_check:
                    if translator:
                        print(f"{translator.get('register.detect_turnstile')}")
                    else:
                        print("检测到验证框...")
                    
                    # 随机延时后点击验证
                    time.sleep(random.uniform(1, 3))
                    challenge_check.click()
                    time.sleep(2)

                    # 检查验证结果
                    if check_verification_success(page, translator):
                        if translator:
                            print(f"{translator.get('register.verification_success')}")
                        else:
                            print("验证通过！")
                        return True

            except Exception as e:
                if translator:
                    print(f"{translator.get('register.verification_failed')}")
                else:
                    print(f"验证尝试失败: {e}")

            # 检查是否已经验证成功
            if check_verification_success(page, translator):
                if translator:
                    print(f"{translator.get('register.verification_success')}")
                else:
                    print("验证通过！")
                return True

            time.sleep(random.uniform(1, 2))

        if translator:
            print(f"{translator.get('register.verification_failed')}")
        else:
            print("超出最大重试次数")
        return False

    except Exception as e:
        if translator:
            print(f"{translator.get('register.verification_error', error=str(e))}")
        else:
            print(f"验证过程出错: {e}")
        return False

def check_verification_success(page, translator=None):
    """检查验证是否成功"""
    try:
        # 检查是否存在后续表单元素，这表示验证已通过
        if (page.ele("@name=password", timeout=0.5) or 
            page.ele("@name=email", timeout=0.5) or
            page.ele("@data-index=0", timeout=0.5) or
            page.ele("Account Settings", timeout=0.5)):
            return True
        
        # 检查是否出现错误消息
        error_messages = [
            'xpath://div[contains(text(), "Can\'t verify the user is human")]',
            'xpath://div[contains(text(), "Error: 600010")]',
            'xpath://div[contains(text(), "Please try again")]'
        ]
        
        for error_xpath in error_messages:
            if page.ele(error_xpath):
                return False
            
        return False
    except:
        return False

def generate_password(length=12):
    """生成随机密码"""
    chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789!@#$%^&*"
    return ''.join(random.choices(chars, k=length))

def fill_password(page, password, translator=None):
    """填写密码"""
    try:
        print("\n正在设置密码...")
        password_input = page.ele("@name=password")
        if password_input:
            password_input.input(password)
            time.sleep(random.uniform(0.5, 1.0))
            
            submit_button = page.ele("@type=submit")
            if submit_button:
                submit_button.click()
                time.sleep(random.uniform(2.0, 3.0))
                
            if translator:
                print(f"{translator.get('register.password_success')}")
            else:
                print(f"密码设置完成: {password}")
            return True
            
    except Exception as e:
        if translator:
            print(f"{translator.get('register.password_error', error=str(e))}")
        else:
            print(f"设置密码时出错: {e}")
        return False

def handle_verification_code(browser_tab, email_tab, controller, email, password, translator=None):
    """处理验证码"""
    try:
        if translator:
            print(f"\n{translator.get('register.waiting_for_verification_code')}")
        else:
            print("\n等待并获取验证码...")
        time.sleep(5)  # 等待验证码邮件

        # 刷新邮箱页面
        email_tab.refresh()
        
        # 获取验证码，设置超时
        verification_code = None
        max_attempts = 20
        retry_interval = 10
        start_time = time.time()
        timeout = 160

        if translator:
            print(f"\n{translator.get('register.start_getting_verification_code')}")
        else:
            print("开始获取验证码...")
        
        for attempt in range(max_attempts):
            # 检查是否超时
            if time.time() - start_time > timeout:
                if translator:
                    print(f"{translator.get('register.verification_timeout')}")
                else:
                    print("获取验证码超时...")
                break
                
            verification_code = controller.get_verification_code()
            if verification_code:
                if translator:
                    print(f"{translator.get('register.verification_success')}")
                else:
                    print(f"成功获取验证码: {verification_code}")
                break
                
            remaining_time = int(timeout - (time.time() - start_time))
            if translator:
                print(f"{translator.get('register.try_get_code', attempt=attempt + 1, time=remaining_time)}")
            else:
                print(f"第 {attempt + 1} 次尝试获取验证码，剩余时间: {remaining_time}秒...")
            
            # 刷新邮箱
            email_tab.refresh()
            time.sleep(retry_interval)
        
        if verification_code:
            # 在注册页面填写验证码
            for i, digit in enumerate(verification_code):
                browser_tab.ele(f"@data-index={i}").input(digit)
                time.sleep(random.uniform(0.1, 0.3))
            
            if translator:
                print(f"{translator.get('register.verification_success')}")
            else:
                print("验证码填写完成")
            time.sleep(3)
            
            # 处理最后一次 Turnstile 验证
            if handle_turnstile(browser_tab, translator):
                if translator:
                    print(f"{translator.get('register.verification_success')}")
                else:
                    print("最后一次验证通过！")
                time.sleep(2)
                
                # 直接访问设置页面
                if translator:
                    print(f"{translator.get('register.visiting_url')}: https://www.cursor.com/settings")
                else:
                    print("访问设置页面...")
                browser_tab.get("https://www.cursor.com/settings")
                time.sleep(3)  # 等待页面加载
                
                # 直接返回成功，让 cursor_register.py 处理账户信息获取
                return True
                
            else:
                if translator:
                    print(f"{translator.get('register.verification_failed')}")
                else:
                    print("最后一次验证失败")
                return False
            
        return False
        
    except Exception as e:
        if translator:
            print(f"{translator.get('register.verification_error', error=str(e))}")
        else:
            print(f"处理验证码时出错: {e}")
        return False

def main(email=None, password=None, first_name=None, last_name=None, email_tab=None, controller=None, translator=None):
    """主函数，可以接收账号信息、邮箱标签页和翻译器"""
    global _translator
    _translator = translator  # 保存到全局变量
    
    signal.signal(signal.SIGINT, signal_handler)
    signal.signal(signal.SIGTERM, signal_handler)
    
    page = None
    success = False
    try:
        page = setup_driver(translator)
        if translator:
            print(f"{translator.get('register.browser_started')}")
        else:
            print("浏览器已启动")
        
        # 访问注册页面
        url = "https://authenticator.cursor.sh/sign-up"
        if translator:
            print(f"\n{translator.get('register.visiting_url')}: {url}")
        else:
            print(f"\n正在访问: {url}")
        
        # 访问页面
        simulate_human_input(page, url, translator)
        if translator:
            print(f"{translator.get('register.waiting_for_page_load')}")
        else:
            print("等待页面加载...")
        time.sleep(5)
        
        # 如果没有提供账号信息，则生成随机信息
        if not all([email, password, first_name, last_name]):
            first_name = ''.join(random.choices('abcdefghijklmnopqrstuvwxyz', k=6)).capitalize()
            last_name = ''.join(random.choices('abcdefghijklmnopqrstuvwxyz', k=6)).capitalize()
            email = f"{first_name.lower()}{random.randint(100,999)}@example.com"
            password = generate_password()
            
            # 保存账号信息
            with open('test_accounts.txt', 'a', encoding='utf-8') as f:
                f.write(f"\n{'='*50}\n")
                f.write(f"Email: {email}\n")
                f.write(f"Password: {password}\n")
                f.write(f"{'='*50}\n")
        
        # 填写表单
        if fill_signup_form(page, first_name, last_name, email, translator):
            if translator:
                print(f"\n{translator.get('register.form_submitted')}")
            else:
                print("\n表单已提交，开始验证...")
            
            # 处理第一次 Turnstile 验证
            if handle_turnstile(page, translator):
                if translator:
                    print(f"\n{translator.get('register.first_verification_passed')}")
                else:
                    print("\n第一阶段验证通过！")
                
                # 填写密码
                if fill_password(page, password, translator):
                    if translator:
                        print(f"\n{translator.get('register.waiting_for_second_verification')}")
                    else:
                        print("\n等待第二次验证...")
                    time.sleep(2)
                    
                    # 处理第二次 Turnstile 验证
                    if handle_turnstile(page, translator):
                        if translator:
                            print(f"\n{translator.get('register.waiting_for_verification_code')}")
                        else:
                            print("\n开始处理验证码...")
                        if handle_verification_code(page, email_tab, controller, email, password, translator):
                            if translator:
                                print(f"\n{translator.get('register.verification_success')}")
                            else:
                                print("\n注册流程完成！")
                            success = True
                            return True, page  # 返回成功状态和浏览器实例
                        else:
                            print("\n验证码处理失败")
                    else:
                        print("\n第二次验证失败")
                else:
                    print("\n密码设置失败")
            else:
                print("\n第一次验证失败")
        
        return False, None
        
    except Exception as e:
        print(f"发生错误: {e}")
        return False, None
    finally:
        if page and not success:  # 只在失败时清理
            try:
                page.quit()
            except:
                pass
            cleanup_chrome_processes(translator)

if __name__ == "__main__":
    main()  # 直接运行时不传参数，使用随机生成的信息