#!/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() logger.info(f"当前操作系统: {system}") if system == "Darwin": # macOS base_path = "/Applications/Cursor.app/Contents/Resources/app" pkg_path = os.path.join(base_path, "package.json") main_path = os.path.join(base_path, "out", "main.js") if not os.path.exists(pkg_path) or not os.path.exists(main_path): raise OSError("在 macOS 系统上未找到 Cursor 安装路径") return pkg_path, main_path elif system == "Windows": base_path = os.path.join( os.getenv("LOCALAPPDATA", ""), "Programs", "Cursor", "resources", "app" ) pkg_path = os.path.join(base_path, "package.json") main_path = os.path.join(base_path, "out", "main.js") if not os.path.exists(pkg_path) or not os.path.exists(main_path): raise OSError("在 Windows 系统上未找到 Cursor 安装路径") return pkg_path, main_path elif system == "Linux": linux_paths = [ "/opt/Cursor/resources/app", "/usr/share/cursor/resources/app" ] for base_path in linux_paths: pkg_path = os.path.join(base_path, "package.json") main_path = os.path.join(base_path, "out", "main.js") if os.path.exists(pkg_path) and os.path.exists(main_path): return pkg_path, main_path raise OSError("在 Linux 系统上未找到 Cursor 安装路径") else: raise OSError(f"不支持的操作系统: {system}") 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: # 获取原始文件的权限和所有者信息 original_stat = os.stat(main_path) original_mode = original_stat.st_mode original_uid = original_stat.st_uid original_gid = original_stat.st_gid 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) # 恢复原始文件的权限和所有者 os.chmod(main_path, original_mode) if os.name != "nt": # 在非Windows系统上设置所有者 os.chown(main_path, original_uid, original_gid) 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 backup_files(pkg_path: str, main_path: str) -> bool: """ 备份原始文件 Args: pkg_path: package.json 文件路径（未使用） main_path: main.js 文件路径 Returns: bool: 备份是否成功 """ try: # 只备份 main.js if os.path.exists(main_path): backup_main = f"{main_path}.bak" shutil.copy2(main_path, backup_main) logger.info(f"已备份 main.js: {backup_main}") return True except Exception as e: logger.error(f"备份文件失败: {str(e)}") return False def restore_backup_files(pkg_path: str, main_path: str) -> bool: """ 恢复备份文件 Args: pkg_path: package.json 文件路径（未使用） main_path: main.js 文件路径 Returns: bool: 恢复是否成功 """ try: # 只恢复 main.js backup_main = f"{main_path}.bak" if os.path.exists(backup_main): shutil.copy2(backup_main, main_path) logger.info(f"已恢复 main.js") return True logger.error("未找到备份文件") return False except Exception as e: logger.error(f"恢复备份失败: {str(e)}") return False def patch_cursor_get_machine_id(restore_mode=False) -> None: """ 主函数 Args: restore_mode: 是否为恢复模式 """ logger.info("开始执行脚本...") try: # 获取路径 pkg_path, main_path = get_cursor_paths() # 检查系统要求 if not check_system_requirements(pkg_path, main_path): sys.exit(1) if restore_mode: # 恢复备份 if restore_backup_files(pkg_path, main_path): logger.info("备份恢复完成") else: logger.error("备份恢复失败") return # 获取版本号 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 backup_files(pkg_path, main_path): logger.error("文件备份失败，终止操作") sys.exit(1) # 修改文件 if not modify_main_js(main_path): sys.exit(1) logger.info("脚本执行完成") except Exception as e: logger.error(f"执行过程中发生错误: {str(e)}") sys.exit(1) # 添加patch函数作为主函数的别名 def patch(restore_mode=False) -> bool: """ patch函数，作为patch_cursor_get_machine_id的别名 Args: restore_mode: 是否为恢复模式 Returns: bool: 修补是否成功 """ try: patch_cursor_get_machine_id(restore_mode) return True except Exception as e: logger.error(f"修补失败: {str(e)}") return False if __name__ == "__main__": patch_cursor_get_machine_id()