128 lines
4.6 KiB
Python
128 lines
4.6 KiB
Python
"""Callback路由处理"""
|
||
from ..Convention.Runtime.GlobalConfig import *
|
||
from ..Convention.Runtime.Architecture import Architecture
|
||
|
||
from fastapi import APIRouter, Request
|
||
from fastapi.responses import JSONResponse
|
||
import re
|
||
|
||
from ..CoreModules.models import CallbackRequest
|
||
from ..CoreModules.plugin_interface import PluginInterface
|
||
|
||
|
||
logger: ProjectConfig = Architecture.Get(ProjectConfig)
|
||
ALWAYS_RETURN_OK = logger.FindItem("always_return_ok", True) # 返回ok，可以避免重试
|
||
logger.SaveProperties()
|
||
|
||
router = APIRouter()
|
||
|
||
@router.get("/callback")
|
||
async def callback_verify():
|
||
"""Callback可用性校验"""
|
||
logger.Log("Info", "收到Callback验证请求")
|
||
return JSONResponse({"result": "ok"})
|
||
|
||
# 机器人名称模式（用于从@消息中提取，兼容半角/全角空格）
|
||
AT_PATTERN = re.compile(r'@[^\s]+[\s\u3000]+(.+)', re.DOTALL)
|
||
|
||
def parse_message_after_at(message: str) -> str:
|
||
# 去除首尾空格
|
||
message = message.strip()
|
||
|
||
# 尝试提取@后的内容
|
||
at_match = AT_PATTERN.search(message)
|
||
if at_match:
|
||
return at_match.group(1).strip()
|
||
return message
|
||
|
||
@router.post("/callback/construct")
|
||
async def callback_receive_construct(callback_data: CallbackRequest):
|
||
"""以构造好的Callback消息进行处理, 已知方式"""
|
||
try:
|
||
# 解析指令
|
||
content = parse_message_after_at(callback_data.content)
|
||
command = content.split(" ")[0]
|
||
message = content[len(command):].strip()
|
||
logger.Log("Info", f"识别指令: command={command}")
|
||
|
||
# 处理指令
|
||
result = await handle_command(command, message, callback_data.chatid, callback_data.creator)
|
||
if result:
|
||
return JSONResponse({"result": "ok", "message": result})
|
||
else:
|
||
return JSONResponse({"result": "ok"})
|
||
|
||
except Exception as e:
|
||
logger.Log("Error", f"处理Callback异常: {e}")
|
||
if ALWAYS_RETURN_OK:
|
||
return JSONResponse({"result": "ok"})
|
||
else:
|
||
return JSONResponse({"result": "error", "message": str(e)})
|
||
|
||
@router.post("/callback")
|
||
async def callback_receive(request: Request):
|
||
"""接受未知的Callback消息并进行处理, 默认方式"""
|
||
try:
|
||
# 解析请求数据
|
||
data = await request.json()
|
||
logger.Log("Info", f"完整callback数据: {data}")
|
||
|
||
# 验证请求
|
||
try:
|
||
callback_data = CallbackRequest(**data)
|
||
except Exception as e:
|
||
logger.Log("Error", f"请求数据验证失败: {e}")
|
||
if ALWAYS_RETURN_OK:
|
||
return JSONResponse({"result": "ok"})
|
||
else:
|
||
return JSONResponse({"result": "error", "message": str(e)})
|
||
|
||
# 解析指令
|
||
content = parse_message_after_at(callback_data.content)
|
||
command = content.split(" ")[0]
|
||
message = content[len(command):].strip()
|
||
logger.Log("Info", f"识别指令: command={command}")
|
||
|
||
# 处理指令
|
||
result = await handle_command(command, message, callback_data.chatid, callback_data.creator)
|
||
if result:
|
||
return JSONResponse({"result": "ok", "message": result})
|
||
else:
|
||
return JSONResponse({"result": "ok"})
|
||
|
||
except Exception as e:
|
||
logger.Log("Error", f"处理Callback异常: {e}")
|
||
if ALWAYS_RETURN_OK:
|
||
return JSONResponse({"result": "ok"})
|
||
else:
|
||
return JSONResponse({"result": "error", "message": str(e)})
|
||
|
||
|
||
async def handle_command(command: str, message: str,
|
||
chat_id: int, user_id: int) -> str|None:
|
||
"""处理指令
|
||
|
||
Args:
|
||
command: 指令
|
||
message: 消息内容
|
||
chat_id: 会话ID
|
||
user_id: 用户ID
|
||
|
||
Returns:
|
||
回复文本或None
|
||
"""
|
||
try:
|
||
plugin = PluginInterface.plugin_instances.get(command, None)
|
||
if plugin:
|
||
logger.Log("Info", f"已找到插件注册指令: {command}, class: {plugin.__class__.__name__}")
|
||
return await plugin.callback(message, chat_id, user_id)
|
||
elif "default" in PluginInterface.plugin_instances:
|
||
logger.Log("Info", f"未找到插件注册指令: {command}, 使用默认插件: {PluginInterface.plugin_instances["default"].__class__.__name__}")
|
||
return await PluginInterface.plugin_instances["default"].callback(command+" "+message, chat_id, user_id)
|
||
else:
|
||
return f"❌ 未识别指令: {command}"
|
||
except Exception as e:
|
||
logger.Log("Error", f"{ConsoleFrontColor.RED}处理指令异常: {e}{ConsoleFrontColor.RESET}")
|
||
return f"❌ 处理指令时出错: {str(e)}"
|
||
|