basic ui

server/routers/agent.py

@@ -0,0 +1,128 @@
+"""
+Agent Router
+============
+
+API endpoints for agent control (start/stop/pause/resume).
+"""
+
+import re
+from pathlib import Path
+
+from fastapi import APIRouter, HTTPException
+
+from ..schemas import AgentStatus, AgentActionResponse
+from ..services.process_manager import get_manager
+
+# Lazy import to avoid sys.path manipulation at module level
+_GENERATIONS_DIR = None
+
+
+def _get_generations_dir():
+    """Lazy import of GENERATIONS_DIR."""
+    global _GENERATIONS_DIR
+    if _GENERATIONS_DIR is None:
+        import sys
+        root = Path(__file__).parent.parent.parent
+        if str(root) not in sys.path:
+            sys.path.insert(0, str(root))
+        from start import GENERATIONS_DIR
+        _GENERATIONS_DIR = GENERATIONS_DIR
+    return _GENERATIONS_DIR
+
+
+router = APIRouter(prefix="/api/projects/{project_name}/agent", tags=["agent"])
+
+# Root directory for process manager
+ROOT_DIR = Path(__file__).parent.parent.parent
+
+
+def validate_project_name(name: str) -> str:
+    """Validate and sanitize project name to prevent path traversal."""
+    if not re.match(r'^[a-zA-Z0-9_-]{1,50}$', name):
+        raise HTTPException(
+            status_code=400,
+            detail="Invalid project name"
+        )
+    return name
+
+
+def get_project_manager(project_name: str):
+    """Get the process manager for a project."""
+    project_name = validate_project_name(project_name)
+    project_dir = _get_generations_dir() / project_name
+
+    if not project_dir.exists():
+        raise HTTPException(status_code=404, detail=f"Project '{project_name}' not found")
+
+    return get_manager(project_name, ROOT_DIR)
+
+
+@router.get("/status", response_model=AgentStatus)
+async def get_agent_status(project_name: str):
+    """Get the current status of the agent for a project."""
+    manager = get_project_manager(project_name)
+
+    # Run healthcheck to detect crashed processes
+    await manager.healthcheck()
+
+    return AgentStatus(
+        status=manager.status,
+        pid=manager.pid,
+        started_at=manager.started_at,
+    )
+
+
+@router.post("/start", response_model=AgentActionResponse)
+async def start_agent(project_name: str):
+    """Start the agent for a project."""
+    manager = get_project_manager(project_name)
+
+    success, message = await manager.start()
+
+    return AgentActionResponse(
+        success=success,
+        status=manager.status,
+        message=message,
+    )
+
+
+@router.post("/stop", response_model=AgentActionResponse)
+async def stop_agent(project_name: str):
+    """Stop the agent for a project."""
+    manager = get_project_manager(project_name)
+
+    success, message = await manager.stop()
+
+    return AgentActionResponse(
+        success=success,
+        status=manager.status,
+        message=message,
+    )
+
+
+@router.post("/pause", response_model=AgentActionResponse)
+async def pause_agent(project_name: str):
+    """Pause the agent for a project."""
+    manager = get_project_manager(project_name)
+
+    success, message = await manager.pause()
+
+    return AgentActionResponse(
+        success=success,
+        status=manager.status,
+        message=message,
+    )
+
+
+@router.post("/resume", response_model=AgentActionResponse)
+async def resume_agent(project_name: str):
+    """Resume a paused agent."""
+    manager = get_project_manager(project_name)
+
+    success, message = await manager.resume()
+
+    return AgentActionResponse(
+        success=success,
+        status=manager.status,
+        message=message,
+    )