Make platform dynamic

serversideup · Oct 7, 2024 · 153b828 · 153b828
1 parent 7a83041
commit 153b828
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/build.sh b/build.sh
@@ -77,7 +77,7 @@ build_docker_image() {
 
     # Set default platform if not specified
     if [ -z "$PLATFORM" ]; then
-        PLATFORM="linux/amd64"
+        PLATFORM=$(detect_platform)
     fi
 
     # Add platform to build args
@@ -119,6 +119,22 @@ check_vars() {
   return 0
 }
 
+detect_platform() {
+    local arch=$(uname -m)
+    case $arch in
+        x86_64)
+            echo "linux/amd64"
+            ;;
+        arm64|aarch64)
+            echo "linux/arm64/v8"
+            ;;
+        *)
+            echo "Unsupported architecture: $arch" >&2
+            exit 1
+            ;;
+    esac
+}
+
 echo_color_message (){
   color=$1
   message=$2