feat: 添加Yggdrasil密码重置功能，更新依赖和配置

2025-11-30 18:56:56 +08:00
parent a4b6c5011e
commit 4188ee1555
18 changed files with 683 additions and 95 deletions
--- a/scripts/generate_test_account.py
+++ b/scripts/generate_test_account.py
@@ -0,0 +1,293 @@
+#!/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()
+