crackerjack 0.29.0__py3-none-any.whl → 0.31.4__py3-none-any.whl

crackerjack/config/hooks.py

@@ -0,0 +1,218 @@
+from dataclasses import dataclass
+from enum import Enum
+from pathlib import Path
+
+
+class HookStage(Enum):
+    FAST = "fast"
+    COMPREHENSIVE = "comprehensive"
+
+
+class RetryPolicy(Enum):
+    NONE = "none"
+    FORMATTING_ONLY = "formatting_only"
+    ALL_HOOKS = "all_hooks"
+
+
+@dataclass
+class HookDefinition:
+    name: str
+    command: list[str]
+    timeout: int = 60
+    stage: HookStage = HookStage.FAST
+    retry_on_failure: bool = False
+    is_formatting: bool = False
+    manual_stage: bool = False
+    config_path: Path | None = None
+
+    def get_command(self) -> list[str]:
+        cmd = ["pre-commit", "run"]
+        if self.config_path:
+            cmd.extend(["-c", str(self.config_path)])
+        if self.manual_stage:
+            cmd.extend(["--hook-stage", "manual"])
+        cmd.extend([self.name, "--all-files"])
+        return cmd
+
+
+@dataclass
+class HookStrategy:
+    name: str
+    hooks: list[HookDefinition]
+    timeout: int = 300
+    retry_policy: RetryPolicy = RetryPolicy.NONE
+    parallel: bool = False
+    max_workers: int = 3
+
+
+FAST_HOOKS = [
+    HookDefinition(
+        name="trailing-whitespace",
+        command=["pre-commit", "run", "trailing-whitespace", "--all-files"],
+        is_formatting=True,
+        retry_on_failure=True,
+    ),
+    HookDefinition(
+        name="end-of-file-fixer",
+        command=["pre-commit", "run", "end-of-file-fixer", "--all-files"],
+        is_formatting=True,
+        retry_on_failure=True,
+    ),
+    HookDefinition(
+        name="check-yaml",
+        command=["pre-commit", "run", "check-yaml", "--all-files"],
+    ),
+    HookDefinition(
+        name="check-toml",
+        command=["pre-commit", "run", "check-toml", "--all-files"],
+    ),
+    HookDefinition(
+        name="check-added-large-files",
+        command=["pre-commit", "run", "check-added-large-files", "--all-files"],
+    ),
+    HookDefinition(
+        name="uv-lock",
+        command=["pre-commit", "run", "uv-lock", "--all-files"],
+    ),
+    HookDefinition(
+        name="gitleaks",
+        command=["pre-commit", "run", "gitleaks", "--all-files"],
+    ),
+    HookDefinition(
+        name="codespell",
+        command=["pre-commit", "run", "codespell", "--all-files"],
+    ),
+    HookDefinition(
+        name="ruff-check",
+        command=["pre-commit", "run", "ruff-check", "--all-files"],
+    ),
+    HookDefinition(
+        name="ruff-format",
+        command=["pre-commit", "run", "ruff-format", "--all-files"],
+        is_formatting=True,
+        retry_on_failure=True,
+    ),
+    HookDefinition(
+        name="mdformat",
+        command=["pre-commit", "run", "mdformat", "--all-files"],
+        is_formatting=True,
+        retry_on_failure=True,
+    ),
+]
+
+COMPREHENSIVE_HOOKS = [
+    HookDefinition(
+        name="pyright",
+        command=[
+            "pre-commit",
+            "run",
+            "--hook-stage",
+            "manual",
+            "pyright",
+            "--all-files",
+        ],
+        timeout=120,
+        stage=HookStage.COMPREHENSIVE,
+        manual_stage=True,
+    ),
+    HookDefinition(
+        name="bandit",
+        command=[
+            "pre-commit",
+            "run",
+            "--hook-stage",
+            "manual",
+            "bandit",
+            "--all-files",
+        ],
+        timeout=120,
+        stage=HookStage.COMPREHENSIVE,
+        manual_stage=True,
+    ),
+    HookDefinition(
+        name="vulture",
+        command=[
+            "pre-commit",
+            "run",
+            "--hook-stage",
+            "manual",
+            "vulture",
+            "--all-files",
+        ],
+        timeout=120,
+        stage=HookStage.COMPREHENSIVE,
+        manual_stage=True,
+    ),
+    HookDefinition(
+        name="refurb",
+        command=[
+            "pre-commit",
+            "run",
+            "--hook-stage",
+            "manual",
+            "refurb",
+            "--all-files",
+        ],
+        timeout=120,
+        stage=HookStage.COMPREHENSIVE,
+        manual_stage=True,
+    ),
+    HookDefinition(
+        name="creosote",
+        command=[
+            "pre-commit",
+            "run",
+            "--hook-stage",
+            "manual",
+            "creosote",
+            "--all-files",
+        ],
+        timeout=120,
+        stage=HookStage.COMPREHENSIVE,
+        manual_stage=True,
+    ),
+    HookDefinition(
+        name="complexipy",
+        command=[
+            "pre-commit",
+            "run",
+            "--hook-stage",
+            "manual",
+            "complexipy",
+            "--all-files",
+        ],
+        timeout=120,
+        stage=HookStage.COMPREHENSIVE,
+        manual_stage=True,
+    ),
+]
+
+
+FAST_STRATEGY = HookStrategy(
+    name="fast",
+    hooks=FAST_HOOKS,
+    timeout=60,
+    retry_policy=RetryPolicy.FORMATTING_ONLY,
+)
+
+COMPREHENSIVE_STRATEGY = HookStrategy(
+    name="comprehensive",
+    hooks=COMPREHENSIVE_HOOKS,
+    timeout=120,
+    retry_policy=RetryPolicy.NONE,
+)
+
+
+class HookConfigLoader:
+    @staticmethod
+    def load_strategy(name: str, _: Path | None = None) -> HookStrategy:
+        if name == "fast":
+            return FAST_STRATEGY
+        if name == "comprehensive":
+            return COMPREHENSIVE_STRATEGY
+        msg = f"Unknown hook strategy: {name}"
+        raise ValueError(msg)
+
+    @staticmethod
+    def get_all_strategies() -> dict[str, HookStrategy]:
+        return {"fast": FAST_STRATEGY, "comprehensive": COMPREHENSIVE_STRATEGY}

crackerjack/core/async_workflow_orchestrator.py

@@ -0,0 +1,406 @@
+import asyncio
+import logging
+import time
+import typing as t
+from pathlib import Path
+
+from rich.console import Console
+
+from crackerjack.models.protocols import OptionsProtocol
+
+from .phase_coordinator import PhaseCoordinator
+from .session_coordinator import SessionCoordinator
+
+
+class AsyncWorkflowPipeline:
+    def __init__(
+        self,
+        console: Console,
+        pkg_path: Path,
+        session: SessionCoordinator,
+        phases: PhaseCoordinator,
+    ) -> None:
+        self.console = console
+        self.pkg_path = pkg_path
+        self.session = session
+        self.phases = phases
+        self.logger = logging.getLogger("crackerjack.async_pipeline")
+
+    async def run_complete_workflow_async(self, options: OptionsProtocol) -> bool:
+        start_time = time.time()
+        self.session.initialize_session_tracking(options)
+        self.session.track_task("workflow", "Complete async crackerjack workflow")
+
+        try:
+            if hasattr(options, "ai_agent") and options.ai_agent:
+                success = await self._execute_ai_agent_workflow_async(options)
+            else:
+                success = await self._execute_workflow_phases_async(options)
+            self.session.finalize_session(start_time, success)
+            return success
+        except KeyboardInterrupt:
+            self.console.print("Interrupted by user")
+            self.session.fail_task("workflow", "Interrupted by user")
+            return False
+        except Exception as e:
+            self.console.print(f"Error: {e}")
+            self.session.fail_task("workflow", f"Unexpected error: {e}")
+            return False
+        finally:
+            self.session.cleanup_resources()
+
+    async def _execute_workflow_phases_async(self, options: OptionsProtocol) -> bool:
+        success = True
+
+        self.phases.run_configuration_phase(options)
+
+        if not await self._execute_cleaning_phase_async(options):
+            success = False
+
+            self.session.fail_task("workflow", "Cleaning phase failed")
+            return False
+
+        if not await self._execute_quality_phase_async(options):
+            success = False
+
+            return False
+
+        if not self.phases.run_publishing_phase(options):
+            success = False
+            self.session.fail_task("workflow", "Publishing failed")
+            return False
+
+        if not self.phases.run_commit_phase(options):
+            success = False
+
+        return success
+
+    async def _execute_cleaning_phase_async(self, options: OptionsProtocol) -> bool:
+        if not options.clean:
+            return True
+
+        return await asyncio.to_thread(self.phases.run_cleaning_phase, options)
+
+    async def _execute_quality_phase_async(self, options: OptionsProtocol) -> bool:
+        if hasattr(options, "fast") and options.fast:
+            return await self._run_fast_hooks_async(options)
+        if hasattr(options, "comp") and options.comp:
+            return await self._run_comprehensive_hooks_async(options)
+        if options.test:
+            return await self._execute_test_workflow_async(options)
+        return await self._execute_standard_hooks_workflow_async(options)
+
+    async def _execute_test_workflow_async(self, options: OptionsProtocol) -> bool:
+        overall_success = True
+
+        if not await self._run_fast_hooks_async(options):
+            overall_success = False
+
+            self.session.fail_task("workflow", "Fast hooks failed")
+            return False
+
+        test_task = asyncio.create_task(self._run_testing_phase_async(options))
+        hooks_task = asyncio.create_task(self._run_comprehensive_hooks_async(options))
+
+        test_success, hooks_success = await asyncio.gather(
+            test_task,
+            hooks_task,
+            return_exceptions=True,
+        )
+
+        if isinstance(test_success, Exception):
+            self.logger.error(f"Test execution error: {test_success}")
+            test_success = False
+
+        if isinstance(hooks_success, Exception):
+            self.logger.error(f"Hooks execution error: {hooks_success}")
+            hooks_success = False
+
+        if not test_success:
+            overall_success = False
+
+            self.session.fail_task("workflow", "Testing failed")
+            return False
+
+        if not hooks_success:
+            overall_success = False
+
+            self.session.fail_task("workflow", "Comprehensive hooks failed")
+            return False
+
+        return overall_success
+
+    async def _execute_standard_hooks_workflow_async(
+        self,
+        options: OptionsProtocol,
+    ) -> bool:
+        hooks_success = await self._run_hooks_phase_async(options)
+        if not hooks_success:
+            self.session.fail_task("workflow", "Hooks failed")
+            return False
+        return True
+
+    async def _run_fast_hooks_async(self, options: OptionsProtocol) -> bool:
+        return await asyncio.to_thread(self.phases.run_fast_hooks_only, options)
+
+    async def _run_comprehensive_hooks_async(self, options: OptionsProtocol) -> bool:
+        return await asyncio.to_thread(
+            self.phases.run_comprehensive_hooks_only,
+            options,
+        )
+
+    async def _run_hooks_phase_async(self, options: OptionsProtocol) -> bool:
+        return await asyncio.to_thread(self.phases.run_hooks_phase, options)
+
+    async def _run_testing_phase_async(self, options: OptionsProtocol) -> bool:
+        return await asyncio.to_thread(self.phases.run_testing_phase, options)
+
+    async def _execute_ai_agent_workflow_async(
+        self, options: OptionsProtocol, max_iterations: int = 10
+    ) -> bool:
+        """Execute AI agent workflow with iterative fixing between iterations."""
+        self.console.print(
+            f"🤖 Starting AI Agent workflow (max {max_iterations} iterations)"
+        )
+
+        # Always run configuration phase first
+        self.phases.run_configuration_phase(options)
+
+        # Run cleaning phase if requested
+        if not await self._execute_cleaning_phase_async(options):
+            self.session.fail_task("workflow", "Cleaning phase failed")
+            return False
+
+        # Iterative quality improvement with AI fixing
+        iteration_success = await self._run_iterative_quality_improvement(
+            options, max_iterations
+        )
+        if not iteration_success:
+            return False
+
+        # Run remaining phases
+        return await self._run_final_workflow_phases(options)
+
+    async def _run_iterative_quality_improvement(
+        self, options: OptionsProtocol, max_iterations: int
+    ) -> bool:
+        """Run iterative quality improvement until all checks pass."""
+        for iteration in range(1, max_iterations + 1):
+            self.console.print(f"\n🔄 Iteration {iteration}/{max_iterations}")
+
+            iteration_result = await self._execute_single_iteration(options, iteration)
+
+            if iteration_result == "success":
+                self.console.print("✅ All quality checks passed!")
+                return True
+            elif iteration_result == "failed":
+                return False
+            # Continue to next iteration if result == "continue"
+
+        # If we exhausted all iterations without success
+        self.console.print(
+            f"❌ Failed to achieve code quality after {max_iterations} iterations"
+        )
+        self.session.fail_task("workflow", f"Failed after {max_iterations} iterations")
+        return False
+
+    async def _execute_single_iteration(
+        self, options: OptionsProtocol, iteration: int
+    ) -> str:
+        """Execute a single AI agent iteration. Returns 'success', 'failed', or 'continue'."""
+        # Step 1: Fast hooks with retry logic
+        fast_hooks_success = await self._run_fast_hooks_with_retry_async(options)
+
+        # Step 2 & 3: Collect ALL issues
+        test_issues = await self._collect_test_issues_async(options)
+        hook_issues = await self._collect_comprehensive_hook_issues_async(options)
+
+        # If everything passes, we're done
+        if fast_hooks_success and not test_issues and not hook_issues:
+            return "success"
+
+        # Step 4: Apply AI fixes for ALL collected issues
+        if test_issues or hook_issues:
+            fix_success = await self._apply_ai_fixes_async(
+                options, test_issues, hook_issues, iteration
+            )
+            if not fix_success:
+                self.console.print(f"❌ AI fixing failed in iteration {iteration}")
+                self.session.fail_task(
+                    "workflow", f"AI fixing failed in iteration {iteration}"
+                )
+                return "failed"
+
+        return "continue"
+
+    async def _run_final_workflow_phases(self, options: OptionsProtocol) -> bool:
+        """Run the final publishing and commit phases."""
+        if not self.phases.run_publishing_phase(options):
+            self.session.fail_task("workflow", "Publishing failed")
+            return False
+
+        if not self.phases.run_commit_phase(options):
+            self.session.fail_task("workflow", "Commit failed")
+            return False
+
+        return True
+
+    async def _run_fast_hooks_with_retry_async(self, options: OptionsProtocol) -> bool:
+        """Run fast hooks with one retry if they fail."""
+        success = await self._run_fast_hooks_async(options)
+        if not success:
+            self.console.print("⚠️ Fast hooks failed, retrying once...")
+            success = await self._run_fast_hooks_async(options)
+        return success
+
+    async def _collect_test_issues_async(self, options: OptionsProtocol) -> list[str]:
+        """Collect all test failures without stopping on first failure."""
+        if not options.test:
+            return []
+
+        try:
+            success = await self._run_testing_phase_async(options)
+            if success:
+                return []
+            else:
+                return ["Test failures detected - see logs for details"]
+        except Exception as e:
+            return [f"Test execution error: {e}"]
+
+    async def _collect_comprehensive_hook_issues_async(
+        self, options: OptionsProtocol
+    ) -> list[str]:
+        """Collect all comprehensive hook issues without stopping on first failure."""
+        try:
+            success = await self._run_comprehensive_hooks_async(options)
+            if success:
+                return []
+            else:
+                return ["Comprehensive hook failures detected - see logs for details"]
+        except Exception as e:
+            return [f"Comprehensive hooks error: {e}"]
+
+    async def _apply_ai_fixes_async(
+        self,
+        options: OptionsProtocol,
+        test_issues: list[str],
+        hook_issues: list[str],
+        iteration: int,
+    ) -> bool:
+        """Apply AI fixes for all collected issues in batch."""
+        all_issues = test_issues + hook_issues
+        if not all_issues:
+            return True
+
+        self.console.print(
+            f"🔧 Applying AI fixes for {len(all_issues)} issues in iteration {iteration}"
+        )
+
+        # This would integrate with the AI agent system to actually apply fixes
+        # For now, we'll simulate the fixing process
+        try:
+            # In a real implementation, this would:
+            # 1. Analyze all collected issues
+            # 2. Generate fixes using AI agents
+            # 3. Apply the fixes to source code
+            # 4. Return success/failure status
+
+            # Placeholder for actual AI fixing logic
+            await asyncio.sleep(0.1)  # Simulate processing time
+
+            self.console.print(f"✅ AI fixes applied for iteration {iteration}")
+            return True
+
+        except Exception as e:
+            self.logger.error(f"AI fixing failed: {e}")
+            return False
+
+
+class AsyncWorkflowOrchestrator:
+    def __init__(
+        self,
+        console: Console | None = None,
+        pkg_path: Path | None = None,
+        dry_run: bool = False,
+        web_job_id: str | None = None,
+    ) -> None:
+        self.console = console or Console(force_terminal=True)
+        self.pkg_path = pkg_path or Path.cwd()
+        self.dry_run = dry_run
+        self.web_job_id = web_job_id
+
+        from crackerjack.models.protocols import (
+            FileSystemInterface,
+            GitInterface,
+            HookManager,
+            PublishManager,
+            TestManagerProtocol,
+        )
+
+        from .container import create_container
+
+        self.container = create_container(
+            console=self.console,
+            pkg_path=self.pkg_path,
+            dry_run=self.dry_run,
+        )
+
+        self.session = SessionCoordinator(self.console, self.pkg_path, self.web_job_id)
+        self.phases = PhaseCoordinator(
+            console=self.console,
+            pkg_path=self.pkg_path,
+            session=self.session,
+            filesystem=self.container.get(FileSystemInterface),
+            git_service=self.container.get(GitInterface),
+            hook_manager=self.container.get(HookManager),
+            test_manager=self.container.get(TestManagerProtocol),
+            publish_manager=self.container.get(PublishManager),
+        )
+
+        self.async_pipeline = AsyncWorkflowPipeline(
+            console=self.console,
+            pkg_path=self.pkg_path,
+            session=self.session,
+            phases=self.phases,
+        )
+
+        self.logger = logging.getLogger("crackerjack.async_orchestrator")
+
+    async def run_complete_workflow_async(self, options: OptionsProtocol) -> bool:
+        return await self.async_pipeline.run_complete_workflow_async(options)
+
+    def run_complete_workflow(self, options: OptionsProtocol) -> bool:
+        return asyncio.run(self.run_complete_workflow_async(options))
+
+    def run_cleaning_phase(self, options: OptionsProtocol) -> bool:
+        return self.phases.run_cleaning_phase(options)
+
+    def run_fast_hooks_only(self, options: OptionsProtocol) -> bool:
+        return self.phases.run_fast_hooks_only(options)
+
+    def run_comprehensive_hooks_only(self, options: OptionsProtocol) -> bool:
+        return self.phases.run_comprehensive_hooks_only(options)
+
+    def run_hooks_phase(self, options: OptionsProtocol) -> bool:
+        return self.phases.run_hooks_phase(options)
+
+    def run_testing_phase(self, options: OptionsProtocol) -> bool:
+        return self.phases.run_testing_phase(options)
+
+    def run_publishing_phase(self, options: OptionsProtocol) -> bool:
+        return self.phases.run_publishing_phase(options)
+
+    def run_commit_phase(self, options: OptionsProtocol) -> bool:
+        return self.phases.run_commit_phase(options)
+
+    def run_configuration_phase(self, options: OptionsProtocol) -> bool:
+        return self.phases.run_configuration_phase(options)
+
+    def _cleanup_resources(self) -> None:
+        self.session.cleanup_resources()
+
+    def _register_cleanup(self, cleanup_handler: t.Callable[[], None]) -> None:
+        self.session.register_cleanup(cleanup_handler)
+
+    def _track_lock_file(self, lock_file_path: Path) -> None:
+        self.session.track_lock_file(lock_file_path)