ErisPulse 2.1.15.dev3__py3-none-any.whl → 2.2.0.dev1__py3-none-any.whl

ErisPulse/Core/Event/__init__.py

@@ -0,0 +1,48 @@
+"""
+ErisPulse 事件处理模块
+
+提供统一的事件处理接口，支持命令、消息、通知、请求和元事件处理
+"""
+
+from .cmd import command
+from .message import message
+from .notice import notice
+from .request import request
+from .meta import meta
+from .manager import event_manager
+from . import exceptions
+from .. import config
+
+# 初始化默认配置
+def _setup_default_config():
+    """
+    设置默认配置
+    
+    {!--< internal-use >!--}
+    内部使用的方法，用于初始化默认配置
+    """
+    default_config = {
+        "command": {
+            "prefix": "/",
+            "case_sensitive": True
+        },
+        "message": {
+            "ignore_self": True
+        }
+    }
+    
+    current_config = config.getConfig("ErisPulse.event")
+    if current_config is None:
+        config.setConfig("ErisPulse.event", default_config)
+
+_setup_default_config()
+
+__all__ = [
+    "command",
+    "message", 
+    "notice",
+    "request",
+    "meta",
+    "event_manager",
+    "exceptions"
+]

ErisPulse/Core/Event/base.py

@@ -0,0 +1,117 @@
+"""
+ErisPulse 事件处理基础模块
+
+提供事件处理的核心功能，包括事件注册、处理和中间件支持
+"""
+
+from .. import adapter, logger
+from typing import Callable, Any, Dict, List, Optional, Union
+import asyncio
+
+class BaseEventHandler:
+    """
+    基础事件处理器
+    
+    提供事件处理的基本功能，包括处理器注册、中间件支持等
+    """
+    
+    def __init__(self, event_type: str, module_name: str = None):
+        """
+        初始化事件处理器
+        
+        :param event_type: 事件类型
+        :param module_name: 模块名称
+        """
+        self.event_type = event_type
+        self.module_name = module_name
+        self.handlers: List[Dict] = []
+        self.middlewares: List[Callable] = []
+    
+    def middleware(self, func: Callable) -> Callable:
+        """
+        添加中间件
+        
+        :param func: 中间件函数
+        :return: 中间件函数
+        """
+        self.middlewares.append(func)
+        return func
+    
+    def register(self, handler: Callable, priority: int = 0, condition: Callable = None):
+        """
+        注册事件处理器
+        
+        :param handler: 事件处理器函数
+        :param priority: 处理器优先级，数值越小优先级越高
+        :param condition: 处理器条件函数，返回True时才会执行处理器
+        """
+        handler_info = {
+            "func": handler,
+            "priority": priority,
+            "condition": condition,
+            "module": self.module_name
+        }
+        self.handlers.append(handler_info)
+        # 按优先级排序
+        self.handlers.sort(key=lambda x: x["priority"])
+        
+        # 注册到适配器
+        if self.event_type:
+            adapter.on(self.event_type)(self._process_event)
+    
+    def __call__(self, priority: int = 0, condition: Callable = None):
+        """
+        装饰器方式注册事件处理器
+        
+        :param priority: 处理器优先级
+        :param condition: 处理器条件函数
+        :return: 装饰器函数
+        """
+        def decorator(func: Callable):
+            self.register(func, priority, condition)
+            return func
+        return decorator
+    
+    async def _process_event(self, event: Dict[str, Any]):
+        """
+        处理事件
+        
+        {!--< internal-use >!--}
+        内部使用的方法，用于处理事件
+        
+        :param event: 事件数据
+        """
+        # 执行中间件
+        processed_event = event
+        for middleware in self.middlewares:
+            try:
+                if asyncio.iscoroutinefunction(middleware):
+                    processed_event = await middleware(processed_event)
+                else:
+                    processed_event = middleware(processed_event)
+            except Exception as e:
+                logger.error(f"中间件执行错误: {e}")
+        
+        # 执行处理器
+        for handler_info in self.handlers:
+            condition = handler_info.get("condition")
+            # 检查条件
+            if condition and not condition(processed_event):
+                continue
+                
+            handler = handler_info["func"]
+            try:
+                if asyncio.iscoroutinefunction(handler):
+                    await handler(processed_event)
+                else:
+                    handler(processed_event)
+            except Exception as e:
+                logger.error(f"事件处理器执行错误: {e}")
+    
+    def unregister(self, handler: Callable):
+        """
+        注销事件处理器
+        
+        :param handler: 要注销的事件处理器
+        """
+        self.handlers = [h for h in self.handlers if h["func"] != handler]

ErisPulse/Core/Event/cmd.py

@@ -0,0 +1,210 @@
+"""
+ErisPulse 命令处理模块
+
+提供基于装饰器的命令注册和处理功能
+"""
+
+from .base import BaseEventHandler
+from .manager import event_manager
+from .. import config, logger
+from typing import Callable, Union, List, Dict, Any, Optional
+import asyncio
+
+class CommandHandler:
+    """
+    命令处理器
+    
+    提供命令注册、解析和执行功能
+    """
+    
+    def __init__(self):
+        """
+        初始化命令处理器
+        """
+        self.commands: Dict[str, Dict] = {}
+        self.aliases: Dict[str, str] = {}  # 别名映射
+        self.groups: Dict[str, List[str]] = {}  # 命令组
+        self.prefix = config.getConfig("ErisPulse.event.command.prefix", "/")
+        self.handler = event_manager.create_event_handler("message", "command")
+        
+        # 注册消息处理器
+        self.handler.register(self._handle_message)
+    
+    def __call__(self, 
+                 name: Union[str, List[str]] = None, 
+                 aliases: List[str] = None,
+                 group: str = None,
+                 priority: int = 0,
+                 help: str = None,
+                 usage: str = None):
+        """
+        命令装饰器
+        
+        :param name: 命令名称，可以是字符串或字符串列表
+        :param aliases: 命令别名列表
+        :param group: 命令组名称
+        :param priority: 处理器优先级
+        :param help: 命令帮助信息
+        :param usage: 命令使用方法
+        :return: 装饰器函数
+        """
+        def decorator(func: Callable):
+            cmd_names = []
+            if isinstance(name, str):
+                cmd_names = [name]
+            elif isinstance(name, list):
+                cmd_names = name
+            else:
+                # 使用函数名作为命令名
+                cmd_names = [func.__name__]
+            
+            main_name = cmd_names[0]
+            
+            # 添加别名
+            alias_list = aliases or []
+            if len(cmd_names) > 1:
+                alias_list.extend(cmd_names[1:])
+            
+            # 注册命令
+            for cmd_name in cmd_names:
+                self.commands[cmd_name] = {
+                    "func": func,
+                    "help": help,
+                    "usage": usage,
+                    "group": group,
+                    "main_name": main_name
+                }
+                
+                # 注册别名映射
+                if cmd_name != main_name:
+                    self.aliases[cmd_name] = main_name
+            
+            # 添加到命令组
+            if group:
+                if group not in self.groups:
+                    self.groups[group] = []
+                for cmd_name in cmd_names:
+                    if cmd_name not in self.groups[group]:
+                        self.groups[group].append(cmd_name)
+            
+            return func
+        return decorator
+    
+    async def _handle_message(self, event: Dict[str, Any]):
+        """
+        处理消息事件中的命令
+        
+        {!--< internal-use >!--}
+        内部使用的方法，用于从消息中解析并执行命令
+        
+        :param event: 消息事件数据
+        """
+        # 检查是否为文本消息
+        if event.get("type") != "message":
+            return
+        
+        message_segments = event.get("message", [])
+        text_content = ""
+        for segment in message_segments:
+            if segment.get("type") == "text":
+                text_content = segment.get("data", {}).get("text", "")
+                break
+        
+        if not text_content:
+            return
+        
+        # 检查前缀
+        if not text_content.startswith(self.prefix):
+            return
+        
+        # 解析命令和参数
+        command_text = text_content[len(self.prefix):].strip()
+        parts = command_text.split()
+        if not parts:
+            return
+        
+        cmd_name = parts[0]
+        args = parts[1:] if len(parts) > 1 else []
+        
+        # 处理别名
+        actual_cmd_name = self.aliases.get(cmd_name, cmd_name)
+        
+        # 查找命令处理器
+        if actual_cmd_name in self.commands:
+            cmd_info = self.commands[actual_cmd_name]
+            handler = cmd_info["func"]
+            
+            # 添加命令相关信息到事件
+            command_info = {
+                "name": actual_cmd_name,
+                "main_name": cmd_info["main_name"],
+                "args": args,
+                "raw": command_text,
+                "help": cmd_info["help"],
+                "usage": cmd_info["usage"],
+                "group": cmd_info["group"]
+            }
+            
+            event["command"] = command_info
+            
+            try:
+                if asyncio.iscoroutinefunction(handler):
+                    await handler(event)
+                else:
+                    handler(event)
+            except Exception as e:
+                logger.error(f"命令执行错误: {e}")
+                # 可以发送错误信息给用户
+    
+    def get_command(self, name: str) -> Optional[Dict]:
+        """
+        获取命令信息
+        
+        :param name: 命令名称
+        :return: 命令信息字典，如果不存在则返回None
+        """
+        actual_name = self.aliases.get(name, name)
+        return self.commands.get(actual_name)
+    
+    def get_commands(self) -> Dict[str, Dict]:
+        """
+        获取所有命令
+        
+        :return: 命令信息字典
+        """
+        return self.commands
+    
+    def get_group_commands(self, group: str) -> List[str]:
+        """
+        获取命令组中的命令
+        
+        :param group: 命令组名称
+        :return: 命令名称列表
+        """
+        return self.groups.get(group, [])
+    
+    def help(self, command_name: str = None) -> str:
+        """
+        生成帮助信息
+        
+        :param command_name: 命令名称，如果为None则生成所有命令的帮助
+        :return: 帮助信息字符串
+        """
+        if command_name:
+            cmd_info = self.get_command(command_name)
+            if cmd_info:
+                help_text = cmd_info.get("help", "无帮助信息")
+                usage = cmd_info.get("usage", f"{self.prefix}{command_name}")
+                return f"命令: {command_name}\n用法: {usage}\n说明: {help_text}"
+            else:
+                return f"未找到命令: {command_name}"
+        else:
+            # 生成所有命令的帮助
+            help_lines = ["可用命令:"]
+            for cmd_name, cmd_info in self.commands.items():
+                if cmd_name == cmd_info["main_name"]:  # 只显示主命令
+                    help_text = cmd_info.get("help", "无说明")
+                    help_lines.append(f"  {self.prefix}{cmd_name} - {help_text}")
+            return "\n".join(help_lines)
+
+command = CommandHandler()

ErisPulse/Core/Event/exceptions.py

@@ -0,0 +1,37 @@
+"""
+ErisPulse 事件系统异常处理模块
+
+提供事件系统中可能发生的各种异常类型定义
+"""
+
+class EventException(Exception):
+    """
+    事件系统基础异常
+    
+    所有事件系统相关异常的基类
+    """
+    pass
+
+class CommandException(EventException):
+    """
+    命令处理异常
+    
+    当命令处理过程中发生错误时抛出
+    """
+    pass
+
+class EventHandlerException(EventException):
+    """
+    事件处理器异常
+    
+    当事件处理器执行过程中发生错误时抛出
+    """
+    pass
+
+class EventNotFoundException(EventException):
+    """
+    事件未找到异常
+    
+    当尝试获取不存在的事件处理器时抛出
+    """
+    pass

ErisPulse/Core/Event/manager.py

@@ -0,0 +1,127 @@
+"""
+ErisPulse 事件管理器
+
+提供全局事件管理功能，包括事件处理器创建、模块事件管理等
+"""
+
+from .. import logger
+from .base import BaseEventHandler
+from .exceptions import EventNotFoundException
+from typing import Dict, List, Optional, Callable, Any
+import asyncio
+
+class EventManager:
+    """
+    事件管理器
+    
+    管理所有事件处理器，提供全局事件处理功能
+    """
+    
+    def __init__(self):
+        """
+        初始化事件管理器
+        """
+        self.event_handlers: Dict[str, BaseEventHandler] = {}
+        self.module_events: Dict[str, List[str]] = {}
+        self.global_middlewares: List[Callable] = []
+    
+    def create_event_handler(self, event_type: str, module_name: str = None) -> BaseEventHandler:
+        """
+        创建事件处理器
+        
+        :param event_type: 事件类型
+        :param module_name: 模块名称
+        :return: 事件处理器实例
+        """
+        if event_type not in self.event_handlers:
+            handler = BaseEventHandler(event_type, module_name)
+            self.event_handlers[event_type] = handler
+        return self.event_handlers[event_type]
+    
+    def get_event_handler(self, event_type: str) -> Optional[BaseEventHandler]:
+        """
+        获取事件处理器
+        
+        :param event_type: 事件类型
+        :return: 事件处理器实例，如果不存在则返回None
+        """
+        return self.event_handlers.get(event_type)
+    
+    def register_module_event(self, module_name: str, event_type: str):
+        """
+        注册模块事件
+        
+        :param module_name: 模块名称
+        :param event_type: 事件类型
+        """
+        if module_name not in self.module_events:
+            self.module_events[module_name] = []
+        if event_type not in self.module_events[module_name]:
+            self.module_events[module_name].append(event_type)
+    
+    def middleware(self, func: Callable):
+        """
+        添加全局中间件
+        
+        :param func: 中间件函数
+        :return: 中间件函数
+        """
+        self.global_middlewares.append(func)
+        return func
+    
+    async def emit(self, event_type: str, event_data: Dict[str, Any]):
+        """
+        触发事件
+        
+        :param event_type: 事件类型
+        :param event_data: 事件数据
+        """
+        # 执行全局中间件
+        processed_data = event_data
+        for middleware in self.global_middlewares:
+            try:
+                if asyncio.iscoroutinefunction(middleware):
+                    processed_data = await middleware(processed_data)
+                else:
+                    processed_data = middleware(processed_data)
+            except Exception as e:
+                logger.error(f"全局中间件执行错误: {e}")
+        
+        # 触发事件处理器
+        if event_type in self.event_handlers:
+            handler = self.event_handlers[event_type]
+            await handler._process_event(processed_data)
+        
+        # 触发通配符处理器
+        if "*" in self.event_handlers:
+            handler = self.event_handlers["*"]
+            await handler._process_event(processed_data)
+    
+    def get_module_events(self, module_name: str) -> List[str]:
+        """
+        获取模块注册的事件
+        
+        :param module_name: 模块名称
+        :return: 事件类型列表
+        """
+        return self.module_events.get(module_name, [])
+    
+    def cleanup_module_events(self, module_name: str):
+        """
+        清理模块事件
+        
+        :param module_name: 模块名称
+        """
+        if module_name in self.module_events:
+            event_types = self.module_events[module_name]
+            for event_type in event_types:
+                if event_type in self.event_handlers:
+                    # 移除该模块注册的处理器
+                    handler = self.event_handlers[event_type]
+                    handler.handlers = [
+                        h for h in handler.handlers 
+                        if h.get("module") != module_name
+                    ]
+            del self.module_events[module_name]
+
+event_manager = EventManager()

ErisPulse/Core/Event/message.py

@@ -0,0 +1,89 @@
+"""
+ErisPulse 消息处理模块
+
+提供基于装饰器的消息事件处理功能
+"""
+
+from .base import BaseEventHandler
+from .manager import event_manager
+from typing import Callable, Dict, Any
+import asyncio
+
+class MessageHandler:
+    """
+    消息处理器
+    
+    提供不同类型消息事件的处理功能
+    """
+    
+    def __init__(self):
+        """
+        初始化消息处理器
+        """
+        self.handler = event_manager.create_event_handler("message", "message")
+    
+    def on_message(self, priority: int = 0):
+        """
+        消息事件装饰器
+        
+        :param priority: 处理器优先级
+        :return: 装饰器函数
+        """
+        def decorator(func: Callable):
+            self.handler.register(func, priority)
+            return func
+        return decorator
+    
+    def on_private_message(self, priority: int = 0):
+        """
+        私聊消息事件装饰器
+        
+        :param priority: 处理器优先级
+        :return: 装饰器函数
+        """
+        def condition(event: Dict[str, Any]) -> bool:
+            return event.get("detail_type") == "private"
+        
+        def decorator(func: Callable):
+            self.handler.register(func, priority, condition)
+            return func
+        return decorator
+    
+    def on_group_message(self, priority: int = 0):
+        """
+        群聊消息事件装饰器
+        
+        :param priority: 处理器优先级
+        :return: 装饰器函数
+        """
+        def condition(event: Dict[str, Any]) -> bool:
+            return event.get("detail_type") == "group"
+        
+        def decorator(func: Callable):
+            self.handler.register(func, priority, condition)
+            return func
+        return decorator
+    
+    def on_at_message(self, priority: int = 0):
+        """
+        @消息事件装饰器
+        
+        :param priority: 处理器优先级
+        :return: 装饰器函数
+        """
+        def condition(event: Dict[str, Any]) -> bool:
+            # 检查消息中是否有@机器人
+            message_segments = event.get("message", [])
+            self_id = event.get("self", {}).get("user_id")
+            
+            for segment in message_segments:
+                if segment.get("type") == "mention" and segment.get("data", {}).get("user_id") == self_id:
+                    return True
+            return False
+        
+        def decorator(func: Callable):
+            self.handler.register(func, priority, condition)
+            return func
+        return decorator
+
+message = MessageHandler()