claude-mpm 5.4.96__py3-none-any.whl → 5.6.17__py3-none-any.whl

claude_mpm/commander/runtime/monitor.py

@@ -0,0 +1,346 @@
+"""Runtime monitor for continuous output monitoring and event detection.
+
+This module provides RuntimeMonitor which continuously polls tmux pane output
+and detects events using OutputParser.
+"""
+
+import asyncio
+import logging
+from typing import TYPE_CHECKING, Dict, List, Optional
+
+from ..events.manager import EventManager
+from ..models.events import Event
+from ..parsing.output_parser import OutputParser
+from ..tmux_orchestrator import TmuxOrchestrator
+
+if TYPE_CHECKING:
+    from ..core.block_manager import BlockManager
+
+logger = logging.getLogger(__name__)
+
+
+class RuntimeMonitor:
+    """Monitors tmux pane output and detects events.
+
+    This class continuously polls tmux pane output, uses OutputParser to detect
+    events, and emits them via EventManager. Supports starting/stopping monitoring
+    for individual panes and tracking which panes are actively monitored.
+
+    Attributes:
+        orchestrator: TmuxOrchestrator for capturing output
+        parser: OutputParser for event detection
+        event_manager: EventManager for emitting events
+        poll_interval: Seconds between polls (default: 2.0)
+        capture_lines: Number of lines to capture from tmux (default: 1000)
+
+    Example:
+        >>> monitor = RuntimeMonitor(orchestrator, parser, event_manager)
+        >>> await monitor.start_monitoring("%5", "proj_123")
+        >>> events = await monitor.poll_once("%5")
+        >>> await monitor.stop_monitoring("%5")
+    """
+
+    def __init__(
+        self,
+        orchestrator: TmuxOrchestrator,
+        parser: OutputParser,
+        event_manager: EventManager,
+        poll_interval: float = 2.0,
+        capture_lines: int = 1000,
+        block_manager: Optional["BlockManager"] = None,
+    ):
+        """Initialize runtime monitor.
+
+        Args:
+            orchestrator: TmuxOrchestrator for capturing output
+            parser: OutputParser for event detection
+            event_manager: EventManager for emitting events
+            poll_interval: Seconds between polls (default: 2.0)
+            capture_lines: Number of lines to capture (default: 1000)
+            block_manager: Optional BlockManager for automatic work blocking
+
+        Raises:
+            ValueError: If any required parameter is None
+        """
+        if orchestrator is None:
+            raise ValueError("Orchestrator cannot be None")
+        if parser is None:
+            raise ValueError("Parser cannot be None")
+        if event_manager is None:
+            raise ValueError("EventManager cannot be None")
+
+        self.orchestrator = orchestrator
+        self.parser = parser
+        self.event_manager = event_manager
+        self.poll_interval = poll_interval
+        self.capture_lines = capture_lines
+        self.block_manager = block_manager
+
+        # Track active monitors: pane_target -> (project_id, task, last_output_hash)
+        self._monitors: Dict[str, tuple[str, Optional[asyncio.Task], int]] = {}
+        self._running = False
+
+        logger.debug(
+            "RuntimeMonitor initialized (interval: %.2fs, lines: %d, block_manager: %s)",
+            poll_interval,
+            capture_lines,
+            "enabled" if block_manager else "disabled",
+        )
+
+    async def start_monitoring(self, pane_target: str, project_id: str) -> None:
+        """Start monitoring a pane for events.
+
+        Creates a background task that continuously polls the pane output
+        and detects events. Only one monitor per pane is allowed.
+
+        Args:
+            pane_target: Tmux pane target to monitor (e.g., '%5')
+            project_id: Project ID for event attribution
+
+        Raises:
+            ValueError: If pane_target or project_id is None/empty
+            RuntimeError: If monitoring already active for this pane
+
+        Example:
+            >>> await monitor.start_monitoring("%5", "proj_123")
+        """
+        if not pane_target:
+            raise ValueError("Pane target cannot be None or empty")
+        if not project_id:
+            raise ValueError("Project ID cannot be None or empty")
+
+        if pane_target in self._monitors:
+            raise RuntimeError(f"Monitoring already active for pane {pane_target}")
+
+        logger.info(
+            "Starting monitoring for pane %s (project: %s)", pane_target, project_id
+        )
+
+        # Create background polling task
+        task = asyncio.create_task(
+            self._monitor_loop(pane_target, project_id), name=f"monitor-{pane_target}"
+        )
+
+        # Track monitor with initial output hash of 0
+        self._monitors[pane_target] = (project_id, task, 0)
+
+    async def stop_monitoring(self, pane_target: str) -> None:
+        """Stop monitoring a pane.
+
+        Cancels the background polling task and removes the monitor.
+
+        Args:
+            pane_target: Tmux pane target to stop monitoring
+
+        Raises:
+            ValueError: If pane_target is None/empty
+
+        Example:
+            >>> await monitor.stop_monitoring("%5")
+        """
+        if not pane_target:
+            raise ValueError("Pane target cannot be None or empty")
+
+        if pane_target not in self._monitors:
+            logger.debug("No monitoring active for pane %s", pane_target)
+            return
+
+        logger.info("Stopping monitoring for pane %s", pane_target)
+
+        _project_id, task, _ = self._monitors[pane_target]
+
+        # Cancel the monitoring task
+        if task and not task.done():
+            task.cancel()
+            try:
+                await task
+            except asyncio.CancelledError:
+                pass
+
+        # Remove from monitors
+        del self._monitors[pane_target]
+        logger.debug("Stopped monitoring pane %s", pane_target)
+
+    async def poll_once(self, pane_target: str) -> List[Event]:
+        """Poll pane output once and return any detected events.
+
+        This is a one-time poll that doesn't require starting a monitor.
+        Useful for on-demand event checking.
+
+        Args:
+            pane_target: Tmux pane target to poll
+
+        Returns:
+            List of Event objects detected in the output
+
+        Raises:
+            ValueError: If pane_target is None/empty
+
+        Example:
+            >>> events = await monitor.poll_once("%5")
+            >>> for event in events:
+            ...     print(f"Event: {event.title}")
+        """
+        if not pane_target:
+            raise ValueError("Pane target cannot be None or empty")
+
+        # Determine project_id if this pane is being monitored
+        project_id = "unknown"
+        session_id = None
+        if pane_target in self._monitors:
+            project_id, _, _ = self._monitors[pane_target]
+
+        logger.debug("Polling pane %s once", pane_target)
+
+        try:
+            # Capture output from tmux
+            output = self.orchestrator.capture_output(
+                pane_target, lines=self.capture_lines
+            )
+
+            # Parse for events - don't create events automatically
+            parse_results = self.parser.parse(
+                content=output,
+                project_id=project_id,
+                session_id=session_id,
+                create_events=False,
+            )
+
+            # Create events manually so we can return them
+            events = []
+            for result in parse_results:
+                event = self.event_manager.create(
+                    project_id=project_id,
+                    session_id=session_id,
+                    event_type=result.event_type,
+                    title=result.title,
+                    content=result.content,
+                    options=result.options,
+                    context=result.context,
+                )
+                events.append(event)
+
+            logger.debug(
+                "Poll detected %d events from pane %s", len(events), pane_target
+            )
+            return events
+
+        except Exception as e:
+            logger.warning("Failed to poll pane %s: %s", pane_target, e)
+            return []
+
+    @property
+    def active_monitors(self) -> Dict[str, str]:
+        """Get map of pane_target -> project_id for active monitors.
+
+        Returns:
+            Dict mapping pane targets to their project IDs
+
+        Example:
+            >>> monitors = monitor.active_monitors
+            >>> print(monitors)
+            {'%5': 'proj_123', '%6': 'proj_456'}
+        """
+        return {pane: project_id for pane, (project_id, _, _) in self._monitors.items()}
+
+    async def _monitor_loop(self, pane_target: str, project_id: str) -> None:
+        """Background monitoring loop for a single pane.
+
+        Continuously polls the pane output at the configured interval and
+        detects events. Uses output hashing to avoid reprocessing identical output.
+
+        Args:
+            pane_target: Tmux pane target to monitor
+            project_id: Project ID for event attribution
+        """
+        logger.debug("Monitor loop started for pane %s", pane_target)
+
+        while pane_target in self._monitors:
+            try:
+                # Capture output from tmux
+                output = self.orchestrator.capture_output(
+                    pane_target, lines=self.capture_lines
+                )
+
+                # Check if output has changed since last poll
+                output_hash = hash(output)
+                _, task, last_hash = self._monitors.get(
+                    pane_target, (project_id, None, 0)
+                )
+
+                if output_hash == last_hash:
+                    # No change in output, skip parsing
+                    await asyncio.sleep(self.poll_interval)
+                    continue
+
+                # Update hash
+                self._monitors[pane_target] = (project_id, task, output_hash)
+
+                # Parse for events
+                parse_results = self.parser.parse(
+                    content=output,
+                    project_id=project_id,
+                    session_id=None,
+                    create_events=True,  # Create events via EventManager
+                )
+
+                if parse_results:
+                    logger.info(
+                        "Detected %d events from pane %s",
+                        len(parse_results),
+                        pane_target,
+                    )
+
+                    # Automatically block work for blocking events
+                    if self.block_manager:
+                        for parse_result in parse_results:
+                            # Get the created event from EventManager
+                            # Events are created with matching titles, so find by title
+                            pending_events = self.event_manager.get_pending(project_id)
+                            for event in pending_events:
+                                if (
+                                    event.title == parse_result.title
+                                    and event.is_blocking
+                                ):
+                                    blocked_work = (
+                                        await self.block_manager.check_and_block(event)
+                                    )
+                                    if blocked_work:
+                                        logger.info(
+                                            "Event %s blocked %d work items: %s",
+                                            event.id,
+                                            len(blocked_work),
+                                            blocked_work,
+                                        )
+                                    break
+
+            except Exception as e:
+                logger.error(
+                    "Error in monitor loop for pane %s: %s",
+                    pane_target,
+                    e,
+                    exc_info=True,
+                )
+
+            # Wait before next poll
+            await asyncio.sleep(self.poll_interval)
+
+        logger.debug("Monitor loop stopped for pane %s", pane_target)
+
+    async def stop_all(self) -> None:
+        """Stop all active monitors.
+
+        Cancels all background polling tasks and clears the monitor registry.
+
+        Example:
+            >>> await monitor.stop_all()
+        """
+        logger.info("Stopping all %d active monitors", len(self._monitors))
+
+        # Get all pane targets before iteration (avoid dict size change during iteration)
+        pane_targets = list(self._monitors.keys())
+
+        for pane_target in pane_targets:
+            await self.stop_monitoring(pane_target)
+
+        logger.info("All monitors stopped")

claude_mpm/commander/session/__init__.py

@@ -0,0 +1,6 @@
+"""Commander session management."""
+
+from .context import Message, SessionContext
+from .manager import SessionManager
+
+__all__ = ["Message", "SessionContext", "SessionManager"]

claude_mpm/commander/session/context.py

@@ -0,0 +1,81 @@
+"""Session context for Commander chat interface."""
+
+from dataclasses import dataclass, field
+from datetime import datetime, timezone
+from typing import Optional
+
+
+@dataclass
+class Message:
+    """A single message in the conversation."""
+
+    role: str  # "user" or "assistant"
+    content: str
+    timestamp: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
+
+
+@dataclass
+class SessionContext:
+    """Tracks the state of a Commander chat session.
+
+    Attributes:
+        connected_instance: Name of currently connected instance, or None.
+        messages: Conversation history.
+        session_id: Unique session identifier.
+        created_at: When session was created.
+
+    Example:
+        >>> context = SessionContext()
+        >>> context.is_connected
+        False
+        >>> context.connect_to("myapp")
+        >>> context.is_connected
+        True
+        >>> context.connected_instance
+        'myapp'
+    """
+
+    connected_instance: Optional[str] = None
+    messages: list[Message] = field(default_factory=list)
+    session_id: str = field(
+        default_factory=lambda: datetime.now(timezone.utc).isoformat()
+    )
+    created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
+
+    @property
+    def is_connected(self) -> bool:
+        """Check if connected to an instance."""
+        return self.connected_instance is not None
+
+    def connect_to(self, instance_name: str) -> None:
+        """Connect to an instance.
+
+        Args:
+            instance_name: Name of instance to connect to.
+        """
+        self.connected_instance = instance_name
+
+    def disconnect(self) -> None:
+        """Disconnect from current instance."""
+        self.connected_instance = None
+
+    def add_message(self, role: str, content: str) -> None:
+        """Add a message to conversation history.
+
+        Args:
+            role: "user" or "assistant".
+            content: Message content.
+        """
+        self.messages.append(Message(role=role, content=content))
+
+    def get_messages_for_llm(self) -> list[dict]:
+        """Get messages formatted for LLM API.
+
+        Returns:
+            List of message dicts with 'role' and 'content' keys.
+        """
+        return [{"role": msg.role, "content": msg.content} for msg in self.messages]
+
+    def clear_history(self) -> None:
+        """Clear conversation history."""
+        self.messages.clear()

claude_mpm/commander/session/manager.py

@@ -0,0 +1,59 @@
+"""Session manager for Commander chat interface."""
+
+from .context import SessionContext
+
+
+class SessionManager:
+    """Manages Commander chat session state.
+
+    Coordinates session context, connection state, and conversation history.
+
+    Example:
+        >>> manager = SessionManager()
+        >>> manager.connect_to("myapp")
+        >>> manager.context.is_connected
+        True
+        >>> manager.add_user_message("Fix the bug")
+        >>> len(manager.context.messages)
+        1
+    """
+
+    def __init__(self):
+        """Initialize session manager with fresh context."""
+        self.context = SessionContext()
+
+    def connect_to(self, instance_name: str) -> None:
+        """Connect to an instance.
+
+        Args:
+            instance_name: Name of instance to connect to.
+        """
+        self.context.connect_to(instance_name)
+
+    def disconnect(self) -> None:
+        """Disconnect from current instance."""
+        self.context.disconnect()
+
+    def add_user_message(self, content: str) -> None:
+        """Add a user message to conversation history.
+
+        Args:
+            content: User message content.
+        """
+        self.context.add_message("user", content)
+
+    def add_assistant_message(self, content: str) -> None:
+        """Add an assistant message to conversation history.
+
+        Args:
+            content: Assistant message content.
+        """
+        self.context.add_message("assistant", content)
+
+    def clear_history(self) -> None:
+        """Clear conversation history."""
+        self.context.clear_history()
+
+    def reset(self) -> None:
+        """Reset session to initial state."""
+        self.context = SessionContext()