""" 插件管理器模块 负责扫描、加载和管理 `plugins` 目录下的所有插件。 """ import importlib import os import pkgutil import sys from typing import Set from .command_manager import CommandManager from ..utils.exceptions import SyncHandlerError, PluginLoadError, PluginReloadError, PluginNotFoundError from ..utils.logger import logger, ModuleLogger from ..utils.singleton import Singleton # 确保logger在模块级别可见 __all__ = ['PluginManager', 'logger'] class PluginManager(Singleton): """ 插件管理器类 """ def __init__(self, command_manager: "CommandManager" | None = None) -> None: """ 初始化插件管理器 :param command_manager: CommandManager的实例 """ # 检查是否已经初始化 if hasattr(self, '_command_manager'): return # 只有首次初始化时才执行 if command_manager: self._command_manager = command_manager self.loaded_plugins: Set[str] = set() # 创建模块专用日志记录器 self.logger = ModuleLogger("PluginManager") @property def command_manager(self): """ 获取命令管理器实例 """ return self._command_manager def load_all_plugins(self) -> None: """ 扫描并加载 `plugins` 目录下的所有插件。 """ # 使用 pathlib 获取更可靠的路径 # 当前文件: core/managers/plugin_manager.py # 目标: plugins/ current_dir = os.path.dirname(os.path.abspath(__file__)) # 回退两级到项目根目录 (core/managers -> core -> root) root_dir = os.path.dirname(os.path.dirname(current_dir)) plugin_dir = os.path.join(root_dir, "plugins") package_name = "plugins" if not os.path.exists(plugin_dir): self.logger.error(f"插件目录不存在: {plugin_dir}") return self.logger.info(f"正在从 {package_name} 加载插件 (路径: {plugin_dir})...") for _, module_name, is_pkg in pkgutil.iter_modules([plugin_dir]): full_module_name = f"{package_name}.{module_name}" action = "加载" # 初始化默认值 try: if full_module_name in self.loaded_plugins: self.command_manager.unload_plugin(full_module_name) module = importlib.reload(sys.modules[full_module_name]) action = "重载" else: module = importlib.import_module(full_module_name) action = "加载" if hasattr(module, "__plugin_meta__"): meta = getattr(module, "__plugin_meta__") self.command_manager.plugins[full_module_name] = meta self.loaded_plugins.add(full_module_name) type_str = "包" if is_pkg else "文件" self.logger.success(f" [{type_str}] 成功{action}: {module_name}") except SyncHandlerError as e: error = PluginLoadError( plugin_name=module_name, message=f"同步处理器错误: {str(e)}", original_error=e ) self.logger.error(f" 插件 {module_name} 加载失败: {error.message} (跳过此插件)") self.logger.log_custom_exception(error) except Exception as e: error = PluginLoadError( plugin_name=module_name, message=f"未知错误: {str(e)}", original_error=e ) self.logger.exception(f" 加载插件 {module_name} 失败: {error.message}") self.logger.log_custom_exception(error) def reload_plugin(self, full_module_name: str) -> None: """ 精确重载单个插件。 """ if full_module_name not in self.loaded_plugins: self.logger.warning(f"尝试重载一个未被加载的插件: {full_module_name},将按首次加载处理。") if full_module_name not in sys.modules: reload_error = PluginNotFoundError( plugin_name=full_module_name, message="模块未在sys.modules中找到" ) self.logger.error(f"重载失败: {reload_error.message}") self.logger.log_custom_exception(reload_error) return try: self.command_manager.unload_plugin(full_module_name) module = importlib.reload(sys.modules[full_module_name]) if hasattr(module, "__plugin_meta__"): meta = getattr(module, "__plugin_meta__") self.command_manager.plugins[full_module_name] = meta self.logger.success(f"插件 {full_module_name} 已成功重载。") except SyncHandlerError as e: error = PluginReloadError( plugin_name=full_module_name, message=f"同步处理器错误: {str(e)}", original_error=e ) self.logger.error(f"重载插件 {full_module_name} 失败: {error.message}") self.logger.log_custom_exception(error) except Exception as e: error = PluginReloadError( plugin_name=full_module_name, message=f"未知错误: {str(e)}", original_error=e ) self.logger.exception(f"重载插件 {full_module_name} 时发生错误: {error.message}") self.logger.log_custom_exception(error)