Auto-detect arm platform in build.sh & rebuild.sh

From #355

Author: pgleeson

build.sh

@@ -1,4 +1,14 @@
 #!/bin/bash
 
 version=$(<VERSION) # Read version of Dockerfile from file VERSION
-docker build "$@" -t "openworm/openworm:$version" .
+
+# Set the platform flag if we're on ARM
+arch=$(uname -m)
+if [[ "$arch" == "arm64" || "$arch" == "aarch64" ]]; then
+    platform_flag="--platform linux/amd64"
+else
+    platform_flag=""
+fi
+
+# Build the Docker image
+docker build $platform_flag "$@" -t "openworm/openworm:$version" .

rebuild.sh

@@ -1,4 +1,14 @@
 #!/bin/bash
 
 version=$(<VERSION) # Read version of Dockerfile from file VERSION
-docker build "$@" -t "openworm/openworm:$version"  --no-cache .
+
+# Set the platform flag if we're on ARM
+arch=$(uname -m)
+if [[ "$arch" == "arm64" || "$arch" == "aarch64" ]]; then
+    platform_flag="--platform linux/amd64"
+else
+    platform_flag=""
+fi
+
+# Rebuild the Docker image
+docker build $platform_flag "$@" -t "openworm/openworm:$version"  --no-cache .