CG-10667: Add local codegen server daemon

Author: caroljung-cg

Dockerfile-runner

@@ -0,0 +1,68 @@
+FROM ghcr.io/astral-sh/uv:python3.13-bookworm-slim
+
+# Set environment variables to prevent interactive prompts during installation
+ENV NVM_DIR=/root/.nvm \
+    NODE_VERSION=18.17.0 \
+    DEBIAN_FRONTEND=noninteractive \
+    NODE_OPTIONS="--max-old-space-size=8192" \
+    PYTHONUNBUFFERED=1 \
+    COREPACK_ENABLE_DOWNLOAD_PROMPT=0 \
+    PYTHONPATH="/usr/local/lib/python3.13/site-packages" \
+    IS_SANDBOX=True \
+    HATCH_BUILD_HOOKS_ENABLE=1
+# Update packages lists and install git and curl
+RUN apt-get update && apt-get install -y \
+    git \
+    curl \
+    gcc \
+    build-essential \
+    python3-dev \
+    # Cleanup apt cache to reduce image size
+    && rm -rf /var/lib/apt/lists/*
+
+# Install nvm and Node.js
+SHELL ["/bin/bash", "-c"]
+RUN curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.5/install.sh | bash \
+    && source $NVM_DIR/nvm.sh \
+    && nvm install $NODE_VERSION \
+    && nvm use default \
+    && npm install -g yarn pnpm \
+    && corepack enable \
+    && corepack prepare yarn@stable --activate \
+    && corepack prepare pnpm@latest --activate \
+    && uv pip install --system uvicorn[standard]
+
+# Add node and npm to PATH
+ENV PATH=$NVM_DIR/versions/node/v$NODE_VERSION/bin:$PATH
+
+# Install codegen from source instead of PyPI
+WORKDIR /codegen-sdk
+COPY . .
+
+# Install dependencies and build codegen with entry points
+RUN --mount=type=cache,target=/root/.cache/uv \
+    uv venv && source .venv/bin/activate \
+    && uv sync --frozen --no-dev --all-extras \
+    && uv pip install --system -e . --no-deps \
+    && uv pip install --system .
+
+# Change back to app directory for running
+WORKDIR /app
+
+# Verify all installations
+RUN codegen --version \
+    && node --version \
+    && corepack --version \
+    && npm --version \
+    && yarn --version \
+    && pnpm --version \
+    && python --version
+
+# Create a non-root user for local development + debugging
+RUN useradd -m -s /bin/bash user
+USER root
+RUN chown -R user:user /home/user
+USER user
+
+WORKDIR /app
+ENTRYPOINT ["/bin/bash", "-c"]

src/codegen/cli/cli.py

@@ -13,6 +13,7 @@
 from codegen.cli.commands.reset.main import reset_command
 from codegen.cli.commands.run.main import run_command
 from codegen.cli.commands.run_on_pr.main import run_on_pr_command
+from codegen.cli.commands.start.main import start_command
 from codegen.cli.commands.style_debug.main import style_debug_command
 
 click.rich_click.USE_RICH_MARKUP = True
@@ -39,6 +40,7 @@ def main():
 main.add_command(run_on_pr_command)
 main.add_command(notebook_command)
 main.add_command(reset_command)
+main.add_command(start_command)
 
 
 if __name__ == "__main__":

src/codegen/cli/commands/start/main.py

@@ -0,0 +1,59 @@
+from pathlib import Path
+
+import click
+import rich
+from rich.box import ROUNDED
+from rich.panel import Panel
+
+
+@click.command(name="start")
+@click.option("--platform", "-t", type=click.Choice(["linux/amd64", "linux/arm64", "linux/amd64,linux/arm64"]), default="linux/amd64,linux/arm64", help="Target platform(s) for the Docker image")
+@click.option("--port", "-p", type=int, default=8000)
+@click.option("--detached", "-d", is_flag=True, default=False, help="Starts up the server as detached background process")
+def start_command(port: int, platform: str, detached: bool):
+    """Starts a local codegen server"""
+    import subprocess
+    from importlib.metadata import version
+
+    codegen_version = version("codegen")
+    rich.print(codegen_version)
+    repo_root = Path(__file__).parent.parent.parent.parent.parent.parent
+    dockerfile_path = repo_root / "Dockerfile-runner"
+
+    # Build the Docker image
+    rich.print("[bold blue]Building Docker image...[/bold blue]")
+    build_cmd = [
+        "docker",
+        "buildx",
+        "build",
+        "--platform",
+        platform,
+        "-f",
+        str(dockerfile_path),
+        "-t",
+        "codegen-runner",
+        "--load",
+        str(repo_root),
+    ]
+    rich.print(f"build_cmd: {str.join(' ', build_cmd)}")
+
+    try:
+        subprocess.run(build_cmd, check=True)
+
+        # Run the Docker container
+        rich.print("[bold blue]Starting Docker container...[/bold blue]")
+        run_mode = "-d" if detached else "-it"
+        entry_point = f'"uv run --frozen uvicorn codegen.runner.sandbox.server:app --host 0.0.0.0 --port {port}"'
+        run_cmd = ["docker", "run", run_mode, "-p", f"8000:{port}", "codegen-runner", entry_point]
+
+        rich.print(f"run_cmd: {str.join(' ', run_cmd)}")
+        subprocess.run(run_cmd, check=True)
+
+        rich.print(Panel(f"[green]Server started successfully![/green]\nAccess the server at: [bold]http://0.0.0.0:{port}[/bold]", box=ROUNDED, title="Codegen Server"))
+
+    except subprocess.CalledProcessError as e:
+        rich.print(f"[bold red]Error:[/bold red] Failed to {e.cmd[0]} Docker container")
+        raise click.Abort()
+    except Exception as e:
+        rich.print(f"[bold red]Error:[/bold red] {e!s}")
+        raise click.Abort()