Class PluginHookManager

RoverSign/utils/bot_send_hook.py:14–51

插件 hook 管理器

class PluginHookManager:
    """插件 hook 管理器"""

    def __init__(self, plugin_name: str):
        self.plugin_name = plugin_name
        self.target_send_hooks: list[Callable] = []
        self.user_activity_hooks: list[Callable] = []
        self.group_activity_hooks: list[Callable] = []

    def register_target_send_hook(self, func: Callable):
        """注册 target_send 方法 hook"""
        existing = [h for h in self.target_send_hooks if h.__name__ == func.__name__]
        if existing:
            self.target_send_hooks[:] = [h for h in self.target_send_hooks if h.__name__ != func.__name__]
            logger.debug(f"[库洛签到·BotHook] 更新 target_send hook: {func.__name__}")
        else:
            logger.debug(f"[库洛签到·BotHook] 注册 target_send hook: {func.__name__}")
        self.target_send_hooks.append(func)

    def register_user_activity_hook(self, func: Callable):
        """注册用户活跃度 hook"""
        existing = [h for h in self.user_activity_hooks if h.__name__ == func.__name__]
        if existing:
            self.user_activity_hooks[:] = [h for h in self.user_activity_hooks if h.__name__ != func.__name__]
            logger.debug(f"[库洛签到·BotHook] 更新 user_activity hook: {func.__name__}")
        else:
            logger.debug(f"[库洛签到·BotHook] 注册 user_activity hook: {func.__name__}")
        self.user_activity_hooks.append(func)

    def register_group_activity_hook(self, func: Callable):
        """注册群活跃度 hook"""
        existing = [h for h in self.group_activity_hooks if h.__name__ == func.__name__]
        if existing:
            self.group_activity_hooks[:] = [h for h in self.group_activity_hooks if h.__name__ != func.__name__]
            logger.debug(f"[库洛签到·BotHook] 更新 group_activity hook: {func.__name__}")
        else:
            logger.debug(f"[库洛签到·BotHook] 注册 group_activity hook: {func.__name__}")
        self.group_activity_hooks.append(func)


def get_or_create_hook_manager(plugin_name: str) -> PluginHookManager: