auto resolve platform

Author: caroljung-cg

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

@@ -1,3 +1,4 @@
+import platform as py_platform
 import subprocess
 from importlib.metadata import version
 from pathlib import Path
@@ -18,12 +19,11 @@
 
 
 @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=None, help="Port to run the server on")
 @click.option("--detached", "-d", is_flag=True, help="Run the server in detached mode")
 @click.option("--skip-build", is_flag=True, help="Skip building the Docker image")
 @click.option("--force", "-f", is_flag=True, help="Force start the server even if it is already running")
-def start_command(port: int | None, platform: str, detached: bool = False, skip_build: bool = False, force: bool = False) -> None:
+def start_command(port: int | None, detached: bool = False, skip_build: bool = False, force: bool = False) -> None:
     """Starts a local codegen server"""
     repo_path = Path.cwd().resolve()
     repo_config = RepoConfig.from_repo_path(str(repo_path))
@@ -42,15 +42,10 @@ def start_command(port: int | None, platform: str, detached: bool = False, skip_
 
     try:
         if not skip_build:
-            rich.print("[bold blue]Building Docker image...[/bold blue]")
-            _build_docker_image(codegen_root, platform)
-        rich.print("[bold blue]Starting Docker container...[/bold blue]")
+            _build_docker_image(codegen_root)
         _run_docker_container(repo_config, port, detached)
         rich.print(Panel(f"[green]Server started successfully![/green]\nAccess the server at: [bold]http://{_default_host}:{port}[/bold]", box=ROUNDED, title="Codegen Server"))
         # TODO: memory snapshot here
-    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()
@@ -75,7 +70,8 @@ def _handle_existing_container(repo_config: RepoConfig, container: DockerContain
     click.Abort()
 
 
-def _build_docker_image(codegen_root: Path, platform: str) -> None:
+def _build_docker_image(codegen_root: Path) -> None:
+    platform = _get_platform()
     build_cmd = [
         "docker",
         "buildx",
@@ -89,11 +85,31 @@ def _build_docker_image(codegen_root: Path, platform: str) -> None:
         "--load",
         str(codegen_root),
     ]
-    rich.print(f"build_cmd: {str.join(' ', build_cmd)}")
+    rich.print(
+        Panel(
+            f"{str.join(' ', build_cmd)}",
+            box=ROUNDED,
+            title="Running Build Command",
+            style="blue",
+            padding=(1, 1),
+        )
+    )
     subprocess.run(build_cmd, check=True)
 
 
+def _get_platform() -> str:
+    machine = py_platform.machine().lower()
+    if machine in ("x86_64", "amd64"):
+        return "linux/amd64"
+    elif machine in ("arm64", "aarch64"):
+        return "linux/arm64"
+    else:
+        rich.print(f"[yellow]Warning: Unknown architecture {machine}, defaulting to linux/amd64[/yellow]")
+        return "linux/amd64"
+
+
 def _run_docker_container(repo_config: RepoConfig, port: int, detached: bool) -> None:
+    rich.print("[bold blue]Starting Docker container...[/bold blue]")
     container_repo_path = f"/app/git/{repo_config.name}"
     name_args = ["--name", f"{repo_config.name}"]
     envvars = {
@@ -110,7 +126,15 @@ def _run_docker_container(repo_config: RepoConfig, port: int, detached: bool) ->
     detached_args = ["-d"] if detached else []
     run_cmd = ["docker", "run", "--rm", *detached_args, *port_args, *name_args, *mount_args, *envvars_args, CODEGEN_RUNNER_IMAGE, entry_point]
 
-    rich.print(f"run_cmd: {str.join(' ', run_cmd)}")
+    rich.print(
+        Panel(
+            f"{str.join(' ', run_cmd)}",
+            box=ROUNDED,
+            title="Running Run Command",
+            style="blue",
+            padding=(1, 1),
+        )
+    )
     subprocess.run(run_cmd, check=True)
 
     if detached: