minion-code 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

minion_code/agents/hooks.py

@@ -0,0 +1,378 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+Hook system for MinionCodeAgent.
+
+Re-exports base hooks from minion framework and provides
+minion-code specific hook implementations.
+"""
+
+import logging
+from typing import Any, Dict, List, Optional, Union, Callable, Set
+
+# Re-export base types from minion framework
+from minion.main.tool_hooks import (
+    PermissionDecision,
+    PreToolUseResult,
+    PostToolUseResult,
+    ToolCallInfo,
+    PreToolUseHook,
+    PostToolUseHook,
+    HookMatcher,
+    PostHookMatcher,
+    HookConfig,
+    ToolHooks,
+    NoOpToolHooks,
+    HookedTool,
+    wrap_tools_with_hooks,
+    create_auto_accept_hook,
+    create_auto_deny_hook,
+    create_dangerous_command_check_hook,
+    create_logging_hook,
+)
+
+logger = logging.getLogger(__name__)
+
+
+# ============================================================================
+# Spinner Controller for CLI
+# ============================================================================
+
+
+class SpinnerController:
+    """
+    Controller to pause/resume spinners during user prompts.
+
+    Usage:
+        controller = SpinnerController()
+        hook = create_cli_confirm_hook(spinner_controller=controller)
+
+        # In your main loop:
+        with Progress(...) as progress:
+            controller.set_progress(progress)
+            # ... run agent ...
+            controller.clear_progress()
+    """
+
+    def __init__(self):
+        self._progress = None
+        self._is_paused = False
+
+    def set_progress(self, progress: Any) -> None:
+        """Set the current Progress instance."""
+        self._progress = progress
+        self._is_paused = False
+
+    def clear_progress(self) -> None:
+        """Clear the current Progress instance."""
+        self._progress = None
+        self._is_paused = False
+
+    def pause(self) -> None:
+        """Pause the spinner (stop the progress display)."""
+        if self._progress and not self._is_paused:
+            try:
+                self._progress.stop()
+                self._is_paused = True
+            except Exception as e:
+                logger.debug(f"Failed to pause spinner: {e}")
+
+    def resume(self) -> None:
+        """Resume the spinner (start the progress display)."""
+        if self._progress and self._is_paused:
+            try:
+                self._progress.start()
+                self._is_paused = False
+            except Exception as e:
+                logger.debug(f"Failed to resume spinner: {e}")
+
+
+# ============================================================================
+# minion-code Specific Hook Implementations
+# ============================================================================
+
+
+def _format_tool_input(tool_name: str, tool_input: Dict[str, Any]) -> str:
+    """Format tool input for display in confirmation dialog."""
+    if tool_name == "bash":
+        return f"Command: {tool_input.get('command', '')}"
+
+    if tool_name in ("file_write", "file_edit"):
+        path = tool_input.get("file_path", tool_input.get("path", ""))
+        return f"File: {path}"
+
+    if tool_name == "file_read":
+        return f"File: {tool_input.get('file_path', '')}"
+
+    if tool_name == "Task":
+        desc = tool_input.get("description", "")
+        prompt = tool_input.get("prompt", "")
+        subagent = tool_input.get("subagent_type", "general-purpose")
+        # Truncate prompt if too long
+        if len(prompt) > 200:
+            prompt = prompt[:200] + "..."
+        return f"Subagent: {subagent}\nDescription: {desc}\nPrompt: {prompt}"
+
+    # Default: show all parameters (excluding internal ones like 'state')
+    parts = []
+    skip_keys = {"state"}  # Internal parameters to hide
+    for key, value in tool_input.items():
+        if key in skip_keys:
+            continue
+        if isinstance(value, str) and len(value) > 100:
+            value = value[:100] + "..."
+        parts.append(f"{key}: {value}")
+
+    return "\n".join(parts) if parts else "(no parameters)"
+
+
+def create_confirm_writes_hook(
+    adapter: Any,  # OutputAdapter
+    tools_registry: Optional[Dict[str, Any]] = None,
+    skip_readonly: bool = True,
+) -> PreToolUseHook:
+    """
+    Create a hook that requests user confirmation for non-readonly tools.
+
+    Args:
+        adapter: OutputAdapter instance with confirm() method
+        tools_registry: Optional dict mapping tool names to tool instances
+        skip_readonly: If True, auto-accept readonly tools without confirmation
+    """
+
+    async def confirm_writes(
+        tool_name: str, tool_input: Dict[str, Any], tool_use_id: str
+    ) -> PreToolUseResult:
+        # Skip readonly tools
+        if skip_readonly:
+            # Known readonly tools (hardcoded list)
+            # Task is readonly because subagents have their own permission control
+            readonly_tools = {
+                "file_read",
+                "glob",
+                "grep",
+                "ls",
+                "web_fetch",
+                "web_search",
+                "todo_read",
+                "Task",
+            }
+            if tool_name in readonly_tools:
+                return PreToolUseResult(decision=PermissionDecision.ACCEPT)
+
+            # Check if tool is readonly from registry
+            if tools_registry:
+                tool = tools_registry.get(tool_name)
+                if tool and getattr(tool, "readonly", False):
+                    return PreToolUseResult(decision=PermissionDecision.ACCEPT)
+
+        # Format tool input for display
+        input_summary = _format_tool_input(tool_name, tool_input)
+
+        # Request confirmation
+        try:
+            confirmed = await adapter.confirm(
+                message=f"Allow {tool_name}?\n{input_summary}",
+                title="Tool Permission",
+                resource_type="tool",
+                resource_name=tool_name,
+                resource_args=tool_input,
+            )
+        except Exception as e:
+            logger.error(f"Error during confirmation: {e}")
+            return PreToolUseResult(
+                decision=PermissionDecision.DENY, reason=f"Confirmation error: {e}"
+            )
+
+        if confirmed:
+            return PreToolUseResult(decision=PermissionDecision.ACCEPT)
+        else:
+            return PreToolUseResult(
+                decision=PermissionDecision.DENY, reason="User denied permission"
+            )
+
+    return confirm_writes
+
+
+def create_cli_confirm_hook(
+    allowed_tools: Optional[Set[str]] = None,
+    session_allowed: Optional[Set[str]] = None,
+    console: Optional[Any] = None,
+    spinner_controller: Optional[SpinnerController] = None,
+) -> PreToolUseHook:
+    """
+    Create a hook that prompts for confirmation in CLI/terminal.
+
+    Args:
+        allowed_tools: Set of tool names that are always allowed (persistent)
+        session_allowed: Set of tool names allowed for this session only
+        console: Optional Rich Console for better formatting
+        spinner_controller: Optional SpinnerController to pause spinner during prompts
+    """
+    if allowed_tools is None:
+        allowed_tools = set()
+    if session_allowed is None:
+        session_allowed = set()
+
+    # Task is readonly because subagents have their own permission control
+    readonly_tools = {
+        "file_read",
+        "glob",
+        "grep",
+        "ls",
+        "web_fetch",
+        "web_search",
+        "todo_read",
+        "Task",
+    }
+
+    # Use Rich console if provided, otherwise create one
+    if console is None:
+        from rich.console import Console
+
+        console = Console()
+
+    async def cli_confirm(
+        tool_name: str, tool_input: Dict[str, Any], tool_use_id: str
+    ) -> PreToolUseResult:
+        # Skip readonly tools
+        if tool_name in readonly_tools:
+            return PreToolUseResult(decision=PermissionDecision.ACCEPT)
+
+        if tool_name in allowed_tools or tool_name in session_allowed:
+            return PreToolUseResult(decision=PermissionDecision.ACCEPT)
+
+        input_summary = _format_tool_input(tool_name, tool_input)
+
+        # Pause spinner before prompting user
+        if spinner_controller:
+            spinner_controller.pause()
+
+        try:
+            # Use Rich console to print
+            console.print()
+            console.print(f"[bold yellow]{'='*60}[/]")
+            console.print(f"[bold cyan]🔧 Tool: {tool_name}[/]")
+            console.print(f"[bold yellow]{'='*60}[/]")
+            console.print(input_summary)
+            console.print(f"[bold yellow]{'='*60}[/]")
+            console.print("[dim]Options:[/]")
+            console.print("  [green][y][/] Yes, allow this once")
+            console.print("  [red][n][/] No, deny")
+            console.print("  [yellow][a][/] Always allow this tool (session)")
+            console.print("  [blue][A][/] Always allow this tool (permanent)")
+
+            try:
+                # Don't lowercase - we need to distinguish 'a' from 'A'
+                response = console.input("[bold]Allow? [y/n/a/A]: [/]").strip()
+            except (EOFError, KeyboardInterrupt):
+                return PreToolUseResult(
+                    decision=PermissionDecision.DENY, reason="User cancelled"
+                )
+
+            if response.lower() == "y":
+                return PreToolUseResult(decision=PermissionDecision.ACCEPT)
+            elif response == "a":  # Session allow (lowercase only)
+                session_allowed.add(tool_name)
+                logger.info(f"Tool '{tool_name}' allowed for this session")
+                return PreToolUseResult(decision=PermissionDecision.ACCEPT)
+            elif response == "A":  # Permanent allow (uppercase only)
+                allowed_tools.add(tool_name)
+                logger.info(f"Tool '{tool_name}' permanently allowed")
+                return PreToolUseResult(decision=PermissionDecision.ACCEPT)
+            else:
+                return PreToolUseResult(
+                    decision=PermissionDecision.DENY, reason="User denied permission"
+                )
+        finally:
+            # Resume spinner after user responds
+            if spinner_controller:
+                spinner_controller.resume()
+
+    return cli_confirm
+
+
+def create_cli_hooks(
+    auto_accept: bool = False,
+    spinner_controller: Optional[SpinnerController] = None,
+    console: Optional[Any] = None,
+) -> HookConfig:
+    """
+    Create hook configuration for CLI usage.
+
+    Args:
+        auto_accept: If True, auto-accept all tools (no confirmation prompts)
+        spinner_controller: Optional SpinnerController to pause spinner during prompts
+        console: Optional Rich Console for output
+    """
+    if auto_accept:
+        return create_autonomous_hooks()
+
+    return HookConfig(
+        pre_tool_use=[
+            HookMatcher("bash", create_dangerous_command_check_hook()),
+            HookMatcher(
+                "*",
+                create_cli_confirm_hook(
+                    spinner_controller=spinner_controller,
+                    console=console,
+                ),
+            ),
+        ]
+    )
+
+
+def create_default_hooks(adapter: Any) -> HookConfig:
+    """
+    Create default hook configuration with:
+    - Dangerous command blocking for bash
+    - User confirmation for write operations
+    """
+    return HookConfig(
+        pre_tool_use=[
+            HookMatcher("bash", create_dangerous_command_check_hook()),
+            HookMatcher("*", create_confirm_writes_hook(adapter)),
+        ]
+    )
+
+
+def create_autonomous_hooks() -> HookConfig:
+    """
+    Create hook configuration for autonomous/unattended mode.
+    Blocks dangerous commands but auto-accepts everything else.
+    """
+    return HookConfig(
+        pre_tool_use=[
+            HookMatcher("bash", create_dangerous_command_check_hook()),
+            HookMatcher("*", create_auto_accept_hook()),
+        ]
+    )
+
+
+__all__ = [
+    # Re-exported from minion.main.tool_hooks
+    "PermissionDecision",
+    "PreToolUseResult",
+    "PostToolUseResult",
+    "ToolCallInfo",
+    "PreToolUseHook",
+    "PostToolUseHook",
+    "HookMatcher",
+    "PostHookMatcher",
+    "HookConfig",
+    "ToolHooks",
+    "NoOpToolHooks",
+    "HookedTool",
+    "wrap_tools_with_hooks",
+    "create_auto_accept_hook",
+    "create_auto_deny_hook",
+    "create_dangerous_command_check_hook",
+    "create_logging_hook",
+    # minion-code specific
+    "SpinnerController",
+    "create_confirm_writes_hook",
+    "create_cli_confirm_hook",
+    "create_cli_hooks",
+    "create_default_hooks",
+    "create_autonomous_hooks",
+]