Support selecting the architecture for Linux developer environments (#65)

Co-authored-by: Ofek Lev <ofekmeister@gmail.com>

Author: KevinFairise2

src/dda/env/dev/types/linux_container.py

@@ -49,6 +49,14 @@ class LinuxContainerConfig(DeveloperEnvironmentConfig):
             }
         ),
     ] = "zsh"
+    arch: Annotated[
+        str | None,
+        msgspec.Meta(
+            extra={
+                "help": "The architecture to use e.g. `amd64` or `arm64`",
+            }
+        ),
+    ] = None
 
 
 class LinuxContainer(DeveloperEnvironmentInterface[LinuxContainerConfig]):
@@ -81,7 +89,10 @@ def start(self) -> None:
             from dda.utils.retry import wait_for
 
             if not self.config.no_pull:
-                self.docker.wait(["pull", self.config.image], message=f"Pulling image: {self.config.image}")
+                pull_command = ["pull", self.config.image]
+                if self.config.arch is not None:
+                    pull_command.extend(("--platform", f"linux/{self.config.arch}"))
+                self.docker.wait(pull_command, message=f"Pulling image: {self.config.image}")
 
             self.shared_dir.ensure_dir()
             command = [
@@ -98,6 +109,9 @@ def start(self) -> None:
                 "-e",
                 AppEnvVars.TELEMETRY_API_KEY,
             ]
+            if self.config.arch is not None:
+                command.extend(("--platform", f"linux/{self.config.arch}"))
+
             for shared_shell_file in self.shell.collect_shared_files():
                 unix_path = shared_shell_file.relative_to(self.global_shared_dir).as_posix()
                 command.extend(("-v", f"{shared_shell_file}:{self.home_dir}/.shared/{unix_path}"))

tests/env/dev/types/test_linux_container.py

@@ -38,6 +38,7 @@ def test_default_config(app):
     container = LinuxContainer(app=app, name="linux-container", instance="default")
 
     assert msgspec.to_builtins(container.config) == {
+        "arch": None,
         "cli": "docker",
         "clone": False,
         "image": "datadog/agent-dev-env-linux",