新增冒险系统

games/adventure.py

@@ -0,0 +1,218 @@
+"""冒险系统游戏模块"""
+import random
+import time
+import logging
+from datetime import datetime
+from games.base import BaseGame
+from utils.parser import CommandParser
+from core.database import get_db
+from typing import *
+
+logger = logging.getLogger(__name__)
+
+
+class AdventureGame(BaseGame):
+    """冒险系统游戏"""
+    
+    def __init__(self):
+        """初始化游戏"""
+        super().__init__()
+        self.db = get_db()
+        
+        # 奖品池配置
+        self.prize_pool: List[Tuple[int, float, str]] = [
+            # (权重, 倍率, 描述) 
+            (500, 0.5, "少量积分"),       
+            (350, 1, "中等积分"),        
+            (200, 2, "大量积分"),      
+            (100, 5, "丰厚积分"),      
+            (50, 10, "丰厚积分"),    
+            (10, 100, "🌟 巨额积分"),   
+            (1, 1000, "💎 传说积分"),  
+        ]
+
+        self.total_weight: int = 0
+        for weight,_ in self.prize_pool:
+            self.total_weight += weight
+    
+    async def handle(self, command: str, chat_id: int, user_id: int) -> str:
+        """处理冒险相关指令
+        
+        Args:
+            command: 指令，如 ".adventure", ".adventure 10", ".adventure stats"
+            chat_id: 会话ID
+            user_id: 用户ID
+            
+        Returns:
+            回复消息
+        """
+        try:
+            # 提取参数
+            _, args = CommandParser.extract_command_args(command)
+            args = args.strip().lower()
+            
+            
+            # 冒险说明
+            if args in ['help', '帮助', 'info']:
+                return self._get_adventure_help()
+            
+            # 默认：冒险耗时1分钟
+            else:
+                # 解析消耗时间
+                cost_time = 1  # 默认消耗1分钟
+                if args.isdigit():
+                    cost_time = int(args)
+                
+                return await self._perform_adventure(chat_id, user_id, cost_time)
+            
+        except Exception as e:
+            logger.error(f"处理冒险指令错误: {e}", exc_info=True)
+            return f"❌ 处理指令出错: {str(e)}"
+    
+    async def _perform_adventure(self, chat_id: int, user_id: int, cost_time: int) -> str:
+        """执行冒险耗时
+        
+        Args:
+            chat_id: 会话ID（使用0作为用户级标识）
+            user_id: 用户ID
+            cost_time: 消耗时间（分钟）
+            
+        Returns:
+            抽奖结果消息
+        """
+        # 参数验证
+        if cost_time < 1:
+            return "❌ 冒险时间至少需要1分钟！"
+        
+        # 查询冒险状态（使用chat_id=0表示用户级状态）
+        state = self.db.get_game_state(0, user_id, 'adventure')
+        current_time = int(time.time())
+        
+        # 情况1：用户已有未完成的冒险
+        if state:
+            try:
+                state_data = state['state_data']
+                start_time = state_data.get('start_time', 0)
+                saved_cost_time = state_data.get('cost_time', 1)
+                end_time = start_time + saved_cost_time * 60
+                remaining_seconds = end_time - current_time
+                
+                # 情况1.1：冒险已完成（时间已到或过期）
+                if remaining_seconds <= 0:
+                    # 选择奖品池
+                    prize_pool = self.prize_pool
+                    
+                    # 执行抽奖
+                    reward = self._draw_prize(prize_pool)
+                    reward_points = int(reward['value'] * saved_cost_time)
+                    
+                    # 处理奖励
+                    self.db.add_points(user_id, reward_points, "adventure", f"冒险奖励")
+                    
+                    # 删除冒险状态
+                    self.db.delete_game_state(0, user_id, 'adventure')
+                    
+                    # 获取更新后的积分信息
+                    updated_points = self.db.get_user_points(user_id)
+                    
+                    # 格式化输出
+                    text = f"## ⚡️ 冒险结果\n\n"
+                    text += f"**消耗时间**: {saved_cost_time} 分钟\n\n"
+                    text += f"**{reward['description']}**: 获得 {reward_points} 积分\n\n"
+                    text += f"**当前积分**: {updated_points['points']} 分\n\n"
+                    text += "---\n\n"
+                    text += "💡 提示：冒险进行时不能炼金！"
+                    
+                    return text
+                
+                # 情况1.2：冒险未完成，返回等待提示
+                remaining_minutes = remaining_seconds // 60
+                remaining_secs = remaining_seconds % 60
+                
+                if remaining_minutes > 0:
+                    wait_msg = f"{remaining_minutes} 分 {remaining_secs} 秒"
+                else:
+                    wait_msg = f"{remaining_secs} 秒"
+                
+                text = f"## ⚡️ 冒险进行中\n\n"
+                text += f"你正在进行一次冒险，还需等待 **{wait_msg}** 才能完成。\n\n"
+                text += f"**当前冒险时长**: {saved_cost_time} 分钟\n\n"
+                text += "---\n\n"
+                text += "💡 提示：冒险期间无法进行炼金，请耐心等待！"
+                
+                return text
+                
+            except Exception as e:
+                # 状态数据异常，删除损坏状态并允许重新开始
+                logger.error(f"冒险状态数据异常: {e}", exc_info=True)
+                self.db.delete_game_state(0, user_id, 'adventure')
+                # 继续执行到情况3（开始新冒险）
+        
+        # 情况3：用户没有冒险状态，开始新的冒险
+        state_data = {
+            'start_time': current_time,
+            'cost_time': cost_time
+        }
+        
+        # 保存冒险状态
+        self.db.save_game_state(0, user_id, 'adventure', state_data)
+        
+        # 计算预计完成时间
+        end_time = current_time + cost_time * 60
+        end_datetime = datetime.fromtimestamp(end_time)
+        end_time_str = end_datetime.strftime('%H:%M:%S')
+        
+        text = f"## ⚡️ 冒险开始\n\n"
+        text += f"你已经开始了冒险之旅，本次冒险将持续 **{cost_time}** 分钟。\n\n"
+        text += f"**预计完成时间**: {end_time_str}\n\n"
+        text += "---\n\n"
+        text += "💡 提示：冒险期间无法进行炼金，完成后使用 `.adventure` 获取奖励！"
+        
+        return text
+    
+    def _draw_prize(self, prize_pool: list) -> dict:
+        """从奖品池中抽取奖品
+        
+        Args:
+            prize_pool: 奖品池，格式为 (权重, 倍率, 描述)
+            
+        Returns:
+            奖品信息，格式为 {'value': 倍率, 'description': 描述}
+        """
+        # 生成随机数
+        rand = random.random()*self.total_weight
+        cumulative_prob = 0.0
+        
+        for weight, multiplier, description in prize_pool:
+            cumulative_prob += weight
+            if rand <= cumulative_prob:
+                return {
+                    'value': multiplier,
+                    'description': description
+                }
+        
+        # 兜底返回第一个奖品
+        return {
+            'value': prize_pool[0][1],
+            'description': prize_pool[0][2]
+        }
+    
+    def _get_adventure_help(self) -> str:
+        """获取冒险帮助信息
+        
+        Returns:
+            帮助信息消息
+        """
+        text = f"## ⚡️ 冒险系统\n\n"
+        text += f"### 基础用法\n"
+        text += f"- `.adventure` - 消耗1分钟进行冒险\n"
+        text += f"- `.adventure time` - 消耗time分钟进行冒险, 最少一分钟\n"
+        
+        text += f"### 其他功能\n"
+        text += f"- `.adventure help` - 查看帮助\n\n"
+        
+        return text
+    
+    def get_help(self) -> str:
+        """获取帮助信息"""
+        return self._get_adventure_help()