Merge branch 'main' of https://github.com/Fairy-Oracle-Sanctuary/NeoBot into dev

2026-01-07 23:16:34 +08:00
parent c3b3541694 aaf4a896dd
commit 6d7dfc179d
3 changed files with 298 additions and 0 deletions
--- a/README.md
+++ b/README.md
@@ -13,7 +13,36 @@
 本项目旨在提供一个稳定、高性能且开发体验优秀的机器人平台，服务于我们的社群管理和日常自动化需求。

 ### ✨ 核心特性
+> **[INTERNAL USE ONLY]**
+>
+> 本仓库为 Calglau BOT 的内部开发版本，请遵守相关保密协议。

+**Powered by NEO Bot Framework**
+
+## 📖 项目概述
+
+**Calglau BOT** 是一个基于 NEO Bot Framework 构建的、功能丰富的 QQ 机器人。它被设计为一个模块化、易于扩展的内部工具，通过插件化的方式集成了多种实用与娱乐功能。
+
+本项目旨在提供一个稳定、高性能且开发体验优秀的机器人平台，服务于我们的社群管理和日常自动化需求。
+
+### ✨ 核心特性
+
+*   **模块化插件架构**：所有功能均以独立插件形式存在于 `plugins/` 目录，易于开发、维护和热重载。
+*   **高性能异步核心**：基于 `asyncio` 和 `websockets`，确保在高并发消息下依然响应迅速。
+*   **开发者友好**：内置插件热重载，修改代码无需重启；完整的类型提示和清晰的 API 设计，提升开发效率。
+*   **集成 Redis 缓存**：自动缓存常用 API 调用（如群信息），减少重复请求，提升响应速度。
+*   **内置帮助系统**：通过 `/help` 指令可自动生成并展示所有已加载插件的功能说明。
+
+### 🛠️ 技术栈
+
+*   **核心框架**: Python 3.8+ & NEO Bot Framework
+*   **异步库**: `asyncio`
+*   **网络通信**: `websockets` (OneBot v11)
+*   **缓存**: `Redis`
+*   **日志**: `Loguru`
+*   **文件监控**: `watchdog` (用于热重载)
+
+---
 *   **模块化插件架构**：所有功能均以独立插件形式存在于 `plugins/` 目录，易于开发、维护和热重载。
 *   **高性能异步核心**：基于 `asyncio` 和 `websockets`，确保在高并发消息下依然响应迅速。
 *   **开发者友好**：内置插件热重载，修改代码无需重启；完整的类型提示和清晰的 API 设计，提升开发效率。
@@ -52,10 +81,29 @@
 │   ├── admin.json
 │   └── permissions.json
 ├── html/                   # 静态网页文件
+├── plugins/                # 插件目录，所有机器人的功能模块都在这里
+│   ├── admin.py
+│   ├── bili_parser.py
+│   ├── code_py.py
+│   ├── echo.py
+│   ├── forward_test.py
+│   ├── jrcd.py
+│   └── thpic.py
+├── core/                   # NEO 框架核心代码，通常无需修改
+│   ├── api/
+│   ├── bot.py
+│   ├── ...
+│   └── ws.py
+├── data/                   # 数据存储目录 (管理员列表, 权限配置)
+│   ├── admin.json
+│   └── permissions.json
+├── html/                   # 静态网页文件
 │   ├── 404.html
 │   └── index.html
 ├── models/                 # 数据模型 (事件, 消息段等)
 │   ├── ...
+├── models/                 # 数据模型 (事件, 消息段等)
+│   ├── ...
 ├── .gitignore
 ├── config.toml             # 主配置文件
 ├── main.py                 # 项目启动入口
@@ -76,6 +124,11 @@

 ### 1. 环境准备

+*   **Python 3.12 或更高版本**
+    *   **我觉得**: 在开发和调试阶段使用官方的 **CPython** 解释器，以获得最佳的第三方库兼容性和调试体验。
+    *   **你也可以觉得**: 在生产环境部署时，可以考虑使用 **PyPy** 以获取潜在的性能提升，但这可能会牺牲一定的兼容性。
+*   Redis 服务
+*   一个正在运行的 OneBot v11 实现端 (推荐 **NapCatQQ**)
 *   **Python 3.12 或更高版本**
    *   **我觉得**: 在开发和调试阶段使用官方的 **CPython** 解释器，以获得最佳的第三方库兼容性和调试体验。
    *   **你也可以觉得**: 在生产环境部署时，可以考虑使用 **PyPy** 以获取潜在的性能提升，但这可能会牺牲一定的兼容性。
@@ -84,6 +137,7 @@

 ### 2. 安装依赖

+克隆本项目后，在项目根目录执行：
 克隆本项目后，在项目根目录执行：
 ```bash
 pip install -r requirements.txt
@@ -97,11 +151,22 @@ pip install -r requirements.txt

 **因此，在拉取仓库后，您通常无需对 `config.toml` 文件进行任何修改即可直接运行。**

+如果您需要连接到本地或其他特定环境，可以参考以下配置结构进行修改。配置示例：
+### 3. 配置
+
+**[内部开发]**
+
+为了方便内部开发和调试，项目中的 `config.toml` 文件已预先配置为连接到官方的 DEV 调试服务器。
+
+**因此，在拉取仓库后，您通常无需对 `config.toml` 文件进行任何修改即可直接运行。**
+
 如果您需要连接到本地或其他特定环境，可以参考以下配置结构进行修改。配置示例：

 ```toml
 # config.toml

+# config.toml
+
 [napcat_ws]
 # OneBot 实现端的 WebSocket 地址
 uri = "ws://127.0.0.1:3001"  
@@ -109,11 +174,26 @@ uri = "ws://127.0.0.1:3001"
 token = ""          
 # 断线重连间隔（秒）
 reconnect_interval = 5        
+# OneBot 实现端的 WebSocket 地址
+uri = "ws://127.0.0.1:3001"  
+# Access Token (如果有)
+token = ""          
+# 断线重连间隔（秒）
+reconnect_interval = 5        

 [bot]
 # 机器人指令的起始符号，可配置多个
 command_prefixes = ["/", "!", "！"] 

+[redis]
+# Redis 连接信息
+host = "127.0.0.1"
+port = 6379
+db = 0
+password = ""
+# 机器人指令的起始符号，可配置多个
+command_prefixes = ["/", "!", "！"] 
+
 [redis]
 # Redis 连接信息
 host = "127.0.0.1"
@@ -141,7 +221,29 @@ Calglau BOT 的所有功能都通过插件实现。开发新功能非常简单
 2.  在 `plugins/` 目录下创建或修改任意 `.py` 文件。
 3.  **保存文件**。
 4.  观察控制台输出 `[HotReload] 插件重载完成` 的提示。你的新代码已即时生效。
+机器人启动后，将自动连接到 OneBot 实现端。控制台会输出加载的插件列表和连接状态。

+---
+
+## 🛠️ 插件开发指南
+
+Calglau BOT 的所有功能都通过插件实现。开发新功能非常简单，并且得益于热重载，你无需在开发过程中频繁重启机器人。
+
+### 🔥 热重载工作流
+
+1.  保持 `python main.py` 进程运行。
+2.  在 `plugins/` 目录下创建或修改任意 `.py` 文件。
+3.  **保存文件**。
+4.  观察控制台输出 `[HotReload] 插件重载完成` 的提示。你的新代码已即时生效。
+
+### 创建一个新插件
+
+1.  在 `plugins/` 目录下新建一个 Python 文件，例如 `weather.py`。
+2.  在该文件中编写你的逻辑。
+
+#### 1. 定义插件元数据 (`__plugin_meta__`)
+
+为了让 `/help` 指令能自动发现你的插件，请在文件顶部定义 `__plugin_meta__` 字典：
 ### 创建一个新插件

 1.  在 `plugins/` 目录下新建一个 Python 文件，例如 `weather.py`。
@@ -153,6 +255,7 @@ Calglau BOT 的所有功能都通过插件实现。开发新功能非常简单

 ```python
 # plugins/weather.py
+# plugins/weather.py

 __plugin_meta__ = {
    "name": "天气查询",
@@ -163,23 +266,39 @@ __plugin_meta__ = {

 #### 2. 编写指令处理器

+使用 `@matcher.command()` 装饰器来注册一个聊天指令。
+    "name": "天气查询",
+    "description": "提供城市天气查询功能。",
+    "usage": "/weather [城市名] - 查询指定城市的实时天气。",
+}
+```
+
+#### 2. 编写指令处理器
+
 使用 `@matcher.command()` 装饰器来注册一个聊天指令。

 ```python
 # plugins/weather.py
+# plugins/weather.py
 from core.command_manager import matcher
 from models import MessageEvent

+# ... (元数据定义) ...
 # ... (元数据定义) ...

@matcher.command("weather")
+async def handle_weather_command(event: MessageEvent, args: list[str]):
 async def handle_weather_command(event: MessageEvent, args: list[str]):
    """
    处理 /weather 指令
    :param event: 消息事件对象，用于回复等操作
    :param args: 用户发送的参数列表 (已按空格分割)
+    处理 /weather 指令
+    :param event: 消息事件对象，用于回复等操作
+    :param args: 用户发送的参数列表 (已按空格分割)
    """
    if not args:
+        await event.reply("请输入要查询的城市名，例如：/weather 北京")
        await event.reply("请输入要查询的城市名，例如：/weather 北京")
        return

@@ -189,19 +308,35 @@ async def handle_weather_command(event: MessageEvent, args: list[str]):
    # (示例代码，省略了真实 API 调用)
    weather_data = f"{city}的天气是：晴，25°C。"
    
+    await event.reply(weather_data)
+    city = args[0]
+    
+    # 此处应调用天气 API 获取数据
+    # (示例代码，省略了真实 API 调用)
+    weather_data = f"{city}的天气是：晴，25°C。"
+    
    await event.reply(weather_data)
 ```

 #### 3. 监听事件

+除了指令，你还可以监听各种事件，例如新成员入群。
+#### 3. 监听事件
+
 除了指令，你还可以监听各种事件，例如新成员入群。

 ```python
 from core.command_manager import matcher
 from models import GroupIncreaseNoticeEvent
+from models import GroupIncreaseNoticeEvent
 from core.bot import Bot

+
@matcher.on_notice("group_increase")
+async def welcome_new_member(bot: Bot, event: GroupIncreaseNoticeEvent):
+    """当有新成员加入群聊时触发"""
+    welcome_message = f"欢迎新成员 @{event.user_id} 加入本群！"
+    await bot.send_group_msg(event.group_id, welcome_message)
 async def welcome_new_member(bot: Bot, event: GroupIncreaseNoticeEvent):
    """当有新成员加入群聊时触发"""
    welcome_message = f"欢迎新成员 @{event.user_id} 加入本群！"
@@ -226,6 +361,31 @@ async def welcome_new_member(bot: Bot, event: GroupIncreaseNoticeEvent):

 ## 🗺️ 路线图 (Roadmap)

+-   [ ] **Web 仪表盘**: 开发一个简单的 Web 页面，用于查看机器人状态和插件列表。
+-   [ ] **权限系统重构**: 引入更精细化的权限节点，允许按插件或指令控制用户权限。
+-   [ ] **数据库集成**: 引入 `SQLite` 或其他数据库，用于需要持久化存储数据的功能。
+-   [ ] **新插件开发**:
+    -   [ ] 天气查询插件
+    -   [ ] GIL实现
+    -   [ ] coming soon...
+---
+
+## 📦 当前功能插件
+
+| 插件文件 (`plugins/`) | 功能描述 |
+|-----------------------|----------|
+| `admin.py`            | 机器人管理员权限管理 |
+| `bili_parser.py`      | 自动解析 Bilibili 视频链接分享卡片 |
+| `code_py.py`          | 执行 Python 代码片段 (高危，仅限管理员) |
+| `echo.py`             | 提供 `/echo` 复读和 `/赞我` 功能 |
+| `forward_test.py`     | 演示如何发送合并转发消息 |
+| `jrcd.py`             | 娱乐功能：今日人品、牛牛词典 |
+| `thpic.py`            | 发送一张随机的东方 Project 图片 |
+
+---
+
+## 🗺️ 路线图 (Roadmap)
+
 -   [ ] **Web 仪表盘**: 开发一个简单的 Web 页面，用于查看机器人状态和插件列表。
 -   [ ] **权限系统重构**: 引入更精细化的权限节点，允许按插件或指令控制用户权限。
 -   [ ] **数据库集成**: 引入 `SQLite` 或其他数据库，用于需要持久化存储数据的功能。
--- a/core/handlers/event_handler.py
+++ b/core/handlers/event_handler.py
@@ -82,6 +82,7 @@ class MessageHandler(BaseHandler):
    def command(
        self,
        *names: str,
+        *names: str,
        permission: Optional[Permission] = None,
        override_permission_check: bool = False
    ) -> Callable:
--- a/plugins/code_py.py
+++ b/plugins/code_py.py
@@ -1,6 +1,9 @@
 # -*- coding: utf-8 -*-
 import html
 import textwrap
+# -*- coding: utf-8 -*-
+import html
+import textwrap
 import asyncio
 from typing import Dict

@@ -13,12 +16,20 @@ __plugin_meta__ = {
    "name": "Python 代码执行",
    "description": "在安全的沙箱环境中执行 Python 代码片段，支持单行、多行和转发回复。",
    "usage": "/py <单行代码>\n/code_py <单行代码>\n/py (进入多行输入模式)",
+    "name": "Python 代码执行",
+    "description": "在安全的沙箱环境中执行 Python 代码片段，支持单行、多行和转发回复。",
+    "usage": "/py <单行代码>\n/code_py <单行代码>\n/py (进入多行输入模式)",
 }

 # --- 会话状态管理 ---
 # 结构: {(user_id, group_id): asyncio.TimerHandle}
 multi_line_sessions: Dict[tuple, asyncio.TimerHandle] = {}

+async def reply_as_forward(event: MessageEvent, input_code: str, output_result: str):
+# --- 会话状态管理 ---
+# 结构: {(user_id, group_id): asyncio.TimerHandle}
+multi_line_sessions: Dict[tuple, asyncio.TimerHandle] = {}
+
 async def reply_as_forward(event: MessageEvent, input_code: str, output_result: str):
    """
    将输入和输出打包成转发消息进行回复。
@@ -48,9 +59,38 @@ async def reply_as_forward(event: MessageEvent, input_code: str, output_result:
        # 降级为普通消息回复
        await event.reply(f"--- 你的代码 ---\n{input_code}\n--- 执行结果 ---\n{output_result}")

+async def execute_code(event: MessageEvent, code: str):
+    将输入和输出打包成转发消息进行回复。
+    参考 forward_test.py 的实现，兼容私聊和群聊。
+    """
+    bot = event.bot
+    
+    # 1. 构建消息节点列表
+    nodes = [
+        bot.build_forward_node(
+            user_id=event.user_id, 
+            nickname=event.sender.nickname or str(event.user_id), 
+            message=f"--- Your Code ---\n{input_code}"
+        ),
+        bot.build_forward_node(
+            user_id=event.self_id, 
+            nickname="Code Executor", 
+            message=f"--- Execution Result ---\n{output_result}"
+        )
+    ]
+    
+    try:
+        # 2. 发送合并转发消息
+        await bot.send_forwarded_messages(event, nodes)
+    except Exception as e:
+        logger.error(f"[code_py] 发送转发消息失败: {e}")
+        # 降级为普通消息回复
+        await event.reply(f"--- 你的代码 ---\n{input_code}\n--- 执行结果 ---\n{output_result}")
+
 async def execute_code(event: MessageEvent, code: str):
    """
    核心代码执行逻辑。
+    核心代码执行逻辑。
    """
    code_executor = getattr(event.bot, 'code_executor', None)
    if not code_executor or not code_executor.docker_client:
@@ -97,15 +137,74 @@ def normalize_code(code: str) -> str:
    return code.strip()


+@matcher.command("py", "python", "code_py", permission=ADMIN)
+async def code_py_main(event: MessageEvent, args: list[str]):
+    code_executor = getattr(event.bot, 'code_executor', None)
+    if not code_executor or not code_executor.docker_client:
+        await event.reply("代码执行服务当前不可用，请检查 Docker 连接配置。")
+        return
+
+    # 修改 add_task，让它能直接接收回复函数
+    await code_executor.add_task(
+        code, 
+        lambda result: reply_as_forward(event, code, result)
+    )
+    await event.reply("代码已提交至沙箱执行队列，请稍候...")
+
+def cleanup_session(session_key: tuple):
+    """
+    清理超时的会话。
+    """
+    if session_key in multi_line_sessions:
+        del multi_line_sessions[session_key]
+        logger.info(f"[code_py] 会话 {session_key} 已超时，自动取消。")
+
+def normalize_code(code: str) -> str:
+    """
+    规范化用户输入的 Python 代码字符串。
+
+    主要处理两个问题：
+    1. 对消息中可能存在的 HTML 实体进行解码 (e.g., &#91; -> [)。
+    2. 移除整个代码块的公共前导缩进，以修复因复制粘贴导致的多余缩进。
+
+    :param code: 原始代码字符串。
+    :return: 规范化后的代码字符串。
+    """
+    # 1. 解码 HTML 实体
+    code = html.unescape(code)
+
+    # 2. 移除公共前导缩进
+    try:
+        code = textwrap.dedent(code)
+    except Exception:
+        # 在某些情况下（例如，不一致的缩进），dedent 可能会失败，
+        # 但我们不希望因此中断流程，所以捕获异常并继续。
+        pass
+        
+    return code.strip()
+
+
@matcher.command("py", "python", "code_py", permission=ADMIN)
 async def code_py_main(event: MessageEvent, args: list[str]):
    """
    /py 命令的主入口。
    - 如果有参数，直接执行。
    - 如果没有参数，开启多行输入模式。
+    /py 命令的主入口。
+    - 如果有参数，直接执行。
+    - 如果没有参数，开启多行输入模式。
    """
    code_to_run = " ".join(args)

+    if code_to_run:
+        # 单行模式，对代码进行规范化处理
+        normalized_code = normalize_code(code_to_run)
+        if not normalized_code:
+            await event.reply("代码为空或格式错误，请输入有效的代码。")
+            return
+        await execute_code(event, normalized_code)
+    code_to_run = " ".join(args)
+
    if code_to_run:
        # 单行模式，对代码进行规范化处理
        normalized_code = normalize_code(code_to_run)
@@ -132,6 +231,24 @@ async def code_py_main(event: MessageEvent, args: list[str]):
            session_key
        )
        multi_line_sessions[session_key] = timeout_handler
+        # 多行模式
+        # 使用 getattr 兼容私聊和群聊
+        session_key = (event.user_id, getattr(event, 'group_id', 'private'))
+        
+        # 如果上一个会话的超时任务还在，先取消它
+        if session_key in multi_line_sessions:
+            multi_line_sessions[session_key].cancel()
+
+        await event.reply("已进入多行代码输入模式，请直接发送你的代码。\n(60秒内无操作将自动取消)")
+        
+        # 设置 60 秒超时
+        loop = asyncio.get_running_loop()
+        timeout_handler = loop.call_later(
+            60, 
+            cleanup_session, 
+            session_key
+        )
+        multi_line_sessions[session_key] = timeout_handler

@matcher.on_message()
 async def handle_multi_line_code(event: MessageEvent):
@@ -148,6 +265,26 @@ async def handle_multi_line_code(event: MessageEvent):
        # 对多行代码进行规范化处理
        normalized_code = normalize_code(event.raw_message)
        
+        if not normalized_code:
+            await event.reply("捕获到的代码为空或格式错误，已取消输入。")
+            return
+
+        await execute_code(event, normalized_code)
+        return True # 消费事件，防止其他处理器响应
+async def handle_multi_line_code(event: MessageEvent):
+    """
+    通用消息处理器，用于捕获多行模式下的代码输入。
+    """
+    # 使用 getattr 兼容私聊和群聊
+    session_key = (event.user_id, getattr(event, 'group_id', 'private'))
+    if session_key in multi_line_sessions:
+        # 取消超时任务
+        multi_line_sessions[session_key].cancel()
+        del multi_line_sessions[session_key]
+
+        # 对多行代码进行规范化处理
+        normalized_code = normalize_code(event.raw_message)
+        
        if not normalized_code:
            await event.reply("捕获到的代码为空或格式错误，已取消输入。")
            return