855 协议版本-调整完毕内容

plugins/game_task/main.py

@@ -1,8 +1,9 @@
 import random
-import logging
 from datetime import datetime
 from typing import Dict, Any, List, Optional, Tuple
 
+from loguru import logger
+
 from message_util import MessageUtil
 from plugin_common.message_plugin_interface import MessagePluginInterface
 from plugin_common.plugin_interface import PluginStatus
@@ -12,6 +13,7 @@ from utils.decorator.points_decorator import points_reward_decorator
 from utils.ai.game_chatgpt_qa import game_question_json, game_answer_json
 from db.connection import DBConnectionManager
 from db.encyclopedia import EncyclopediaDB
+from wechat_ipad import WechatAPIClient
 
 
 class GameTaskPlugin(MessagePluginInterface):
@@ -43,10 +45,11 @@ class GameTaskPlugin(MessagePluginInterface):
 
     def __init__(self):
         super().__init__()
+        self.LOG = logger
+        self.bot: WechatAPIClient = None
 
     def initialize(self, context: Dict[str, Any]) -> bool:
         """初始化插件"""
-        self.LOG = logging.getLogger(f"Plugin.{self.name}")
         self.LOG.info(f"正在初始化 {self.name} 插件...")
 
         # 保存上下文对象
@@ -70,7 +73,6 @@ class GameTaskPlugin(MessagePluginInterface):
         # 初始化数据库连接
         self.db_manager = DBConnectionManager.get_instance()
         self.encyclopedia_db = EncyclopediaDB(self.db_manager)
-
         self.LOG.info(f"[{self.name}] 插件初始化完成，指令：{self._commands}")
         return True
 
@@ -106,7 +108,7 @@ class GameTaskPlugin(MessagePluginInterface):
             return 0
 
     @plugin_stats_decorator(plugin_name="百科问答")
-    def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
+    async def process_message(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
         """处理消息"""
         content = str(message.get("content", "")).strip()
         command = content.split(" ")[0].lower()
@@ -115,6 +117,7 @@ class GameTaskPlugin(MessagePluginInterface):
         gbm: GroupBotManager = message.get("gbm")
         all_contacts = message.get("all_contacts", {})
 
+        self.bot = message.get("bot")
         self.LOG.info(f"插件执行： {self.name}：{content}")
 
         # 检查权限
@@ -126,22 +129,22 @@ class GameTaskPlugin(MessagePluginInterface):
             wx_nick_name = all_contacts.get(sender, sender)
 
             if command == "/s":
-                self._handle_join_game(sender, roomid, wx_nick_name)
+                await self._handle_join_game(sender, roomid, wx_nick_name)
                 return True, "加入游戏成功"
             elif command == "/t":
-                self._handle_get_task(sender, roomid)
+                await self._handle_get_task(sender, roomid)
                 return True, "获取任务成功"
             elif command == "/a":
                 # 这里传递整个消息对象给处理方法
-                return self._handle_submit_answer(message)
+                return await self._handle_submit_answer(message)
             elif command == "/r":
-                self._handle_show_rank(sender, roomid)
+                await self._handle_show_rank(sender, roomid)
                 return True, "显示排行榜成功"
             elif command == "/l":
-                self._handle_show_active_tasks(sender, roomid)
+                await self._handle_show_active_tasks(sender, roomid)
                 return True, "显示活跃任务成功"
             elif command == "/h":
-                self._handle_list_uncompleted_tasks(sender, roomid)
+                await self._handle_list_uncompleted_tasks(sender, roomid)
                 return True, "列举未完成任务成功"
             else:
                 self.message_util.send_text(f"❌未知命令！\n{self.command_format}",
@@ -152,15 +155,15 @@ class GameTaskPlugin(MessagePluginInterface):
             self.LOG.error(f"处理消息出错: {e}")
             return False, f"处理出错: {e}"
 
-    def _handle_join_game(self, sender: str, roomid: str, wx_nick_name: str) -> None:
+    async def _handle_join_game(self, sender: str, roomid: str, wx_nick_name: str) -> None:
         """处理加入游戏请求"""
         try:
             # 检查并添加群聊
             if not self.encyclopedia_db.check_group_exists(roomid):
                 self.encyclopedia_db.add_group(roomid)
-                self.message_util.send_text(
-                    f"🎉 群 {roomid} 已就位，准备开燥！",
+                await self.bot.send_text_message(
                     (roomid if roomid else sender),
+                    f"🎉 群 {roomid} 已就位，准备开燥！",
                     sender
                 )
 
@@ -168,22 +171,22 @@ class GameTaskPlugin(MessagePluginInterface):
             player = self.encyclopedia_db.get_player(sender, roomid)
             if not player:
                 self.encyclopedia_db.add_player(sender, roomid, wx_nick_name)
-                self.message_util.send_text(
+                await self.bot.send_text_message(
+                    (roomid if roomid else sender),
                     f"🎉 哇塞，{wx_nick_name} 你来啦！\n"
                     f"🌟 群 {roomid} 瞬间燃爆！\n"
                     f"🎈 快来接任务，秀翻全场！指令： /t",
-                    (roomid if roomid else sender),
                     sender
                 )
         except Exception as e:
             self.LOG.error(f"加入游戏出错: {e}")
-            self.message_util.send_text(
-                f"😔 加入游戏出错，请稍后再试！",
+            await self.bot.send_text_message(
                 (roomid if roomid else sender),
+                f"😔 加入游戏出错，请稍后再试！",
                 sender
             )
 
-    def _handle_get_task(self, sender: str, roomid: str) -> None:
+    async def _handle_get_task(self, sender: str, roomid: str) -> None:
         """处理获取任务请求"""
         try:
             # 获取群内所有玩家
@@ -194,35 +197,35 @@ class GameTaskPlugin(MessagePluginInterface):
                 # 检查并添加群聊
                 if not self.encyclopedia_db.check_group_exists(roomid):
                     self.encyclopedia_db.add_group(roomid)
-                    self.message_util.send_text(
-                        f"🎉 群 {roomid} 已就位，准备开燥！",
+                    await self.bot.send_text_message(
                         (roomid if roomid else sender),
+                        f"🎉 群 {roomid} 已就位，准备开燥！",
                         sender
                     )
-                
+
                 # 获取用户昵称 (从all_contacts中获取可能不可行，因为这里没有all_contacts参数)
                 # 使用sender作为临时昵称
                 wx_nick_name = sender
-                
+
                 # 添加当前用户为玩家
                 self.encyclopedia_db.add_player(sender, roomid, wx_nick_name)
-                self.message_util.send_text(
+                await self.bot.send_text_message(
+                    (roomid if roomid else sender),
                     f"🎉 哇塞，{wx_nick_name} 你是第一个玩家！\n"
                     f"🌟 已自动为你加入游戏！\n"
                     f"🎈 现在就为你准备题目...",
-                    (roomid if roomid else sender),
                     sender
                 )
-                
+
                 # 更新玩家列表
                 players = self.encyclopedia_db.get_all_players_in_group(roomid)
-            
+
             player_dict = {p['player_id']: p['player_name'] for p in players}
             if sender not in player_dict:
-                self.message_util.send_text(
+                await self.bot.send_text_message(
+                    (roomid if roomid else sender),
                     f"😅 嘿，你谁啊？\n"
                     f"🌟 先用 /s 报名，不然没法玩哦！",
-                    (roomid if roomid else sender),
                     sender
                 )
                 return
@@ -241,32 +244,32 @@ class GameTaskPlugin(MessagePluginInterface):
             )
 
             if not active_task_id:
-                self.message_util.send_text(
-                    f"😔 任务创建失败，请稍后再试！",
+                await self.bot.send_text_message(
                     (roomid if roomid else sender),
+                    f"😔 任务创建失败，请稍后再试！",
                     sender
                 )
                 return
 
-            self.message_util.send_text(
+            await self.bot.send_text_message(
+                (roomid if roomid else sender),
                 f"🎁 {player_dict[sender]}，你的专属任务闪亮登场！\n"
                 f"🎀 任务ID: {active_task_id}\n"
                 f"🎈 问题：[{category}]{question}\n"
                 f"🌼 积分：{score}\n"
                 f"🌈 快上答案：/a {active_task_id} 答案",
-                (roomid if roomid else sender),
                 sender
             )
         except Exception as e:
             self.LOG.error(f"获取任务出错: {e}")
-            self.message_util.send_text(
-                f"😔 获取任务出错，请稍后再试！",
+            await self.bot.send_text_message(
                 (roomid if roomid else sender),
+                f"😔 获取任务出错，请稍后再试！",
                 sender
             )
 
     @points_reward_decorator(calculate_game_points, "game", "百科答题奖励", Feature.TASK_GAME)
-    def _handle_submit_answer(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
+    async def _handle_submit_answer(self, message: Dict[str, Any]) -> Tuple[bool, Optional[str]]:
         """处理提交答案请求"""
         try:
             content = str(message.get("content", "")).strip()
@@ -274,11 +277,11 @@ class GameTaskPlugin(MessagePluginInterface):
             roomid = message.get("roomid", "")
             parts = content.split(" ", 2)
             if len(parts) < 3:
-                self.message_util.send_text(
+                await self.bot.send_text_message(
+                    (roomid if roomid else sender),
                     f"😅 喂，格式不对啊！\n"
                     f"🌟 正确姿势：/a [任务ID] [答案]\n"
                     f"🎈 比如：/a 1 钒",
-                    (roomid if roomid else sender),
                     sender
                 )
                 return False, "0"
@@ -289,10 +292,10 @@ class GameTaskPlugin(MessagePluginInterface):
             # 获取玩家信息
             player = self.encyclopedia_db.get_player(sender, roomid)
             if not player:
-                self.message_util.send_text(
+                await self.bot.send_text_message(
+                    (roomid if roomid else sender),
                     f"😅 嘿，你是路人甲吗？\n"
                     f"🌟 用 /s 先加入群 {roomid} 吧！",
-                    (roomid if roomid else sender),
                     sender
                 )
                 return False, "0"
@@ -300,11 +303,11 @@ class GameTaskPlugin(MessagePluginInterface):
             player_name = player['player_name']
 
             if not task_id.isdigit():
-                self.message_util.send_text(
+                await self.bot.send_text_message(
+                    (roomid if roomid else sender),
                     f"😅 喂，任务ID得是数字好吗？\n"
                     f"🌟 比如：1\n"
                     f"🎈 别瞎搞，重新来！",
-                    (roomid if roomid else sender),
                     sender
                 )
                 return False, "0"
@@ -315,20 +318,20 @@ class GameTaskPlugin(MessagePluginInterface):
             task_data = self.encyclopedia_db.get_task_by_id(roomid, active_task_id)
 
             if not task_data:
-                self.message_util.send_text(
+                await self.bot.send_text_message(
+                    (roomid if roomid else sender),
                     f"😔 哎哟，任务 task_{active_task_id} 不翼而飞啦！\n"
                     f"🌼 可能被别人抢先一步咯！",
-                    (roomid if roomid else sender),
                     sender
                 )
                 return False, "0"
 
             if task_data['status'] == 'completed':
-                self.message_util.send_text(
+                await self.bot.send_text_message(
+                    (roomid if roomid else sender),
                     f"😄 哈哈，你慢了一步！\n"
                     f"🌟 任务 task_{active_task_id} 已经完结\n"
                     f"🎈 快去抢新任务吧！",
-                    (roomid if roomid else sender),
                     sender
                 )
                 return False, "0"
@@ -358,30 +361,31 @@ class GameTaskPlugin(MessagePluginInterface):
                 self.encyclopedia_db.complete_task(active_task_id)
 
                 if sender == holder_id:
-                    self.message_util.send_text(
+                    await self.bot.send_text_message(
+                        (roomid if roomid else sender),
                         f"🎉 {player_name} 你是天才吗？\n"
                         f"🌟 任务：{question}\n"
                         f"🎈 答对啦，简直无敌！\n"
                         f"🌈 奖励：{points} 分\n"
                         f"🎀 彩蛋：{description}",
-                        (roomid if roomid else sender),
                         sender
                     )
                 else:
-                    self.message_util.send_text(
+                    await self.bot.send_text_message(
+                        (roomid if roomid else sender),
                         f"🎉 {player_name} 抢答王上线！\n"
                         f"🌟 任务：{question}\n"
                         f"🎈 原主：{holder_name} 被你截胡啦！\n"
                         f"🌈 狂揽 {points} 分，太骚了！\n"
                         f"🎀 彩蛋：{description}",
-                        (roomid if roomid else sender),
                         sender
                     )
             else:
                 # 扣除积分
                 self.encyclopedia_db.update_player_points(sender, roomid, -1)
                 points = -1
-                self.message_util.send_text(
+                await self.bot.send_text_message(
+                    (roomid if roomid else sender),
                     f"😅 {player_name} 你这是要笑死我吗？\n"
                     f"🌼 任务：{question}\n"
                     f"🎈 你答：{answer}\n"
@@ -389,7 +393,6 @@ class GameTaskPlugin(MessagePluginInterface):
                     f"🌈 扣 1 分，别哭哦！\n"
                     f"🎀 提示：{description}\n"
                     f"🌟 任务ID: {active_task_id} 还能抢救一下！",
-                    (roomid if roomid else sender),
                     sender
                 )
 
@@ -398,17 +401,17 @@ class GameTaskPlugin(MessagePluginInterface):
             self.LOG.error(f"提交答案出错: {e}")
             return False, "0"
 
-    def _handle_show_rank(self, sender: str, roomid: str) -> None:
+    async def _handle_show_rank(self, sender: str, roomid: str) -> None:
         """处理显示排行榜请求"""
         try:
             # 获取排行榜
             ranks = self.encyclopedia_db.get_player_ranking(roomid, 10)
 
             if not ranks:
-                self.message_util.send_text(
+                await self.bot.send_text_message(
+                    (roomid if roomid else sender),
                     f"😔 群 {roomid} 冷冷清清\n"
                     f"🌟 快来一起燥起来吧！",
-                    (roomid if roomid else sender),
                     sender
                 )
                 return
@@ -424,23 +427,23 @@ class GameTaskPlugin(MessagePluginInterface):
             )
         except Exception as e:
             self.LOG.error(f"显示排行榜出错: {e}")
-            self.message_util.send_text(
-                f"😔 获取排行榜出错，请稍后再试！",
+            await self.bot.send_text_message(
                 (roomid if roomid else sender),
+                f"😔 获取排行榜出错，请稍后再试！",
                 sender
             )
 
-    def _handle_show_active_tasks(self, sender: str, roomid: str) -> None:
+    async def _handle_show_active_tasks(self, sender: str, roomid: str) -> None:
         """处理显示活跃任务请求"""
         try:
             # 获取活跃任务
             tasks = self.encyclopedia_db.get_active_tasks_in_group(roomid)
 
             if not tasks:
-                self.message_util.send_text(
+                await self.bot.send_text_message(
+                    (roomid if roomid else sender),
                     f"😄 群 {roomid} 现在一片祥和\n"
                     f"🌟 没任务？快用 /t 搞一个！",
-                    (roomid if roomid else sender),
                     sender
                 )
                 return
@@ -453,30 +456,30 @@ class GameTaskPlugin(MessagePluginInterface):
                     f"🌼 大佬：{task['player_name']}\n"
                 )
 
-            self.message_util.send_text(
-                task_text,
+            await self.bot.send_text_message(
                 (roomid if roomid else sender),
+                task_text,
                 sender
             )
         except Exception as e:
             self.LOG.error(f"显示活跃任务出错: {e}")
-            self.message_util.send_text(
-                f"😔 获取活跃任务出错，请稍后再试！",
+            await self.bot.send_text_message(
                 (roomid if roomid else sender),
+                f"😔 获取活跃任务出错，请稍后再试！",
                 sender
             )
 
-    def _handle_list_uncompleted_tasks(self, sender: str, roomid: str) -> None:
+    async def _handle_list_uncompleted_tasks(self, sender: str, roomid: str) -> None:
         """处理列举未完成任务请求"""
         try:
             # 获取未完成任务
             tasks = self.encyclopedia_db.get_active_tasks_in_group(roomid)
 
             if not tasks:
-                self.message_util.send_text(
+                await self.bot.send_text_message(
+                    (roomid if roomid else sender),
                     f"😄 群 {roomid} 全员开挂？\n"
                     f"🌟 没未完成任务，快用 /t 再战！",
-                    (roomid if roomid else sender),
                     sender
                 )
                 return
@@ -489,20 +492,20 @@ class GameTaskPlugin(MessagePluginInterface):
                     f"🌼 主人：{task['player_name']}\n"
                 )
 
-            self.message_util.send_text(
-                task_text,
+            await self.bot.send_text_message(
                 (roomid if roomid else sender),
+                task_text,
                 sender
             )
         except Exception as e:
             self.LOG.error(f"列举未完成任务出错: {e}")
-            self.message_util.send_text(
-                f"😔 获取未完成任务出错，请稍后再试！",
+            await self.bot.send_text_message(
                 (roomid if roomid else sender),
+                f"😔 获取未完成任务出错，请稍后再试！",
                 sender
             )
 
-    def run_random_task_assignment(self) -> None:
+    async def run_random_task_assignment(self) -> None:
         """定时任务：整点触发，排除23:00-08:00"""
         current_hour = datetime.now().hour
         if current_hour >= 23 or current_hour < 9:
@@ -538,14 +541,14 @@ class GameTaskPlugin(MessagePluginInterface):
                 )
 
                 if active_task_id:
-                    self.message_util.send_text(
+                    await self.bot.send_text_message(
+                        group_id,
                         f"🎁 新任务来袭，够不够刺激？\n"
                         f"🎀 任务ID: {active_task_id}\n"
                         f"🌟 幸运鹅：{holder_name}\n"
                         f"🎈 问题：[{category}]{question}\n"
                         f"🌼 积分：{score}\n"
-                        f"🌈 抢答格式：/a {active_task_id} 答案",
-                        group_id
+                        f"🌈 抢答格式：/a {active_task_id} 答案"
                     )
         except Exception as e:
             self.LOG.error(f"定时任务出错: {e}")