48 lines
1.8 KiB
Python
48 lines
1.8 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import subprocess
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from control_plane.projects.models import Decision, Task
|
||
|
|
|
||
|
|
|
||
|
|
class WorkerContextBuilder:
|
||
|
|
def build_for_task(self, task: Task, worktree_path: Path | None = None) -> dict[str, object]:
|
||
|
|
decisions = Decision.objects.filter(project=task.project).order_by("-created_at")[:10]
|
||
|
|
files: dict[str, str] = {}
|
||
|
|
tests: dict[str, str] = {}
|
||
|
|
current_diff = ""
|
||
|
|
if worktree_path is not None:
|
||
|
|
safe_root = worktree_path.resolve()
|
||
|
|
for path in sorted(safe_root.rglob("*.py"))[:40]:
|
||
|
|
if ".git" in path.parts or "__pycache__" in path.parts or "migrations" in path.parts:
|
||
|
|
continue
|
||
|
|
relative = path.relative_to(safe_root).as_posix()
|
||
|
|
content = path.read_text(encoding="utf-8")
|
||
|
|
if "SECRET" in relative.upper() or ".env" in relative:
|
||
|
|
continue
|
||
|
|
if relative.startswith("tests") or "test" in path.name:
|
||
|
|
tests[relative] = content[:4000]
|
||
|
|
else:
|
||
|
|
files[relative] = content[:4000]
|
||
|
|
current_diff = subprocess.run(
|
||
|
|
["git", "diff", "--", "."],
|
||
|
|
cwd=safe_root,
|
||
|
|
check=False,
|
||
|
|
capture_output=True,
|
||
|
|
text=True,
|
||
|
|
).stdout[:12000]
|
||
|
|
context = {
|
||
|
|
"task": {
|
||
|
|
"id": str(task.id),
|
||
|
|
"goal": task.goal,
|
||
|
|
"acceptance_criteria": task.acceptance_criteria,
|
||
|
|
},
|
||
|
|
"architecture_summary": task.project.architecture_summary,
|
||
|
|
"decisions": [decision.decision for decision in decisions],
|
||
|
|
"files": files,
|
||
|
|
"tests": tests,
|
||
|
|
"current_diff": current_diff,
|
||
|
|
}
|
||
|
|
return context
|