@jahanxu/trellis 0.4.1

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

@@ -0,0 +1,189 @@
+#!/usr/bin/env python3
+"""
+Project-specific path configuration loader.
+
+Reads .trellis/paths.yaml to get project-specific directory structure.
+Falls back to defaults if config file doesn't exist.
+
+This separates framework layer from business layer, allowing different
+projects to use different directory structures.
+"""
+
+import sys
+from pathlib import Path
+from typing import Optional, Dict
+
+try:
+    import yaml
+    HAS_YAML = True
+except ImportError:
+    HAS_YAML = False
+
+# =============================================================================
+# Default Configuration (Framework Layer)
+# =============================================================================
+
+DEFAULT_CONFIG = {
+    "deliverables_base": "deliverables",
+    "role_outputs": {
+        "pm": {"dir": "requirements"},
+        "designer": {"dir": "prototypes"},
+        "frontend": {"dir": "production/features"},
+    },
+    "pools": {
+        "pm": "requirements",
+        "designer": "prototypes",
+        "frontend": "implementations",
+    },
+}
+
+# =============================================================================
+# Configuration Loader
+# =============================================================================
+
+
+def load_paths_config(repo_root: Path) -> Dict:
+    """Load paths configuration from .trellis/paths.yaml.
+
+    Args:
+        repo_root: Repository root path
+
+    Returns:
+        Configuration dict with paths
+    """
+    config_file = repo_root / ".trellis" / "paths.yaml"
+
+    # If config file doesn't exist, use defaults
+    if not config_file.exists():
+        return DEFAULT_CONFIG.copy()
+
+    # If yaml not available, use defaults
+    if not HAS_YAML:
+        print(
+            "Warning: PyYAML not installed, using default paths. "
+            "Install with: pip install pyyaml",
+            file=sys.stderr,
+        )
+        return DEFAULT_CONFIG.copy()
+
+    # Load config from yaml
+    try:
+        with open(config_file, "r", encoding="utf-8") as f:
+            config = yaml.safe_load(f) or {}
+
+        # Merge with defaults (in case some keys are missing)
+        merged = DEFAULT_CONFIG.copy()
+        merged.update(config)
+
+        # Ensure role_outputs has all roles
+        for role in ["pm", "designer", "frontend"]:
+            if role not in merged["role_outputs"]:
+                merged["role_outputs"][role] = DEFAULT_CONFIG["role_outputs"][role]
+
+        return merged
+    except Exception as e:
+        print(
+            f"Warning: Failed to load {config_file}: {e}. Using defaults.",
+            file=sys.stderr,
+        )
+        return DEFAULT_CONFIG.copy()
+
+
+def get_deliverables_base(repo_root: Path) -> str:
+    """Get deliverables base directory.
+
+    Args:
+        repo_root: Repository root path
+
+    Returns:
+        Deliverables base directory name
+    """
+    config = load_paths_config(repo_root)
+    return config.get("deliverables_base", "deliverables")
+
+
+def get_role_output_dir(repo_root: Path, role: str) -> str:
+    """Get output directory for a specific role.
+
+    Args:
+        repo_root: Repository root path
+        role: Role name (pm, designer, frontend)
+
+    Returns:
+        Full output directory path (relative to repo_root)
+    """
+    config = load_paths_config(repo_root)
+    base = config.get("deliverables_base", "deliverables")
+    role_config = config.get("role_outputs", {}).get(role, {})
+    role_dir = role_config.get("dir", "")
+
+    if not role_dir:
+        # Fallback to default
+        role_dir = DEFAULT_CONFIG["role_outputs"].get(role, {}).get("dir", "")
+
+    return f"{base}/{role_dir}"
+
+
+def get_pool_name(role: str) -> str:
+    """Get pool name for a specific role.
+
+    Args:
+        role: Role name (pm, designer, frontend)
+
+    Returns:
+        Pool name (e.g., 'requirements', 'prototypes', 'implementations')
+    """
+    # Pool names are usually fixed in framework layer
+    # But can be overridden in config if needed
+    return DEFAULT_CONFIG["pools"].get(role, role)
+
+
+def get_output_dir_for_task(repo_root: Path, role: str, task_id: str) -> str:
+    """Get complete output directory for a specific task.
+
+    Args:
+        repo_root: Repository root path
+        role: Role name (pm, designer, frontend)
+        task_id: Task ID
+
+    Returns:
+        Full output directory path (e.g., "deliverables/requirements/user-login")
+    """
+    role_dir = get_role_output_dir(repo_root, role)
+    return f"{role_dir}/{task_id}"
+
+
+# =============================================================================
+# CLI for testing
+# =============================================================================
+
+
+def main():
+    """Test configuration loading."""
+    # Add parent directory to path for imports
+    sys.path.insert(0, str(Path(__file__).parent))
+    from paths import get_repo_root
+
+    repo_root = get_repo_root()
+    config = load_paths_config(repo_root)
+
+    print("=== Loaded Configuration ===")
+    print(f"Deliverables base: {config['deliverables_base']}")
+    print("\nRole output directories:")
+    for role in ["pm", "designer", "frontend"]:
+        full_path = get_role_output_dir(repo_root, role)
+        print(f"  {role}: {full_path}")
+
+    print("\nPool names:")
+    for role in ["pm", "designer", "frontend"]:
+        pool = get_pool_name(role)
+        print(f"  {role}: {pool}")
+
+    print("\nExample output paths:")
+    for role in ["pm", "designer", "frontend"]:
+        path = get_output_dir_for_task(repo_root, role, "user-login")
+        print(f"  {role}/user-login: {path}")
+
+
+if __name__ == "__main__":
+    main()

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

@@ -0,0 +1,365 @@
+#!/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
+"""
+
+
+import json
+from datetime import datetime
+from pathlib import Path
+
+from .paths import get_repo_root
+from .worktree import get_agents_dir
+
+
+def _read_json_file(path: Path) -> Optional[dict]:
+    """Read and parse a JSON file."""
+    try:
+        return json.loads(path.read_text(encoding="utf-8"))
+    except (FileNotFoundError, json.JSONDecodeError, OSError):
+        return None
+
+
+def _write_json_file(path: Path, data: dict) -> bool:
+    """Write dict to JSON file."""
+    try:
+        path.write_text(json.dumps(data, indent=2, ensure_ascii=False), encoding="utf-8")
+        return True
+    except (OSError, IOError):
+        return False
+
+
+# =============================================================================
+# Registry File Access
+# =============================================================================
+
+def registry_get_file(repo_root: Optional[Path] = None) -> Optional[Path]:
+    """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: Optional[Path] = None) -> Optional[Path]:
+    """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_file(registry_file, {"agents": []})
+
+        return registry_file
+    except (OSError, IOError):
+        return None
+
+
+# =============================================================================
+# Agent Lookup
+# =============================================================================
+
+def registry_get_agent_by_id(
+    agent_id: str,
+    repo_root: Optional[Path] = None
+) -> Optional[dict]:
+    """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.
+    """
+    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 None
+
+    data = _read_json_file(registry_file)
+    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: Optional[Path] = None
+) -> Optional[dict]:
+    """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.
+    """
+    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 None
+
+    data = _read_json_file(registry_file)
+    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: Optional[Path] = None
+) -> Optional[dict]:
+    """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.
+    """
+    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 None
+
+    data = _read_json_file(registry_file)
+    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: Optional[Path] = None
+) -> Optional[str]:
+    """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.
+    """
+    agent = registry_get_agent_by_worktree(worktree_path, repo_root)
+    if agent:
+        return agent.get("task_dir")
+    return None
+
+
+# =============================================================================
+# Agent Modification
+# =============================================================================
+
+def registry_remove_by_id(agent_id: str, repo_root: Optional[Path] = None) -> bool:
+    """Remove agent by ID.
+
+    Args:
+        agent_id: Agent ID.
+        repo_root: Repository root path. Defaults to auto-detected.
+
+    Returns:
+        True on success.
+    """
+    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 True  # Nothing to remove
+
+    data = _read_json_file(registry_file)
+    if not data:
+        return True
+
+    agents = data.get("agents", [])
+    data["agents"] = [a for a in agents if a.get("id") != agent_id]
+
+    return _write_json_file(registry_file, data)
+
+
+def registry_remove_by_worktree(
+    worktree_path: str,
+    repo_root: Optional[Path] = 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.
+    """
+    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 True  # Nothing to remove
+
+    data = _read_json_file(registry_file)
+    if not data:
+        return True
+
+    agents = data.get("agents", [])
+    data["agents"] = [a for a in agents if a.get("worktree_path") != worktree_path]
+
+    return _write_json_file(registry_file, data)
+
+
+def registry_add_agent(
+    agent_id: str,
+    worktree_path: str,
+    pid: int,
+    task_dir: str,
+    repo_root: Optional[Path] = 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_file(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_file(registry_file, data)
+
+
+def registry_list_agents(repo_root: Optional[Path] = None) -> list[dict]:
+    """List all agents.
+
+    Args:
+        repo_root: Repository root path. Defaults to auto-detected.
+
+    Returns:
+        List of agent dicts.
+    """
+    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 []
+
+    data = _read_json_file(registry_file)
+    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))