WPSBot/games/gift.py

"""积分赠送系统游戏模块"""
import logging
from datetime import datetime
from games.base import BaseGame
from utils.parser import CommandParser
from core.database import get_db

logger = logging.getLogger(__name__)


class GiftGame(BaseGame):
    """积分赠送系统游戏"""
    
    def __init__(self):
        """初始化游戏"""
        super().__init__()
        self.db = get_db()
    
    async def handle(self, command: str, chat_id: int, user_id: int) -> str:
        """处理积分赠送相关指令
        
        Args:
            command: 指令，如 ".gift 123 50 生日快乐", ".gift sent", ".gift received"
            chat_id: 会话ID
            user_id: 用户ID
            
        Returns:
            回复消息
        """
        try:
            # 提取参数
            _, args = CommandParser.extract_command_args(command)
            args = args.strip()
            
            
            # 赠送帮助
            if args in ['help', '帮助']:
                return self._get_gift_help()
            
            # 默认：执行赠送
            else:
                return self._process_gift_command(args, user_id)
            
        except Exception as e:
            logger.error(f"处理积分赠送指令错误: {e}", exc_info=True)
            return f"❌ 处理指令出错: {str(e)}"
    
    def _process_gift_command(self, args: str, sender_id: int) -> str:
        """处理赠送指令
        
        Args:
            args: 指令参数
            sender_id: 发送者ID
            
        Returns:
            处理结果消息
        """
        # 解析参数：.gift <receiver_identifier> <points> [message]
        parts = args.split(maxsplit=2)
        
        if len(parts) < 2:
            return "❌ 指令格式错误！\n\n正确格式：`.gift <用户名或ID> <积分数量> [附赠消息]`\n\n示例：\n`.gift 张三 50 生日快乐`\n`.gift 123 50`\n`.gift 456 100`"
        
        # 解析积分数量
        try:
            points = int(parts[1])
            message = parts[2] if len(parts) > 2 else None
        except ValueError:
            return "❌ 积分数量必须是数字！"
        
        # 检查第一部分是用户名还是ID
        receiver_input = parts[0]
        
        if receiver_input.isdigit():
            # 是数字，作为用户ID处理
            receiver_id = int(receiver_input)
        else:
            # 是用户名，通过数据库查找
            user = self.db.get_user_by_name(receiver_input)
            if not user:
                return f"❌ 未找到用户: {receiver_input}\n\n请确认用户名是否正确，或使用用户ID。"
            receiver_id = user['user_id']
        
        # 获取接收者名称用于显示
        receiver_name = self.db.get_user_display_name(receiver_id)
        
        # 获取发送者名称用于显示
        sender_name = self.db.get_user_display_name(sender_id)
        
        # 验证参数
        if points <= 0:
            return "❌ 赠送积分数量必须大于0！"
        
        if sender_id == receiver_id:
            return "❌ 不能赠送积分给自己！"
        
        # 检查赠送者积分是否足够
        sender_points = self.db.get_user_points(sender_id)
        if sender_points['points'] < points:
            return f"❌ 积分不足！需要 {points} 积分，当前可用 {sender_points['points']} 积分"
        
        # 执行赠送（消费赠送者积分，增加接收者积分）
        if (self.db.consume_points(sender_id, points, "gift_send", f"赠送积分给用户{receiver_id}") and
            self.db.add_points(receiver_id, points, "gift_receive", f"收到用户{sender_id}的积分赠送")):
            
            # 获取更新后的积分信息
            sender_points_after = self.db.get_user_points(sender_id)
            receiver_points_after = self.db.get_user_points(receiver_id)
            
            text = f"## 🎁 积分赠送成功！\n\n"
            text += f"**赠送者**：{sender_name}\n\n"
            text += f"**接收者**：{receiver_name}\n\n"
            text += f"**赠送积分**：{points} 分\n\n"
            
            if message:
                text += f"**附赠消息**：{message}\n\n"
            
            text += f"**赠送者剩余积分**：{sender_points_after['points']} 分\n\n"
            text += f"**接收者当前积分**：{receiver_points_after['points']} 分\n\n"
            text += "---\n\n"
            text += "💝 感谢您的慷慨赠送！"
            
            return text
        else:
            return "❌ 赠送失败！请检查积分是否足够或用户ID是否正确。"
    
    def _get_gift_stats(self, user_id: int) -> str:
        """获取赠送统计信息
        
        Args:
            user_id: 用户ID
            
        Returns:
            统计信息消息
        """
        stats = self.db.get_gift_stats(user_id)
        
        if stats['sent_count'] == 0 and stats['received_count'] == 0:
            return "📊 你还没有任何积分赠送记录哦～"
        
        text = f"## 🎁 积分赠送统计\n\n"
        text += f"**发送统计**：\n"
        text += f"- 赠送次数：{stats['sent_count']} 次\n"
        text += f"- 赠送积分：{stats['total_sent']} 分\n\n"
        
        text += f"**接收统计**：\n"
        text += f"- 接收次数：{stats['received_count']} 次\n"
        text += f"- 接收积分：{stats['total_received']} 分\n\n"
        
        net_gift = stats['net_gift']
        if net_gift > 0:
            text += f"**净收益**：+{net_gift} 分 🎉\n\n"
        elif net_gift < 0:
            text += f"**净收益**：{net_gift} 分 😅\n\n"
        else:
            text += f"**净收益**：0 分 ⚖️\n\n"
        
        text += "---\n\n"
        text += "💡 提示：使用 `.gift sent` 查看发送记录，`.gift received` 查看接收记录"
        
        return text
    
    def _get_gift_records_sent(self, user_id: int, limit: int = 10) -> str:
        """获取发送的赠送记录
        
        Args:
            user_id: 用户ID
            limit: 限制数量
            
        Returns:
            记录信息消息
        """
        records = self.db.get_gift_records_sent(user_id, limit)
        
        if not records:
            return "📝 暂无发送记录"
        
        text = f"## 🎁 发送记录（最近 {len(records)} 条）\n\n"
        
        for record in records:
            timestamp = datetime.fromtimestamp(record['created_at']).strftime('%m-%d %H:%M')
            receiver_name = self.db.get_user_display_name(record['receiver_id'])
            points = record['points']
            message = record.get('message', '')
            
            text += f"**{timestamp}** 赠送 {points} 分给 {receiver_name}\n"
            if message:
                text += f"  💬 {message}\n"
            text += "\n"
        
        return text
    
    def _get_gift_records_received(self, user_id: int, limit: int = 10) -> str:
        """获取接收的赠送记录
        
        Args:
            user_id: 用户ID
            limit: 限制数量
            
        Returns:
            记录信息消息
        """
        records = self.db.get_gift_records_received(user_id, limit)
        
        if not records:
            return "📝 暂无接收记录"
        
        text = f"## 🎁 接收记录（最近 {len(records)} 条）\n\n"
        
        for record in records:
            timestamp = datetime.fromtimestamp(record['created_at']).strftime('%m-%d %H:%M')
            sender_name = self.db.get_user_display_name(record['sender_id'])
            points = record['points']
            message = record.get('message', '')
            
            text += f"**{timestamp}** 收到 {sender_name} 的 {points} 分\n"
            if message:
                text += f"  💬 {message}\n"
            text += "\n"
        
        return text
    
    def _get_gift_help(self) -> str:
        """获取赠送帮助信息
        
        Returns:
            帮助信息消息
        """
        text = f"## 🎁 积分赠送系统\n\n"
        text += f"### 基础用法\n"
        text += f"- `.gift <用户名或ID> <积分数量> [附赠消息]` - 赠送积分\n"
        text += f"- `.gift stats` - 查看赠送统计\n"
        text += f"- `.gift sent` - 查看发送记录\n"
        text += f"- `.gift received` - 查看接收记录\n"
        text += f"- `.gift help` - 查看帮助\n\n"
        
        text += f"### 赠送规则\n"
        text += f"- **积分限制**：单次最多赠送1000积分\n"
        text += f"- **自赠限制**：不能赠送给自己\n"
        text += f"- **积分检查**：必须有足够积分才能赠送\n"
        text += f"- **附赠消息**：可选，最多100字符\n\n"
        
        text += f"### 示例\n"
        text += f"```\n"
        text += f".gift 张三 50 生日快乐\n"
        text += f".gift 123 50 (使用用户ID)\n"
        text += f".gift 李四 100 感谢你的帮助\n"
        text += f".gift 王五 200\n"
        text += f".gift stats\n"
        text += f"```\n\n"
        
        text += f"### 说明\n"
        text += f"- 支持使用用户名或用户ID进行赠送\n"
        text += f"- 使用用户名需要先通过 `.register` 注册名称\n"
        text += f"- 所有赠送都有完整记录\n"
        text += f"- 赠送和接收都会在积分记录中显示\n"
        text += f"- 赠送是单向的，无需对方确认\n"
        
        return text
    
    def get_help(self) -> str:
        """获取帮助信息"""
        return self._get_gift_help()