cursor-auto-register/browser_utils.py at main · VictorKTO/cursor-auto-register · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
from DrissionPage import ChromiumOptions, Chromium
import sys
import os
from logger import info, warning, error
from config import (
    BROWSER_USER_AGENT,
    BROWSER_PATH,
    BROWSER_HEADLESS,
    DYNAMIC_USERAGENT,
    USE_PROXY,
    PROXY_TYPE,
    PROXY_HOST,
    PROXY_PORT,
    PROXY_USERNAME,
    PROXY_PASSWORD,
    PROXY_TIMEOUT
)
import random
import time

from fake_useragent import UserAgent


def get_random_user_agent():
    ua = UserAgent()
    return ua.random

class BrowserManager:
    def __init__(self):
        self.browser = None

    def init_browser(self):
        try:
            info("正在初始化浏览器...")
            co = ChromiumOptions()

            # 如果配置了特定的浏览器路径，则使用
            if BROWSER_PATH and os.path.exists(BROWSER_PATH):
                co.set_browser_path(BROWSER_PATH)
                info(f"使用自定义浏览器路径: {BROWSER_PATH}")

            try:
                extension_path = self._get_extension_path()
                co.add_extension(extension_path)
            except FileNotFoundError as e:
                error(f"加载插件失败，警告: {e}")

            # 设置User-Agent
            if DYNAMIC_USERAGENT:
                # 随机选择一个User-Agent
                user_agent = get_random_user_agent()
                info(f"使用动态User-Agent: {user_agent}")
                co.set_user_agent(user_agent)
            elif BROWSER_USER_AGENT:
                info(f"使用固定User-Agent: {BROWSER_USER_AGENT}")
                co.set_user_agent(BROWSER_USER_AGENT)
            else:
                info("不配置User-Agent")

            co.set_pref("credentials_enable_service", False)
            co.set_argument("--hide-crash-restore-bubble")
            # 禁用自动化特征（关键参数）
            co.set_argument("--disable-blink-features=AutomationControlled")
            co.set_argument("--disable-features=AutomationControlled")
            co.set_argument("--disable-automation-extension")

            # 随机化指纹参数
            co.set_pref("webgl.vendor", "NVIDIA Corporation")
            co.set_pref(
                "webgl.renderer",
                "ANGLE (NVIDIA, NVIDIA GeForce RTX 3060 Direct3D11 vs_5_0 ps_5_0)",
            )
            co.set_pref("navigator.plugins.length", 5)
            co.set_pref("navigator.hardwareConcurrency", 8)

            # 覆盖自动化特征（关键）
            co.set_pref("dom.webdriver.enabled", False)
            co.set_pref("useAutomationExtension", False)

            # 设置时区参数
            co.set_argument("--timezone=Asia/Shanghai")
            co.set_pref("timezone.override", "Asia/Shanghai")

            # 设置更真实的屏幕参数
            co.set_pref("screen.width", 1920)
            co.set_pref("screen.height", 1080)
            co.set_pref("screen.pixelDepth", 24)
            co.auto_port()

            # 生产环境使用无头模式
            co.headless(BROWSER_HEADLESS)

            # Mac 系统特殊处理
            if sys.platform == "darwin" or sys.platform == "linux":
                co.set_argument("--no-sandbox")
                co.set_argument("--disable-gpu")

            # 添加代理设置
            if USE_PROXY and PROXY_HOST and PROXY_PORT:
                proxy_string = f"{PROXY_TYPE}://"

                # 如果有认证信息
                if PROXY_USERNAME and PROXY_PASSWORD:
                    proxy_string += f"{PROXY_USERNAME}:{PROXY_PASSWORD}@"

                proxy_string += f"{PROXY_HOST}:{PROXY_PORT}"

                info(f"使用代理: {PROXY_TYPE} {proxy_string}")
                co.set_argument(f'--proxy-server={proxy_string}')

            self.browser = Chromium(co)
            info("浏览器初始化成功")
        except Exception as e:
            error(f"浏览器初始化失败: {str(e)}")
        return self.browser

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

        if hasattr(sys, "_MEIPASS"):
            extension_path = os.path.join(sys._MEIPASS, "turnstilePatch")

        if not os.path.exists(extension_path):
            raise FileNotFoundError(f"插件不存在: {extension_path}")
        info(f"插件路径: {extension_path}")
        return extension_path

    def quit(self):
        info("正在关闭浏览器...")
        try:
            if self.browser:
                self.browser.quit()
            info("浏览器已关闭")
        except Exception as e:
            error(f"关闭浏览器出错: {str(e)}")