@mindfoldhq/trellis 0.4.0 → 0.5.0-beta.10

package/dist/templates/trellis/scripts/common/phase.py

@@ -1,254 +0,0 @@
-#!/usr/bin/env python3
-"""
-Phase Management Utilities.
-
-Centralized phase tracking for multi-agent pipeline.
-
-Provides:
-    get_current_phase     - Returns current phase number
-    get_total_phases      - Returns total phase count
-    get_phase_action      - Returns action name for phase
-    get_phase_info        - Returns "N/M (action)" format
-    set_phase             - Sets current_phase
-    advance_phase         - Advances to next phase
-    get_phase_for_action  - Returns phase number for action
-    map_subagent_to_action - Map subagent type to action name
-    is_phase_completed    - Check if phase is completed
-    is_current_action     - Check if at specific action
-"""
-
-from __future__ import annotations
-
-from pathlib import Path
-
-from .io import read_json, write_json
-
-
-# =============================================================================
-# Internal Helpers (operate on pre-loaded data dict)
-# =============================================================================
-
-def _total_phases(data: dict) -> int:
-    """Get total phases from pre-loaded data."""
-    next_action = data.get("next_action", [])
-    return len(next_action) if isinstance(next_action, list) else 0
-
-
-def _phase_action(data: dict, phase: int) -> str:
-    """Get action name for a phase from pre-loaded data."""
-    next_action = data.get("next_action", [])
-    if isinstance(next_action, list):
-        for item in next_action:
-            if isinstance(item, dict) and item.get("phase") == phase:
-                return item.get("action", "unknown")
-    return "unknown"
-
-
-def _phase_for_action(data: dict, action: str) -> int:
-    """Get phase number for an action name from pre-loaded data."""
-    next_action = data.get("next_action", [])
-    if isinstance(next_action, list):
-        for item in next_action:
-            if isinstance(item, dict) and item.get("action") == action:
-                return item.get("phase", 0)
-    return 0
-
-
-# =============================================================================
-# Phase Functions
-# =============================================================================
-
-def get_current_phase(task_json: Path) -> int:
-    """Get current phase number.
-
-    Args:
-        task_json: Path to task.json file.
-
-    Returns:
-        Current phase number, or 0 if not found.
-    """
-    data = read_json(task_json)
-    if not data:
-        return 0
-    return data.get("current_phase", 0) or 0
-
-
-def get_total_phases(task_json: Path) -> int:
-    """Get total number of phases.
-
-    Args:
-        task_json: Path to task.json file.
-
-    Returns:
-        Total phase count, or 0 if not found.
-    """
-    data = read_json(task_json)
-    if not data:
-        return 0
-    return _total_phases(data)
-
-
-def get_phase_action(task_json: Path, phase: int) -> str:
-    """Get action name for a specific phase.
-
-    Args:
-        task_json: Path to task.json file.
-        phase: Phase number.
-
-    Returns:
-        Action name, or "unknown" if not found.
-    """
-    data = read_json(task_json)
-    if not data:
-        return "unknown"
-    return _phase_action(data, phase)
-
-
-def get_phase_info(task_json: Path) -> str:
-    """Get formatted phase info: "N/M (action)".
-
-    Args:
-        task_json: Path to task.json file.
-
-    Returns:
-        Formatted string like "1/4 (implement)".
-    """
-    data = read_json(task_json)
-    if not data:
-        return "N/A"
-
-    current_phase = data.get("current_phase", 0) or 0
-    total = _total_phases(data)
-    action_name = _phase_action(data, current_phase)
-
-    if current_phase == 0 or current_phase is None:
-        return f"0/{total} (pending)"
-    else:
-        return f"{current_phase}/{total} ({action_name})"
-
-
-def set_phase(task_json: Path, phase: int) -> bool:
-    """Set current phase to a specific value.
-
-    Args:
-        task_json: Path to task.json file.
-        phase: Phase number to set.
-
-    Returns:
-        True on success, False on error.
-    """
-    data = read_json(task_json)
-    if not data:
-        return False
-
-    data["current_phase"] = phase
-    return write_json(task_json, data)
-
-
-def advance_phase(task_json: Path) -> bool:
-    """Advance to next phase.
-
-    Args:
-        task_json: Path to task.json file.
-
-    Returns:
-        True on success, False on error or at final phase.
-    """
-    data = read_json(task_json)
-    if not data:
-        return False
-
-    current = data.get("current_phase", 0) or 0
-    total = _total_phases(data)
-    next_phase = current + 1
-
-    if next_phase > total:
-        return False  # Already at final phase
-
-    data["current_phase"] = next_phase
-    return write_json(task_json, data)
-
-
-def get_phase_for_action(task_json: Path, action: str) -> int:
-    """Get phase number for a specific action name.
-
-    Args:
-        task_json: Path to task.json file.
-        action: Action name.
-
-    Returns:
-        Phase number, or 0 if not found.
-    """
-    data = read_json(task_json)
-    if not data:
-        return 0
-    return _phase_for_action(data, action)
-
-
-def map_subagent_to_action(subagent_type: str) -> str:
-    """Map subagent type to action name.
-
-    Used by hooks to determine which action a subagent corresponds to.
-
-    Args:
-        subagent_type: Subagent type string.
-
-    Returns:
-        Corresponding action name.
-    """
-    mapping = {
-        "implement": "implement",
-        "check": "check",
-        "debug": "debug",
-        "research": "research",
-    }
-    return mapping.get(subagent_type, subagent_type)
-
-
-def is_phase_completed(task_json: Path, phase: int) -> bool:
-    """Check if a phase is completed (current_phase > phase).
-
-    Args:
-        task_json: Path to task.json file.
-        phase: Phase number to check.
-
-    Returns:
-        True if phase is completed.
-    """
-    current = get_current_phase(task_json)
-    return current > phase
-
-
-def is_current_action(task_json: Path, action: str) -> bool:
-    """Check if we're at a specific action.
-
-    Args:
-        task_json: Path to task.json file.
-        action: Action name to check.
-
-    Returns:
-        True if current phase matches the action.
-    """
-    data = read_json(task_json)
-    if not data:
-        return False
-    current = data.get("current_phase", 0) or 0
-    action_phase = _phase_for_action(data, action)
-    return current == action_phase
-
-
-# =============================================================================
-# Main Entry (for testing)
-# =============================================================================
-
-if __name__ == "__main__":
-    import sys
-
-    if len(sys.argv) > 1:
-        path = Path(sys.argv[1])
-        print(f"Task JSON: {path}")
-        print(f"Phase info: {get_phase_info(path)}")
-        print(f"Current phase: {get_current_phase(path)}")
-        print(f"Total phases: {get_total_phases(path)}")
-    else:
-        print("Usage: python3 phase.py <task.json>")

package/dist/templates/trellis/scripts/common/registry.py

@@ -1,335 +0,0 @@
-#!/usr/bin/env python3
-"""
-Registry utility functions for multi-agent pipeline.
-
-Provides:
-    registry_get_file        - Get registry file path
-    registry_get_agent_by_id - Find agent by ID
-    registry_get_agent_by_worktree - Find agent by worktree path
-    registry_get_task_dir    - Get task dir for a worktree
-    registry_remove_by_id    - Remove agent by ID
-    registry_remove_by_worktree - Remove agent by worktree path
-    registry_add_agent       - Add agent to registry
-    registry_search_agent    - Search agent by ID or task_dir
-    registry_list_agents     - List all agents
-"""
-
-from __future__ import annotations
-
-from datetime import datetime
-from pathlib import Path
-
-from .io import read_json, write_json
-from .paths import get_repo_root
-from .worktree import get_agents_dir
-
-
-# =============================================================================
-# Internal Helpers
-# =============================================================================
-
-def _load_registry(
-    repo_root: Path | None = None,
-) -> tuple[Path | None, dict | None]:
-    """Load registry file and data in one step.
-
-    Returns:
-        (registry_file_path, data_dict) — either may be None.
-    """
-    if repo_root is None:
-        repo_root = get_repo_root()
-
-    registry_file = registry_get_file(repo_root)
-    if not registry_file or not registry_file.is_file():
-        return registry_file, None
-
-    data = read_json(registry_file)
-    return registry_file, data
-
-
-# =============================================================================
-# Registry File Access
-# =============================================================================
-
-def registry_get_file(repo_root: Path | None = None) -> Path | None:
-    """Get registry file path.
-
-    Args:
-        repo_root: Repository root path. Defaults to auto-detected.
-
-    Returns:
-        Path to registry.json, or None if agents dir not found.
-    """
-    if repo_root is None:
-        repo_root = get_repo_root()
-
-    agents_dir = get_agents_dir(repo_root)
-    if agents_dir:
-        return agents_dir / "registry.json"
-    return None
-
-
-def _ensure_registry(repo_root: Path | None = None) -> Path | None:
-    """Ensure registry file exists with valid structure.
-
-    Args:
-        repo_root: Repository root path. Defaults to auto-detected.
-
-    Returns:
-        Path to registry file, or None if cannot create.
-    """
-    if repo_root is None:
-        repo_root = get_repo_root()
-
-    registry_file = registry_get_file(repo_root)
-    if not registry_file:
-        return None
-
-    agents_dir = registry_file.parent
-
-    try:
-        agents_dir.mkdir(parents=True, exist_ok=True)
-
-        if not registry_file.exists():
-            write_json(registry_file, {"agents": []})
-
-        return registry_file
-    except (OSError, IOError):
-        return None
-
-
-# =============================================================================
-# Agent Lookup
-# =============================================================================
-
-def registry_get_agent_by_id(
-    agent_id: str,
-    repo_root: Path | None = None
-) -> dict | None:
-    """Get agent by ID.
-
-    Args:
-        agent_id: Agent ID.
-        repo_root: Repository root path. Defaults to auto-detected.
-
-    Returns:
-        Agent dict, or None if not found.
-    """
-    _, data = _load_registry(repo_root)
-    if not data:
-        return None
-
-    for agent in data.get("agents", []):
-        if agent.get("id") == agent_id:
-            return agent
-
-    return None
-
-
-def registry_get_agent_by_worktree(
-    worktree_path: str,
-    repo_root: Path | None = None
-) -> dict | None:
-    """Get agent by worktree path.
-
-    Args:
-        worktree_path: Worktree path.
-        repo_root: Repository root path. Defaults to auto-detected.
-
-    Returns:
-        Agent dict, or None if not found.
-    """
-    _, data = _load_registry(repo_root)
-    if not data:
-        return None
-
-    for agent in data.get("agents", []):
-        if agent.get("worktree_path") == worktree_path:
-            return agent
-
-    return None
-
-
-def registry_search_agent(
-    search: str,
-    repo_root: Path | None = None
-) -> dict | None:
-    """Search agent by ID or task_dir containing search term.
-
-    Args:
-        search: Search term.
-        repo_root: Repository root path. Defaults to auto-detected.
-
-    Returns:
-        First matching agent dict, or None if not found.
-    """
-    _, data = _load_registry(repo_root)
-    if not data:
-        return None
-
-    for agent in data.get("agents", []):
-        # Exact ID match
-        if agent.get("id") == search:
-            return agent
-        # Partial match on task_dir
-        task_dir = agent.get("task_dir", "")
-        if search in task_dir:
-            return agent
-
-    return None
-
-
-def registry_get_task_dir(
-    worktree_path: str,
-    repo_root: Path | None = None
-) -> str | None:
-    """Get task directory for a worktree.
-
-    Args:
-        worktree_path: Worktree path.
-        repo_root: Repository root path. Defaults to auto-detected.
-
-    Returns:
-        Task directory path, or None if not found.
-    """
-    _, data = _load_registry(repo_root)
-    if not data:
-        return None
-
-    for agent in data.get("agents", []):
-        if agent.get("worktree_path") == worktree_path:
-            return agent.get("task_dir")
-
-    return None
-
-
-# =============================================================================
-# Agent Modification
-# =============================================================================
-
-def registry_remove_by_id(agent_id: str, repo_root: Path | None = None) -> bool:
-    """Remove agent by ID.
-
-    Args:
-        agent_id: Agent ID.
-        repo_root: Repository root path. Defaults to auto-detected.
-
-    Returns:
-        True on success.
-    """
-    registry_file, data = _load_registry(repo_root)
-    if not registry_file or not data:
-        return True  # Nothing to remove
-
-    agents = data.get("agents", [])
-    data["agents"] = [a for a in agents if a.get("id") != agent_id]
-
-    return write_json(registry_file, data)
-
-
-def registry_remove_by_worktree(
-    worktree_path: str,
-    repo_root: Path | None = None
-) -> bool:
-    """Remove agent by worktree path.
-
-    Args:
-        worktree_path: Worktree path.
-        repo_root: Repository root path. Defaults to auto-detected.
-
-    Returns:
-        True on success.
-    """
-    registry_file, data = _load_registry(repo_root)
-    if not registry_file or not data:
-        return True  # Nothing to remove
-
-    agents = data.get("agents", [])
-    data["agents"] = [a for a in agents if a.get("worktree_path") != worktree_path]
-
-    return write_json(registry_file, data)
-
-
-def registry_add_agent(
-    agent_id: str,
-    worktree_path: str,
-    pid: int,
-    task_dir: str,
-    repo_root: Path | None = None,
-    platform: str = "claude",
-) -> bool:
-    """Add agent to registry (replaces if same ID exists).
-
-    Args:
-        agent_id: Agent ID.
-        worktree_path: Worktree path.
-        pid: Process ID.
-        task_dir: Task directory path.
-        repo_root: Repository root path. Defaults to auto-detected.
-        platform: Platform used (e.g., 'claude', 'opencode', 'codex', 'kiro', 'antigravity'). Defaults to 'claude'.
-
-    Returns:
-        True on success.
-    """
-    if repo_root is None:
-        repo_root = get_repo_root()
-
-    registry_file = _ensure_registry(repo_root)
-    if not registry_file:
-        return False
-
-    data = read_json(registry_file)
-    if not data:
-        data = {"agents": []}
-
-    # Remove existing agent with same ID
-    agents = data.get("agents", [])
-    agents = [a for a in agents if a.get("id") != agent_id]
-
-    # Create new agent record
-    started_at = datetime.now().isoformat()
-    new_agent = {
-        "id": agent_id,
-        "worktree_path": worktree_path,
-        "pid": pid,
-        "started_at": started_at,
-        "task_dir": task_dir,
-        "platform": platform,
-    }
-
-    agents.append(new_agent)
-    data["agents"] = agents
-
-    return write_json(registry_file, data)
-
-
-def registry_list_agents(repo_root: Path | None = None) -> list[dict]:
-    """List all agents.
-
-    Args:
-        repo_root: Repository root path. Defaults to auto-detected.
-
-    Returns:
-        List of agent dicts.
-    """
-    _, data = _load_registry(repo_root)
-    if not data:
-        return []
-
-    return data.get("agents", [])
-
-
-# =============================================================================
-# Main Entry (for testing)
-# =============================================================================
-
-if __name__ == "__main__":
-    import json as json_mod
-
-    repo = get_repo_root()
-    print(f"Repository root: {repo}")
-    print(f"Registry file: {registry_get_file(repo)}")
-    print()
-    print("Agents:")
-    agents = registry_list_agents(repo)
-    print(json_mod.dumps(agents, indent=2))