feat: Add version-specific machine ID reset mechanism for Cursor; 感谢 linuxdo
This commit is contained in:
cheng zhen
@@ -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("获取会话令牌失败，注册流程未完成")
|
||||
|
||||
222
patch_cursor_get_machine_id.py
Normal file
222
patch_cursor_get_machine_id.py
Normal file
@@ -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()
|
||||
Reference in New Issue
Block a user