163 lines
5.2 KiB
Python
163 lines
5.2 KiB
Python
"""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.debug(f"消息内容: {data.get('content')}")
|
||
|
||
# 验证请求
|
||
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)
|
||
|
||
# 未知游戏类型
|
||
logger.warning(f"未知游戏类型: {game_type}")
|
||
return "❌ 未知的游戏类型"
|
||
|
||
except Exception as e:
|
||
logger.error(f"处理游戏指令异常: {e}", exc_info=True)
|
||
return f"❌ 处理指令时出错: {str(e)}"
|
||
|