AndyMik90
117 lines
3.5 KiB
Python
117 lines
3.5 KiB
Python
"""
|
|
Utility Functions for Agent System
|
|
===================================
|
|
|
|
Helper functions for git operations, plan management, and file syncing.
|
|
"""
|
|
|
|
import json
|
|
import logging
|
|
import shutil
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def get_latest_commit(project_dir: Path) -> str | None:
|
|
"""Get the hash of the latest git commit."""
|
|
try:
|
|
result = subprocess.run(
|
|
["git", "rev-parse", "HEAD"],
|
|
cwd=project_dir,
|
|
capture_output=True,
|
|
text=True,
|
|
check=True,
|
|
)
|
|
return result.stdout.strip()
|
|
except subprocess.CalledProcessError:
|
|
return None
|
|
|
|
|
|
def get_commit_count(project_dir: Path) -> int:
|
|
"""Get the total number of commits."""
|
|
try:
|
|
result = subprocess.run(
|
|
["git", "rev-list", "--count", "HEAD"],
|
|
cwd=project_dir,
|
|
capture_output=True,
|
|
text=True,
|
|
check=True,
|
|
)
|
|
return int(result.stdout.strip())
|
|
except (subprocess.CalledProcessError, ValueError):
|
|
return 0
|
|
|
|
|
|
def load_implementation_plan(spec_dir: Path) -> dict | None:
|
|
"""Load the implementation plan JSON."""
|
|
plan_file = spec_dir / "implementation_plan.json"
|
|
if not plan_file.exists():
|
|
return None
|
|
try:
|
|
with open(plan_file) as f:
|
|
return json.load(f)
|
|
except (OSError, json.JSONDecodeError):
|
|
return None
|
|
|
|
|
|
def find_subtask_in_plan(plan: dict, subtask_id: str) -> dict | None:
|
|
"""Find a subtask by ID in the plan."""
|
|
for phase in plan.get("phases", []):
|
|
for subtask in phase.get("subtasks", []):
|
|
if subtask.get("id") == subtask_id:
|
|
return subtask
|
|
return None
|
|
|
|
|
|
def find_phase_for_subtask(plan: dict, subtask_id: str) -> dict | None:
|
|
"""Find the phase containing a subtask."""
|
|
for phase in plan.get("phases", []):
|
|
for subtask in phase.get("subtasks", []):
|
|
if subtask.get("id") == subtask_id:
|
|
return phase
|
|
return None
|
|
|
|
|
|
def sync_plan_to_source(spec_dir: Path, source_spec_dir: Path | None) -> bool:
|
|
"""
|
|
Sync implementation_plan.json from worktree back to source spec directory.
|
|
|
|
When running in isolated mode (worktrees), the agent updates the implementation
|
|
plan inside the worktree. This function syncs those changes back to the main
|
|
project's spec directory so the frontend/UI can see the progress.
|
|
|
|
Args:
|
|
spec_dir: Current spec directory (may be inside worktree)
|
|
source_spec_dir: Original spec directory in main project (outside worktree)
|
|
|
|
Returns:
|
|
True if sync was performed, False if not needed or failed
|
|
"""
|
|
# Skip if no source specified or same path (not in worktree mode)
|
|
if not source_spec_dir:
|
|
return False
|
|
|
|
# Resolve paths and check if they're different
|
|
spec_dir_resolved = spec_dir.resolve()
|
|
source_spec_dir_resolved = source_spec_dir.resolve()
|
|
|
|
if spec_dir_resolved == source_spec_dir_resolved:
|
|
return False # Same directory, no sync needed
|
|
|
|
# Sync the implementation plan
|
|
plan_file = spec_dir / "implementation_plan.json"
|
|
if not plan_file.exists():
|
|
return False
|
|
|
|
source_plan_file = source_spec_dir / "implementation_plan.json"
|
|
|
|
try:
|
|
shutil.copy2(plan_file, source_plan_file)
|
|
logger.debug(f"Synced implementation plan to source: {source_plan_file}")
|
|
return True
|
|
except Exception as e:
|
|
logger.warning(f"Failed to sync implementation plan to source: {e}")
|
|
return False
|