claude-mpm 4.0.31__py3-none-any.whl → 4.0.34__py3-none-any.whl

claude_mpm/services/event_bus/event_bus.py

@@ -0,0 +1,334 @@
+"""Event Bus implementation using pyee.
+
+WHY pyee over alternatives:
+- AsyncIOEventEmitter supports both sync and async handlers
+- Battle-tested library with minimal dependencies
+- Simple EventEmitter pattern familiar to developers
+- Thread-safe for multi-threaded environments
+- Efficient event dispatch with minimal overhead
+"""
+
+import asyncio
+import logging
+import threading
+from datetime import datetime
+from typing import Any, Callable, Dict, List, Optional, Set
+from pyee import AsyncIOEventEmitter
+
+# Configure logger
+logger = logging.getLogger(__name__)
+
+
+class EventBus:
+    """Singleton Event Bus for decoupled event handling.
+    
+    WHY singleton pattern:
+    - Ensures single point of event coordination
+    - Prevents duplicate event processing
+    - Simplifies configuration and management
+    - Thread-safe initialization with proper locking
+    """
+    
+    _instance: Optional["EventBus"] = None
+    _lock = threading.Lock()
+    
+    def __new__(cls) -> "EventBus":
+        """Ensure singleton instance creation."""
+        if cls._instance is None:
+            with cls._lock:
+                if cls._instance is None:
+                    cls._instance = super().__new__(cls)
+        return cls._instance
+    
+    def __init__(self):
+        """Initialize the event bus once."""
+        # Only initialize once
+        if hasattr(self, "_initialized"):
+            return
+            
+        self._initialized = True
+        self._emitter = AsyncIOEventEmitter()
+        self._enabled = True
+        self._event_filters: Set[str] = set()
+        self._stats = {
+            "events_published": 0,
+            "events_filtered": 0,
+            "events_failed": 0,
+            "last_event_time": None
+        }
+        self._debug = False
+        
+        # Event history for debugging (limited size)
+        self._event_history: List[Dict[str, Any]] = []
+        self._max_history_size = 100
+        
+        logger.info("EventBus initialized")
+    
+    @classmethod
+    def get_instance(cls) -> "EventBus":
+        """Get the singleton EventBus instance.
+        
+        Returns:
+            EventBus: The singleton instance
+        """
+        return cls()
+    
+    def enable(self) -> None:
+        """Enable event bus processing."""
+        self._enabled = True
+        logger.info("EventBus enabled")
+    
+    def disable(self) -> None:
+        """Disable event bus processing (for testing or maintenance)."""
+        self._enabled = False
+        logger.info("EventBus disabled")
+    
+    def set_debug(self, debug: bool) -> None:
+        """Enable or disable debug logging.
+        
+        Args:
+            debug: Whether to enable debug logging
+        """
+        self._debug = debug
+        if debug:
+            logger.setLevel(logging.DEBUG)
+        else:
+            logger.setLevel(logging.INFO)
+    
+    def add_filter(self, pattern: str) -> None:
+        """Add an event filter pattern.
+        
+        Only events matching filter patterns will be processed.
+        Use wildcards: 'hook.*' matches all hook events.
+        
+        Args:
+            pattern: Event name pattern to allow
+        """
+        self._event_filters.add(pattern)
+        logger.debug(f"Added event filter: {pattern}")
+    
+    def remove_filter(self, pattern: str) -> None:
+        """Remove an event filter pattern.
+        
+        Args:
+            pattern: Event name pattern to remove
+        """
+        self._event_filters.discard(pattern)
+        logger.debug(f"Removed event filter: {pattern}")
+    
+    def clear_filters(self) -> None:
+        """Clear all event filters (allow all events)."""
+        self._event_filters.clear()
+        logger.debug("Cleared all event filters")
+    
+    def _should_process_event(self, event_type: str) -> bool:
+        """Check if an event should be processed based on filters.
+        
+        Args:
+            event_type: The event type to check
+            
+        Returns:
+            bool: True if event should be processed
+        """
+        # If no filters, process all events
+        if not self._event_filters:
+            return True
+            
+        # Check if event matches any filter
+        for filter_pattern in self._event_filters:
+            if filter_pattern.endswith("*"):
+                # Wildcard pattern
+                prefix = filter_pattern[:-1]
+                if event_type.startswith(prefix):
+                    return True
+            elif event_type == filter_pattern:
+                # Exact match
+                return True
+                
+        return False
+    
+    def publish(self, event_type: str, data: Any) -> bool:
+        """Publish an event synchronously (for use from sync contexts like hooks).
+        
+        This method is thread-safe and can be called from any thread.
+        Events are dispatched asynchronously to handlers.
+        
+        Args:
+            event_type: The event type (e.g., 'hook.pre_tool')
+            data: The event data
+            
+        Returns:
+            bool: True if event was published, False if filtered or disabled
+        """
+        if not self._enabled:
+            if self._debug:
+                logger.debug(f"EventBus disabled, dropping event: {event_type}")
+            return False
+        
+        # Check filters
+        if not self._should_process_event(event_type):
+            self._stats["events_filtered"] += 1
+            if self._debug:
+                logger.debug(f"Event filtered out: {event_type}")
+            return False
+        
+        try:
+            # Record event in history
+            self._record_event(event_type, data)
+            
+            # Emit event (pyee handles thread safety)
+            self._emitter.emit(event_type, data)
+            
+            # Update stats
+            self._stats["events_published"] += 1
+            self._stats["last_event_time"] = datetime.now().isoformat()
+            
+            if self._debug:
+                logger.debug(f"Published event: {event_type}")
+            
+            return True
+            
+        except Exception as e:
+            self._stats["events_failed"] += 1
+            logger.error(f"Failed to publish event {event_type}: {e}")
+            return False
+    
+    async def publish_async(self, event_type: str, data: Any) -> bool:
+        """Publish an event from an async context.
+        
+        Args:
+            event_type: The event type
+            data: The event data
+            
+        Returns:
+            bool: True if event was published
+        """
+        # Just delegate to sync publish (pyee handles both)
+        return self.publish(event_type, data)
+    
+    def on(self, event_type: str, handler: Callable) -> None:
+        """Register an event handler.
+        
+        The handler can be sync or async. For async handlers,
+        they will be scheduled on the event loop.
+        
+        Args:
+            event_type: The event type to listen for (supports wildcards)
+            handler: The handler function
+        """
+        if event_type.endswith("*"):
+            # Register for wildcard pattern
+            prefix = event_type[:-1]
+            
+            # Create a wrapper that checks event names
+            def wildcard_wrapper(actual_event_type: str):
+                def wrapper(data):
+                    if actual_event_type.startswith(prefix):
+                        if asyncio.iscoroutinefunction(handler):
+                            return handler(actual_event_type, data)
+                        else:
+                            handler(actual_event_type, data)
+                return wrapper
+            
+            # Register for all possible events (we'll filter in the wrapper)
+            # For now, register common prefixes
+            for common_event in ["hook", "socketio", "system", "agent"]:
+                if common_event.startswith(prefix) or prefix.startswith(common_event):
+                    self._emitter.on(f"{common_event}.*", wildcard_wrapper(f"{common_event}.*"))
+        else:
+            # Regular event registration
+            self._emitter.on(event_type, handler)
+        
+        logger.debug(f"Registered handler for: {event_type}")
+    
+    def once(self, event_type: str, handler: Callable) -> None:
+        """Register a one-time event handler.
+        
+        Args:
+            event_type: The event type to listen for
+            handler: The handler function
+        """
+        self._emitter.once(event_type, handler)
+        logger.debug(f"Registered one-time handler for: {event_type}")
+    
+    def remove_listener(self, event_type: str, handler: Callable) -> None:
+        """Remove an event handler.
+        
+        Args:
+            event_type: The event type
+            handler: The handler to remove
+        """
+        self._emitter.remove_listener(event_type, handler)
+        logger.debug(f"Removed handler for: {event_type}")
+    
+    def remove_all_listeners(self, event_type: Optional[str] = None) -> None:
+        """Remove all listeners for an event type, or all listeners.
+        
+        Args:
+            event_type: Optional event type. If None, removes all listeners.
+        """
+        if event_type:
+            self._emitter.remove_all_listeners(event_type)
+            logger.debug(f"Removed all handlers for: {event_type}")
+        else:
+            self._emitter.remove_all_listeners()
+            logger.debug("Removed all event handlers")
+    
+    def _record_event(self, event_type: str, data: Any) -> None:
+        """Record event in history for debugging.
+        
+        Args:
+            event_type: The event type
+            data: The event data
+        """
+        event_record = {
+            "timestamp": datetime.now().isoformat(),
+            "type": event_type,
+            "data": data
+        }
+        
+        self._event_history.append(event_record)
+        
+        # Trim history if too large
+        if len(self._event_history) > self._max_history_size:
+            self._event_history = self._event_history[-self._max_history_size:]
+    
+    def get_stats(self) -> Dict[str, Any]:
+        """Get event bus statistics.
+        
+        Returns:
+            dict: Statistics about event processing
+        """
+        return {
+            **self._stats,
+            "enabled": self._enabled,
+            "filters_active": len(self._event_filters) > 0,
+            "filter_count": len(self._event_filters),
+            "history_size": len(self._event_history)
+        }
+    
+    def get_recent_events(self, limit: int = 10) -> List[Dict[str, Any]]:
+        """Get recent events from history.
+        
+        Args:
+            limit: Maximum number of events to return
+            
+        Returns:
+            list: Recent events
+        """
+        return self._event_history[-limit:]
+    
+    def clear_history(self) -> None:
+        """Clear the event history."""
+        self._event_history.clear()
+        logger.debug("Cleared event history")
+    
+    def reset_stats(self) -> None:
+        """Reset event statistics."""
+        self._stats = {
+            "events_published": 0,
+            "events_filtered": 0,
+            "events_failed": 0,
+            "last_event_time": None
+        }
+        logger.debug("Reset event statistics")

claude_mpm/services/event_bus/relay.py

@@ -0,0 +1,301 @@
+"""Socket.IO Relay - Consumes events from EventBus and relays to Socket.IO.
+
+WHY separate relay component:
+- Single point of Socket.IO connection management
+- Isolates Socket.IO failures from event producers
+- Enables graceful degradation when Socket.IO unavailable
+- Simplifies testing by mocking just the relay
+- Supports batching and retry logic in one place
+"""
+
+import asyncio
+import logging
+import os
+import time
+from typing import Any, Dict, Optional
+from datetime import datetime
+
+# Socket.IO imports
+try:
+    import socketio
+    SOCKETIO_AVAILABLE = True
+except ImportError:
+    SOCKETIO_AVAILABLE = False
+    socketio = None
+
+from .event_bus import EventBus
+
+# Configure logger
+logger = logging.getLogger(__name__)
+
+
+class SocketIORelay:
+    """Relay events from EventBus to Socket.IO clients.
+    
+    WHY relay pattern:
+    - Decouples event production from Socket.IO emission
+    - Handles connection failures without affecting producers
+    - Provides single point for Socket.IO configuration
+    - Enables event batching and optimization
+    - Simplifies debugging with centralized logging
+    """
+    
+    def __init__(self, port: Optional[int] = None):
+        """Initialize the Socket.IO relay.
+        
+        Args:
+            port: Socket.IO server port (defaults to env var or 8765)
+        """
+        self.port = port or int(os.environ.get("CLAUDE_MPM_SOCKETIO_PORT", "8765"))
+        self.event_bus = EventBus.get_instance()
+        self.client: Optional[Any] = None
+        self.connected = False
+        self.enabled = True
+        self.debug = os.environ.get("CLAUDE_MPM_RELAY_DEBUG", "false").lower() == "true"
+        
+        # Connection retry settings
+        self.max_retries = 3
+        self.retry_delay = 0.5
+        self.last_connection_attempt = 0
+        self.connection_cooldown = 5.0  # Seconds between connection attempts
+        
+        # Statistics
+        self.stats = {
+            "events_relayed": 0,
+            "events_failed": 0,
+            "connection_failures": 0,
+            "last_relay_time": None
+        }
+        
+        if not SOCKETIO_AVAILABLE:
+            logger.warning("Socket.IO not available, relay will be disabled")
+            self.enabled = False
+    
+    def enable(self) -> None:
+        """Enable the relay."""
+        if not SOCKETIO_AVAILABLE:
+            logger.warning("Cannot enable relay: Socket.IO not available")
+            return
+        self.enabled = True
+        logger.info("SocketIO relay enabled")
+    
+    def disable(self) -> None:
+        """Disable the relay."""
+        self.enabled = False
+        if self.client and self.connected:
+            try:
+                self.client.disconnect()
+            except:
+                pass
+        logger.info("SocketIO relay disabled")
+    
+    def _create_client(self) -> bool:
+        """Create and connect Socket.IO client.
+        
+        Returns:
+            bool: True if connection successful
+        """
+        if not SOCKETIO_AVAILABLE or not self.enabled:
+            return False
+        
+        # Check connection cooldown
+        current_time = time.time()
+        if current_time - self.last_connection_attempt < self.connection_cooldown:
+            return False
+        
+        self.last_connection_attempt = current_time
+        
+        try:
+            # Create new client
+            self.client = socketio.Client(
+                reconnection=True,
+                reconnection_attempts=3,
+                reconnection_delay=1,
+                logger=False,
+                engineio_logger=False
+            )
+            
+            # Connect to server
+            self.client.connect(
+                f"http://localhost:{self.port}",
+                wait=True,
+                wait_timeout=2.0,
+                transports=['websocket', 'polling']
+            )
+            
+            self.connected = True
+            logger.info(f"SocketIO relay connected to port {self.port}")
+            return True
+            
+        except Exception as e:
+            self.stats["connection_failures"] += 1
+            if self.debug:
+                logger.debug(f"Failed to connect to Socket.IO server: {e}")
+            self.connected = False
+            self.client = None
+            return False
+    
+    def _ensure_connection(self) -> bool:
+        """Ensure Socket.IO client is connected.
+        
+        Returns:
+            bool: True if connected or reconnected
+        """
+        if not self.enabled:
+            return False
+        
+        # Check existing connection
+        if self.client and self.connected:
+            try:
+                # Verify connection is still alive
+                if self.client.connected:
+                    return True
+            except:
+                pass
+        
+        # Need to create or reconnect
+        return self._create_client()
+    
+    async def relay_event(self, event_type: str, data: Any) -> bool:
+        """Relay an event to Socket.IO.
+        
+        Args:
+            event_type: The event type
+            data: The event data
+            
+        Returns:
+            bool: True if successfully relayed
+        """
+        if not self.enabled:
+            return False
+        
+        # Ensure we have a connection
+        if not self._ensure_connection():
+            self.stats["events_failed"] += 1
+            return False
+        
+        try:
+            # Emit to Socket.IO
+            self.client.emit("claude_event", {
+                "event": "claude_event",
+                "type": event_type.split(".")[0] if "." in event_type else event_type,
+                "subtype": event_type.split(".", 1)[1] if "." in event_type else "generic",
+                "timestamp": data.get("timestamp", datetime.now().isoformat()),
+                "data": data,
+                "source": "event_bus"
+            })
+            
+            # Update statistics
+            self.stats["events_relayed"] += 1
+            self.stats["last_relay_time"] = datetime.now().isoformat()
+            
+            if self.debug:
+                logger.debug(f"Relayed event to Socket.IO: {event_type}")
+            
+            return True
+            
+        except Exception as e:
+            self.stats["events_failed"] += 1
+            if self.debug:
+                logger.debug(f"Failed to relay event {event_type}: {e}")
+            # Mark connection as failed for retry
+            self.connected = False
+            return False
+    
+    def start(self) -> None:
+        """Start the relay by subscribing to EventBus events.
+        
+        This sets up listeners for all hook events and relays them
+        to Socket.IO clients.
+        """
+        if not self.enabled:
+            logger.warning("Cannot start relay: disabled or Socket.IO not available")
+            return
+        
+        # Define async handler for events
+        async def handle_hook_event(event_type: str, data: Any):
+            """Handle events from the event bus."""
+            # Only relay hook events by default
+            if event_type.startswith("hook."):
+                await self.relay_event(event_type, data)
+        
+        # Subscribe to all hook events
+        self.event_bus.on("hook.*", handle_hook_event)
+        
+        # Also subscribe to specific high-priority events
+        for event in ["hook.pre_tool", "hook.post_tool", "hook.subagent_stop", 
+                      "hook.user_prompt", "hook.assistant_response"]:
+            self.event_bus.on(event, lambda data, evt=event: 
+                             asyncio.create_task(self.relay_event(evt, data)))
+        
+        logger.info("SocketIO relay started and subscribed to events")
+    
+    def stop(self) -> None:
+        """Stop the relay and clean up resources."""
+        # Disconnect Socket.IO client
+        if self.client and self.connected:
+            try:
+                self.client.disconnect()
+            except:
+                pass
+        
+        # Could remove event bus listeners here if needed
+        # For now, let them be cleaned up naturally
+        
+        self.connected = False
+        self.client = None
+        logger.info("SocketIO relay stopped")
+    
+    def get_stats(self) -> Dict[str, Any]:
+        """Get relay statistics.
+        
+        Returns:
+            dict: Statistics about relay operation
+        """
+        return {
+            **self.stats,
+            "enabled": self.enabled,
+            "connected": self.connected,
+            "port": self.port
+        }
+
+
+# Global relay instance
+_relay_instance: Optional[SocketIORelay] = None
+
+
+def get_relay(port: Optional[int] = None) -> SocketIORelay:
+    """Get or create the global SocketIO relay instance.
+    
+    Args:
+        port: Optional port number
+        
+    Returns:
+        SocketIORelay: The relay instance
+    """
+    global _relay_instance
+    if _relay_instance is None:
+        _relay_instance = SocketIORelay(port)
+    return _relay_instance
+
+
+def start_relay(port: Optional[int] = None) -> SocketIORelay:
+    """Start the global SocketIO relay.
+    
+    Args:
+        port: Optional port number
+        
+    Returns:
+        SocketIORelay: The started relay instance
+    """
+    relay = get_relay(port)
+    relay.start()
+    return relay
+
+
+def stop_relay() -> None:
+    """Stop the global SocketIO relay."""
+    global _relay_instance
+    if _relay_instance:
+        _relay_instance.stop()
+        _relay_instance = None

claude_mpm/services/events/__init__.py

@@ -0,0 +1,44 @@
+"""
+Event Bus System for Claude MPM
+===============================
+
+A decoupled event system that separates event producers from consumers,
+providing reliable, testable, and maintainable event handling.
+
+Key Components:
+- EventBus: Core pub/sub system
+- Event: Standard event format
+- IEventProducer: Interface for event producers
+- IEventConsumer: Interface for event consumers
+- Various consumer implementations
+"""
+
+from .core import Event, EventBus, EventMetadata, EventPriority
+from .interfaces import IEventConsumer, IEventProducer, ConsumerConfig
+from .consumers import (
+    SocketIOConsumer,
+    LoggingConsumer,
+    MetricsConsumer,
+    DeadLetterConsumer,
+)
+from .producers import HookEventProducer, SystemEventProducer
+
+__all__ = [
+    # Core
+    "Event",
+    "EventBus",
+    "EventMetadata",
+    "EventPriority",
+    # Interfaces
+    "IEventConsumer",
+    "IEventProducer",
+    "ConsumerConfig",
+    # Consumers
+    "SocketIOConsumer",
+    "LoggingConsumer",
+    "MetricsConsumer",
+    "DeadLetterConsumer",
+    # Producers
+    "HookEventProducer",
+    "SystemEventProducer",
+]

claude_mpm/services/events/consumers/__init__.py

@@ -0,0 +1,18 @@
+"""
+Event Bus Consumers
+==================
+
+Various consumer implementations for processing events from the event bus.
+"""
+
+from .dead_letter import DeadLetterConsumer
+from .logging import LoggingConsumer
+from .metrics import MetricsConsumer
+from .socketio import SocketIOConsumer
+
+__all__ = [
+    "SocketIOConsumer",
+    "LoggingConsumer",
+    "MetricsConsumer",
+    "DeadLetterConsumer",
+]