K2cr2O1
5d07a84283
本次提交对核心模块进行了深度重构，引入 Pydantic 增强配置管理的类型安全性，并全面优化了插件管理系统。 主要变更详情： 1. 核心架构与配置 - 重构配置加载模块：引入 Pydantic 模型 (`core/config_models.py`)，提供严格的配置项类型检查、验证及默认值管理。 - 统一模块结构：规范化模块导入路径，移除冗余的 `__init__.py` 文件，提升项目结构的清晰度。 - 性能优化：集成 Redis 缓存支持 (`RedisManager`)，有效降低高频 API 调用开销，提升响应速度。 2. 插件系统升级 - 实现热重载机制：新增插件文件变更监听功能，支持开发过程中自动重载插件，提升开发效率。 - 优化生命周期管理：改进插件加载与卸载逻辑，支持精确卸载指定插件及其关联的命令、事件处理器和定时任务。 3. 功能特性增强 - 新增媒体 API：引入 `MediaAPI` 模块，封装图片、语音等富媒体资源的获取与处理接口。 - 完善权限体系：重构权限管理系统，实现管理员与操作员的分级控制，支持更细粒度的命令权限校验。 4. 代码质量与稳定性 - 全面类型修复：解决 `mypy` 静态类型检查发现的大量类型错误（包括 `CommandManager`、`EventFactory` 及 `Bot` API 签名不匹配问题）。 - 增强错误处理：优化消息处理管道的异常捕获机制，完善关键路径的日志记录，提升系统运行稳定性。
215 lines
8.3 KiB
Python
215 lines
8.3 KiB
Python
"""
|
||
WebSocket 核心通信模块
|
||
|
||
该模块定义了 `WS` 类，负责与 OneBot v11 实现（如 NapCat）建立和管理
|
||
WebSocket 连接。它是整个机器人框架的底层通信基础。
|
||
|
||
主要职责包括：
|
||
- 建立 WebSocket 连接并处理认证。
|
||
- 实现断线自动重连机制。
|
||
- 监听并接收来自 OneBot 的事件和 API 响应。
|
||
- 分发事件给 `CommandManager` 进行处理。
|
||
- 提供 `call_api` 方法，用于异步发送 API 请求并等待响应。
|
||
"""
|
||
import asyncio
|
||
import json
|
||
from typing import Any, Dict, Optional
|
||
import uuid
|
||
|
||
import websockets
|
||
|
||
from models.events.factory import EventFactory
|
||
|
||
from .bot import Bot
|
||
from .config_loader import global_config
|
||
from .managers.command_manager import matcher
|
||
from .utils.logger import logger
|
||
|
||
|
||
class WS:
|
||
"""
|
||
WebSocket 客户端，负责与 OneBot v11 实现进行底层通信。
|
||
"""
|
||
|
||
def __init__(self, code_executor=None):
|
||
"""
|
||
初始化 WebSocket 客户端。
|
||
|
||
从全局配置中读取 WebSocket URI、访问令牌（Token）和重连间隔。
|
||
"""
|
||
# 读取参数
|
||
cfg = global_config.napcat_ws
|
||
self.url = cfg.uri
|
||
self.token = cfg.token
|
||
self.reconnect_interval = cfg.reconnect_interval
|
||
|
||
self.ws = None
|
||
self._pending_requests = {}
|
||
self.bot: Bot | None = None
|
||
self.self_id: int | None = None
|
||
self.code_executor = code_executor
|
||
|
||
async def connect(self):
|
||
"""
|
||
启动并管理 WebSocket 连接。
|
||
|
||
这是一个无限循环，负责建立连接。如果连接断开，它会根据配置的
|
||
`reconnect_interval` 时间间隔后自动尝试重新连接。
|
||
"""
|
||
headers = {"Authorization": f"Bearer {self.token}"} if self.token else {}
|
||
|
||
while True:
|
||
try:
|
||
logger.info(f"正在尝试连接至 NapCat: {self.url}")
|
||
async with websockets.connect(
|
||
self.url, additional_headers=headers
|
||
) as websocket:
|
||
self.ws = websocket
|
||
logger.success("连接成功！")
|
||
await self._listen_loop(websocket)
|
||
|
||
except (
|
||
websockets.exceptions.ConnectionClosed,
|
||
ConnectionRefusedError,
|
||
) as e:
|
||
logger.warning(f"连接断开或服务器拒绝访问: {e}")
|
||
except Exception as e:
|
||
logger.exception(f"运行异常: {e}")
|
||
|
||
logger.info(f"{self.reconnect_interval}秒后尝试重连...")
|
||
await asyncio.sleep(self.reconnect_interval)
|
||
|
||
async def _listen_loop(self, websocket_connection):
|
||
"""
|
||
核心监听循环，处理所有接收到的 WebSocket 消息。
|
||
|
||
此循环会持续从 WebSocket 连接中读取消息，并根据消息内容
|
||
判断是 API 响应还是上报的事件，然后分发给相应的处理逻辑。
|
||
|
||
Args:
|
||
websocket_connection: 当前活动的 WebSocket 连接对象。
|
||
"""
|
||
async for message in websocket_connection:
|
||
try:
|
||
data = json.loads(message)
|
||
|
||
# 1. 处理 API 响应
|
||
# 如果消息中包含 echo 字段，说明是 API 调用的响应
|
||
echo_id = data.get("echo")
|
||
if echo_id and echo_id in self._pending_requests:
|
||
future = self._pending_requests.pop(echo_id)
|
||
if not future.done():
|
||
future.set_result(data)
|
||
continue
|
||
|
||
# 2. 处理上报事件
|
||
# 如果消息中包含 post_type 字段，说明是 OneBot 上报的事件
|
||
if "post_type" in data:
|
||
# 使用 create_task 异步执行，避免阻塞 WebSocket 接收循环
|
||
asyncio.create_task(self.on_event(data))
|
||
|
||
except Exception as e:
|
||
logger.exception(f"解析消息异常: {e}")
|
||
|
||
async def on_event(self, event_data: dict):
|
||
"""
|
||
事件处理和分发层。
|
||
|
||
当接收到一个 OneBot 事件时，此方法负责：
|
||
1. 使用 `EventFactory` 将原始 JSON 数据解析成对应的事件对象。
|
||
2. 为事件对象注入 `Bot` 实例，以便在插件中可以调用 API。
|
||
3. 打印格式化的事件日志。
|
||
4. 将事件对象传递给 `CommandManager` (`matcher`) 进行后续处理。
|
||
|
||
Args:
|
||
event_data (dict): 从 WebSocket 接收到的原始事件字典。
|
||
"""
|
||
try:
|
||
# 使用工厂创建事件对象
|
||
event = EventFactory.create_event(event_data)
|
||
|
||
# 在收到第一个 meta_event 时，初始化 Bot 实例
|
||
if event.post_type == "meta_event" and self.bot is None:
|
||
self.self_id = event.self_id
|
||
self.bot = Bot(self)
|
||
logger.success(f"Bot 实例初始化完成: self_id={self.self_id}")
|
||
|
||
# 将代码执行器注入到 Bot 和执行器自身
|
||
if self.code_executor:
|
||
self.bot.code_executor = self.code_executor
|
||
self.code_executor.bot = self.bot
|
||
logger.info("代码执行器已成功注入 Bot 实例。")
|
||
|
||
# 如果 bot 尚未初始化，则不处理后续事件
|
||
if self.bot is None:
|
||
logger.warning("Bot 尚未初始化，跳过事件处理。")
|
||
return
|
||
|
||
event.bot = self.bot # 注入 Bot 实例
|
||
|
||
# 打印日志
|
||
if event.post_type == "message":
|
||
sender_name = event.sender.nickname if hasattr(event, "sender") and event.sender else "Unknown"
|
||
message_type = getattr(event, "message_type", "Unknown")
|
||
user_id = getattr(event, "user_id", "Unknown")
|
||
raw_message = getattr(event, "raw_message", "")
|
||
logger.info(f"[消息] {message_type} | {user_id}({sender_name}): {raw_message}")
|
||
elif event.post_type == "notice":
|
||
notice_type = getattr(event, "notice_type", "Unknown")
|
||
logger.info(f"[通知] {notice_type}")
|
||
elif event.post_type == "request":
|
||
request_type = getattr(event, "request_type", "Unknown")
|
||
logger.info(f"[请求] {request_type}")
|
||
elif event.post_type == "meta_event":
|
||
meta_event_type = getattr(event, "meta_event_type", "Unknown")
|
||
logger.debug(f"[元事件] {meta_event_type}")
|
||
|
||
|
||
# 分发事件
|
||
await matcher.handle_event(self.bot, event)
|
||
|
||
except Exception as e:
|
||
logger.exception(f"事件处理异常: {e}")
|
||
|
||
async def call_api(self, action: str, params: Optional[Dict[Any, Any]] = None) -> Dict[Any, Any]:
|
||
"""
|
||
向 OneBot v11 实现端发送一个 API 请求。
|
||
|
||
该方法通过 WebSocket 发送请求，并使用 `echo` 字段来匹配对应的响应。
|
||
它创建了一个 `Future` 对象来异步等待响应，并设置了超时机制。
|
||
|
||
Args:
|
||
action (str): API 的动作名称，例如 "send_group_msg"。
|
||
params (dict, optional): API 请求的参数字典。 Defaults to None.
|
||
|
||
Returns:
|
||
dict: OneBot API 的响应数据。如果超时或连接断开，则返回一个
|
||
表示失败的字典。
|
||
"""
|
||
if not self.ws:
|
||
logger.error("调用 API 失败: WebSocket 未初始化")
|
||
return {"status": "failed", "msg": "websocket not initialized"}
|
||
|
||
from websockets.protocol import State
|
||
|
||
if getattr(self.ws, "state", None) is not State.OPEN:
|
||
logger.error("调用 API 失败: WebSocket 连接未打开")
|
||
return {"status": "failed", "msg": "websocket is not open"}
|
||
|
||
echo_id = str(uuid.uuid4())
|
||
payload = {"action": action, "params": params or {}, "echo": echo_id}
|
||
|
||
loop = asyncio.get_running_loop()
|
||
future = loop.create_future()
|
||
self._pending_requests[echo_id] = future
|
||
|
||
await self.ws.send(json.dumps(payload))
|
||
|
||
try:
|
||
return await asyncio.wait_for(future, timeout=30.0)
|
||
except asyncio.TimeoutError:
|
||
self._pending_requests.pop(echo_id, None)
|
||
logger.warning(f"API 调用超时: action={action}, params={params}")
|
||
return {"status": "failed", "retcode": -1, "msg": "api timeout"}
|
||
|