294 lines
8.2 KiB
Python
294 lines
8.2 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
CarrotSkin 测试账户生成器
|
||
此脚本创建新用户、角色，并输出所有测试信息
|
||
"""
|
||
|
||
import json
|
||
import random
|
||
import sys
|
||
from datetime import datetime
|
||
from pathlib import Path
|
||
|
||
try:
|
||
import requests
|
||
except ImportError:
|
||
print("错误: 需要安装 requests 库")
|
||
print("请运行: pip install requests")
|
||
sys.exit(1)
|
||
|
||
# 颜色输出支持（Windows 和 Unix 都支持）
|
||
try:
|
||
from colorama import init, Fore, Style
|
||
init(autoreset=True)
|
||
HAS_COLORAMA = True
|
||
# colorama 没有 GRAY，使用 WHITE 代替
|
||
Fore.GRAY = Fore.WHITE
|
||
except ImportError:
|
||
# 如果没有 colorama，使用空字符串
|
||
class Fore:
|
||
CYAN = YELLOW = GREEN = RED = GRAY = WHITE = RESET = ""
|
||
class Style:
|
||
RESET_ALL = ""
|
||
HAS_COLORAMA = False
|
||
|
||
# API 基础 URL
|
||
BASE_URL = "http://localhost:8080/api/v1"
|
||
|
||
|
||
def print_colored(text, color=Fore.RESET):
|
||
"""打印彩色文本"""
|
||
print(f"{color}{text}{Style.RESET_ALL}")
|
||
|
||
|
||
def print_header(text):
|
||
"""打印标题"""
|
||
print_colored(f"\n{'=' * 40}", Fore.CYAN)
|
||
print_colored(f" {text}", Fore.CYAN)
|
||
print_colored(f"{'=' * 40}\n", Fore.CYAN)
|
||
|
||
|
||
def print_step(text):
|
||
"""打印步骤标题"""
|
||
print_colored(f"\n=== {text} ===", Fore.YELLOW)
|
||
|
||
|
||
def print_success(text):
|
||
"""打印成功消息"""
|
||
print_colored(f"✓ {text}", Fore.GREEN)
|
||
|
||
|
||
def print_error(text):
|
||
"""打印错误消息"""
|
||
print_colored(f"✗ {text}", Fore.RED)
|
||
|
||
|
||
def print_info(text):
|
||
"""打印信息消息"""
|
||
print_colored(f" {text}", Fore.GRAY)
|
||
|
||
|
||
def register_user():
|
||
"""步骤1: 注册新用户"""
|
||
print_step("步骤1: 注册新用户")
|
||
|
||
random_num = random.randint(10000, 99999)
|
||
username = f"testuser{random_num}"
|
||
email = f"test{random_num}@example.com"
|
||
login_password = "password123"
|
||
verification_code = "123456"
|
||
|
||
print_info(f"用户名: {username}")
|
||
print_info(f"邮箱: {email}")
|
||
print_info(f"密码: {login_password}")
|
||
|
||
register_data = {
|
||
"username": username,
|
||
"email": email,
|
||
"password": login_password,
|
||
"verification_code": verification_code
|
||
}
|
||
|
||
try:
|
||
response = requests.post(
|
||
f"{BASE_URL}/auth/register",
|
||
json=register_data,
|
||
headers={"Content-Type": "application/json"},
|
||
timeout=10
|
||
)
|
||
response.raise_for_status()
|
||
result = response.json()
|
||
|
||
if result.get("code") == 200:
|
||
jwt_token = result["data"]["token"]
|
||
user_id = result["data"]["user_info"]["id"]
|
||
user_role = result["data"]["user_info"]["role"]
|
||
|
||
print_success("注册成功!")
|
||
print_info(f"用户ID: {user_id}")
|
||
print_info(f"角色: {user_role}")
|
||
|
||
return {
|
||
"username": username,
|
||
"email": email,
|
||
"login_password": login_password,
|
||
"jwt_token": jwt_token,
|
||
"user_id": user_id,
|
||
"user_role": user_role
|
||
}
|
||
else:
|
||
print_error(f"注册失败: {result.get('message', '未知错误')}")
|
||
sys.exit(1)
|
||
except requests.exceptions.RequestException as e:
|
||
print_error(f"注册失败: {str(e)}")
|
||
sys.exit(1)
|
||
|
||
|
||
def create_profile(jwt_token):
|
||
"""步骤2: 创建Minecraft角色"""
|
||
print_step("步骤2: 创建Minecraft角色")
|
||
|
||
profile_name = f"TestPlayer{random.randint(100, 999)}"
|
||
print_info(f"角色名: {profile_name}")
|
||
|
||
profile_data = {
|
||
"name": profile_name
|
||
}
|
||
|
||
headers = {
|
||
"Authorization": f"Bearer {jwt_token}",
|
||
"Content-Type": "application/json"
|
||
}
|
||
|
||
try:
|
||
response = requests.post(
|
||
f"{BASE_URL}/profile/",
|
||
json=profile_data,
|
||
headers=headers,
|
||
timeout=10
|
||
)
|
||
response.raise_for_status()
|
||
result = response.json()
|
||
|
||
if result.get("code") == 200:
|
||
profile_uuid = result["data"]["uuid"]
|
||
print_success("角色创建成功!")
|
||
print_info(f"角色UUID: {profile_uuid}")
|
||
print_info(f"角色名: {result['data']['name']}")
|
||
|
||
return {
|
||
"profile_name": profile_name,
|
||
"profile_uuid": profile_uuid
|
||
}
|
||
else:
|
||
print_error(f"角色创建失败: {result.get('message', '未知错误')}")
|
||
sys.exit(1)
|
||
except requests.exceptions.RequestException as e:
|
||
print_error(f"角色创建失败: {str(e)}")
|
||
sys.exit(1)
|
||
|
||
|
||
def reset_yggdrasil_password(jwt_token):
|
||
"""步骤3: 重置并获取Yggdrasil密码"""
|
||
print_step("步骤3: 重置并获取Yggdrasil密码")
|
||
|
||
headers = {
|
||
"Authorization": f"Bearer {jwt_token}",
|
||
"Content-Type": "application/json"
|
||
}
|
||
|
||
try:
|
||
response = requests.post(
|
||
f"{BASE_URL}/user/yggdrasil-password/reset",
|
||
headers=headers,
|
||
timeout=10
|
||
)
|
||
response.raise_for_status()
|
||
result = response.json()
|
||
|
||
if result.get("code") == 200:
|
||
yggdrasil_password = result["data"]["password"]
|
||
print_success("Yggdrasil密码重置成功!")
|
||
print_info(f"Yggdrasil密码: {yggdrasil_password}")
|
||
return yggdrasil_password
|
||
else:
|
||
print_error(f"Yggdrasil密码重置失败: {result.get('message', '未知错误')}")
|
||
return ""
|
||
except requests.exceptions.RequestException as e:
|
||
print_error(f"Yggdrasil密码重置失败: {str(e)}")
|
||
return ""
|
||
|
||
|
||
def generate_output(user_info, profile_info, yggdrasil_password):
|
||
"""生成输出信息"""
|
||
output = f"""========================================
|
||
CarrotSkin 测试账户信息
|
||
========================================
|
||
|
||
=== 账户信息 ===
|
||
用户名: {user_info['username']}
|
||
邮箱: {user_info['email']}
|
||
登录密码: {user_info['login_password']}
|
||
用户ID: {user_info['user_id']}
|
||
|
||
=== JWT Token (API认证) ===
|
||
Token: {user_info['jwt_token']}
|
||
|
||
=== 角色信息 ===
|
||
角色名: {profile_info['profile_name']}
|
||
角色UUID: {profile_info['profile_uuid']}
|
||
|
||
=== Yggdrasil信息 ===
|
||
Yggdrasil密码: {yggdrasil_password}
|
||
|
||
=== 测试命令 ===
|
||
|
||
# 1. API登录
|
||
curl -X POST http://localhost:8080/api/v1/auth/login \\
|
||
-H 'Content-Type: application/json' \\
|
||
-d '{{"username":"{user_info['username']}","password":"{user_info['login_password']}"}}'
|
||
|
||
# 2. 创建角色
|
||
curl -X POST http://localhost:8080/api/v1/profile/ \\
|
||
-H 'Content-Type: application/json' \\
|
||
-H 'Authorization: Bearer {user_info['jwt_token']}' \\
|
||
-d '{{"name":"NewProfile"}}'
|
||
|
||
# 3. 重置Yggdrasil密码
|
||
curl -X POST http://localhost:8080/api/v1/user/yggdrasil-password/reset \\
|
||
-H 'Content-Type: application/json' \\
|
||
-H 'Authorization: Bearer {user_info['jwt_token']}'
|
||
|
||
# 4. Yggdrasil认证
|
||
curl -X POST http://localhost:8080/api/v1/yggdrasil/authserver/authenticate \\
|
||
-H 'Content-Type: application/json' \\
|
||
-d '{{
|
||
"username": "{user_info['username']}",
|
||
"password": "{yggdrasil_password}",
|
||
"requestUser": true
|
||
}}'
|
||
|
||
# 5. 获取角色信息
|
||
curl -X GET http://localhost:8080/api/v1/profile/{profile_info['profile_uuid']}
|
||
|
||
========================================
|
||
"""
|
||
return output
|
||
|
||
|
||
def main():
|
||
"""主函数"""
|
||
print_header("CarrotSkin 测试账户生成器")
|
||
|
||
# 步骤1: 注册用户
|
||
user_info = register_user()
|
||
|
||
# 步骤2: 创建角色
|
||
profile_info = create_profile(user_info["jwt_token"])
|
||
|
||
# 步骤3: 重置Yggdrasil密码
|
||
yggdrasil_password = reset_yggdrasil_password(user_info["jwt_token"])
|
||
|
||
# 步骤4: 输出所有信息
|
||
print_header("测试账户信息汇总")
|
||
|
||
output = generate_output(user_info, profile_info, yggdrasil_password)
|
||
print(output)
|
||
|
||
# 保存到文件
|
||
output_file = f"test_account_{user_info['username']}.txt"
|
||
try:
|
||
with open(output_file, "w", encoding="utf-8") as f:
|
||
f.write(output)
|
||
print_success(f"信息已保存到文件: {output_file}")
|
||
except Exception as e:
|
||
print_error(f"保存文件失败: {str(e)}")
|
||
|
||
print_header("测试完成!")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|
||
|