newest reset machine

reset_machine_manual.py

@@ -5,19 +5,11 @@ import uuid
 import hashlib
 import shutil
 import sqlite3
-import logging
+import platform
+import re
+import tempfile
 from colorama import Fore, Style, init
-
-# 设置日志记录
-logging.basicConfig(
-    level=logging.INFO,
-    format='%(asctime)s - %(levelname)s - %(message)s',
-    handlers=[
-        logging.StreamHandler(),
-        logging.FileHandler('reset_machine.log', encoding='utf-8')
-    ]
-)
-logger = logging.getLogger(__name__)
+from typing import Tuple
 
 # 初始化colorama
 init()
@@ -32,6 +24,173 @@ EMOJI = {
     "RESET": "🔄",
 }
 
+def get_cursor_paths(translator=None) -> Tuple[str, str]:
+    """根据不同操作系统获取 Cursor 相关路径"""
+    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(translator.get('reset.unsupported_os', system=system) if translator else 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(translator.get('reset.linux_path_not_found') if translator else "在 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 version_check(version: str, min_version: str = "", max_version: str = "", translator=None) -> bool:
+    """版本号检查"""
+    version_pattern = r"^\d+\.\d+\.\d+$"
+    try:
+        if not re.match(version_pattern, version):
+            print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('reset.invalid_version_format', version=version)}{Style.RESET_ALL}")
+            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):
+            print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('reset.version_too_low', version=version, min_version=min_version)}{Style.RESET_ALL}")
+            return False
+
+        if max_version and current > parse_version(max_version):
+            print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('reset.version_too_high', version=version, max_version=max_version)}{Style.RESET_ALL}")
+            return False
+
+        return True
+
+    except Exception as e:
+        print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('reset.version_check_error', error=str(e))}{Style.RESET_ALL}")
+        return False
+
+def check_cursor_version(translator) -> bool:
+    """检查 Cursor 版本"""
+    try:
+        pkg_path, _ = get_cursor_paths(translator)
+        with open(pkg_path, "r", encoding="utf-8") as f:
+            version = json.load(f)["version"]
+        return version_check(version, min_version="0.45.0", translator=translator)
+    except Exception as e:
+        print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('reset.check_version_failed', error=str(e))}{Style.RESET_ALL}")
+        return False
+
+def modify_main_js(main_path: str, translator) -> bool:
+    """修改 main.js 文件"""
+    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(main_path, main_path + ".old")
+        shutil.move(tmp_path, main_path)
+
+        os.chmod(main_path, original_mode)
+        if os.name != "nt":
+            os.chown(main_path, original_uid, original_gid)
+
+        print(f"{Fore.GREEN}{EMOJI['SUCCESS']} {translator.get('reset.file_modified')}{Style.RESET_ALL}")
+        return True
+
+    except Exception as e:
+        print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('reset.modify_file_failed', error=str(e))}{Style.RESET_ALL}")
+        if "tmp_path" in locals():
+            os.unlink(tmp_path)
+        return False
+
+def patch_cursor_get_machine_id(translator) -> bool:
+    """修补 Cursor getMachineId 函数"""
+    try:
+        print(f"{Fore.CYAN}{EMOJI['INFO']} {translator.get('reset.start_patching')}...{Style.RESET_ALL}")
+        
+        # 获取路径
+        pkg_path, main_path = get_cursor_paths(translator)
+        
+        # 检查文件权限
+        for file_path in [pkg_path, main_path]:
+            if not os.path.isfile(file_path):
+                print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('reset.file_not_found', path=file_path)}{Style.RESET_ALL}")
+                return False
+            if not os.access(file_path, os.W_OK):
+                print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('reset.no_write_permission', path=file_path)}{Style.RESET_ALL}")
+                return False
+
+        # 获取版本号
+        try:
+            with open(pkg_path, "r", encoding="utf-8") as f:
+                version = json.load(f)["version"]
+            print(f"{Fore.CYAN}{EMOJI['INFO']} {translator.get('reset.current_version', version=version)}{Style.RESET_ALL}")
+        except Exception as e:
+            print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('reset.read_version_failed', error=str(e))}{Style.RESET_ALL}")
+            return False
+
+        # 检查版本
+        if not version_check(version, min_version="0.45.0", translator=translator):
+            print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('reset.version_not_supported')}{Style.RESET_ALL}")
+            return False
+
+        print(f"{Fore.CYAN}{EMOJI['INFO']} {translator.get('reset.version_check_passed')}{Style.RESET_ALL}")
+
+        # 备份文件
+        backup_path = main_path + ".bak"
+        if not os.path.exists(backup_path):
+            shutil.copy2(main_path, backup_path)
+            print(f"{Fore.GREEN}{EMOJI['SUCCESS']} {translator.get('reset.backup_created', path=backup_path)}{Style.RESET_ALL}")
+
+        # 修改文件
+        if not modify_main_js(main_path, translator):
+            return False
+
+        print(f"{Fore.GREEN}{EMOJI['SUCCESS']} {translator.get('reset.patch_completed')}{Style.RESET_ALL}")
+        return True
+
+    except Exception as e:
+        print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('reset.patch_failed', error=str(e))}{Style.RESET_ALL}")
+        return False
+
 class MachineIDResetter:
     def __init__(self, translator=None):
         self.translator = translator
@@ -135,7 +294,6 @@ class MachineIDResetter:
             return True
         except Exception as e:
             print(f"{Fore.RED}{EMOJI['ERROR']} {self.translator.get('reset.system_ids_update_failed', error=str(e))}{Style.RESET_ALL}")
-            logger.error(f"System IDs update failed: {e}")
             return False
 
     def _update_windows_machine_guid(self):
@@ -151,12 +309,12 @@ class MachineIDResetter:
             new_guid = str(uuid.uuid4())
             winreg.SetValueEx(key, "MachineGuid", 0, winreg.REG_SZ, new_guid)
             winreg.CloseKey(key)
-            logger.info("Windows MachineGuid updated successfully")
+            print("Windows MachineGuid updated successfully")
         except PermissionError:
-            logger.error("Permission denied: Run as administrator to update Windows MachineGuid")
+            print("Permission denied: Run as administrator to update Windows MachineGuid")
             raise
         except Exception as e:
-            logger.error(f"Failed to update Windows MachineGuid: {e}")
+            print(f"Failed to update Windows MachineGuid: {e}")
             raise
 
     def _update_macos_platform_uuid(self, new_ids):
@@ -168,11 +326,11 @@ class MachineIDResetter:
                 cmd = f'sudo plutil -replace "UUID" -string "{new_ids["telemetry.macMachineId"]}" "{uuid_file}"'
                 result = os.system(cmd)
                 if result == 0:
-                    logger.info("macOS Platform UUID updated successfully")
+                    print("macOS Platform UUID updated successfully")
                 else:
                     raise Exception("Failed to execute plutil command")
         except Exception as e:
-            logger.error(f"Failed to update macOS Platform UUID: {e}")
+            print(f"Failed to update macOS Platform UUID: {e}")
             raise
 
     def reset_machine_ids(self):
@@ -215,6 +373,14 @@ class MachineIDResetter:
             # 更新系统ID
             self.update_system_ids(new_ids)
 
+            # 检查 Cursor 版本并执行相应的操作
+            greater_than_0_45 = check_cursor_version(self.translator)
+            if greater_than_0_45:
+                print(f"{Fore.CYAN}{EMOJI['INFO']} 检测到 Cursor 版本 >= 0.45.0，正在修补 getMachineId...{Style.RESET_ALL}")
+                patch_cursor_get_machine_id(self.translator)
+            else:
+                print(f"{Fore.YELLOW}{EMOJI['INFO']} Cursor 版本 < 0.45.0，跳过 getMachineId 修补{Style.RESET_ALL}")
+
             print(f"{Fore.GREEN}{EMOJI['SUCCESS']} {self.translator.get('reset.success')}{Style.RESET_ALL}")
             print(f"\n{Fore.CYAN}{self.translator.get('reset.new_id')}:{Style.RESET_ALL}")
             for key, value in new_ids.items():