From 874f2dfaf4cd0f4a6ea9f097a44bfe1c62d58ccf Mon Sep 17 00:00:00 2001
From: Jose Diaz-Gonzalez <email@josediazgonzalez.com>
Date: Fri, 27 Oct 2023 23:02:08 -0400
Subject: [PATCH] feat: set the platform flag in order to run amd64 images on
 arm64 deploy targets

This allows folks to use herokuish, pack, and lambda built-images on arm64 servers.

Closes #6312
---
 docs/deployment/schedulers/docker-local.md                  | 6 ++++++
 .../bin/scheduler-deploy-process-container                  | 6 ++++++
 plugins/scheduler-docker-local/scheduler-run                | 6 ++++++
 3 files changed, 18 insertions(+)

diff --git a/docs/deployment/schedulers/docker-local.md b/docs/deployment/schedulers/docker-local.md
index 614bc98c8..4e0a861c9 100644
--- a/docs/deployment/schedulers/docker-local.md
+++ b/docs/deployment/schedulers/docker-local.md
@@ -25,6 +25,12 @@ dokku scheduler:set node-js-app selected
 
 ## Usage
 
+### Deploying amd64 images on arm64
+
+> New as of 0.33.0
+
+Many builders only produce amd64-compatible images. The docker-local scheduler will automatically detect these and run them via the `--platform=linux/amd64` on arm64 deploy targets.
+
 ### Disabling chown of persistent storage
 
 The `scheduler-docker-local` plugin will ensure your storage mounts are owned by either `herokuishuser` or the overridden value you have set in `DOKKU_APP_USER`. You may disable this by running the following `scheduler-docker-local:set` command for your application:
diff --git a/plugins/scheduler-docker-local/bin/scheduler-deploy-process-container b/plugins/scheduler-docker-local/bin/scheduler-deploy-process-container
index 511e52348..0ec2b6d18 100755
--- a/plugins/scheduler-docker-local/bin/scheduler-deploy-process-container
+++ b/plugins/scheduler-docker-local/bin/scheduler-deploy-process-container
@@ -49,6 +49,12 @@ main() {
 
   DOCKER_ARGS+=" --env=PORT=$DOKKU_PORT"
 
+  local DOKKU_IMAGE_ARCHITECTURE="$("$DOCKER_BIN" image inspect --format '{{.Architecture}}' "$IMAGE")"
+  if [[ "$DOKKU_IMAGE_ARCHITECTURE" == "amd64" ]] && [[ "$(dpkg --print-architecture 2>/dev/null || true)" != "amd64" ]]; then
+    dokku_log_warn "Detected linux/amd64 image, forcing --platform=linux/amd64"
+    DOCKER_ARGS+=" --platform=linux/amd64"
+  fi
+
   START_CMD=$(fn-scheduler-docker-local-extract-start-cmd "$APP" "$PROC_TYPE" "$START_CMD" "$DOKKU_HEROKUISH" "$DOKKU_PORT")
   DOCKER_ARGS+=" $IMAGE"
   DOCKER_ARGS+=" $START_CMD"
diff --git a/plugins/scheduler-docker-local/scheduler-run b/plugins/scheduler-docker-local/scheduler-run
index 8620341a3..cb6bbfb74 100755
--- a/plugins/scheduler-docker-local/scheduler-run
+++ b/plugins/scheduler-docker-local/scheduler-run
@@ -87,6 +87,12 @@ trigger-scheduler-docker-local-scheduler-run() {
     fi
   fi
 
+  local DOKKU_IMAGE_ARCHITECTURE="$("$DOCKER_BIN" image inspect --format '{{.Architecture}}' "$IMAGE")"
+  if [[ "$DOKKU_IMAGE_ARCHITECTURE" == "amd64" ]] && [[ "$(dpkg --print-architecture 2>/dev/null || true)" != "amd64" ]]; then
+    dokku_log_warn "Detected linux/amd64 image, forcing --platform=linux/amd64"
+    DOCKER_ARGS+=" --platform=linux/amd64"
+  fi
+
   # shellcheck disable=SC2124
   DOCKER_ARGS+=" --label=com.dokku.container-type=$PROCESS_TYPE"
   DOCKER_ARGS+=" --label=com.dokku.app-name=$APP"