262 lines
8.7 KiB
Python
262 lines
8.7 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.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()
|
||
|
||
# AI 对话：统一按 Markdown 发送（按任务决策）
|
||
if game_type == 'ai_chat':
|
||
try:
|
||
await sender.send_markdown(response_text)
|
||
except Exception as send_md_err:
|
||
logger.error(f"发送Markdown消息失败，改用文本发送: {send_md_err}")
|
||
await sender.send_text(response_text)
|
||
else:
|
||
# 其他模块保持原有启发式：以 # 开头视为 Markdown，否则文本
|
||
if response_text.startswith('#'):
|
||
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 == 'register':
|
||
return await handle_register_command(command, chat_id, 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 == 'sanguosha':
|
||
from games.sanguosha import SanguoshaGame
|
||
game = SanguoshaGame()
|
||
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 == 'adventure':
|
||
from games.adventure import AdventureGame
|
||
game = AdventureGame()
|
||
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)
|
||
|
||
# 复述功能
|
||
if game_type == 'say':
|
||
# 提取参数并原样返回
|
||
_, args = CommandParser.extract_command_args(command)
|
||
args = args.strip()
|
||
if not args:
|
||
return "用法：.say 你想让我说的话\n别名：.说 / .复述"
|
||
return args
|
||
|
||
# AI对话系统
|
||
if game_type == 'ai_chat':
|
||
from games.ai_chat import AIChatGame
|
||
game = AIChatGame()
|
||
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)}"
|
||
|
||
|
||
async def handle_register_command(command: str, chat_id: int, user_id: int) -> str:
|
||
"""处理注册命令
|
||
|
||
Args:
|
||
command: 完整指令 ".register name"
|
||
chat_id: 会话ID
|
||
user_id: 用户ID
|
||
|
||
Returns:
|
||
注册结果消息
|
||
"""
|
||
try:
|
||
# 提取参数
|
||
_, args = CommandParser.extract_command_args(command)
|
||
args = args.strip()
|
||
|
||
# 验证参数
|
||
if not args:
|
||
return "❌ 请提供要注册的名称！\n\n正确格式：`.register <名称>`\n\n示例：\n`.register 张三`\n`.register 小明`"
|
||
|
||
if len(args) > 20:
|
||
return "❌ 名称过长！最多支持20个字符。"
|
||
|
||
# 更新用户名称
|
||
db = get_db()
|
||
success = db.update_user_name(user_id, args)
|
||
|
||
if success:
|
||
return f"✅ 注册成功！\n\n**您的名称**：{args}\n\n之后您可以使用这个名称参与各种游戏和功能。"
|
||
else:
|
||
return "❌ 注册失败！请稍后重试。"
|
||
|
||
except Exception as e:
|
||
logger.error(f"处理注册指令错误: {e}", exc_info=True)
|
||
return f"❌ 处理指令出错: {str(e)}"
|
||
|