首次提交

2025-12-26 22:41:42 +08:00
commit 4a944316fe
143 changed files with 17550 additions and 0 deletions
--- a/danding_bot/plugins/onmyoji_gacha/api_utils.py
+++ b/danding_bot/plugins/onmyoji_gacha/api_utils.py
@@ -0,0 +1,216 @@
+import requests
+import json
+from typing import Dict, Optional, Tuple
+from nonebot import logger
+from .config import Config
+
+def mask_username(username: str) -> str:
+    """
+    对用户名进行脱敏处理，只显示前两位和后两位，中间用*号隐藏
+    
+    Args:
+        username: 原始用户名
+        
+    Returns:
+        脱敏后的用户名
+    """
+    if not username:
+        return username
+    
+    # 如果用户名长度小于等于4，直接显示前两位和后两位（可能重叠）
+    if len(username) <= 4:
+        return username
+    
+    # 显示前两位和后两位，中间用*号填充
+    return f"{username[:2]}{'*' * (len(username) - 4)}{username[-2:]}"
+
+# 获取配置
+config = Config()
+
+# API 端点配置
+DD_API_HOST = "https://api.danding.vip/DD/"  # 蛋定服务器连接地址
+BOT_TOKEN = "3340e353a49447f1be640543cbdcd937"  # 对接服务器的Token
+BOT_USER_ID = "1424473282"  # 机器人用户ID
+
+async def query_qq_binding(qq: str) -> Tuple[bool, Optional[str], Optional[str]]:
+    """
+    查询QQ号是否绑定了蛋定用户名
+    
+    Args:
+        qq: 要查询的QQ号
+        
+    Returns:
+        Tuple[是否绑定, 用户名, VIP到期时间]
+    """
+    try:
+        url = f"{DD_API_HOST}query_qq_binding"
+        data = {"qq": qq}
+        
+        response = requests.post(url=url, json=data)
+        logger.debug(f"查询QQ绑定状态响应: {response}")
+        
+        if response.status_code != 200:
+            logger.error(f"查询QQ绑定状态失败，状态码: {response.status_code}")
+            return False, None, None
+            
+        result = response.json()
+        logger.debug(f"查询QQ绑定状态结果: {result}")
+        
+        if result.get("code") == 200:
+            data = result.get("data", {})
+            is_bound = data.get("is_bound", False)
+            
+            if is_bound:
+                username = data.get("username")
+                vip_time = data.get("vip_time")
+                return True, username, vip_time
+            else:
+                return False, None, None
+        else:
+            logger.error(f"查询QQ绑定状态失败，错误信息: {result.get('message')}")
+            return False, None, None
+            
+    except Exception as e:
+        logger.error(f"查询QQ绑定状态异常: {str(e)}")
+        return False, None, None
+
+async def add_user_viptime(username: str, time_class: str = "Day") -> Tuple[bool, str]:
+    """
+    为用户添加VIP时间
+    
+    Args:
+        username: 蛋定用户名
+        time_class: 时间类型 (Hour/Day/Week/Month/Season/Year)
+        
+    Returns:
+        Tuple[是否成功, 响应消息]
+    """
+    try:
+        url = f"{DD_API_HOST}bot_add_user_viptime"
+        data = {
+            "user": BOT_USER_ID,
+            "token": BOT_TOKEN,
+            "username": username,
+            "classes": time_class
+        }
+        
+        response = requests.post(url=url, json=data)
+        logger.debug(f"添加VIP时间响应: {response}")
+        
+        if response.status_code != 200:
+            error_msg = f"添加VIP时间失败，状态码: {response.status_code}"
+            logger.error(error_msg)
+            return False, error_msg
+            
+        result = response.json()
+        logger.debug(f"添加VIP时间结果: {result}")
+        
+        if result.get("code") == 200:
+            return True, result.get("msg", "添加VIP时间成功")
+        else:
+            error_msg = result.get("msg", "添加VIP时间失败")
+            logger.error(f"添加VIP时间失败: {error_msg}")
+            return False, error_msg
+            
+    except Exception as e:
+        error_msg = f"添加VIP时间异常: {str(e)}"
+        logger.error(error_msg)
+        return False, error_msg
+
+async def process_ssr_sp_reward(user_id: str) -> Tuple[bool, str]:
+    """
+    处理SSR/SP奖励发放
+    
+    Args:
+        user_id: QQ用户ID
+        
+    Returns:
+        Tuple[是否自动发放成功, 消息内容]
+    """
+    # 查询QQ绑定状态
+    is_bound, username, vip_time = await query_qq_binding(user_id)
+    
+    if not is_bound:
+        # 用户未绑定，返回提示信息
+        msg = (f"🎉恭喜您抽中了SSR/SP稀有度式神！🎉\n"
+               f"获得奖励：蛋定助手天卡一张\n"
+               f"获取奖励请联系管理员，或前往蛋定云服务中绑定QQ号即可体验自动加时！")
+        return False, msg
+    else:
+        # 用户已绑定，自动加时
+        success, message = await add_user_viptime(username, "Day")
+        
+        if success:
+            masked_username = mask_username(username)
+            msg = (f"🎉恭喜您抽中了SSR/SP稀有度式神！🎉\n"
+                   f"🎁已自动为您的蛋定账号({masked_username})添加天卡时长！\n"
+                   f"📅到期时间: {message.split('到期时间为:')[-1] if '到期时间为:' in message else '请查看您的账号详情'}")
+            return True, msg
+        else:
+            # 自动加时失败，返回错误信息和手动领取提示
+            msg = (f"🎉恭喜您抽中了SSR/SP稀有度式神！🎉\n"
+                   f"获得奖励：蛋定助手天卡一张\n"
+                   f"⚠️自动加时失败: {message}\n"
+                   f"请联系管理员手动领取奖励！")
+            return False, msg
+
+async def process_achievement_reward(user_id: str, achievement_id: str) -> Tuple[bool, str]:
+    """
+    处理成就奖励发放
+    
+    Args:
+        user_id: QQ用户ID
+        achievement_id: 成就ID
+        
+    Returns:
+        Tuple[是否自动发放成功, 消息内容]
+    """
+    # 获取成就配置
+    achievement_config = config.ACHIEVEMENTS.get(achievement_id)
+    if not achievement_config:
+        # 检查是否是重复奖励
+        if "_repeat_" in achievement_id:
+            base_achievement_id = achievement_id.split("_repeat_")[0]
+            base_config = config.ACHIEVEMENTS.get(base_achievement_id)
+            if base_config:
+                reward_type = base_config.get("repeat_reward", "天卡")
+            else:
+                reward_type = "天卡"
+        else:
+            return False, f"未找到成就配置: {achievement_id}"
+    else:
+        reward_type = achievement_config.get("reward", "天卡")
+    
+    # 查询QQ绑定状态
+    is_bound, username, vip_time = await query_qq_binding(user_id)
+    
+    if not is_bound:
+        # 用户未绑定，返回提示信息
+        msg = (f"🏆 恭喜解锁成就奖励！\n"
+               f"获得奖励：蛋定助手{reward_type}一张\n"
+               f"获取奖励请联系管理员，或前往蛋定云服务中绑定QQ号即可体验自动加时！")
+        return False, msg
+    else:
+        # 用户已绑定，自动加时
+        # 将奖励类型转换为API需要的时间类型
+        time_class = "Day"  # 默认为天卡
+        if "周卡" in reward_type:
+            time_class = "Week"
+        elif "月卡" in reward_type:
+            time_class = "Month"
+        
+        success, message = await add_user_viptime(username, time_class)
+        
+        if success:
+            masked_username = mask_username(username)
+            msg = (f"🏆 恭喜解锁成就奖励！\n"
+                   f"🎁已自动为您的蛋定账号({masked_username})添加{reward_type}时长！\n"
+                   f"📅到期时间: {message.split('到期时间为:')[-1] if '到期时间为:' in message else '请查看您的账号详情'}")
+            return True, msg
+        else:
+            # 自动加时失败，返回错误信息和手动领取提示
+            msg = (f"🏆 恭喜解锁成就奖励！\n"
+                   f"获得奖励：蛋定助手{reward_type}一张\n"
+                   f"⚠️自动加时失败: {message}\n"
+                   f"请联系管理员手动领取奖励！")
+            return False, msg