113 lines
3.6 KiB
Python
113 lines
3.6 KiB
Python
|
import os
|
|||
|
|
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 os.name == "nt": # Windows
|
|||
|
|
self.db_path = os.path.join(
|
|||
|
|
os.getenv("APPDATA"), "Cursor", "User", "globalStorage", "storage.json"
|
|||
|
|
)
|
|||
|
|
else: # macOS
|
|||
|
|
self.db_path = os.path.expanduser(
|
|||
|
|
"~/Library/Application Support/Cursor/User/globalStorage/storage.json"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
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
|
|||
|
|
|
|||
|
|
# 读取现有配置
|
|||
|
|
print(f"{Fore.CYAN}{EMOJI['FILE']} 读取当前配置...{Style.RESET_ALL}")
|
|||
|
|
with open(self.db_path, "r", encoding="utf-8") as f:
|
|||
|
|
config = json.load(f)
|
|||
|
|
|
|||
|
|
# 备份原文件
|
|||
|
|
backup_path = self.db_path + ".bak"
|
|||
|
|
print(
|
|||
|
|
f"{Fore.YELLOW}{EMOJI['BACKUP']} 创建配置备份: {backup_path}{Style.RESET_ALL}"
|
|||
|
|
)
|
|||
|
|
shutil.copy2(self.db_path, backup_path)
|
|||
|
|
|
|||
|
|
# 生成新的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 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']} 按回车键退出...")
|