feat: add --local optional to mount repo to docker

Author: caroljung-cg

Dockerfile-runner

@@ -1,3 +1,6 @@
+ARG LOCAL_REPO_PATH
+ARG REPO_NAME
+
 FROM ghcr.io/astral-sh/uv:python3.13-bookworm-slim
 
 # Set environment variables to prevent interactive prompts during installation
@@ -59,4 +62,7 @@ RUN chown -R user:user /home/user
 USER user
 
 WORKDIR /app
+
+COPY ${LOCAL_REPO_PATH} /app/git/${REPO_NAME}
+
 ENTRYPOINT ["/bin/bash", "-c"]

Dockerfile-runner-mount

@@ -0,0 +1,62 @@
+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
+RUN node --version \
+    && corepack --version \
+    && npm --version \
+    && yarn --version \
+    && pnpm --version \
+    && python --version
+
+# Build codegen
+WORKDIR /codegen-sdk
+COPY . .
+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 .
+RUN codegen --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/commands/start/main.py

@@ -17,7 +17,8 @@
 @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=None, help="Port to run the server on")
 @click.option("--detached", "-d", is_flag=True, default=False, help="Starts up the server as detached background process")
-def start_command(port: int | None, platform: str, detached: bool):
+@click.option("--local", is_flag=True, default=False, help="If true, interacts with the mounted local repository. If false, interacts with the repository copied into the image.")
+def start_command(port: int | None, platform: str, detached: bool, local: bool):
     """Starts a local codegen server"""
     codegen_version = version("codegen")
     rich.print(f"[bold green]Codegen version:[/bold green] {codegen_version}")
@@ -26,10 +27,12 @@ def start_command(port: int | None, platform: str, detached: bool):
         port = get_free_port()
 
     try:
+        repo_path = Path.cwd().resolve()
+        repo_config = RepoConfig.from_repo_path(repo_path)
         rich.print("[bold blue]Building Docker image...[/bold blue]")
-        _build_docker_image(codegen_root, platform)
+        _build_docker_image(repo_config, codegen_root, platform, local)
         rich.print("[bold blue]Starting Docker container...[/bold blue]")
-        _run_docker_container(port, detached)
+        _run_docker_container(repo_config, port, detached, local)
         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")
@@ -39,15 +42,18 @@ def start_command(port: int | None, platform: str, detached: bool):
         raise click.Abort()
 
 
-def _build_docker_image(codegen_root: Path, platform: str):
+def _build_docker_image(repo_config: RepoConfig, codegen_root: Path, platform: str, local: bool):
+    build_arg = [] if local else ["--build-arg", f"LOCAL_REPO_PATH={repo_config.repo_path}", "--build-arg", f"REPO_NAME={repo_config.name}"]
+    dockerfile = "Dockerfile-runner-mount" if local else "Dockerfile-runner"
     build_cmd = [
         "docker",
         "buildx",
         "build",
+        *build_arg,
         "--platform",
         platform,
         "-f",
-        str(codegen_root / "Dockerfile-runner"),
+        str(codegen_root / dockerfile),
         "-t",
         "codegen-runner",
         "--load",
@@ -57,18 +63,16 @@ def _build_docker_image(codegen_root: Path, platform: str):
     subprocess.run(build_cmd, check=True)
 
 
-def _run_docker_container(port: int, detached: bool):
-    repo_path = Path.cwd().resolve()
-    repo_config = RepoConfig.from_repo_path(repo_path)
+def _run_docker_container(repo_config: RepoConfig, port: int, detached: bool, local: bool):
     container_repo_path = f"/app/git/{repo_config.name}"
     envvars = {
         "REPOSITORY_LANGUAGE": repo_config.language.value,
-        "REPOSITORY_OWNER": LocalGitRepo(repo_path).owner,
+        "REPOSITORY_OWNER": LocalGitRepo(repo_config.repo_path).owner,
         "REPOSITORY_PATH": container_repo_path,
         "GITHUB_TOKEN": SecretsConfig().github_token,
     }
     envvars_args = [arg for k, v in envvars.items() for arg in ("--env", f"{k}={v}")]
-    mount_args = ["-v", f"{repo_path}:{container_repo_path}"]
+    mount_args = ["-v", f"{repo_config.repo_path}:{container_repo_path}"] if local else []
     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"{port}:{port}", *mount_args, *envvars_args, "codegen-runner", entry_point]