diff --git a/README.md b/README.md index 55b9d6b..5e14e47 100644 --- a/README.md +++ b/README.md @@ -74,12 +74,12 @@ │ └── thpic.py ├── core/ # NEO 框架核心代码,通常无需修改 │ ├── api/ +│ ├── data/ # 数据存储目录 (管理员列表, 权限配置) +│ │ ├── admin.json +│ │ └── permissions.json │ ├── bot.py │ ├── ... │ └── ws.py -├── data/ # 数据存储目录 (管理员列表, 权限配置) -│ ├── admin.json -│ └── permissions.json ├── html/ # 静态网页文件 ├── plugins/ # 插件目录,所有机器人的功能模块都在这里 │ ├── admin.py diff --git a/core/managers/__init__.py b/core/managers/__init__.py index 5be6af9..843b996 100644 --- a/core/managers/__init__.py +++ b/core/managers/__init__.py @@ -4,7 +4,6 @@ 这个包集中了机器人核心的单例管理器。 通过从这里导入,可以确保在整个应用中访问到的都是同一个实例。 """ -from ..config_loader import global_config from .admin_manager import AdminManager from .command_manager import matcher as command_manager from .permission_manager import PermissionManager diff --git a/core/managers/permission_manager.py b/core/managers/permission_manager.py index b7904c3..0e83055 100644 --- a/core/managers/permission_manager.py +++ b/core/managers/permission_manager.py @@ -13,7 +13,7 @@ """ import json import os -from typing import Dict, Optional +from typing import Dict from ..utils.logger import logger from ..utils.singleton import Singleton diff --git a/docs/core-concepts/event-flow.md b/docs/core-concepts/event-flow.md index 5d7275e..e38f497 100644 --- a/docs/core-concepts/event-flow.md +++ b/docs/core-concepts/event-flow.md @@ -8,15 +8,51 @@ ```mermaid graph TD - A[OneBot v11 实现端] -- WebSocket Message --> B(core/ws.py); - B -- Raw JSON Data --> C(models/events/factory.py); - C -- Event Object --> D(core/ws.py on_event); - D -- Event Object --> E(core/managers/command_manager.py); - E -- Event & Command Match --> F(core/handlers/event_handler.py); - F -- Matched Handler --> G(plugins/echo.py); - G -- Call API --> H(core/bot.py); - H -- Send Request --> B; - B -- WebSocket Send --> A; + %% 定义样式 + classDef external fill:#e1f5fe,stroke:#01579b,stroke-width:2px; + classDef network fill:#fff9c4,stroke:#fbc02d,stroke-width:2px; + classDef core fill:#e8f5e9,stroke:#2e7d32,stroke-width:2px; + classDef plugin fill:#fce4ec,stroke:#c2185b,stroke-width:2px; + + subgraph External [外部环境] + OneBot[OneBot v11 实现端
(如 NapCatQQ)]:::external + end + + subgraph NeoBot [NEO Bot Framework] + direction TB + + subgraph Network [网络接入层] + WS[WebSocket 连接
core/ws.py]:::network + end + + subgraph Processing [核心处理层] + Factory[事件工厂
models/events/factory.py]:::core + Dispatcher[命令管理器
core/managers/command_manager.py]:::core + Handler[事件处理器
core/handlers/event_handler.py]:::core + BotAPI[Bot API 封装
core/bot.py]:::core + end + + subgraph Plugins [业务插件层] + UserPlugin[用户插件
plugins/*.py]:::plugin + end + end + + %% 事件上报流程 (实线) + OneBot -- 1. WebSocket 消息 --> WS + WS -- 2. 原始 JSON --> Factory + Factory -- 3. Event 对象 --> WS + WS -- 4. 分发事件 --> Dispatcher + Dispatcher -- 5. 匹配指令/事件 --> Handler + Handler -- 6. 调用处理函数 --> UserPlugin + + %% API 调用流程 (虚线) + UserPlugin -. 7. 调用 bot.send() .-> BotAPI + BotAPI -. 8. 封装 API 请求 .-> WS + WS -. 9. 发送 JSON .-> OneBot + + %% 链接样式 + linkStyle 0,1,2,3,4,5 stroke:#333,stroke-width:2px; + linkStyle 6,7,8 stroke:#666,stroke-width:2px,stroke-dasharray: 5 5; ``` ## 详细步骤 diff --git a/docs/core-concepts/singleton-managers.md b/docs/core-concepts/singleton-managers.md index 41d64b5..5d9541f 100644 --- a/docs/core-concepts/singleton-managers.md +++ b/docs/core-concepts/singleton-managers.md @@ -36,7 +36,7 @@ * **核心职责**: * **权限定义与检查**: 定义了 `ADMIN`, `OP`, `USER` 等权限等级,并提供了 `check_permission` 方法来验证用户权限。 * **数据持久化**: 负责从 `core/data/permissions.json` 文件中加载和保存用户权限设置。 - * **与 `AdminManager` 联动**: 在检查权限时,会自动将机器人管理员(来自 `AdminManager`)识别为最高权限 `ADMIN`。 + * **与 `AdminManager` 联动**: 在检查权限和获取所有用户权限时,会自动合并机器人管理员(来自 `AdminManager`)的数据,将其识别为最高权限 `ADMIN`。 ### 3. `AdminManager` (全局实例: `admin_manager`) diff --git a/docs/project-structure.md b/docs/project-structure.md index 4690b15..037360d 100644 --- a/docs/project-structure.md +++ b/docs/project-structure.md @@ -40,6 +40,7 @@ * `utils/`: 提供被广泛使用的工具类,如 `logger` (日志)、`singleton` (单例模式基类)。 * `bot.py`: 定义了 `Bot` 类,这是插件开发者最常与之交互的对象,用于调用所有 OneBot API。 * `config_loader.py`: 负责解析 `config.toml` 文件,并提供一个全局的 `global_config` 对象。 +* `config_models.py`: 使用 Pydantic 定义了配置文件的结构和类型验证。 * `ws.py`: 实现了与 OneBot v11 实现端的 WebSocket 连接、心跳、重连和消息收发。 ### `docs/` diff --git a/plugins/admin.py b/plugins/admin.py index 6dd0c18..e518bb3 100644 --- a/plugins/admin.py +++ b/plugins/admin.py @@ -1,4 +1,3 @@ -from core.handlers.event_handler import MessageHandler from core.managers import command_manager, permission_manager from core.permission import Permission from models.events.message import MessageEvent diff --git a/plugins/thpic.py b/plugins/thpic.py index 2112cf6..0512118 100644 --- a/plugins/thpic.py +++ b/plugins/thpic.py @@ -12,7 +12,7 @@ from models.events.message import MessageEvent, MessageSegment __plugin_meta__ = { "name": "thpic", "description": "来看看东方Project的图片吧!", - "usage": "/thpic", + "usage": "/thpic [nums](1~10)", } @@ -25,7 +25,38 @@ async def handle_echo(bot: Bot, event: MessageEvent, args: list[str]): :param event: 消息事件对象。 :param args: 指令参数列表(未使用)。 """ - try: - await event.reply(str(MessageSegment.image("https://img.paulzzh.com/touhou/random"))) - except Exception as e: - await event.reply(f"报错了。。。{e}") + parts = args + print(parts) + if not parts: + try: + await event.reply( + str(MessageSegment.image("https://img.paulzzh.com/touhou/random")) + ) + except Exception as e: + await event.reply(f"报错了。。。{e}") + else: + if parts[0].isdigit(): + nums = int(parts[0]) + if nums <= 0: + await event.reply("请输入一个大于0的整数。") + return + elif nums > 10: + await event.reply("请输入一个不大于10的整数。") + return + try: + nodes = [] + for _ in range(nums): + nodes.append( + bot.build_forward_node( + user_id=event.self_id, + nickname="机器人", + message=MessageSegment.image( + "https://img.paulzzh.com/touhou/random" + ), + ) + ) + await bot.send_forwarded_messages(event, nodes) + except Exception as e: + await event.reply(f"报错了。。。{e}") + else: + await event.reply(f"用法不正确。\n\n{__plugin_meta__['usage']}") diff --git a/tests/test_plugin_reload_meta.py b/tests/test_plugin_reload_meta.py index 92a9e93..8eb0f81 100644 --- a/tests/test_plugin_reload_meta.py +++ b/tests/test_plugin_reload_meta.py @@ -1,6 +1,4 @@ -import pytest -from unittest.mock import MagicMock from core.managers.command_manager import CommandManager class TestPluginReloadMeta: