refined

Author: codegen-bot

Dockerfile-runner

@@ -34,14 +34,16 @@ RUN curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.5/install.sh | b
 
 # 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
 
 # Install codegen from source instead of PyPI
 WORKDIR /codegen-sdk
-
-# Add build argument for repo path
-ARG REPO_PATH=.
-# Copy from specified repo path instead of current directory
-COPY ${REPO_PATH} .
+COPY . .
 
 # Install dependencies and build codegen with entry points
 RUN --mount=type=cache,target=/root/.cache/uv \
@@ -50,17 +52,7 @@ RUN --mount=type=cache,target=/root/.cache/uv \
     && 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
+RUN codegen --version
 
 # Create a non-root user for local development + debugging
 RUN useradd -m -s /bin/bash user

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

@@ -7,6 +7,9 @@
 from rich.box import ROUNDED
 from rich.panel import Panel
 
+from codegen.configs.models.secrets import SecretsConfig
+from codegen.git.schemas.repo_config import RepoConfig
+
 
 @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")
@@ -15,44 +18,57 @@
 def start_command(port: int, platform: str, detached: bool):
     """Starts a local codegen server"""
     codegen_version = version("codegen")
-    rich.print(codegen_version)
-    repo_root = Path(__file__).parent.parent.parent.parent.parent.parent
-    dockerfile_path = repo_root / "Dockerfile-runner"
+    rich.print(f"[bold green]Codegen version:[/bold green] {codegen_version}")
+    codegen_root = Path(__file__).parent.parent.parent.parent.parent.parent
+
+    try:
+        rich.print("[bold blue]Building Docker image...[/bold blue]")
+        _build_docker_image(codegen_root, platform)
+        rich.print("[bold blue]Starting Docker container...[/bold blue]")
+        _run_docker_container(port, platform, detached)
+        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()
 
-    # Build the Docker image
-    rich.print("[bold blue]Building Docker image...[/bold blue]")
+
+def _build_docker_image(codegen_root: Path, platform: str):
     build_cmd = [
         "docker",
         "buildx",
         "build",
         "--platform",
         platform,
         "-f",
-        str(dockerfile_path),
+        str(codegen_root / "Dockerfile-runner"),
         "-t",
         "codegen-runner",
         "--load",
-        str(repo_root),
+        str(codegen_root),
     ]
     rich.print(f"build_cmd: {str.join(' ', build_cmd)}")
+    subprocess.run(build_cmd, check=True)
 
-    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)
+def _run_docker_container(port: int, detached: bool):
+    repo_path = Path.cwd().resolve()
+    repo_config = RepoConfig.from_repo_path(repo_path)
+    container_repo_path = f"/app/git/{repo_config.name}"
+    envvars = {
+        "REPOSITORY_LANGUAGE": repo_config.language.value,
+        "REPOSITORY_OWNER": repo_config.organization_name,
+        "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}")]
 
-        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"))
+    mount_args = ["-v", f"{repo_path}:{container_repo_path}"]
+    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}", *mount_args, *envvars_args, "codegen-runner", entry_point]
 
-    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()
+    rich.print(f"run_cmd: {str.join(' ', run_cmd)}")
+    subprocess.run(run_cmd, check=True)