claude-mpm 4.1.6__py3-none-any.whl → 4.24.0__py3-none-any.whl

claude_mpm/tools/code_tree_events.py

@@ -0,0 +1,420 @@
+#!/usr/bin/env python3
+"""
+Code Tree Event Emitter
+=======================
+
+WHY: Provides incremental event emission for real-time code tree visualization
+in the dashboard. Uses Socket.IO to stream events as code is analyzed.
+
+DESIGN DECISIONS:
+- Support lazy loading with directory discovery
+- Emit events only for structural elements (directories, files, main functions)
+- Filter out internal functions and handlers
+- Batch events for performance (emit every 10 nodes or 100ms)
+- Use clean event subtypes without colons
+"""
+
+import json
+import threading
+import time
+from collections import deque
+from dataclasses import asdict, dataclass
+from datetime import datetime, timezone
+from typing import Any, Dict, List, Optional
+
+from ..core.enums import OperationResult
+
+try:
+    import socketio
+
+    SOCKETIO_AVAILABLE = True
+except ImportError:
+    SOCKETIO_AVAILABLE = False
+    socketio = None
+
+from pathlib import Path
+
+from ..core.logging_config import get_logger
+
+
+@dataclass
+class CodeNodeEvent:
+    """Represents a code node discovery event."""
+
+    file_path: str
+    node_type: str  # 'class', 'function', 'method', 'module', 'import'
+    name: str
+    line_start: int
+    line_end: int
+    complexity: int = 0
+    has_docstring: bool = False
+    decorators: List[str] = None
+    parent: Optional[str] = None
+    children_count: int = 0
+    language: str = "python"
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Convert to dictionary for JSON serialization."""
+        data = asdict(self)
+        data["timestamp"] = datetime.now(timezone.utc).isoformat()
+        return data
+
+
+class CodeTreeEventEmitter:
+    """Emits code analysis events to dashboard via Socket.IO.
+
+    WHY: Real-time streaming of analysis progress allows users to see
+    the tree building incrementally, providing feedback for large codebases.
+    """
+
+    # Event types - using underscores for clean subtypes
+    EVENT_DIRECTORY_DISCOVERED = "code:directory:discovered"
+    EVENT_FILE_DISCOVERED = "code:file:discovered"
+    EVENT_FILE_ANALYZED = "code:file:analyzed"
+    EVENT_NODE_FOUND = "code:node:found"
+    EVENT_ANALYSIS_START = "code:analysis:start"
+    EVENT_ANALYSIS_COMPLETE = "code:analysis:complete"
+    EVENT_PROGRESS = "code:analysis:progress"
+    EVENT_ERROR = "code:analysis:error"
+
+    def __init__(
+        self,
+        socketio_url: str = "http://localhost:8765",
+        batch_size: int = 10,
+        batch_timeout: float = 0.1,
+        use_stdout: bool = False,
+    ):
+        """Initialize event emitter.
+
+        Args:
+            socketio_url: Socket.IO server URL
+            batch_size: Number of events to batch before emitting
+            batch_timeout: Maximum time to wait before emitting batch (seconds)
+            use_stdout: If True, emit to stdout instead of Socket.IO
+        """
+        self.logger = get_logger(__name__)
+        self.socketio_url = socketio_url
+        self.batch_size = batch_size
+        self.batch_timeout = batch_timeout
+        self.use_stdout = use_stdout
+
+        # Event buffer for batching
+        self.event_buffer = deque()
+        self.buffer_lock = threading.Lock()
+        self.last_emit_time = time.time()
+
+        # Socket.IO client
+        self.sio = None
+        self.connected = False
+        if not use_stdout:
+            self._init_socketio()
+
+        # Statistics
+        self.stats = {
+            "events_sent": 0,
+            "events_buffered": 0,
+            "files_processed": 0,
+            "nodes_found": 0,
+            "errors": 0,
+            "start_time": None,
+        }
+
+        # Background task for periodic emission
+        self._emit_task = None
+        self._stop_event = threading.Event()
+
+    def _init_socketio(self):
+        """Initialize Socket.IO client connection."""
+        if not SOCKETIO_AVAILABLE:
+            self.logger.warning("Socket.IO not available - events will be logged only")
+            return
+
+        try:
+            self.sio = socketio.Client(
+                reconnection=True,
+                reconnection_attempts=3,
+                reconnection_delay=1,
+                logger=False,
+                engineio_logger=False,
+            )
+
+            @self.sio.event
+            def connect():
+                self.connected = True
+                self.logger.info(
+                    f"Connected to Socket.IO server at {self.socketio_url}"
+                )
+
+            @self.sio.event
+            def disconnect():
+                self.connected = False
+                self.logger.info("Disconnected from Socket.IO server")
+
+            # Attempt connection
+            self.sio.connect(self.socketio_url, wait_timeout=2)
+
+        except Exception as e:
+            self.logger.warning(f"Failed to connect to Socket.IO: {e}")
+            self.sio = None
+
+    def start(self):
+        """Start the event emitter and background tasks."""
+        self.stats["start_time"] = datetime.now(timezone.utc)
+        self._stop_event.clear()
+
+        # Start background emit task
+        self._emit_task = threading.Thread(target=self._emit_loop, daemon=True)
+        self._emit_task.start()
+
+        # Emit analysis start event
+        self.emit(
+            self.EVENT_ANALYSIS_START,
+            {
+                "timestamp": datetime.now(timezone.utc).isoformat(),
+                "batch_size": self.batch_size,
+                "batch_timeout": self.batch_timeout,
+            },
+        )
+
+    def stop(self):
+        """Stop the event emitter and flush remaining events."""
+        # Flush remaining events
+        self._flush_events()
+
+        # Emit analysis complete event with statistics
+        self.emit(
+            self.EVENT_ANALYSIS_COMPLETE,
+            {
+                "timestamp": datetime.now(timezone.utc).isoformat(),
+                "duration": (
+                    (
+                        datetime.now(timezone.utc) - self.stats["start_time"]
+                    ).total_seconds()
+                    if self.stats["start_time"]
+                    else 0
+                ),
+                "stats": self.stats,
+            },
+        )
+
+        # Stop background task
+        self._stop_event.set()
+        if self._emit_task:
+            self._emit_task.join(timeout=1)
+
+        # Disconnect Socket.IO
+        if self.sio and self.connected:
+            self.sio.disconnect()
+
+    def emit(self, event_type: str, data: Dict[str, Any], batch: bool = False):
+        """Emit an event, either immediately or batched.
+
+        Args:
+            event_type: Type of event to emit
+            data: Event data
+            batch: Whether to batch this event
+        """
+        event = {
+            "type": event_type,
+            "data": data,
+            "timestamp": datetime.now(timezone.utc).isoformat(),
+        }
+
+        if batch:
+            with self.buffer_lock:
+                self.event_buffer.append(event)
+                self.stats["events_buffered"] += 1
+
+                # Check if we should flush
+                if len(self.event_buffer) >= self.batch_size:
+                    self._flush_events()
+        else:
+            self._emit_event(event)
+
+    def emit_directory_discovered(self, dir_path: str, children: List[Dict[str, Any]]):
+        """Emit directory discovery event."""
+        self.emit(
+            self.EVENT_DIRECTORY_DISCOVERED,
+            {
+                "path": dir_path,
+                "name": Path(dir_path).name,
+                "children": children,
+                "type": "directory",
+            },
+        )
+
+    def emit_file_discovered(
+        self, file_path: str, language: Optional[str] = None, size: int = 0
+    ):
+        """Emit file discovery event."""
+        self.emit(
+            self.EVENT_FILE_DISCOVERED,
+            {
+                "path": file_path,
+                "name": Path(file_path).name,
+                "language": language or "unknown",
+                "size": size,
+                "type": "file",
+            },
+        )
+
+    def emit_file_start(self, file_path: str, language: Optional[str] = None):
+        """Emit file analysis start event.
+
+        WHY: Signals the beginning of file analysis for progress tracking.
+        """
+        self.emit(
+            "code:file_start",
+            {
+                "path": file_path,
+                "name": Path(file_path).name,
+                "language": language or "unknown",
+                "type": "file",
+                "status": OperationResult.PENDING,
+            },
+        )
+
+    def emit_file_complete(
+        self, file_path: str, nodes_count: int = 0, duration: float = 0
+    ):
+        """Emit file analysis complete event.
+
+        WHY: Signals completion of file analysis with summary statistics.
+        """
+        self.emit(
+            "code:file_complete",
+            {
+                "path": file_path,
+                "name": Path(file_path).name,
+                "nodes_count": nodes_count,
+                "duration": duration,
+                "type": "file",
+                "status": OperationResult.COMPLETED,
+            },
+        )
+
+    def emit_file_analyzed(
+        self, file_path: str, nodes: List[Dict[str, Any]], duration: float = 0
+    ):
+        """Emit file analysis complete event."""
+        self.stats["files_processed"] += 1
+        self.emit(
+            self.EVENT_FILE_ANALYZED,
+            {
+                "path": file_path,
+                "nodes": nodes,
+                "nodes_count": len(nodes),
+                "duration": duration,
+            },
+        )
+
+    def emit_node(self, node: CodeNodeEvent):
+        """Emit code node discovery event (batched).
+
+        Filters out internal functions and handlers.
+        """
+        # Filter out internal handler functions
+        if self._is_internal_function(node.name):
+            return
+
+        self.stats["nodes_found"] += 1
+        # In stdout mode, don't batch - emit immediately for real-time updates
+        batch_mode = not self.use_stdout
+        self.emit(self.EVENT_NODE_FOUND, node.to_dict(), batch=batch_mode)
+
+    def _is_internal_function(self, name: str) -> bool:
+        """Check if function is an internal handler that should be filtered."""
+        internal_patterns = [
+            "handle",  # Event handlers
+            "on_",  # Event callbacks
+            "_",  # Private methods
+            "get_",  # Simple getters
+            "set_",  # Simple setters
+            "__",  # Python magic methods
+        ]
+
+        name_lower = name.lower()
+        return any(name_lower.startswith(pattern) for pattern in internal_patterns)
+
+    def emit_progress(self, current: int, total: int, message: str = ""):
+        """Emit progress update event."""
+        self.emit(
+            self.EVENT_PROGRESS,
+            {
+                "current": current,
+                "total": total,
+                "percentage": (current / total * 100) if total > 0 else 0,
+                "message": message,
+            },
+        )
+
+    def emit_error(self, file_path: str, error: str):
+        """Emit error event."""
+        self.stats["errors"] += 1
+        self.emit(self.EVENT_ERROR, {"file": file_path, "error": str(error)})
+
+    def _emit_event(self, event: Dict[str, Any]):
+        """Emit a single event."""
+
+        # Convert datetime objects to ISO strings for JSON serialization
+        def convert_datetime(obj):
+            if isinstance(obj, datetime):
+                return obj.isoformat()
+            if isinstance(obj, dict):
+                return {k: convert_datetime(v) for k, v in obj.items()}
+            if isinstance(obj, list):
+                return [convert_datetime(item) for item in obj]
+            return obj
+
+        event = convert_datetime(event)
+
+        if self.use_stdout:
+            # Emit to stdout as JSON for subprocess communication
+            print(json.dumps(event), flush=True)
+            self.stats["events_sent"] += 1
+        elif self.sio and self.connected:
+            try:
+                self.sio.emit("code_tree_event", event)
+                self.stats["events_sent"] += 1
+            except Exception as e:
+                self.logger.error(f"Failed to emit event: {e}")
+        else:
+            # Fallback to logging
+            self.logger.debug(
+                f"Event: {event['type']} - {json.dumps(event['data'])[:100]}"
+            )
+
+    def _flush_events(self):
+        """Flush all buffered events."""
+        with self.buffer_lock:
+            if not self.event_buffer:
+                return
+
+            # Emit as batch
+            batch_event = {
+                "type": "code:batch",
+                "data": {
+                    "events": list(self.event_buffer),
+                    "count": len(self.event_buffer),
+                },
+                "timestamp": datetime.now(timezone.utc).isoformat(),
+            }
+
+            self._emit_event(batch_event)
+            self.event_buffer.clear()
+            self.last_emit_time = time.time()
+
+    def _emit_loop(self):
+        """Background loop for periodic event emission."""
+        while not self._stop_event.is_set():
+            time.sleep(self.batch_timeout)
+
+            # Check if enough time has passed since last emit
+            if time.time() - self.last_emit_time >= self.batch_timeout:
+                with self.buffer_lock:
+                    if self.event_buffer:
+                        self._flush_events()
+
+    def get_stats(self) -> Dict[str, Any]:
+        """Get current statistics."""
+        return self.stats.copy()