autoforge-ai 0.1.0

package/server/__init__.py

@@ -0,0 +1,17 @@
+"""
+FastAPI Backend Server
+======================
+
+Web UI server for the Autonomous Coding Agent.
+Provides REST API and WebSocket endpoints for project management,
+feature tracking, and agent control.
+"""
+
+# Fix Windows asyncio subprocess support - MUST be before any other imports
+# that might create an event loop
+import sys
+
+if sys.platform == "win32":
+    import asyncio
+
+    asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())

package/server/main.py

@@ -0,0 +1,261 @@
+"""
+FastAPI Main Application
+========================
+
+Main entry point for the Autonomous Coding UI server.
+Provides REST API, WebSocket, and static file serving.
+"""
+
+import asyncio
+import logging
+import os
+import shutil
+import sys
+from contextlib import asynccontextmanager
+from pathlib import Path
+
+# Fix for Windows subprocess support in asyncio
+if sys.platform == "win32":
+    asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
+
+from dotenv import load_dotenv
+
+# Load environment variables from .env file if present
+load_dotenv()
+
+from fastapi import FastAPI, HTTPException, Request, WebSocket
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import FileResponse
+from fastapi.staticfiles import StaticFiles
+
+from .routers import (
+    agent_router,
+    assistant_chat_router,
+    devserver_router,
+    expand_project_router,
+    features_router,
+    filesystem_router,
+    projects_router,
+    schedules_router,
+    settings_router,
+    spec_creation_router,
+    terminal_router,
+)
+from .schemas import SetupStatus
+from .services.assistant_chat_session import cleanup_all_sessions as cleanup_assistant_sessions
+from .services.chat_constants import ROOT_DIR
+from .services.dev_server_manager import (
+    cleanup_all_devservers,
+    cleanup_orphaned_devserver_locks,
+)
+from .services.expand_chat_session import cleanup_all_expand_sessions
+from .services.process_manager import cleanup_all_managers, cleanup_orphaned_locks
+from .services.scheduler_service import cleanup_scheduler, get_scheduler
+from .services.terminal_manager import cleanup_all_terminals
+from .websocket import project_websocket
+
+# Paths
+UI_DIST_DIR = ROOT_DIR / "ui" / "dist"
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """Lifespan context manager for startup and shutdown."""
+    # Startup - clean up orphaned lock files from previous runs
+    cleanup_orphaned_locks()
+    cleanup_orphaned_devserver_locks()
+
+    # Start the scheduler service
+    scheduler = get_scheduler()
+    await scheduler.start()
+
+    yield
+
+    # Shutdown - cleanup scheduler first to stop triggering new starts
+    await cleanup_scheduler()
+    # Then cleanup all running agents, sessions, terminals, and dev servers
+    await cleanup_all_managers()
+    await cleanup_assistant_sessions()
+    await cleanup_all_expand_sessions()
+    await cleanup_all_terminals()
+    await cleanup_all_devservers()
+
+
+# Create FastAPI app
+app = FastAPI(
+    title="Autonomous Coding UI",
+    description="Web UI for the Autonomous Coding Agent",
+    version="1.0.0",
+    lifespan=lifespan,
+)
+
+# Module logger
+logger = logging.getLogger(__name__)
+
+# Check if remote access is enabled via environment variable
+# Set by start_ui.py when --host is not 127.0.0.1
+ALLOW_REMOTE = os.environ.get("AUTOFORGE_ALLOW_REMOTE", "").lower() in ("1", "true", "yes")
+
+if ALLOW_REMOTE:
+    logger.warning(
+        "ALLOW_REMOTE is enabled. Terminal WebSocket is exposed without sandboxing. "
+        "Only use this in trusted network environments."
+    )
+
+# CORS - allow all origins when remote access is enabled, otherwise localhost only
+if ALLOW_REMOTE:
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=["*"],  # Allow all origins for remote access
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
+else:
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=[
+            "http://localhost:5173",      # Vite dev server
+            "http://127.0.0.1:5173",
+            "http://localhost:8888",      # Production
+            "http://127.0.0.1:8888",
+        ],
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
+
+
+# ============================================================================
+# Security Middleware
+# ============================================================================
+
+if not ALLOW_REMOTE:
+    @app.middleware("http")
+    async def require_localhost(request: Request, call_next):
+        """Only allow requests from localhost (disabled when AUTOFORGE_ALLOW_REMOTE=1)."""
+        client_host = request.client.host if request.client else None
+
+        # Allow localhost connections
+        if client_host not in ("127.0.0.1", "::1", "localhost", None):
+            raise HTTPException(status_code=403, detail="Localhost access only")
+
+        return await call_next(request)
+
+
+# ============================================================================
+# Include Routers
+# ============================================================================
+
+app.include_router(projects_router)
+app.include_router(features_router)
+app.include_router(agent_router)
+app.include_router(schedules_router)
+app.include_router(devserver_router)
+app.include_router(spec_creation_router)
+app.include_router(expand_project_router)
+app.include_router(filesystem_router)
+app.include_router(assistant_chat_router)
+app.include_router(settings_router)
+app.include_router(terminal_router)
+
+
+# ============================================================================
+# WebSocket Endpoint
+# ============================================================================
+
+@app.websocket("/ws/projects/{project_name}")
+async def websocket_endpoint(websocket: WebSocket, project_name: str):
+    """WebSocket endpoint for real-time project updates."""
+    await project_websocket(websocket, project_name)
+
+
+# ============================================================================
+# Setup & Health Endpoints
+# ============================================================================
+
+@app.get("/api/health")
+async def health_check():
+    """Health check endpoint."""
+    return {"status": "healthy"}
+
+
+@app.get("/api/setup/status", response_model=SetupStatus)
+async def setup_status():
+    """Check system setup status."""
+    # Check for Claude CLI
+    claude_cli = shutil.which("claude") is not None
+
+    # Check for CLI configuration directory
+    # Note: CLI no longer stores credentials in ~/.claude/.credentials.json
+    # The existence of ~/.claude indicates the CLI has been configured
+    claude_dir = Path.home() / ".claude"
+    has_claude_config = claude_dir.exists() and claude_dir.is_dir()
+
+    # If GLM mode is configured via .env, we have alternative credentials
+    glm_configured = bool(os.getenv("ANTHROPIC_BASE_URL") and os.getenv("ANTHROPIC_AUTH_TOKEN"))
+    credentials = has_claude_config or glm_configured
+
+    # Check for Node.js and npm
+    node = shutil.which("node") is not None
+    npm = shutil.which("npm") is not None
+
+    return SetupStatus(
+        claude_cli=claude_cli,
+        credentials=credentials,
+        node=node,
+        npm=npm,
+    )
+
+
+# ============================================================================
+# Static File Serving (Production)
+# ============================================================================
+
+# Serve React build files if they exist
+if UI_DIST_DIR.exists():
+    # Mount static assets
+    app.mount("/assets", StaticFiles(directory=UI_DIST_DIR / "assets"), name="assets")
+
+    @app.get("/")
+    async def serve_index():
+        """Serve the React app index.html."""
+        return FileResponse(UI_DIST_DIR / "index.html")
+
+    @app.get("/{path:path}")
+    async def serve_spa(path: str):
+        """
+        Serve static files or fall back to index.html for SPA routing.
+        """
+        # Check if the path is an API route (shouldn't hit this due to router ordering)
+        if path.startswith("api/") or path.startswith("ws/"):
+            raise HTTPException(status_code=404)
+
+        # Try to serve the file directly
+        file_path = (UI_DIST_DIR / path).resolve()
+
+        # Ensure resolved path is within UI_DIST_DIR (prevent path traversal)
+        try:
+            file_path.relative_to(UI_DIST_DIR.resolve())
+        except ValueError:
+            raise HTTPException(status_code=404)
+
+        if file_path.exists() and file_path.is_file():
+            return FileResponse(file_path)
+
+        # Fall back to index.html for SPA routing
+        return FileResponse(UI_DIST_DIR / "index.html")
+
+
+# ============================================================================
+# Main Entry Point
+# ============================================================================
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(
+        "server.main:app",
+        host="127.0.0.1",  # Localhost only for security
+        port=8888,
+        reload=True,
+    )

package/server/routers/__init__.py

@@ -0,0 +1,32 @@
+"""
+API Routers
+===========
+
+FastAPI routers for different API endpoints.
+"""
+
+from .agent import router as agent_router
+from .assistant_chat import router as assistant_chat_router
+from .devserver import router as devserver_router
+from .expand_project import router as expand_project_router
+from .features import router as features_router
+from .filesystem import router as filesystem_router
+from .projects import router as projects_router
+from .schedules import router as schedules_router
+from .settings import router as settings_router
+from .spec_creation import router as spec_creation_router
+from .terminal import router as terminal_router
+
+__all__ = [
+    "projects_router",
+    "features_router",
+    "agent_router",
+    "schedules_router",
+    "devserver_router",
+    "spec_creation_router",
+    "expand_project_router",
+    "filesystem_router",
+    "assistant_chat_router",
+    "settings_router",
+    "terminal_router",
+]

package/server/routers/agent.py

@@ -0,0 +1,177 @@
+"""
+Agent Router
+============
+
+API endpoints for agent control (start/stop/pause/resume).
+Uses project registry for path lookups.
+"""
+
+from pathlib import Path
+
+from fastapi import APIRouter, HTTPException
+
+from ..schemas import AgentActionResponse, AgentStartRequest, AgentStatus
+from ..services.chat_constants import ROOT_DIR
+from ..services.process_manager import get_manager
+from ..utils.project_helpers import get_project_path as _get_project_path
+from ..utils.validation import validate_project_name
+
+
+def _get_settings_defaults() -> tuple[bool, str, int, bool, int]:
+    """Get defaults from global settings.
+
+    Returns:
+        Tuple of (yolo_mode, model, testing_agent_ratio, playwright_headless, batch_size)
+    """
+    import sys
+    root = Path(__file__).parent.parent.parent
+    if str(root) not in sys.path:
+        sys.path.insert(0, str(root))
+
+    from registry import DEFAULT_MODEL, get_all_settings
+
+    settings = get_all_settings()
+    yolo_mode = (settings.get("yolo_mode") or "false").lower() == "true"
+    model = settings.get("model", DEFAULT_MODEL)
+
+    # Parse testing agent settings with defaults
+    try:
+        testing_agent_ratio = int(settings.get("testing_agent_ratio", "1"))
+    except (ValueError, TypeError):
+        testing_agent_ratio = 1
+
+    playwright_headless = (settings.get("playwright_headless") or "true").lower() == "true"
+
+    try:
+        batch_size = int(settings.get("batch_size", "3"))
+    except (ValueError, TypeError):
+        batch_size = 3
+
+    return yolo_mode, model, testing_agent_ratio, playwright_headless, batch_size
+
+
+router = APIRouter(prefix="/api/projects/{project_name}/agent", tags=["agent"])
+
+
+def get_project_manager(project_name: str):
+    """Get the process manager for a project."""
+    project_name = validate_project_name(project_name)
+    project_dir = _get_project_path(project_name)
+
+    if not project_dir:
+        raise HTTPException(status_code=404, detail=f"Project '{project_name}' not found in registry")
+
+    if not project_dir.exists():
+        raise HTTPException(status_code=404, detail=f"Project directory not found: {project_dir}")
+
+    return get_manager(project_name, project_dir, 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.isoformat() if manager.started_at else None,
+        yolo_mode=manager.yolo_mode,
+        model=manager.model,
+        parallel_mode=manager.parallel_mode,
+        max_concurrency=manager.max_concurrency,
+        testing_agent_ratio=manager.testing_agent_ratio,
+    )
+
+
+@router.post("/start", response_model=AgentActionResponse)
+async def start_agent(
+    project_name: str,
+    request: AgentStartRequest = AgentStartRequest(),
+):
+    """Start the agent for a project."""
+    manager = get_project_manager(project_name)
+
+    # Get defaults from global settings if not provided in request
+    default_yolo, default_model, default_testing_ratio, playwright_headless, default_batch_size = _get_settings_defaults()
+
+    yolo_mode = request.yolo_mode if request.yolo_mode is not None else default_yolo
+    model = request.model if request.model else default_model
+    max_concurrency = request.max_concurrency or 1
+    testing_agent_ratio = request.testing_agent_ratio if request.testing_agent_ratio is not None else default_testing_ratio
+
+    batch_size = default_batch_size
+
+    success, message = await manager.start(
+        yolo_mode=yolo_mode,
+        model=model,
+        max_concurrency=max_concurrency,
+        testing_agent_ratio=testing_agent_ratio,
+        playwright_headless=playwright_headless,
+        batch_size=batch_size,
+    )
+
+    # Notify scheduler of manual start (to prevent auto-stop during scheduled window)
+    if success:
+        from ..services.scheduler_service import get_scheduler
+        project_dir = _get_project_path(project_name)
+        if project_dir:
+            get_scheduler().notify_manual_start(project_name, project_dir)
+
+    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()
+
+    # Notify scheduler of manual stop (to prevent auto-start during scheduled window)
+    if success:
+        from ..services.scheduler_service import get_scheduler
+        project_dir = _get_project_path(project_name)
+        if project_dir:
+            get_scheduler().notify_manual_stop(project_name, project_dir)
+
+    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,
+    )