From 46fb759137f5dc8d8c25edf57237ff9c38a57dd8 Mon Sep 17 00:00:00 2001
From: Mengxin Zhu <843303+zxkane@users.noreply.github.com>
Date: Wed, 9 Oct 2024 23:39:37 +0800
Subject: [PATCH] chore: use correct Dockerfile for building lambda image

---
 scripts/push-to-ecr.sh | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/scripts/push-to-ecr.sh b/scripts/push-to-ecr.sh
index 76c941c..ae51fae 100755
--- a/scripts/push-to-ecr.sh
+++ b/scripts/push-to-ecr.sh
@@ -16,16 +16,19 @@ AWS_REGIONS=("us-east-1") # List of AWS region, use below liest if you don't ena
 build_and_push_images() {
     local IMAGE_NAME=$1
     local TAG=$2
-    local ENABLE_MULTI_ARCH=${3:-true} # a parameter for enabling multi-arch build or not, the default is true
+    local ENABLE_MULTI_ARCH=${3:-true}  # Parameter for enabling multi-arch build, default is true
+    local DOCKERFILE_PATH=${4:-"../src/Dockerfile_ecs"}  # Parameter for Dockerfile path, default is "../src/Dockerfile_ecs"
 
     # Build Docker image for each architecture
     if [ "$ENABLE_MULTI_ARCH" == "true" ]; then
         for ARCH in "${ARCHS[@]}"
         do
-            docker buildx build --platform linux/$ARCH -t $IMAGE_NAME:$TAG-$ARCH -f ../src/Dockerfile_ecs --load ../src/
+            # Build multi-architecture Docker image
+            docker buildx build --platform linux/$ARCH -t $IMAGE_NAME:$TAG-$ARCH -f $DOCKERFILE_PATH --load ../src/
         done
     else
-        docker buildx build --platform linux/${ARCHS[0]} -t $IMAGE_NAME:$TAG -f ../src/Dockerfile_ecs --load ../src/
+        # Build single architecture Docker image
+        docker buildx build --platform linux/${ARCHS[0]} -t $IMAGE_NAME:$TAG -f $DOCKERFILE_PATH --load ../src/
     fi
 
     # Push Docker image to ECR for each architecture in each AWS region
@@ -70,5 +73,5 @@ build_and_push_images() {
     done
 }
 
-build_and_push_images "bedrock-proxy-api" "$TAG" "false"
+build_and_push_images "bedrock-proxy-api" "$TAG" "false" "../src/Dockerfile"
 build_and_push_images "bedrock-proxy-api-ecs" "$TAG"