feat: support build arm64 image

Author: zhu-xiaowei

server/scripts/push-to-ecr.sh

@@ -97,7 +97,7 @@ build_and_push_image() {
     local TAG=$2
     local DOCKERFILE_PATH=$3
     local REGION=$AWS_REGION
-    local ARCH="amd64"  # Single architecture for simplicity
+    local ARCHS=("amd64" "arm64")  # Build for both architectures
 
     echo "Logging in to AWS Public ECR..."
     # Log in to AWS Public ECR for pulling base images
@@ -106,13 +106,21 @@ build_and_push_image() {
         exit 1
     fi
 
-    echo "Building $IMAGE_NAME:$TAG..."
+    echo "Building multi-architecture images (amd64 and arm64)..."
+    echo "This may take several minutes..."
+    echo ""
 
-    # Build Docker image
-    if ! docker buildx build --platform linux/$ARCH -t $IMAGE_NAME:$TAG -f $DOCKERFILE_PATH --load ../src/; then
-        echo "❌ ERROR: Failed to build Docker image."
-        exit 1
-    fi
+    # Build Docker image for each architecture
+    for ARCH in "${ARCHS[@]}"
+    do
+        echo "Building for linux/$ARCH..."
+        if ! docker buildx build --platform linux/$ARCH -t $IMAGE_NAME:$TAG-$ARCH -f $DOCKERFILE_PATH --load ../src/; then
+            echo "❌ ERROR: Failed to build Docker image for $ARCH."
+            exit 1
+        fi
+        echo "✅ Built $IMAGE_NAME:$TAG-$ARCH"
+    done
+    echo ""
 
     echo "Getting AWS account ID..."
     # Get the account ID
@@ -135,21 +143,51 @@ build_and_push_image() {
         exit 1
     fi
 
-    echo "Tagging image for ECR..."
-    # Tag the image for ECR
-    if ! docker tag $IMAGE_NAME:$TAG $REPOSITORY_URI:$TAG; then
-        echo "❌ ERROR: Failed to tag Docker image."
-        exit 1
-    fi
+    echo "Pushing images to ECR..."
+    # Push the image to ECR for each architecture
+    for ARCH in "${ARCHS[@]}"
+    do
+        echo "Tagging and pushing $ARCH image..."
+        # Tag the image for ECR
+        if ! docker tag $IMAGE_NAME:$TAG-$ARCH $REPOSITORY_URI:$TAG-$ARCH; then
+            echo "❌ ERROR: Failed to tag Docker image for $ARCH."
+            exit 1
+        fi
+
+        # Push the image to ECR
+        if ! docker push $REPOSITORY_URI:$TAG-$ARCH; then
+            echo "❌ ERROR: Failed to push image for $ARCH to ECR."
+            exit 1
+        fi
+        echo "✅ Pushed $REPOSITORY_URI:$TAG-$ARCH"
+    done
+    echo ""
 
-    echo "Pushing image to ECR..."
-    # Push the image to ECR
-    if ! docker push $REPOSITORY_URI:$TAG; then
-        echo "❌ ERROR: Failed to push image to ECR."
+    echo "Creating multi-architecture manifest..."
+    # Create a manifest for the multi-arch image
+    for ARCH in "${ARCHS[@]}"
+    do
+        if ! docker manifest create $REPOSITORY_URI:$TAG $REPOSITORY_URI:$TAG-$ARCH --amend; then
+            echo "❌ ERROR: Failed to create manifest for $ARCH."
+            exit 1
+        fi
+
+        # Annotate the manifest with architecture information
+        if ! docker manifest annotate $REPOSITORY_URI:$TAG "$REPOSITORY_URI:$TAG-$ARCH" --os linux --arch $ARCH; then
+            echo "❌ ERROR: Failed to annotate manifest for $ARCH."
+            exit 1
+        fi
+    done
+
+    echo "Pushing manifest to ECR..."
+    # Push the manifest to ECR
+    if ! docker manifest push $REPOSITORY_URI:$TAG; then
+        echo "❌ ERROR: Failed to push manifest to ECR."
         exit 1
     fi
 
-    echo "✅ Successfully pushed $IMAGE_NAME:$TAG to $REPOSITORY_URI"
+    echo "✅ Successfully pushed multi-architecture $IMAGE_NAME:$TAG to ECR"
+    echo "   Supported architectures: amd64 (x86_64), arm64"
     echo ""
 
     # Return the image URI for later use