import os
os.environ['PYTHONVERBOSE'] = '0'
os.environ['PYINSTALLER_VERBOSE'] = '0'

from DrissionPage import ChromiumOptions, Chromium
from DrissionPage.common import Keys
import re
import time
import random
from cursor_auth_manager import CursorAuthManager
from configparser import ConfigParser
import os
import sys
import logging

# 在文件开头设置日志
logging.basicConfig(
    level=logging.WARNING,
    format='%(asctime)s - %(levelname)s - %(message)s',
    handlers=[
        logging.StreamHandler(),
        logging.FileHandler('cursor_keep_alive.log', encoding='utf-8')
    ]
)


def load_config():
    """加载配置文件"""
    config = ConfigParser()

    # 修改获取配置文件路径的方式
    if getattr(sys, 'frozen', False):
        # 如果是打包后的执行文件
        root_dir = sys._MEIPASS
    else:
        # 如果是直接运行 Python 脚本
        root_dir = os.getcwd()
        
    config_path = os.path.join(root_dir, "config.ini")

    if os.path.exists(config_path):
        config.read(config_path, encoding="utf-8")
        print(f"已加载配置文件: {config_path}")
        return {
            "account": config["Account"]["email"],
            "password": config["Account"]["password"],
            "first_name": config["Account"]["first_name"],
            "last_name": config["Account"]["last_name"],
        }

    raise FileNotFoundError(f"配置文件不存在: {config_path}")


def get_veri_code(tab):
    """获取验证码"""
    username = account.split("@")[0]
    try:
        while True:
            if tab.ele("@id=pre_button"):
                tab.actions.click("@id=pre_button").type(Keys.CTRL_A).key_down(
                    Keys.BACKSPACE
                ).key_up(Keys.BACKSPACE).input(username).key_down(Keys.ENTER).key_up(
                    Keys.ENTER
                )
                break
            time.sleep(1)

        while True:
            new_mail = tab.ele("@class=mail")
            if new_mail:
                if new_mail.text:
                    print("最新的邮件：", new_mail.text)
                    tab.actions.click("@class=mail")
                    break
                else:
                    print(new_mail)
                    break
            time.sleep(1)

        if tab.ele("@class=overflow-auto mb-20"):
            email_content = tab.ele("@class=overflow-auto mb-20").text
            verification_code = re.search(
                r"verification code is (\d{6})", email_content
            )
            if verification_code:
                code = verification_code.group(1)
                print("验证码：", code)
            else:
                print("未找到验证码")

        if tab.ele("@id=delete_mail"):
            tab.actions.click("@id=delete_mail")
            time.sleep(1)

        if tab.ele("@id=confirm_mail"):
            tab.actions.click("@id=confirm_mail")
            print("删除邮件")
        tab.close()
    except Exception as e:
        print(e)

    return code


def handle_turnstile(tab):
    """处理 Turnstile 验证"""
    print("准备处理验证")
    try:
        while True:
            try:
                challengeCheck = (
                    tab.ele("@id=cf-turnstile", timeout=2)
                    .child()
                    .shadow_root.ele("tag:iframe")
                    .ele("tag:body")
                    .sr("tag:input")
                )

                if challengeCheck:
                    print("验证框加载完成")
                    time.sleep(random.uniform(1, 3))
                    challengeCheck.click()
                    print("验证按钮已点击，等待验证完成...")
                    time.sleep(2)
                    return True
            except:
                pass

            if tab.ele("@name=password"):
                print("无需验证")
                break
            if tab.ele("@data-index=0"):
                print("无需验证")
                break
            if tab.ele("Account Settings"):
                print("无需验证")
                break

            time.sleep(random.uniform(1, 2))
    except Exception as e:
        print(e)
        print("跳过验证")
        return False


def delete_account(browser, tab):
    """删除账户流程"""
    print("\n开始删除账户...")

    try:
        if tab.ele("@name=email"):
            tab.ele("@name=email").input(account)
            print("输入账号")
            time.sleep(random.uniform(1, 3))
    except Exception as e:
        print(f"输入账号失败: {str(e)}")

    try:
        if tab.ele("Continue"):
            tab.ele("Continue").click()
            print("点击Continue")
    except Exception as e:
        print(f"点击Continue失败: {str(e)}")

    handle_turnstile(tab)
    time.sleep(5)

    try:
        if tab.ele("@name=password"):
            tab.ele("@name=password").input(password)
            print("输入密码")
            time.sleep(random.uniform(1, 3))
    except Exception as e:
        print("输入密码失败")

    sign_in_button = tab.ele(
        "xpath:/html/body/div[1]/div/div/div[2]/div/form/div/button"
    )
    try:
        if sign_in_button:
            sign_in_button.click(by_js=True)
            print("点击Sign in")
    except Exception as e:
        print(f"点击Sign in失败: {str(e)}")

    handle_turnstile(tab)

    # 处理验证码
    while True:
        try:
            if tab.ele("Account Settings"):
                break
            if tab.ele("@data-index=0"):
                tab_mail = browser.new_tab(mail_url)
                browser.activate_tab(tab_mail)
                print("打开邮箱页面")
                code = get_veri_code(tab_mail)

                if code:
                    print("获取验证码成功：", code)
                    browser.activate_tab(tab)
                else:
                    print("获取验证码失败，程序退出")
                    return False

                i = 0
                for digit in code:
                    tab.ele(f"@data-index={i}").input(digit)
                    time.sleep(random.uniform(0.1, 0.3))
                    i += 1
                break
        except Exception as e:
            print(e)

    handle_turnstile(tab)
    time.sleep(random.uniform(1, 3))
    # tab.get_screenshot('sign-in_success.png')
    # print("登录账户截图")

    tab.get(settings_url)
    print("进入设置页面")

    try:
        if tab.ele("@class=mt-1"):
            tab.ele("@class=mt-1").click()
            print("点击Adavance")
            time.sleep(random.uniform(1, 2))
    except Exception as e:
        print(f"点击Adavance失败: {str(e)}")

    try:
        if tab.ele("Delete Account"):
            tab.ele("Delete Account").click()
            print("点击Delete Account")
            time.sleep(random.uniform(1, 2))
    except Exception as e:
        print(f"点击Delete Account失败: {str(e)}")

    try:
        if tab.ele("tag:input"):
            tab.actions.click("tag:input").type("delete")
            print("输入delete")
            time.sleep(random.uniform(1, 2))
    except Exception as e:
        print(f"输入delete失败: {str(e)}")

    delete_button = tab.ele(
        "xpath:/html/body/main/div/div/div/div/div/div[1]/div[2]/div[3]/div[2]/div/div/div[2]/button[2]"
    )
    try:
        if delete_button:
            print("点击Delete")
            delete_button.click()
            time.sleep(5)
            # tab.get_screenshot('delete_account.png')
            # print("删除账户截图")
            return True
    except Exception as e:
        print(f"点击Delete失败: {str(e)}")
        return False


def get_cursor_session_token(tab):
    """获取cursor session token"""
    cookies = tab.cookies()
    cursor_session_token = None
    for cookie in cookies:
        if cookie["name"] == "WorkosCursorSessionToken":
            cursor_session_token = cookie["value"].split("%3A%3A")[1]
            break
    return cursor_session_token


def update_cursor_auth(email=None, access_token=None, refresh_token=None):
    """
    更新Cursor的认证信息的便捷函数
    """
    auth_manager = CursorAuthManager()
    return auth_manager.update_auth(email, access_token, refresh_token)


def sign_up_account(browser, tab):
    """注册账户流程"""
    print("\n开始注册新账户...")
    tab.get(sign_up_url)

    try:
        if tab.ele("@name=first_name"):
            print("已打开注册页面")
            tab.actions.click("@name=first_name").input(first_name)
            time.sleep(random.uniform(1, 3))

            tab.actions.click("@name=last_name").input(last_name)
            time.sleep(random.uniform(1, 3))

            tab.actions.click("@name=email").input(account)
            print("输入邮箱")
            time.sleep(random.uniform(1, 3))

            tab.actions.click("@type=submit")
            print("点击注册按钮")

    except Exception as e:
        print("打开注册页面失败")
        return False

    handle_turnstile(tab)

    try:
        if tab.ele("@name=password"):
            tab.ele("@name=password").input(password)
            print("输入密码")
            time.sleep(random.uniform(1, 3))

            tab.ele("@type=submit").click()
            print("点击Continue按钮")

    except Exception as e:
        print("输入密码失败")
        return False

    time.sleep(random.uniform(1, 3))
    if tab.ele("This email is not available."):
        print("This email is not available.")
        return False

    handle_turnstile(tab)

    while True:
        try:
            if tab.ele("Account Settings"):
                break
            if tab.ele("@data-index=0"):
                tab_mail = browser.new_tab(mail_url)
                browser.activate_tab(tab_mail)
                print("打开邮箱页面")
                code = get_veri_code(tab_mail)

                if code:
                    print("获取验证码成功：", code)
                    browser.activate_tab(tab)
                else:
                    print("获取验证码失败，程序退出")
                    return False

                i = 0
                for digit in code:
                    tab.ele(f"@data-index={i}").input(digit)
                    time.sleep(random.uniform(0.1, 0.3))
                    i += 1
                break
        except Exception as e:
            print(e)

    handle_turnstile(tab)

    time.sleep(random.uniform(1, 3))
    print("进入设置页面")
    tab.get(settings_url)
    try:
        usage_ele = tab.ele(
            "xpath:/html/body/main/div/div/div/div/div/div[2]/div/div/div/div[1]/div[1]/span[2]"
        )
        if usage_ele:
            usage_info = usage_ele.text
            total_usage = usage_info.split("/")[-1].strip()
            print("可用上限: " + total_usage)
    except Exception as e:
        print("获取可用上限失败")
    # tab.get_screenshot("sign_up_success.png")
    # print("注册账户截图")
    print("注册完成")
    print("Cursor 账号： " + account)
    print("       密码： " + password)
    return True


def get_extension_path():
    """获取插件路径"""
    root_dir = os.getcwd()
    extension_path = os.path.join(root_dir, "turnstilePatch")

    if hasattr(sys, "_MEIPASS"):
        print("运行在打包环境中")
        extension_path = os.path.join(sys._MEIPASS, "turnstilePatch")

    print(f"尝试加载插件路径: {extension_path}")

    if not os.path.exists(extension_path):
        raise FileNotFoundError(
            f"插件不存在: {extension_path}\n请确保 turnstilePatch 文件夹在正确位置"
        )

    return extension_path


def get_browser_options():
    co = ChromiumOptions()
    try:
        extension_path = get_extension_path()
        co.add_extension(extension_path)
    except FileNotFoundError as e:
        print(f"警告: {e}")
        
    co.headless()
    co.set_user_agent("Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/130.0.6723.92 Safari/537.36")
    co.set_pref("credentials_enable_service", False)
    co.set_argument("--hide-crash-restore-bubble")
    co.auto_port()
    
    # Mac 系统特殊处理
    if sys.platform == 'darwin':
        co.set_argument('--no-sandbox')
        co.set_argument('--disable-gpu')
    
    return co


def cleanup_temp_files():
    """清理临时文件和缓存"""
    try:
        temp_dirs = [
            os.path.join(os.getcwd(), '__pycache__'),
            os.path.join(os.getcwd(), 'build'),
        ]
        
        for dir_path in temp_dirs:
            if os.path.exists(dir_path):
                import shutil
                shutil.rmtree(dir_path)
    except Exception as e:
        logging.warning(f"清理临时文件失败: {str(e)}")


if __name__ == "__main__":
    browser = None
    try:
        # 加载配置
        config = load_config()

        # 固定的 URL 配置
        login_url = "https://authenticator.cursor.sh"
        sign_up_url = "https://authenticator.cursor.sh/sign-up"
        settings_url = "https://www.cursor.com/settings"
        mail_url = "https://tempmail.plus"

        # 账号信息
        account = config["account"]
        password = config["password"]
        first_name = config["first_name"]
        last_name = config["last_name"]

        auto_update_cursor_auth = True

        # 浏览器配置
        co = get_browser_options()

        browser = Chromium(co)
        tab = browser.latest_tab
        tab.run_js("try { turnstile.reset() } catch(e) { }")

        print("开始执行删除和注册流程")
        print("***请确认已经用https://tempmail.plus/zh邮箱成功申请过cursor账号！***")
        tab.get(login_url)

        # 执行删除和注册流程
        if delete_account(browser, tab):
            print("账户删除成功")
            time.sleep(3)
            if sign_up_account(browser, tab):
                token = get_cursor_session_token(tab)
                print(f"CursorSessionToken: {token}")
                print("账户注册成功")
                if auto_update_cursor_auth:
                    update_cursor_auth(
                        email=account, access_token=token, refresh_token=token
                    )
            else:
                print("账户注册失败")
        else:
            print("账户删除失败")

        print("脚本执行完毕")

    except Exception as e:
        print(f"程序执行出错: {str(e)}")
        import traceback
        print(traceback.format_exc())
    finally:
        # 确保浏览器实例被正确关闭
        if browser:
            try:
                browser.quit()
            except:
                pass
        input("\n按回车键退出...")