472 lines
17 KiB
Python
472 lines
17 KiB
Python
import os
|
||
import platform
|
||
import json
|
||
import sys
|
||
import hashlib
|
||
import subprocess
|
||
import uuid
|
||
import requests
|
||
from exit_cursor import ExitCursor
|
||
from logger import logging
|
||
from cursor_auth_manager import CursorAuthManager
|
||
import go_cursor_help
|
||
import patch_cursor_get_machine_id
|
||
from reset_machine import MachineIDResetter
|
||
from logo import print_logo
|
||
from typing import Optional, Tuple, Dict, Any
|
||
from urllib3.exceptions import InsecureRequestWarning
|
||
import urllib3
|
||
import time
|
||
|
||
# 禁用不安全请求警告
|
||
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
|
||
|
||
# 定义 EMOJI 字典
|
||
EMOJI = {"ERROR": "❌", "WARNING": "⚠️", "INFO": "ℹ️"}
|
||
|
||
|
||
def check_cursor_version():
|
||
"""检查cursor版本"""
|
||
pkg_path, main_path = patch_cursor_get_machine_id.get_cursor_paths()
|
||
with open(pkg_path, "r", encoding="utf-8") as f:
|
||
version = json.load(f)["version"]
|
||
return patch_cursor_get_machine_id.version_check(version, min_version="0.45.0")
|
||
|
||
|
||
def reset_machine_id(greater_than_0_45):
|
||
if greater_than_0_45:
|
||
# 提示请手动执行脚本 https://github.com/chengazhen/cursor-auto-free/blob/main/patch_cursor_get_machine_id.py
|
||
go_cursor_help.go_cursor_help()
|
||
else:
|
||
MachineIDResetter().reset_machine_ids()
|
||
|
||
|
||
def print_end_message():
|
||
logging.info("\n\n\n\n\n")
|
||
logging.info("=" * 30)
|
||
logging.info("所有操作已完成")
|
||
|
||
|
||
|
||
def get_account_from_api() -> tuple[bool, dict]:
|
||
"""从API获取账号信息
|
||
|
||
Returns:
|
||
tuple[bool, dict]: (是否成功, 账号信息)
|
||
"""
|
||
try:
|
||
# 获取设备唯一ID
|
||
hardware_id = get_mac_unique_id()
|
||
logging.info(f"设备唯一ID: {hardware_id}")
|
||
|
||
endpoint = "https://cursorapi.nosqli.com/admin/api.account/getUnused"
|
||
data = {
|
||
"machine_id": hardware_id
|
||
}
|
||
headers = {
|
||
"Content-Type": "application/json"
|
||
}
|
||
|
||
import urllib3
|
||
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
|
||
|
||
request_kwargs = {
|
||
"json": data,
|
||
"headers": headers,
|
||
"timeout": 30,
|
||
"verify": False
|
||
}
|
||
|
||
try:
|
||
try:
|
||
response = requests.post(endpoint, **request_kwargs)
|
||
except requests.exceptions.SSLError:
|
||
import ssl
|
||
ssl_context = ssl.create_default_context()
|
||
ssl_context.check_hostname = False
|
||
ssl_context.verify_mode = ssl.CERT_NONE
|
||
|
||
session = requests.Session()
|
||
session.verify = False
|
||
response = session.post(endpoint, **request_kwargs)
|
||
|
||
response_data = response.json()
|
||
|
||
if response_data.get("code") == 200:
|
||
account_data = response_data.get("data", {})
|
||
return True, account_data
|
||
else:
|
||
error_msg = response_data.get("msg", "未知错误")
|
||
logging.error(f"获取未使用账号失败: {error_msg}")
|
||
return False, {}
|
||
|
||
except Exception as e:
|
||
logging.error(f"API请求失败: {str(e)}")
|
||
return False, {}
|
||
|
||
except Exception as e:
|
||
logging.error(f"获取账号过程出错: {str(e)}")
|
||
return False, {}
|
||
|
||
|
||
def get_mac_unique_id() -> str:
|
||
"""
|
||
获取Mac设备的唯一ID（32位MD5）
|
||
组合以下信息生成唯一标识：
|
||
1. 硬件UUID
|
||
2. 系统序列号
|
||
3. 主板序列号
|
||
4. CPU信息
|
||
"""
|
||
def run_cmd(cmd: str) -> str:
|
||
try:
|
||
result = subprocess.check_output(cmd, shell=True).decode('utf-8').strip()
|
||
return result
|
||
except:
|
||
return ""
|
||
|
||
# 收集系统信息
|
||
identifiers = []
|
||
|
||
# 1. 获取硬件UUID
|
||
hw_uuid = run_cmd("ioreg -d2 -c IOPlatformExpertDevice | awk -F\\\" '/IOPlatformUUID/{print $(NF-1)}'")
|
||
identifiers.append(hw_uuid)
|
||
|
||
# 2. 获取系统序列号
|
||
serial = run_cmd("ioreg -d2 -c IOPlatformExpertDevice | awk -F\\\" '/IOPlatformSerialNumber/{print $(NF-1)}'")
|
||
identifiers.append(serial)
|
||
|
||
# 3. 获取主板信息
|
||
board_id = run_cmd("ioreg -d2 -c IOPlatformExpertDevice | awk -F\\\" '/board-id/{print $(NF-1)}'")
|
||
identifiers.append(board_id)
|
||
|
||
# 4. 获取CPU信息
|
||
cpu_info = run_cmd("sysctl -n machdep.cpu.brand_string")
|
||
identifiers.append(cpu_info)
|
||
|
||
# 如果以上方法都失败，使用备用方法
|
||
if not any(identifiers):
|
||
# 使用 UUID 模块获取 UUID（不太稳定，仅作为备用）
|
||
identifiers = [
|
||
str(uuid.getnode()), # MAC 地址的整数表示
|
||
platform.machine(), # CPU 架构
|
||
platform.system(), # 操作系统名称
|
||
platform.version() # 操作系统版本
|
||
]
|
||
|
||
# 组合所有标识符并生成MD5
|
||
unique_string = "".join(filter(None, identifiers))
|
||
return hashlib.md5(unique_string.encode()).hexdigest()
|
||
|
||
|
||
class CursorAccountManager:
|
||
def __init__(self):
|
||
self.base_url = "https://cursorapi.nosqli.com"
|
||
self.api_endpoints = {
|
||
"activate": f"{self.base_url}/admin/api.member/activate",
|
||
"status": f"{self.base_url}/admin/api.member/status",
|
||
"get_unused": f"{self.base_url}/admin/api.account/getUnused",
|
||
"heartbeat": f"{self.base_url}/admin/api.member/heartbeat"
|
||
}
|
||
self.hardware_id = get_mac_unique_id()
|
||
|
||
def get_device_info(self) -> dict:
|
||
"""获取设备信息"""
|
||
return {
|
||
"system": platform.system(),
|
||
"device_name": platform.node(),
|
||
"ip": self._get_ip_address(),
|
||
"location": self._get_location()
|
||
}
|
||
|
||
def _get_ip_address(self) -> str:
|
||
"""获取IP地址"""
|
||
try:
|
||
response = requests.get('https://api.ipify.org?format=json', timeout=5)
|
||
return response.json()['ip']
|
||
except:
|
||
return "未知"
|
||
|
||
def _get_location(self) -> str:
|
||
"""获取地理位置"""
|
||
try:
|
||
ip = self._get_ip_address()
|
||
if ip != "未知":
|
||
response = requests.get(f'http://ip-api.com/json/{ip}', timeout=5)
|
||
data = response.json()
|
||
if data.get('status') == 'success':
|
||
return f"{data.get('country', '')} {data.get('city', '')}"
|
||
except:
|
||
pass
|
||
return "未知"
|
||
|
||
def check_member_status(self) -> tuple[bool, dict]:
|
||
"""检查会员状态
|
||
|
||
Returns:
|
||
tuple[bool, dict]: (是否成功, 状态信息)
|
||
"""
|
||
try:
|
||
data = {
|
||
"machine_id": self.hardware_id
|
||
}
|
||
|
||
api_url = self.api_endpoints["status"]
|
||
logging.info(f"正在检查会员状态...")
|
||
|
||
request_kwargs = {
|
||
"json": data,
|
||
"headers": {"Content-Type": "application/json"},
|
||
"timeout": 2,
|
||
"verify": False
|
||
}
|
||
|
||
session = requests.Session()
|
||
session.verify = False
|
||
|
||
try:
|
||
response = session.post(api_url, **request_kwargs)
|
||
except requests.exceptions.Timeout:
|
||
logging.warning("首次请求超时，正在重试...")
|
||
response = session.post(api_url, **request_kwargs)
|
||
|
||
result = response.json()
|
||
logging.info(f"状态检查响应: {result}")
|
||
|
||
if result.get("code") in [1, 200]:
|
||
api_data = result.get("data", {})
|
||
status_data = {
|
||
"is_active": api_data.get("status") == "active",
|
||
"expire_time": api_data.get("expire_time", ""),
|
||
"total_days": api_data.get("total_days", 0),
|
||
"days_left": api_data.get("days_left", 0),
|
||
"device_info": self.get_device_info()
|
||
}
|
||
return True, status_data
|
||
else:
|
||
error_msg = result.get("msg", "未知错误")
|
||
logging.error(f"获取状态失败: {error_msg}")
|
||
return False, {
|
||
"is_active": False,
|
||
"expire_time": "",
|
||
"total_days": 0,
|
||
"days_left": 0,
|
||
"device_info": self.get_device_info()
|
||
}
|
||
|
||
except Exception as e:
|
||
logging.error(f"获取会员状态失败: {str(e)}")
|
||
return False, {
|
||
"is_active": False,
|
||
"expire_time": "",
|
||
"total_days": 0,
|
||
"days_left": 0,
|
||
"device_info": self.get_device_info()
|
||
}
|
||
|
||
def check_activation_code(self, code: str) -> tuple[bool, str, dict | None]:
|
||
"""检查激活码
|
||
|
||
Args:
|
||
code: 激活码
|
||
|
||
Returns:
|
||
tuple: (成功标志, 消息, 账号信息)
|
||
"""
|
||
max_retries = 3 # 最大重试次数
|
||
retry_delay = 1 # 重试间隔(秒)
|
||
|
||
for attempt in range(max_retries):
|
||
try:
|
||
data = {
|
||
"machine_id": self.hardware_id,
|
||
"code": code
|
||
}
|
||
|
||
# 设置请求参数
|
||
request_kwargs = {
|
||
"json": data,
|
||
"headers": {
|
||
"Content-Type": "application/json",
|
||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36",
|
||
"Accept": "*/*",
|
||
"Connection": "keep-alive"
|
||
},
|
||
"timeout": 10,
|
||
"verify": False
|
||
}
|
||
|
||
# 创建session
|
||
session = requests.Session()
|
||
session.verify = False
|
||
|
||
# 设置重试策略
|
||
retry_strategy = urllib3.Retry(
|
||
total=3,
|
||
backoff_factor=0.5,
|
||
status_forcelist=[500, 502, 503, 504]
|
||
)
|
||
adapter = requests.adapters.HTTPAdapter(max_retries=retry_strategy)
|
||
session.mount("http://", adapter)
|
||
session.mount("https://", adapter)
|
||
|
||
try:
|
||
# 尝试发送请求
|
||
response = session.post(
|
||
self.api_endpoints["activate"],
|
||
**request_kwargs
|
||
)
|
||
response.raise_for_status()
|
||
|
||
result = response.json()
|
||
logging.info(f"激活响应: {result}")
|
||
|
||
# 激活成功
|
||
if result["code"] == 200:
|
||
api_data = result["data"]
|
||
account_info = {
|
||
"status": "active",
|
||
"expire_time": api_data.get("expire_time", ""),
|
||
"total_days": api_data.get("total_days", 0),
|
||
"days_left": api_data.get("days_left", 0),
|
||
"device_info": self.get_device_info()
|
||
}
|
||
return True, result["msg"], account_info
|
||
# 激活码无效或已被使用
|
||
elif result["code"] == 400:
|
||
logging.warning(f"激活码无效或已被使用: {result.get('msg', '未知错误')}")
|
||
return False, result.get("msg", "激活码无效或已被使用"), None
|
||
# 其他错误情况
|
||
else:
|
||
error_msg = result.get("msg", "未知错误")
|
||
if attempt < max_retries - 1:
|
||
logging.warning(f"第{attempt + 1}次尝试失败: {error_msg}, 准备重试...")
|
||
time.sleep(retry_delay)
|
||
continue
|
||
logging.error(f"激活失败: {error_msg}")
|
||
return False, error_msg, None
|
||
|
||
except requests.exceptions.RequestException as e:
|
||
if attempt < max_retries - 1:
|
||
logging.warning(f"第{attempt + 1}次网络请求失败: {str(e)}, 准备重试...")
|
||
time.sleep(retry_delay)
|
||
continue
|
||
logging.error(f"网络请求失败: {str(e)}")
|
||
return False, f"网络连接失败: {str(e)}", None
|
||
|
||
except Exception as e:
|
||
if attempt < max_retries - 1:
|
||
logging.warning(f"第{attempt + 1}次请求发生错误: {str(e)}, 准备重试...")
|
||
time.sleep(retry_delay)
|
||
continue
|
||
logging.error(f"激活失败: {str(e)}")
|
||
return False, f"激活失败: {str(e)}", None
|
||
|
||
return False, "多次尝试后激活失败，请检查网络连接或稍后重试", None
|
||
|
||
|
||
def reset_auth_with_password(password: str = None) -> tuple[bool, str]:
|
||
"""
|
||
封装重置授权的完整流程
|
||
|
||
Args:
|
||
password: 系统密码（可选）
|
||
|
||
Returns:
|
||
tuple[bool, str]: (是否成功, 消息)
|
||
"""
|
||
try:
|
||
logging.info("\n=== 初始化重置流程 ===")
|
||
greater_than_0_45 = check_cursor_version()
|
||
|
||
logging.info("正在从API获取账号信息...")
|
||
success, account_data = get_account_from_api()
|
||
|
||
if not success:
|
||
return False, "获取账号信息失败"
|
||
|
||
email = account_data.get("email", "")
|
||
access_token = account_data.get("access_token", "")
|
||
refresh_token = account_data.get("refresh_token", "")
|
||
|
||
if not all([email, access_token, refresh_token]):
|
||
return False, "账号信息不完整"
|
||
|
||
logging.info(f"获取到账号信息：\n邮箱: {email}")
|
||
|
||
# 更新认证信息
|
||
logging.info("正在更新认证信息...")
|
||
auth_manager = CursorAuthManager()
|
||
if auth_manager.update_auth(email, access_token, refresh_token):
|
||
logging.info("认证信息更新成功")
|
||
|
||
# 重置机器码
|
||
logging.info("正在重置机器码...")
|
||
|
||
# 如果提供了密码，设置环境变量
|
||
if password:
|
||
import os
|
||
os.environ['SUDO_PASSWORD'] = password
|
||
|
||
reset_machine_id(greater_than_0_45)
|
||
logging.info("重置完成")
|
||
return True, "重置成功"
|
||
else:
|
||
return False, "更新认证信息失败"
|
||
|
||
except Exception as e:
|
||
logging.error(f"重置过程出错: {str(e)}")
|
||
import traceback
|
||
logging.error(traceback.format_exc())
|
||
return False, f"重置失败: {str(e)}"
|
||
|
||
|
||
if __name__ == "__main__":
|
||
print_logo()
|
||
greater_than_0_45 = check_cursor_version()
|
||
browser_manager = None
|
||
try:
|
||
logging.info("\n=== 初始化程序 ===")
|
||
# ExitCursor()
|
||
|
||
logging.info("正在从API获取账号信息...")
|
||
success, account_data = get_account_from_api()
|
||
|
||
if not success:
|
||
logging.error("获取账号信息失败")
|
||
sys.exit(1)
|
||
|
||
email = account_data.get("email", "")
|
||
access_token = account_data.get("access_token", "")
|
||
refresh_token = account_data.get("refresh_token", "")
|
||
expire_time = account_data.get("expire_time", "")
|
||
days_left = account_data.get("days_left", 0)
|
||
|
||
if not all([email, access_token, refresh_token]):
|
||
logging.error("账号信息不完整")
|
||
sys.exit(1)
|
||
|
||
logging.info(f"获取到账号信息：\n邮箱: {email}\n到期时间: {expire_time}\n剩余天数: {days_left}")
|
||
|
||
# 更新认证信息
|
||
logging.info("正在更新认证信息...")
|
||
auth_manager = CursorAuthManager()
|
||
if auth_manager.update_auth(email, access_token, refresh_token):
|
||
logging.info("认证信息更新成功")
|
||
# 重置机器码
|
||
logging.info("正在重置机器码...")
|
||
reset_machine_id(greater_than_0_45)
|
||
logging.info("所有操作已完成")
|
||
print_end_message()
|
||
else:
|
||
logging.error("更新认证信息失败")
|
||
|
||
except Exception as e:
|
||
logging.error(f"程序执行出现错误: {str(e)}")
|
||
import traceback
|
||
logging.error(traceback.format_exc())
|
||
finally:
|
||
if browser_manager:
|
||
browser_manager.quit()
|
||
input("\n程序执行完毕，按回车键退出...")
|