crackerjack 0.18.2__py3-none-any.whl → 0.45.2__py3-none-any.whl

crackerjack/services/debug.py

@@ -0,0 +1,733 @@
+import json
+import logging
+import os
+import time
+import typing as t
+from contextlib import contextmanager
+from datetime import datetime
+from pathlib import Path
+from typing import Any
+
+from acb.console import Console
+from acb.depends import depends
+from rich.panel import Panel
+from rich.table import Table
+
+from .log_manager import get_log_manager
+from .logging import get_logger
+
+
+class AIAgentDebugger:
+    def __init__(self, enabled: bool = False, verbose: bool = False) -> None:
+        self.enabled = enabled
+        self.verbose = verbose
+        self.console = depends.get_sync(Console)
+        self.logger = get_logger("crackerjack.ai_agent.debug")
+        self.session_id = f"debug_{int(time.time())}"
+
+        self.debug_log_path: Path | None = None
+
+        if self.enabled:
+            log_manager = get_log_manager()
+            self.debug_log_path = log_manager.create_debug_log_file(
+                f"ai-agent -{self.session_id}",
+            )
+
+        self.mcp_operations: list[dict[str, Any]] = []
+        self.agent_activities: list[dict[str, Any]] = []
+        self.workflow_phases: list[dict[str, Any]] = []
+        self.error_events: list[dict[str, Any]] = []
+
+        self.iteration_stats: list[dict[str, Any]] = []
+        self.current_iteration = 0
+        self.total_test_failures = 0
+        self.total_test_fixes = 0
+        self.total_hook_failures = 0
+        self.total_hook_fixes = 0
+        self.workflow_success = False
+
+        self._debug_logging_setup = False
+
+        if self.enabled:
+            self._print_debug_header()
+
+    def _ensure_debug_logging_setup(self) -> None:
+        if not self._debug_logging_setup and self.enabled:
+            self._setup_debug_logging()
+            self._debug_logging_setup = True
+
+    def _setup_debug_logging(self) -> None:
+        if not self.debug_log_path:
+            return
+
+        debug_handler = logging.FileHandler(self.debug_log_path)
+        debug_handler.setLevel(logging.DEBUG)
+        debug_formatter = logging.Formatter(
+            "%(asctime)s | %(name)s | %(levelname)s | %(message)s",
+        )
+        debug_handler.setFormatter(debug_formatter)
+
+        loggers = [
+            "crackerjack.ai_agent",
+            "crackerjack.mcp",
+            "crackerjack.agents",
+            "crackerjack.workflow",
+        ]
+
+        for logger_name in loggers:
+            logger = logging.getLogger(logger_name)
+            logger.addHandler(debug_handler)
+            logger.setLevel(logging.DEBUG)
+
+    def _print_debug_header(self) -> None:
+        debug_log_info = (
+            f"Debug Log: {self.debug_log_path}"
+            if self.debug_log_path
+            else "Debug Log: None (disabled)"
+        )
+        header = Panel(
+            f"[bold cyan]🐛 AI Agent Debug Mode Active[/ bold cyan]\n"
+            f"Session ID: {self.session_id}\n"
+            f"{debug_log_info}\n"
+            f"Verbose Mode: {'✅' if self.verbose else '❌'}",
+            title="Debug Session",
+            border_style="cyan",
+        )
+        self.console.print(header)
+        self.console.print()
+
+    @contextmanager
+    def debug_operation(self, operation: str, **kwargs: Any) -> t.Iterator[str]:
+        if not self.enabled:
+            yield ""
+            return
+
+        self._ensure_debug_logging_setup()
+        op_id = f"{operation}_{int(time.time() * 1000)}"
+        start_time = time.time()
+
+        self.logger.debug(f"Starting operation: {operation}", extra=kwargs)
+
+        if self.verbose:
+            self.console.print(f"[dim]🔍 {operation} starting...[/ dim]")
+
+        try:
+            yield op_id
+            duration = time.time() - start_time
+            self.logger.debug(
+                f"Operation completed: {operation}",
+                extra={"duration": duration} | kwargs,
+            )
+
+            if self.verbose:
+                self.console.print(
+                    f"[dim green]✅ {operation} completed ({duration: .2f}s)[/ dim green]",
+                )
+
+        except Exception as e:
+            duration = time.time() - start_time
+            self.logger.exception(
+                f"Operation failed: {operation}",
+                extra={"error": str(e), "duration": duration} | kwargs,
+            )
+
+            if self.verbose:
+                self.console.print(
+                    f"[dim red]❌ {operation} failed ({duration: .2f}s): {e}[/ dim red]",
+                )
+            raise
+
+    def log_mcp_operation(
+        self,
+        operation_type: str,
+        tool_name: str,
+        params: dict[str, Any] | None = None,
+        result: dict[str, Any] | None = None,
+        error: str | None = None,
+        duration: float | None = None,
+    ) -> None:
+        if not self.enabled:
+            return
+
+        self._ensure_debug_logging_setup()
+
+        event = {
+            "timestamp": datetime.now().isoformat(),
+            "type": "mcp_operation",
+            "operation": operation_type,
+            "tool": tool_name,
+            "params": params or {},
+            "result": result,
+            "error": error,
+            "duration": duration,
+        }
+
+        self.mcp_operations.append(event)
+
+        self.logger.info(
+            f"MCP {operation_type}: {tool_name}",
+            extra={
+                "tool": tool_name,
+                "params_count": len(params) if params else 0,
+                "success": error is None,
+                "duration": duration,
+            },
+        )
+
+        if self.verbose:
+            status_color = "green" if error is None else "red"
+            status_icon = "✅" if error is None else "❌"
+
+            self.console.print(
+                f"[{status_color}]{status_icon} MCP {operation_type}[/{status_color}]: "
+                f"[bold]{tool_name}[/ bold]"
+                + (f" ({duration: .2f}s)" if duration else ""),
+            )
+
+            if error and self.verbose:
+                self.console.print(f" [red]Error: {error}[/ red]")
+                self.console.print()
+
+    def log_agent_activity(
+        self,
+        agent_name: str,
+        activity: str,
+        issue_id: str | None = None,
+        confidence: float | None = None,
+        result: dict[str, Any] | None = None,
+        metadata: dict[str, Any] | None = None,
+    ) -> None:
+        if not self.enabled:
+            return
+
+        self._ensure_debug_logging_setup()
+
+        event = {
+            "timestamp": datetime.now().isoformat(),
+            "type": "agent_activity",
+            "agent": agent_name,
+            "activity": activity,
+            "issue_id": issue_id,
+            "confidence": confidence,
+            "result": result,
+            "metadata": metadata or {},
+        }
+
+        self.agent_activities.append(event)
+
+        self.logger.info(
+            f"Agent {activity}: {agent_name}",
+            extra={
+                "agent": agent_name,
+                "activity": activity,
+                "issue_id": issue_id,
+                "confidence": confidence,
+            },
+        )
+
+        if self.verbose:
+            confidence_text = f" (confidence: {confidence: .2f})" if confidence else ""
+            issue_text = f" [issue: {issue_id}]" if issue_id else ""
+
+            self.console.print(
+                f"[blue]🤖 {agent_name}[/ blue]: {activity}{confidence_text}{issue_text}",
+            )
+
+    def log_workflow_phase(
+        self,
+        phase: str,
+        status: str,
+        details: dict[str, Any] | None = None,
+        duration: float | None = None,
+    ) -> None:
+        if not self.enabled:
+            return
+
+        self._ensure_debug_logging_setup()
+
+        event = {
+            "timestamp": datetime.now().isoformat(),
+            "type": "workflow_phase",
+            "phase": phase,
+            "status": status,
+            "details": details or {},
+            "duration": duration,
+        }
+
+        self.workflow_phases.append(event)
+
+        self.logger.info(
+            f"Workflow {status}: {phase}",
+            extra={"phase": phase, "status": status, "duration": duration},
+        )
+
+        if self.verbose:
+            status_colors = {
+                "started": "yellow",
+                "completed": "green",
+                "failed": "red",
+                "skipped": "dim",
+            }
+
+            color = status_colors.get(status, "white")
+            duration_text = f" ({duration: .2f}s)" if duration else ""
+
+            self.console.print(
+                f"[{color}]📋 Workflow {status}: {phase}{duration_text}[/{color}]",
+            )
+
+    def log_error_event(
+        self,
+        error_type: str,
+        message: str,
+        context: dict[str, Any] | None = None,
+        traceback_info: str | None = None,
+    ) -> None:
+        if not self.enabled:
+            return
+
+        self._ensure_debug_logging_setup()
+
+        event = {
+            "timestamp": datetime.now().isoformat(),
+            "type": "error_event",
+            "error_type": error_type,
+            "message": message,
+            "context": context or {},
+            "traceback": traceback_info,
+        }
+
+        self.error_events.append(event)
+
+        self.logger.error(
+            f"Error: {error_type}",
+            extra={"error_type": error_type, "message": message, "context": context},
+        )
+
+        if self.verbose:
+            self.console.print(f"[red]💥 {error_type}: {message}[/ red]")
+
+    def print_debug_summary(self) -> None:
+        if not self.enabled:
+            return
+
+        border_style = "green" if self.workflow_success else "red"
+        title_style = "green" if self.workflow_success else "red"
+
+        table = Table(
+            title=f"[{title_style}]AI Agent Debug Summary[/{title_style}]",
+            border_style=border_style,
+        )
+        table.add_column("Category", style="cyan")
+        table.add_column("Count", justify="right", style="green")
+        table.add_column("Details", style="dim")
+
+        table.add_row(
+            "MCP Operations",
+            str(len(self.mcp_operations)),
+            f"Tools used: {len({op['tool'] for op in self.mcp_operations})}",
+        )
+
+        table.add_row(
+            "Agent Activities",
+            str(len(self.agent_activities)),
+            f"Agents active: {len({act['agent'] for act in self.agent_activities})}",
+        )
+
+        table.add_row(
+            "Workflow Phases",
+            str(len(self.workflow_phases)),
+            f"Completed: {len([p for p in self.workflow_phases if p['status'] == 'completed'])}",
+        )
+
+        table.add_row(
+            "Error Events",
+            str(len(self.error_events)),
+            f"Types: {len({err['error_type'] for err in self.error_events})}",
+        )
+
+        table.add_row(
+            "Iterations Completed",
+            str(self.current_iteration),
+            f"Total test fixes: {self.total_test_fixes}, hook fixes: {self.total_hook_fixes}",
+        )
+
+        self.console.print(
+            Panel(table, title="AI Agent Debug Summary", border_style=border_style)
+        )
+
+        if self.iteration_stats:
+            self._print_iteration_breakdown(border_style)
+
+        if self.verbose and self.agent_activities:
+            self._print_agent_activity_breakdown(border_style)
+
+        if self.verbose and self.mcp_operations:
+            self._print_mcp_operation_breakdown(border_style)
+
+        self._print_total_statistics(border_style)
+
+        self.console.print(
+            f"\n[dim]📝 Full debug log available at: {self.debug_log_path}[/ dim]"
+            if self.debug_log_path
+            else "",
+        )
+
+    def _print_iteration_breakdown(self, border_style: str = "red") -> None:
+        if not self.iteration_stats:
+            return
+
+        table = Table(
+            title="[cyan]Iteration Breakdown[/ cyan]",
+            border_style=border_style,
+        )
+        table.add_column("Iteration", style="yellow")
+        table.add_column("Test Failures", justify="right", style="red")
+        table.add_column("Test Fixes", justify="right", style="green")
+        table.add_column("Hook Failures", justify="right", style="red")
+        table.add_column("Hook Fixes", justify="right", style="green")
+        table.add_column("Duration", justify="right", style="cyan")
+
+        for iteration in self.iteration_stats:
+            table.add_row(
+                str(iteration["iteration"]),
+                str(iteration["test_failures"]),
+                str(iteration["test_fixes"]),
+                str(iteration["hook_failures"]),
+                str(iteration["hook_fixes"]),
+                f"{iteration['duration']: .1f}s"
+                if iteration.get("duration")
+                else "N / A",
+            )
+
+        self.console.print(
+            Panel(table, title="Iteration Breakdown", border_style=border_style)
+        )
+
+    def _print_agent_activity_breakdown(self, border_style: str = "red") -> None:
+        agent_stats: dict[str, dict[str, t.Any]] = {}
+        for activity in self.agent_activities:
+            agent = activity["agent"]
+            if agent not in agent_stats:
+                agent_stats[agent] = {
+                    "activities": 0,
+                    "avg_confidence": 0.0,
+                    "confidences": [],
+                }
+
+            agent_stats[agent]["activities"] += 1
+            if activity.get("confidence"):
+                agent_stats[agent]["confidences"].append(activity["confidence"])
+
+        for stats in agent_stats.values():
+            if stats["confidences"]:
+                stats["avg_confidence"] = sum(stats["confidences"]) / len(
+                    stats["confidences"],
+                )
+
+        table = Table(
+            title="[cyan]Agent Activity Breakdown[/ cyan]",
+            border_style=border_style,
+        )
+        table.add_column("Agent", style="blue")
+        table.add_column("Activities", justify="right", style="green")
+        table.add_column("Avg Confidence", justify="right", style="yellow")
+
+        for agent, stats in sorted(agent_stats.items()):
+            confidence_text = (
+                f"{stats['avg_confidence']: .2f}"
+                if stats["avg_confidence"] > 0
+                else "N / A"
+            )
+            table.add_row(agent, str(stats["activities"]), confidence_text)
+
+        self.console.print(
+            Panel(table, title="Agent Activity Breakdown", border_style=border_style)
+        )
+
+    def _print_total_statistics(self, border_style: str = "red") -> None:
+        success_icon = "✅" if self.workflow_success else "❌"
+        status_text = "SUCCESS" if self.workflow_success else "IN PROGRESS"
+        status_style = "green" if self.workflow_success else "red"
+
+        table = Table(
+            title=f"[{status_style}]{success_icon} TOTAL WORKFLOW STATISTICS {success_icon}[/{status_style}]",
+            border_style=border_style,
+        )
+        table.add_column("Metric", style="cyan")
+        table.add_column("Total", justify="right", style=status_style)
+        table.add_column("Status", style=status_style)
+
+        table.add_row("Iterations Completed", str(self.current_iteration), status_text)
+
+        table.add_row(
+            "Total Test Failures Found",
+            str(self.total_test_failures),
+            f"Fixed: {self.total_test_fixes}",
+        )
+
+        table.add_row(
+            "Total Hook Failures Found",
+            str(self.total_hook_failures),
+            f"Fixed: {self.total_hook_fixes}",
+        )
+
+        total_issues = self.total_test_failures + self.total_hook_failures
+        total_fixes = self.total_test_fixes + self.total_hook_fixes
+        fix_rate = (total_fixes / total_issues * 100) if total_issues > 0 else 100
+
+        table.add_row(
+            "Overall Fix Rate",
+            f"{fix_rate: .1f}%",
+            f"{total_fixes}/{total_issues} issues resolved",
+        )
+
+        self.console.print(
+            Panel(table, title="Total Workflow Statistics", border_style=border_style)
+        )
+
+    def _print_mcp_operation_breakdown(self, border_style: str = "red") -> None:
+        tool_stats = {}
+        for op in self.mcp_operations:
+            tool = op["tool"]
+            if tool not in tool_stats:
+                tool_stats[tool] = {"calls": 0, "errors": 0, "total_duration": 0.0}
+
+            tool_stats[tool]["calls"] += 1
+            if op.get("error"):
+                tool_stats[tool]["errors"] += 1
+            if op.get("duration"):
+                tool_stats[tool]["total_duration"] += op["duration"]
+
+        table = Table(
+            title="[cyan]MCP Tool Usage[/ cyan]",
+            border_style=border_style,
+        )
+        table.add_column("Tool", style="cyan")
+        table.add_column("Calls", justify="right", style="green")
+        table.add_column("Errors", justify="right", style="red")
+        table.add_column("Avg Duration", justify="right", style="yellow")
+
+        for tool, stats in sorted(tool_stats.items()):
+            avg_duration = (
+                stats["total_duration"] / stats["calls"] if stats["calls"] > 0 else 0
+            )
+            table.add_row(
+                tool,
+                str(stats["calls"]),
+                str(stats["errors"]),
+                f"{avg_duration: .2f}s" if avg_duration > 0 else "N / A",
+            )
+
+        self.console.print(
+            Panel(table, title="MCP Tool Usage", border_style=border_style)
+        )
+
+    def log_iteration_start(self, iteration_number: int) -> None:
+        if not self.enabled:
+            return
+
+        self.current_iteration = iteration_number
+        iteration_data = {
+            "iteration": iteration_number,
+            "start_time": time.time(),
+            "test_failures": 0,
+            "test_fixes": 0,
+            "hook_failures": 0,
+            "hook_fixes": 0,
+            "duration": 0.0,
+        }
+        self.iteration_stats.append(iteration_data)
+
+        if self.verbose:
+            self.console.print(
+                f"[yellow]🔄 Starting Iteration {iteration_number}[/ yellow]",
+            )
+
+    def log_iteration_end(self, iteration_number: int, success: bool) -> None:
+        if not self.enabled or not self.iteration_stats:
+            return
+
+        iteration_data = None
+        for data in self.iteration_stats:
+            if data["iteration"] == iteration_number:
+                iteration_data = data
+                break
+
+        if iteration_data:
+            iteration_data["duration"] = time.time() - iteration_data["start_time"]
+
+        if self.verbose:
+            status = "✅ PASSED" if success else "❌ FAILED"
+            self.console.print(
+                f"[{'green' if success else 'red'}]🏁 Iteration {iteration_number} {status}[/{'green' if success else 'red'}]",
+            )
+
+    def log_test_failures(self, count: int) -> None:
+        if not self.enabled or not self.iteration_stats:
+            return
+
+        if self.iteration_stats:
+            self.iteration_stats[-1]["test_failures"] = count
+            self.total_test_failures += count
+
+    def log_test_fixes(self, count: int) -> None:
+        if not self.enabled or not self.iteration_stats:
+            return
+
+        if self.iteration_stats:
+            self.iteration_stats[-1]["test_fixes"] = count
+            self.total_test_fixes += count
+
+    def log_hook_failures(self, count: int) -> None:
+        if not self.enabled or not self.iteration_stats:
+            return
+
+        if self.iteration_stats:
+            self.iteration_stats[-1]["hook_failures"] = count
+            self.total_hook_failures += count
+
+    def log_hook_fixes(self, count: int) -> None:
+        if not self.enabled or not self.iteration_stats:
+            return
+
+        if self.iteration_stats:
+            self.iteration_stats[-1]["hook_fixes"] = count
+            self.total_hook_fixes += count
+
+    def set_workflow_success(self, success: bool) -> None:
+        if not self.enabled:
+            return
+
+        self.workflow_success = success
+
+    def export_debug_data(self, output_path: Path | None = None) -> Path:
+        if not self.enabled:
+            return Path("debug_not_enabled.json")
+
+        if output_path is None:
+            output_path = Path(f"crackerjack-debug-export-{self.session_id}.json")
+
+        debug_data = {
+            "session_id": self.session_id,
+            "timestamp": datetime.now().isoformat(),
+            "mcp_operations": self.mcp_operations,
+            "agent_activities": self.agent_activities,
+            "workflow_phases": self.workflow_phases,
+            "error_events": self.error_events,
+        }
+
+        with output_path.open("w") as f:
+            json.dump(debug_data, f, indent=2, default=str)
+
+        self.logger.info(f"Debug data exported to {output_path}")
+        return output_path
+
+
+class NoOpDebugger:
+    def __init__(self) -> None:
+        self.enabled = False
+        self.verbose = False
+        self.debug_log_path = None
+        self.session_id = "disabled"
+
+    def debug_operation(self, operation: str, **kwargs: Any) -> t.Iterator[str]:
+        yield ""
+
+    def log_mcp_operation(
+        self,
+        operation_type: str,
+        tool_name: str,
+        params: dict[str, Any] | None = None,
+        result: dict[str, Any] | None = None,
+        error: str | None = None,
+        duration: float | None = None,
+    ) -> None:
+        pass
+
+    def log_agent_activity(
+        self,
+        agent_name: str,
+        activity: str,
+        issue_id: str | None = None,
+        confidence: float | None = None,
+        result: dict[str, Any] | None = None,
+        metadata: dict[str, Any] | None = None,
+    ) -> None:
+        pass
+
+    def log_workflow_phase(
+        self,
+        phase: str,
+        status: str,
+        details: dict[str, Any] | None = None,
+        duration: float | None = None,
+    ) -> None:
+        pass
+
+    def log_error_event(
+        self,
+        error_type: str,
+        message: str,
+        context: dict[str, Any] | None = None,
+        traceback_info: str | None = None,
+    ) -> None:
+        pass
+
+    def print_debug_summary(self) -> None:
+        pass
+
+    def export_debug_data(self, output_path: Path | None = None) -> Path:
+        return Path("debug_not_enabled.json")
+
+    def log_iteration_start(self, iteration_number: int) -> None:
+        pass
+
+    def log_iteration_end(self, iteration_number: int, success: bool) -> None:
+        pass
+
+    def log_test_failures(self, count: int) -> None:
+        pass
+
+    def log_test_fixes(self, count: int) -> None:
+        pass
+
+    def log_hook_failures(self, count: int) -> None:
+        pass
+
+    def log_hook_fixes(self, count: int) -> None:
+        pass
+
+    def set_workflow_success(self, success: bool) -> None:
+        pass
+
+
+_ai_agent_debugger: AIAgentDebugger | NoOpDebugger | None = None
+
+
+def get_ai_agent_debugger() -> AIAgentDebugger | NoOpDebugger:
+    global _ai_agent_debugger
+    if _ai_agent_debugger is None:
+        debug_enabled = os.environ.get("AI_AGENT_DEBUG", "0") == "1"
+        verbose_mode = os.environ.get("AI_AGENT_VERBOSE", "0") == "1"
+
+        if debug_enabled:
+            _ai_agent_debugger = AIAgentDebugger(
+                enabled=debug_enabled,
+                verbose=verbose_mode,
+            )
+        else:
+            _ai_agent_debugger = NoOpDebugger()
+    return _ai_agent_debugger
+
+
+def enable_ai_agent_debugging(verbose: bool = False) -> AIAgentDebugger:
+    global _ai_agent_debugger
+    _ai_agent_debugger = AIAgentDebugger(enabled=True, verbose=verbose)
+    return _ai_agent_debugger
+
+
+def disable_ai_agent_debugging() -> None:
+    global _ai_agent_debugger
+    if _ai_agent_debugger:
+        _ai_agent_debugger.enabled = False