From 825d080a8bb995385d9f66b401d7b5ed205542fa Mon Sep 17 00:00:00 2001 From: cheng zhen Date: Mon, 3 Feb 2025 19:42:54 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20Add=20version-specific=20machine=20ID?= =?UTF-8?q?=20reset=20mechanism=20for=20Cursor;=20=E6=84=9F=E8=B0=A2=20lin?= =?UTF-8?q?uxdo?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cursor_pro_keep_alive.py | 56 ++++++++- patch_cursor_get_machine_id.py | 222 +++++++++++++++++++++++++++++++++ 2 files changed, 277 insertions(+), 1 deletion(-) create mode 100644 patch_cursor_get_machine_id.py diff --git a/cursor_pro_keep_alive.py b/cursor_pro_keep_alive.py index 86e9bfd..5495bb2 100644 --- a/cursor_pro_keep_alive.py +++ b/cursor_pro_keep_alive.py @@ -1,6 +1,10 @@ import os +import platform +import json +from colorama import Fore, Style from exit_cursor import ExitCursor +import patch_cursor_get_machine_id from reset_machine import MachineIDResetter os.environ["PYTHONVERBOSE"] = "0" @@ -17,6 +21,9 @@ from logo import print_logo from config import Config from datetime import datetime +# 定义 EMOJI 字典 +EMOJI = {"ERROR": "❌", "WARNING": "⚠️", "INFO": "ℹ️"} + def save_screenshot(tab, prefix="turnstile"): """保存截图 @@ -344,6 +351,48 @@ def get_user_agent(): return None +def check_cursor_version(): + """检查cursor版本""" + system = platform.system() + try: + if system == "Darwin": # macOS + package_path = ( + "/Applications/Cursor.app/Contents/Resources/app/package.json" + ) + elif system == "Windows": # Windows + program_files = os.environ.get("ProgramFiles") + package_path = os.path.join( + program_files, "Cursor", "resources", "app", "package.json" + ) + else: + logging.error(f"不支持的操作系统: {system}") + return None + + if not os.path.exists(package_path): + logging.warning("未找到 Cursor 安装") + return None + + with open(package_path, "r", encoding="utf-8") as f: + package_data = json.load(f) + version = package_data.get("version") + if version: + logging.info(f"Cursor 版本: {version}") + version_parts = version.split(".") + if len(version_parts) >= 2: + major_minor = float(f"{version_parts[0]}.{version_parts[1]}") + if major_minor > 0.44: + return False + else: + return True + else: + logging.warning("无法获取版本信息") + return None + + except Exception as e: + logging.error(f"检查版本失败: {str(e)}") + return None + + if __name__ == "__main__": print_logo() browser_manager = None @@ -404,7 +453,12 @@ if __name__ == "__main__": ) logging.info("重置机器码...") - MachineIDResetter().reset_machine_ids() + # 判断cursor版本是否大于0.44 + is_cursor_version_greater_than_0_44 = check_cursor_version() + if is_cursor_version_greater_than_0_44: + MachineIDResetter().reset_machine_ids() + else: + patch_cursor_get_machine_id.patch_cursor_get_machine_id() logging.info("所有操作已完成") else: logging.error("获取会话令牌失败，注册流程未完成") diff --git a/patch_cursor_get_machine_id.py b/patch_cursor_get_machine_id.py new file mode 100644 index 0000000..9062aeb --- /dev/null +++ b/patch_cursor_get_machine_id.py @@ -0,0 +1,222 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- + +import json +import logging +import os +import platform +import re +import shutil +import sys +import tempfile +from typing import Tuple + + +# 配置日志 +def setup_logging() -> logging.Logger: + """配置并返回logger实例""" + logger = logging.getLogger(__name__) + logger.setLevel(logging.INFO) + handler = logging.StreamHandler() + formatter = logging.Formatter( + "%(asctime)s - %(levelname)s: %(message)s", datefmt="%Y-%m-%d %H:%M:%S" + ) + handler.setFormatter(formatter) + logger.addHandler(handler) + return logger + + +logger = setup_logging() + + +def get_cursor_paths() -> Tuple[str, str]: + """ + 根据不同操作系统获取 Cursor 相关路径 + + Returns: + Tuple[str, str]: (package.json路径, main.js路径)的元组 + + Raises: + OSError: 当找不到有效路径或系统不支持时抛出 + """ + system = platform.system() + + paths_map = { + "Darwin": { + "base": "/Applications/Cursor.app/Contents/Resources/app", + "package": "package.json", + "main": "out/main.js", + }, + "Windows": { + "base": os.path.join( + os.getenv("LOCALAPPDATA", ""), "Programs", "Cursor", "resources", "app" + ), + "package": "package.json", + "main": "out/main.js", + }, + "Linux": { + "bases": ["/opt/Cursor/resources/app", "/usr/share/cursor/resources/app"], + "package": "package.json", + "main": "out/main.js", + }, + } + + if system not in paths_map: + raise OSError(f"不支持的操作系统: {system}") + + if system == "Linux": + for base in paths_map["Linux"]["bases"]: + pkg_path = os.path.join(base, paths_map["Linux"]["package"]) + if os.path.exists(pkg_path): + return (pkg_path, os.path.join(base, paths_map["Linux"]["main"])) + raise OSError("在 Linux 系统上未找到 Cursor 安装路径") + + base_path = paths_map[system]["base"] + return ( + os.path.join(base_path, paths_map[system]["package"]), + os.path.join(base_path, paths_map[system]["main"]), + ) + + +def check_system_requirements(pkg_path: str, main_path: str) -> bool: + """ + 检查系统要求 + + Args: + pkg_path: package.json 文件路径 + main_path: main.js 文件路径 + + Returns: + bool: 检查是否通过 + """ + for file_path in [pkg_path, main_path]: + if not os.path.isfile(file_path): + logger.error(f"文件不存在: {file_path}") + return False + + if not os.access(file_path, os.W_OK): + logger.error(f"没有文件写入权限: {file_path}") + return False + + return True + + +def version_check(version: str, min_version: str = "", max_version: str = "") -> bool: + """ + 版本号检查 + + Args: + version: 当前版本号 + min_version: 最小版本号要求 + max_version: 最大版本号要求 + + Returns: + bool: 版本号是否符合要求 + """ + version_pattern = r"^\d+\.\d+\.\d+$" + try: + if not re.match(version_pattern, version): + logger.error(f"无效的版本号格式: {version}") + return False + + def parse_version(ver: str) -> Tuple[int, ...]: + return tuple(map(int, ver.split("."))) + + current = parse_version(version) + + if min_version and current < parse_version(min_version): + logger.error(f"版本号 {version} 小于最小要求 {min_version}") + return False + + if max_version and current > parse_version(max_version): + logger.error(f"版本号 {version} 大于最大要求 {max_version}") + return False + + return True + + except Exception as e: + logger.error(f"版本检查失败: {str(e)}") + return False + + +def modify_main_js(main_path: str) -> bool: + """ + 修改 main.js 文件 + + Args: + main_path: main.js 文件路径 + + Returns: + bool: 修改是否成功 + """ + try: + with tempfile.NamedTemporaryFile(mode="w", delete=False) as tmp_file: + with open(main_path, "r", encoding="utf-8") as main_file: + content = main_file.read() + + # 执行替换 + patterns = { + r"async getMachineId\(\)\{return [^??]+\?\?([^}]+)\}": r"async getMachineId(){return \1}", + r"async getMacMachineId\(\)\{return [^??]+\?\?([^}]+)\}": r"async getMacMachineId(){return \1}", + } + + for pattern, replacement in patterns.items(): + content = re.sub(pattern, replacement, content) + + tmp_file.write(content) + tmp_path = tmp_file.name + + # 使用 shutil.copy2 保留文件权限 + shutil.copy2(main_path, main_path + ".old") + shutil.move(tmp_path, main_path) + logger.info("文件修改成功") + return True + + except Exception as e: + logger.error(f"修改文件时发生错误: {str(e)}") + if "tmp_path" in locals(): + os.unlink(tmp_path) + return False + + +def patch_cursor_get_machine_id() -> None: + """主函数""" + logger.info("开始执行脚本...") + + try: + # 获取路径 + pkg_path, main_path = get_cursor_paths() + + # 检查系统要求 + if not check_system_requirements(pkg_path, main_path): + sys.exit(1) + + # 获取版本号 + try: + with open(pkg_path, "r", encoding="utf-8") as f: + version = json.load(f)["version"] + logger.info(f"当前 Cursor 版本: {version}") + except Exception as e: + logger.error(f"无法读取版本号: {str(e)}") + sys.exit(1) + + # 检查版本 + if not version_check(version, min_version="0.45.0"): + logger.error("版本不符合要求（需 >= 0.45.x）") + sys.exit(1) + + logger.info("版本检查通过，准备修改文件") + + # 修改文件 + if not modify_main_js(main_path): + sys.exit(1) + + logger.info("脚本执行完成") + + except Exception as e: + logger.error(f"执行过程中发生错误: {str(e)}") + sys.exit(1) + + +if __name__ == "__main__": + patch_cursor_get_machine_id()