first commit

2025-02-09 22:02:44 +08:00
commit 49fa4d6fd6
35 changed files with 3282 additions and 0 deletions
--- a/reset_machine.py
+++ b/reset_machine.py
@@ -0,0 +1,134 @@
+import os
+import sys
+import json
+import uuid
+import hashlib
+import shutil
+from colorama import Fore, Style, init
+
+# 初始化colorama
+init()
+
+# 定义emoji和颜色常量
+EMOJI = {
+    "FILE": "📄",
+    "BACKUP": "💾",
+    "SUCCESS": "✅",
+    "ERROR": "❌",
+    "INFO": "ℹ️",
+    "RESET": "🔄",
+}
+
+
+class MachineIDResetter:
+    def __init__(self):
+        # 判断操作系统
+        if sys.platform == "win32":  # Windows
+            appdata = os.getenv("APPDATA")
+            if appdata is None:
+                raise EnvironmentError("APPDATA 环境变量未设置")
+            self.db_path = os.path.join(
+                appdata, "Cursor", "User", "globalStorage", "storage.json"
+            )
+        elif sys.platform == "darwin":  # macOS
+            self.db_path = os.path.abspath(
+                os.path.expanduser(
+                    "~/Library/Application Support/Cursor/User/globalStorage/storage.json"
+                )
+            )
+        elif sys.platform == "linux":  # Linux 和其他类Unix系统
+            self.db_path = os.path.abspath(
+                os.path.expanduser("~/.config/Cursor/User/globalStorage/storage.json")
+            )
+        else:
+            raise NotImplementedError(f"不支持的操作系统: {sys.platform}")
+
+    def generate_new_ids(self):
+        """生成新的机器ID"""
+        # 生成新的UUID
+        dev_device_id = str(uuid.uuid4())
+
+        # 生成新的machineId (64个字符的十六进制)
+        machine_id = hashlib.sha256(os.urandom(32)).hexdigest()
+
+        # 生成新的macMachineId (128个字符的十六进制)
+        mac_machine_id = hashlib.sha512(os.urandom(64)).hexdigest()
+
+        # 生成新的sqmId
+        sqm_id = "{" + str(uuid.uuid4()).upper() + "}"
+
+        return {
+            "telemetry.devDeviceId": dev_device_id,
+            "telemetry.macMachineId": mac_machine_id,
+            "telemetry.machineId": machine_id,
+            "telemetry.sqmId": sqm_id,
+        }
+
+    def reset_machine_ids(self):
+        """重置机器ID并备份原文件"""
+        try:
+            print(f"{Fore.CYAN}{EMOJI['INFO']} 正在检查配置文件...{Style.RESET_ALL}")
+
+            # 检查文件是否存在
+            if not os.path.exists(self.db_path):
+                print(
+                    f"{Fore.RED}{EMOJI['ERROR']} 配置文件不存在: {self.db_path}{Style.RESET_ALL}"
+                )
+                return False
+
+            # 检查文件权限
+            if not os.access(self.db_path, os.R_OK | os.W_OK):
+                print(
+                    f"{Fore.RED}{EMOJI['ERROR']} 无法读写配置文件，请检查文件权限！{Style.RESET_ALL}"
+                )
+                print(
+                    f"{Fore.RED}{EMOJI['ERROR']} 如果你使用过 go-cursor-help 来修改 ID; 请修改文件只读权限 {self.db_path} {Style.RESET_ALL}"
+                )
+                return False
+
+            # 读取现有配置
+            print(f"{Fore.CYAN}{EMOJI['FILE']} 读取当前配置...{Style.RESET_ALL}")
+            with open(self.db_path, "r", encoding="utf-8") as f:
+                config = json.load(f)
+
+            # 生成新的ID
+            print(f"{Fore.CYAN}{EMOJI['RESET']} 生成新的机器标识...{Style.RESET_ALL}")
+            new_ids = self.generate_new_ids()
+
+            # 更新配置
+            config.update(new_ids)
+
+            # 保存新配置
+            print(f"{Fore.CYAN}{EMOJI['FILE']} 保存新配置...{Style.RESET_ALL}")
+            with open(self.db_path, "w", encoding="utf-8") as f:
+                json.dump(config, f, indent=4)
+
+            print(f"{Fore.GREEN}{EMOJI['SUCCESS']} 机器标识重置成功！{Style.RESET_ALL}")
+            print(f"\n{Fore.CYAN}新的机器标识:{Style.RESET_ALL}")
+            for key, value in new_ids.items():
+                print(f"{EMOJI['INFO']} {key}: {Fore.GREEN}{value}{Style.RESET_ALL}")
+
+            return True
+
+        except PermissionError as e:
+            print(f"{Fore.RED}{EMOJI['ERROR']} 权限错误: {str(e)}{Style.RESET_ALL}")
+            print(
+                f"{Fore.YELLOW}{EMOJI['INFO']} 请尝试以管理员身份运行此程序{Style.RESET_ALL}"
+            )
+            return False
+        except Exception as e:
+            print(f"{Fore.RED}{EMOJI['ERROR']} 重置过程出错: {str(e)}{Style.RESET_ALL}")
+
+            return False
+
+
+if __name__ == "__main__":
+    print(f"\n{Fore.CYAN}{'='*50}{Style.RESET_ALL}")
+    print(f"{Fore.CYAN}{EMOJI['RESET']} Cursor 机器标识重置工具{Style.RESET_ALL}")
+    print(f"{Fore.CYAN}{'='*50}{Style.RESET_ALL}")
+
+    resetter = MachineIDResetter()
+    resetter.reset_machine_ids()
+
+    print(f"\n{Fore.CYAN}{'='*50}{Style.RESET_ALL}")
+    input(f"{EMOJI['INFO']} 按回车键退出...")