"""Callback路由处理""" import logging from fastapi import APIRouter, Request from fastapi.responses import JSONResponse from core.models import CallbackRequest from core.database import get_db from utils.message import get_message_sender from utils.parser import CommandParser from utils.rate_limit import get_rate_limiter logger = logging.getLogger(__name__) router = APIRouter() @router.get("/callback") async def callback_verify(): """Callback可用性校验 - WPS会发送GET请求验证""" logger.info("收到Callback验证请求") return JSONResponse({"result": "ok"}) @router.post("/callback") async def callback_receive(request: Request): """接收WPS Callback消息""" try: # 解析请求数据 data = await request.json() logger.info(f"收到消息: chatid={data.get('chatid')}, creator={data.get('creator')}") logger.info(f"消息内容: {data.get('content')}") logger.info(f"完整callback数据: {data}") # 验证请求 try: callback_data = CallbackRequest(**data) except Exception as e: logger.error(f"请求数据验证失败: {e}") return JSONResponse({"result": "ok"}) # 仍返回ok以避免重试 # 解析指令 parse_result = CommandParser.parse(callback_data.content) if not parse_result: # 不是有效指令,忽略 logger.debug("非有效指令,忽略") return JSONResponse({"result": "ok"}) game_type, command = parse_result logger.info(f"识别指令: game_type={game_type}, command={command}") # 检查限流 rate_limiter = get_rate_limiter() if not rate_limiter.is_allowed(): remaining = rate_limiter.get_remaining() reset_time = int(rate_limiter.get_reset_time()) sender = get_message_sender() await sender.send_text( f"⚠️ 消息发送过于频繁,请等待 {reset_time} 秒后再试\n" f"剩余配额: {remaining}" ) return JSONResponse({"result": "ok"}) # 更新用户信息 db = get_db() db.get_or_create_user(callback_data.creator) # 处理指令 response_text = await handle_command( game_type=game_type, command=command, chat_id=callback_data.chatid, user_id=callback_data.creator ) # 发送回复 if response_text: sender = get_message_sender() # 根据内容选择消息类型 if response_text.startswith('#'): # Markdown格式 await sender.send_markdown(response_text) else: # 普通文本 await sender.send_text(response_text) return JSONResponse({"result": "ok"}) except Exception as e: logger.error(f"处理Callback异常: {e}", exc_info=True) # 仍然返回ok,避免WPS重试 return JSONResponse({"result": "ok"}) async def handle_command(game_type: str, command: str, chat_id: int, user_id: int) -> str: """处理游戏指令 Args: game_type: 游戏类型 command: 完整指令 chat_id: 会话ID user_id: 用户ID Returns: 回复文本 """ try: # 帮助指令 if game_type == 'help': from games.base import get_help_message return get_help_message() # 统计指令 if game_type == 'stats': from games.base import get_stats_message return get_stats_message(user_id) # 骰娘游戏 if game_type == 'dice': from games.dice import DiceGame game = DiceGame() return await game.handle(command, chat_id, user_id) # 石头剪刀布 if game_type == 'rps': from games.rps import RPSGame game = RPSGame() return await game.handle(command, chat_id, user_id) # 运势占卜 if game_type == 'fortune': from games.fortune import FortuneGame game = FortuneGame() return await game.handle(command, chat_id, user_id) # 猜数字 if game_type == 'guess': from games.guess import GuessGame game = GuessGame() return await game.handle(command, chat_id, user_id) # 问答游戏 if game_type == 'quiz': from games.quiz import QuizGame game = QuizGame() return await game.handle(command, chat_id, user_id) # 成语接龙 if game_type == 'idiom': from games.idiom import IdiomGame game = IdiomGame() return await game.handle(command, chat_id, user_id) # 五子棋 if game_type == 'gomoku': from games.gomoku import GomokuGame game = GomokuGame() return await game.handle(command, chat_id, user_id) # 积分系统 if game_type == 'points': from games.points import PointsGame game = PointsGame() return await game.handle(command, chat_id, user_id) # 炼金系统 if game_type == 'alchemy': from games.alchemy import AlchemyGame game = AlchemyGame() return await game.handle(command, chat_id, user_id) # 积分赠送系统 if game_type == 'gift': from games.gift import GiftGame game = GiftGame() return await game.handle(command, chat_id, user_id) # 未知游戏类型 logger.warning(f"未知游戏类型: {game_type}") return "❌ 未知的游戏类型" except Exception as e: logger.error(f"处理游戏指令异常: {e}", exc_info=True) return f"❌ 处理指令时出错: {str(e)}"