gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(plugin_system): 实现组件的局部启用与禁用功能

Browse Source 
新增了 `/system plugin enable_local` 和 `/system plugin disable_local` 命令，允许管理员在指定的会话（群聊或私聊）中动态地启用或禁用插件组件。

- 通过 stream_id 对组件状态进行局部覆盖，提供了更精细的控制粒度。
- 引入新的 `plugin.manage.local` 权限节点以控制此高级功能。
- 在 API 层面增加了对组件存在性的检查，增强了系统的健壮性。
This commit is contained in:
minecraft1024a
2025-11-22 20:22:46 +08:00
parent 7b8660bb69
commit 4a4175c246
2 changed files with 78 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/plugin_system/apis/plugin_manage_api.py
View File

@@ -102,6 +102,12 @@ def set_component_enabled_local(stream_id: str, name: str, component_type: Compo
Returns:
bool: 操作成功则为 True。
"""
# 首先，检查组件是否存在
component_info = component_registry.get_component_info(name, component_type)
if not component_info:
logger.error(f"尝试设置局部状态失败：未找到组件 {name} ({component_type.value})。")
return False
# Chatter 唯一性保护
if component_type == ComponentType.CHATTER and not enabled:
# 检查当前 stream_id 上下文中的启用状态

74
src/plugins/built_in/system_management/plugin.py
View File

@@ -11,6 +11,7 @@ from typing import ClassVar
from src.chat.utils.prompt_component_manager import prompt_component_manager
from src.chat.utils.prompt_params import PromptParameters
from src.plugin_system.apis import (
chat_api,
plugin_manage_api,
)
from src.plugin_system.apis.logging_api import get_logger
@@ -21,6 +22,7 @@ from src.plugin_system.base.base_plugin import BasePlugin
from src.plugin_system.base.command_args import CommandArgs
from src.plugin_system.base.component_types import (
ChatType,
ComponentType,
PermissionNodeField,
PlusCommandInfo,
)
@@ -103,6 +105,9 @@ class SystemCommand(PlusCommand):
• `/system plugin load <插件名>` - 加载指定插件
• `/system plugin reload <插件名>` - 重新加载指定插件
• `/system plugin reload_all` - 重新加载所有插件
🎯 局部控制 (需要 `system.plugin.manage.local` 权限):
• `/system plugin enable_local <类型> <名称> [group <群号> | private <QQ号>]` - 在指定会话局部启用组件
• `/system plugin disable_local <类型> <名称> [group <群号> | private <QQ号>]` - 在指定会话局部禁用组件
"""
elif target == "permission":
help_text = """📋 权限管理命令帮助
@@ -157,6 +162,10 @@ class SystemCommand(PlusCommand):
await self._reload_plugin(remaining_args[0])
elif action in ["reload_all", "重载全部"]:
await self._reload_all_plugins()
elif action in ["enable_local", "局部启用"] and len(remaining_args) >= 2:
await self._set_local_component_state(remaining_args, enabled=True)
elif action in ["disable_local", "局部禁用"] and len(remaining_args) >= 2:
await self._set_local_component_state(remaining_args, enabled=False)
else:
await self.send_text("❌ 插件管理命令不合法\n使用 /system plugin help 查看帮助")
@@ -309,7 +318,7 @@ class SystemCommand(PlusCommand):
@require_permission("prompt.view", deny_message="❌ 你没有查看提示词注入信息的权限")
async def _list_prompt_components(self):
"""列出所有已注册的提示词组件"""
components = prompt_component_manager.get_registered_prompt_component_info()
components = await prompt_component_manager.get_registered_prompt_component_info()
if not components:
await self.send_text("🧩 当前没有已注册的提示词组件")
return
@@ -391,7 +400,7 @@ class SystemCommand(PlusCommand):
@require_permission("prompt.view", deny_message="❌ 你没有查看提示词组件信息的权限")
async def _show_prompt_component_info(self, component_name: str):
"""显示特定提示词组件的详细信息"""
all_components = prompt_component_manager.get_registered_prompt_component_info()
all_components = await prompt_component_manager.get_registered_prompt_component_info()
target_component = next((comp for comp in all_components if comp.name == component_name), None)
@@ -486,6 +495,63 @@ class SystemCommand(PlusCommand):
else:
await self.send_text("⚠️ 部分插件重载失败，请检查日志。")
@require_permission("plugin.manage.local", deny_message="❌ 你没有局部管理插件组件的权限")
async def _set_local_component_state(self, args: list[str], enabled: bool):
"""在局部范围内启用或禁用一个组件"""
# 命令格式: <component_type> <component_name> [group <group_id> | private <user_id>]
if len(args) < 2:
action = "enable_local" if enabled else "disable_local"
await self.send_text(f"❌ 用法: /system plugin {action} <类型> <名称> [group <群号> | private <QQ号>]")
return
comp_type_str = args[0]
comp_name = args[1]
stream_id = self.message.chat_info.stream_id # 默认作用于当前会话
if len(args) >= 4:
context_type = args[2].lower()
context_id = args[3]
target_stream = None
if context_type == "group":
target_stream = chat_api.get_stream_by_group_id(
group_id=context_id,
platform=self.message.chat_info.platform
)
elif context_type == "private":
target_stream = chat_api.get_stream_by_user_id(
user_id=context_id,
platform=self.message.chat_info.platform
)
else:
await self.send_text("❌ 无效的作用域类型，请使用 'group' 或 'private'。")
return
if not target_stream:
await self.send_text(f"❌ 在当前平台找不到指定的 {context_type}: `{context_id}`。")
return
stream_id = target_stream.stream_id
try:
component_type = ComponentType(comp_type_str.lower())
except ValueError:
await self.send_text(f"❌ 无效的组件类型: '{comp_type_str}'。有效类型: {', '.join([t.value for t in ComponentType])}")
return
success = plugin_manage_api.set_component_enabled_local(
stream_id=stream_id,
name=comp_name,
component_type=component_type,
enabled=enabled
)
action_text = "启用" if enabled else "禁用"
if success:
await self.send_text(f"✅ 在会话 `{stream_id}` 中，已成功将组件 `{comp_name}` ({comp_type_str}) 设置为 {action_text} 状态。")
else:
await self.send_text(f"❌ 操作失败。可能无法禁用最后一个启用的 Chatter，或组件不存在。请检查日志。")
# =================================================================
# Permission Management Section
@@ -731,4 +797,8 @@ class SystemManagementPlugin(BasePlugin):
node_name="schedule.manage",
description="定时任务管理：暂停和恢复定时任务",
),
PermissionNodeField(
node_name="plugin.manage.local",
description="局部插件管理：在指定会话中启用或禁用组件",
),
]