From 5946a6c884e92b83ae1b567ee9a4a9e4f5bded55 Mon Sep 17 00:00:00 2001
From: alemagna-aws <54452613+alemagna-aws@users.noreply.github.com>
Date: Tue, 30 Mar 2021 15:21:22 -0700
Subject: [PATCH] Update codebuild_build.sh to use local image name override

---
 local_builds/codebuild_build.sh | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/local_builds/codebuild_build.sh b/local_builds/codebuild_build.sh
index 9b5ce7fc..8226dc8a 100755
--- a/local_builds/codebuild_build.sh
+++ b/local_builds/codebuild_build.sh
@@ -182,7 +182,12 @@ else
     docker_command+=" -e \"INITIATOR=$USER\""
 fi
 
-docker_command+=" amazon/aws-codebuild-local:latest"
+if [ -n $local_agent_image ]
+then
+    docker_command+=" $local_agent_image"
+else
+    docker_command+=" amazon/aws-codebuild-local:latest"
+fi
 
 # Note we do not expose the AWS_SECRET_ACCESS_KEY or the AWS_SESSION_TOKEN
 exposed_command=$docker_command