mirror of
https://github.com/yeongpin/cursor-free-vip.git
synced 2025-08-03 04:57:36 +08:00
469 lines
18 KiB
Python
469 lines
18 KiB
Python
import os
|
||
import sys
|
||
import json
|
||
import requests
|
||
import sqlite3
|
||
from typing import Dict, Optional
|
||
import platform
|
||
from colorama import Fore, Style, init
|
||
import logging
|
||
import re
|
||
|
||
# Initialize colorama
|
||
init()
|
||
|
||
# Setup logger
|
||
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# Define emoji constants
|
||
EMOJI = {
|
||
"USER": "👤",
|
||
"USAGE": "📊",
|
||
"PREMIUM": "⭐",
|
||
"BASIC": "📝",
|
||
"SUBSCRIPTION": "💳",
|
||
"INFO": "ℹ️",
|
||
"ERROR": "❌",
|
||
"SUCCESS": "✅",
|
||
"WARNING": "⚠️",
|
||
"TIME": "🕒"
|
||
}
|
||
|
||
class Config:
|
||
"""Config"""
|
||
NAME_LOWER = "cursor"
|
||
NAME_CAPITALIZE = "Cursor"
|
||
BASE_HEADERS = {
|
||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
|
||
"Accept": "application/json",
|
||
"Content-Type": "application/json"
|
||
}
|
||
|
||
class UsageManager:
|
||
"""Usage Manager"""
|
||
|
||
@staticmethod
|
||
def get_proxy():
|
||
"""get proxy"""
|
||
# from config import get_config
|
||
proxy = os.environ.get("HTTP_PROXY") or os.environ.get("HTTPS_PROXY")
|
||
if proxy:
|
||
return {"http": proxy, "https": proxy}
|
||
return None
|
||
|
||
@staticmethod
|
||
def get_usage(token: str) -> Optional[Dict]:
|
||
"""get usage"""
|
||
url = f"https://www.{Config.NAME_LOWER}.com/api/usage"
|
||
headers = Config.BASE_HEADERS.copy()
|
||
headers.update({"Cookie": f"Workos{Config.NAME_CAPITALIZE}SessionToken=user_01OOOOOOOOOOOOOOOOOOOOOOOO%3A%3A{token}"})
|
||
try:
|
||
proxies = UsageManager.get_proxy()
|
||
response = requests.get(url, headers=headers, timeout=10, proxies=proxies)
|
||
response.raise_for_status()
|
||
data = response.json()
|
||
|
||
# 获取 GPT-4 使用量和限制
|
||
gpt4_data = data.get("gpt-4", {})
|
||
premium_usage = gpt4_data.get("numRequestsTotal", 0)
|
||
max_premium_usage = gpt4_data.get("maxRequestUsage", 999)
|
||
|
||
# 获取 GPT-3.5 使用量，但将限制设为 "No Limit"
|
||
gpt35_data = data.get("gpt-3.5-turbo", {})
|
||
basic_usage = gpt35_data.get("numRequestsTotal", 0)
|
||
|
||
return {
|
||
'premium_usage': premium_usage,
|
||
'max_premium_usage': max_premium_usage,
|
||
'basic_usage': basic_usage,
|
||
'max_basic_usage': "No Limit" # 将 GPT-3.5 的限制设为 "No Limit"
|
||
}
|
||
except requests.RequestException as e:
|
||
logger.error(f"获取使用量失败: {str(e)}")
|
||
return None
|
||
|
||
@staticmethod
|
||
def get_stripe_profile(token: str) -> Optional[Dict]:
|
||
"""get user subscription info"""
|
||
url = f"https://api2.{Config.NAME_LOWER}.sh/auth/full_stripe_profile"
|
||
headers = Config.BASE_HEADERS.copy()
|
||
headers.update({"Authorization": f"Bearer {token}"})
|
||
try:
|
||
proxies = UsageManager.get_proxy()
|
||
response = requests.get(url, headers=headers, timeout=10, proxies=proxies)
|
||
response.raise_for_status()
|
||
return response.json()
|
||
except requests.RequestException as e:
|
||
logger.error(f"获取订阅信息失败: {str(e)}")
|
||
return None
|
||
|
||
def get_token_from_config():
|
||
"""get path info from config"""
|
||
try:
|
||
from config import get_config
|
||
config = get_config()
|
||
if not config:
|
||
return None
|
||
|
||
system = platform.system()
|
||
if system == "Windows" and config.has_section('WindowsPaths'):
|
||
return {
|
||
'storage_path': config.get('WindowsPaths', 'storage_path'),
|
||
'sqlite_path': config.get('WindowsPaths', 'sqlite_path'),
|
||
'session_path': os.path.join(os.getenv("APPDATA"), "Cursor", "Session Storage")
|
||
}
|
||
elif system == "Darwin" and config.has_section('MacPaths'): # macOS
|
||
return {
|
||
'storage_path': config.get('MacPaths', 'storage_path'),
|
||
'sqlite_path': config.get('MacPaths', 'sqlite_path'),
|
||
'session_path': os.path.expanduser("~/Library/Application Support/Cursor/Session Storage")
|
||
}
|
||
elif system == "Linux" and config.has_section('LinuxPaths'):
|
||
return {
|
||
'storage_path': config.get('LinuxPaths', 'storage_path'),
|
||
'sqlite_path': config.get('LinuxPaths', 'sqlite_path'),
|
||
'session_path': os.path.expanduser("~/.config/Cursor/Session Storage")
|
||
}
|
||
except Exception as e:
|
||
logger.error(f"获取配置路径失败: {str(e)}")
|
||
|
||
return None
|
||
|
||
def get_token_from_storage(storage_path):
|
||
"""get token from storage.json"""
|
||
if not os.path.exists(storage_path):
|
||
return None
|
||
|
||
try:
|
||
with open(storage_path, 'r', encoding='utf-8') as f:
|
||
data = json.load(f)
|
||
# try to get accessToken
|
||
if 'cursorAuth/accessToken' in data:
|
||
return data['cursorAuth/accessToken']
|
||
|
||
# try other possible keys
|
||
for key in data:
|
||
if 'token' in key.lower() and isinstance(data[key], str) and len(data[key]) > 20:
|
||
return data[key]
|
||
except Exception as e:
|
||
logger.error(f"get token from storage.json failed: {str(e)}")
|
||
|
||
return None
|
||
|
||
def get_token_from_sqlite(sqlite_path):
|
||
"""get token from sqlite"""
|
||
if not os.path.exists(sqlite_path):
|
||
return None
|
||
|
||
try:
|
||
conn = sqlite3.connect(sqlite_path)
|
||
cursor = conn.cursor()
|
||
cursor.execute("SELECT value FROM ItemTable WHERE key LIKE '%token%'")
|
||
rows = cursor.fetchall()
|
||
conn.close()
|
||
|
||
for row in rows:
|
||
try:
|
||
value = row[0]
|
||
if isinstance(value, str) and len(value) > 20:
|
||
return value
|
||
# try to parse JSON
|
||
data = json.loads(value)
|
||
if isinstance(data, dict) and 'token' in data:
|
||
return data['token']
|
||
except:
|
||
continue
|
||
except Exception as e:
|
||
logger.error(f"get token from sqlite failed: {str(e)}")
|
||
|
||
return None
|
||
|
||
def get_token_from_session(session_path):
|
||
"""get token from session"""
|
||
if not os.path.exists(session_path):
|
||
return None
|
||
|
||
try:
|
||
# try to find all possible session files
|
||
for file in os.listdir(session_path):
|
||
if file.endswith('.log'):
|
||
file_path = os.path.join(session_path, file)
|
||
try:
|
||
with open(file_path, 'rb') as f:
|
||
content = f.read().decode('utf-8', errors='ignore')
|
||
# find token pattern
|
||
token_match = re.search(r'"token":"([^"]+)"', content)
|
||
if token_match:
|
||
return token_match.group(1)
|
||
except:
|
||
continue
|
||
except Exception as e:
|
||
logger.error(f"get token from session failed: {str(e)}")
|
||
|
||
return None
|
||
|
||
def get_token():
|
||
"""get Cursor token"""
|
||
# get path from config
|
||
paths = get_token_from_config()
|
||
if not paths:
|
||
return None
|
||
|
||
# try to get token from different locations
|
||
token = get_token_from_storage(paths['storage_path'])
|
||
if token:
|
||
return token
|
||
|
||
token = get_token_from_sqlite(paths['sqlite_path'])
|
||
if token:
|
||
return token
|
||
|
||
token = get_token_from_session(paths['session_path'])
|
||
if token:
|
||
return token
|
||
|
||
return None
|
||
|
||
def format_subscription_type(subscription_data: Dict) -> str:
|
||
"""format subscription type"""
|
||
if not subscription_data:
|
||
return "Free"
|
||
|
||
# handle new API response format
|
||
if "membershipType" in subscription_data:
|
||
membership_type = subscription_data.get("membershipType", "").lower()
|
||
subscription_status = subscription_data.get("subscriptionStatus", "").lower()
|
||
|
||
if subscription_status == "active":
|
||
if membership_type == "pro":
|
||
return "Pro"
|
||
elif membership_type == "pro_trial":
|
||
return "Pro Trial"
|
||
elif membership_type == "team":
|
||
return "Team"
|
||
elif membership_type == "enterprise":
|
||
return "Enterprise"
|
||
elif membership_type:
|
||
return membership_type.capitalize()
|
||
else:
|
||
return "Active Subscription"
|
||
elif subscription_status:
|
||
return f"{membership_type.capitalize()} ({subscription_status})"
|
||
|
||
# compatible with old API response format
|
||
subscription = subscription_data.get("subscription")
|
||
if subscription:
|
||
plan = subscription.get("plan", {}).get("nickname", "Unknown")
|
||
status = subscription.get("status", "unknown")
|
||
|
||
if status == "active":
|
||
if "pro" in plan.lower():
|
||
return "Pro"
|
||
elif "pro_trial" in plan.lower():
|
||
return "Pro Trial"
|
||
elif "team" in plan.lower():
|
||
return "Team"
|
||
elif "enterprise" in plan.lower():
|
||
return "Enterprise"
|
||
else:
|
||
return plan
|
||
else:
|
||
return f"{plan} ({status})"
|
||
|
||
return "Free"
|
||
|
||
def get_email_from_storage(storage_path):
|
||
"""get email from storage.json"""
|
||
if not os.path.exists(storage_path):
|
||
return None
|
||
|
||
try:
|
||
with open(storage_path, 'r', encoding='utf-8') as f:
|
||
data = json.load(f)
|
||
# try to get email
|
||
if 'cursorAuth/cachedEmail' in data:
|
||
return data['cursorAuth/cachedEmail']
|
||
|
||
# try other possible keys
|
||
for key in data:
|
||
if 'email' in key.lower() and isinstance(data[key], str) and '@' in data[key]:
|
||
return data[key]
|
||
except Exception as e:
|
||
logger.error(f"get email from storage.json failed: {str(e)}")
|
||
|
||
return None
|
||
|
||
def get_email_from_sqlite(sqlite_path):
|
||
"""get email from sqlite"""
|
||
if not os.path.exists(sqlite_path):
|
||
return None
|
||
|
||
try:
|
||
conn = sqlite3.connect(sqlite_path)
|
||
cursor = conn.cursor()
|
||
# try to query records containing email
|
||
cursor.execute("SELECT value FROM ItemTable WHERE key LIKE '%email%' OR key LIKE '%cursorAuth%'")
|
||
rows = cursor.fetchall()
|
||
conn.close()
|
||
|
||
for row in rows:
|
||
try:
|
||
value = row[0]
|
||
# if it's a string and contains @, it might be an email
|
||
if isinstance(value, str) and '@' in value:
|
||
return value
|
||
|
||
# try to parse JSON
|
||
try:
|
||
data = json.loads(value)
|
||
if isinstance(data, dict):
|
||
# check if there's an email field
|
||
if 'email' in data:
|
||
return data['email']
|
||
# check if there's a cachedEmail field
|
||
if 'cachedEmail' in data:
|
||
return data['cachedEmail']
|
||
except:
|
||
pass
|
||
except:
|
||
continue
|
||
except Exception as e:
|
||
logger.error(f"get email from sqlite failed: {str(e)}")
|
||
|
||
return None
|
||
|
||
def display_account_info(translator=None):
|
||
"""display account info"""
|
||
print(f"\n{Fore.CYAN}{'─' * 40}{Style.RESET_ALL}")
|
||
print(f"{Fore.CYAN}{EMOJI['USER']} {translator.get('account_info.title') if translator else 'Cursor Account Information'}{Style.RESET_ALL}")
|
||
print(f"{Fore.CYAN}{'─' * 40}{Style.RESET_ALL}")
|
||
|
||
# get token
|
||
token = get_token()
|
||
if not token:
|
||
print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('account_info.token_not_found') if translator else 'Token not found. Please login to Cursor first.'}{Style.RESET_ALL}")
|
||
return
|
||
|
||
# get path info
|
||
paths = get_token_from_config()
|
||
if not paths:
|
||
print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('account_info.config_not_found') if translator else 'Configuration not found.'}{Style.RESET_ALL}")
|
||
return
|
||
|
||
# get email info - try multiple sources
|
||
email = get_email_from_storage(paths['storage_path'])
|
||
|
||
# if not found in storage, try from sqlite
|
||
if not email:
|
||
email = get_email_from_sqlite(paths['sqlite_path'])
|
||
|
||
# get subscription info
|
||
subscription_info = UsageManager.get_stripe_profile(token)
|
||
|
||
# if not found in storage and sqlite, try from subscription info
|
||
if not email and subscription_info:
|
||
# try to get email from subscription info
|
||
if 'customer' in subscription_info and 'email' in subscription_info['customer']:
|
||
email = subscription_info['customer']['email']
|
||
|
||
# get usage info
|
||
usage_info = UsageManager.get_usage(token)
|
||
|
||
# display account info
|
||
if email:
|
||
print(f"{Fore.GREEN}{EMOJI['USER']} {translator.get('account_info.email') if translator else 'Email'}: {Fore.WHITE}{email}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}{EMOJI['WARNING']} {translator.get('account_info.email_not_found') if translator else 'Email not found'}{Style.RESET_ALL}")
|
||
|
||
# display subscription type
|
||
if subscription_info:
|
||
subscription_type = format_subscription_type(subscription_info)
|
||
print(f"{Fore.GREEN}{EMOJI['SUBSCRIPTION']} {translator.get('account_info.subscription') if translator else 'Subscription'}: {Fore.WHITE}{subscription_type}{Style.RESET_ALL}")
|
||
|
||
# display remaining trial days
|
||
days_remaining = subscription_info.get("daysRemainingOnTrial")
|
||
if days_remaining is not None and days_remaining > 0:
|
||
print(f"{Fore.GREEN}{EMOJI['TIME']} {translator.get('account_info.trial_remaining') if translator else 'Remaining Pro Trial'}: {Fore.WHITE}{days_remaining} {translator.get('account_info.days') if translator else 'days'}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}{EMOJI['WARNING']} {translator.get('account_info.subscription_not_found') if translator else 'Subscription information not found'}{Style.RESET_ALL}")
|
||
|
||
# display usage info
|
||
if usage_info:
|
||
print(f"\n{Fore.GREEN}{EMOJI['USAGE']} {translator.get('account_info.usage') if translator else 'Usage Statistics'}:{Style.RESET_ALL}")
|
||
|
||
# GPT-4 usage
|
||
premium_usage = usage_info.get('premium_usage', 0)
|
||
max_premium_usage = usage_info.get('max_premium_usage', "No Limit")
|
||
|
||
# make sure the value is not None
|
||
if premium_usage is None:
|
||
premium_usage = 0
|
||
|
||
# handle "No Limit" case
|
||
if isinstance(max_premium_usage, str) and max_premium_usage == "No Limit":
|
||
premium_color = Fore.GREEN # when there is no limit, use green
|
||
premium_display = f"{premium_usage}/{max_premium_usage}"
|
||
else:
|
||
# calculate percentage when the value is a number
|
||
if max_premium_usage is None or max_premium_usage == 0:
|
||
max_premium_usage = 999
|
||
premium_percentage = 0
|
||
else:
|
||
premium_percentage = (premium_usage / max_premium_usage) * 100
|
||
|
||
# select color based on usage percentage
|
||
premium_color = Fore.GREEN
|
||
if premium_percentage > 70:
|
||
premium_color = Fore.YELLOW
|
||
if premium_percentage > 90:
|
||
premium_color = Fore.RED
|
||
|
||
premium_display = f"{premium_usage}/{max_premium_usage} ({premium_percentage:.1f}%)"
|
||
|
||
print(f"{Fore.YELLOW}{EMOJI['PREMIUM']} {translator.get('account_info.premium_usage') if translator else 'Fast Response'}: {premium_color}{premium_display}{Style.RESET_ALL}")
|
||
|
||
# Slow Response
|
||
basic_usage = usage_info.get('basic_usage', 0)
|
||
max_basic_usage = usage_info.get('max_basic_usage', "No Limit")
|
||
|
||
# make sure the value is not None
|
||
if basic_usage is None:
|
||
basic_usage = 0
|
||
|
||
# handle "No Limit" case
|
||
if isinstance(max_basic_usage, str) and max_basic_usage == "No Limit":
|
||
basic_color = Fore.GREEN # when there is no limit, use green
|
||
basic_display = f"{basic_usage}/{max_basic_usage}"
|
||
else:
|
||
# calculate percentage when the value is a number
|
||
if max_basic_usage is None or max_basic_usage == 0:
|
||
max_basic_usage = 999
|
||
basic_percentage = 0
|
||
else:
|
||
basic_percentage = (basic_usage / max_basic_usage) * 100
|
||
|
||
# select color based on usage percentage
|
||
basic_color = Fore.GREEN
|
||
if basic_percentage > 70:
|
||
basic_color = Fore.YELLOW
|
||
if basic_percentage > 90:
|
||
basic_color = Fore.RED
|
||
|
||
basic_display = f"{basic_usage}/{max_basic_usage} ({basic_percentage:.1f}%)"
|
||
|
||
print(f"{Fore.BLUE}{EMOJI['BASIC']} {translator.get('account_info.basic_usage') if translator else 'Slow Response'}: {basic_color}{basic_display}{Style.RESET_ALL}")
|
||
else:
|
||
print(f"{Fore.YELLOW}{EMOJI['WARNING']} {translator.get('account_info.usage_not_found') if translator else 'Usage information not found'}{Style.RESET_ALL}")
|
||
|
||
print(f"{Fore.CYAN}{'─' * 40}{Style.RESET_ALL}")
|
||
|
||
def main(translator=None):
|
||
"""main function"""
|
||
try:
|
||
display_account_info(translator)
|
||
except Exception as e:
|
||
print(f"{Fore.RED}{EMOJI['ERROR']} {translator.get('account_info.error') if translator else 'Error'}: {str(e)}{Style.RESET_ALL}")
|
||
|
||
if __name__ == "__main__":
|
||
main() |