chore: allow aws credential generation for ecr-public

Set `ECR_PUBLIC` in onder to login with public.ecr.aws Signed-off-by: Andrew Bayer <andrew.bayer@gmail.com>
codeship-library · Jul 13, 2021 · 670f56d · 670f56d
1 parent 3198ace
commit 670f56d
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/dockercfg-generator/aws_docker_creds.sh b/dockercfg-generator/aws_docker_creds.sh
@@ -37,11 +37,20 @@ else
   AWS_ACCOUNT=$(aws sts get-caller-identity | jq -r ".Account")
 fi
 
+# For public images stored in public.ecr.aws, we need to use a different CLI call
+if [[ -v ECR_PUBLIC ]]; then
+  COMMAND=ecr-public
+  SERVER=public.ecr.aws
+else
+  COMMAND=ecr
+  SERVER=${AWS_ACCOUNT}.dkr.ecr.${AWS_REGION}.amazonaws.com
+fi
+
 # fetching aws docker login
 # aws deprecated the get-login function in favor of get-login-password
 # https://docs.aws.amazon.com/cli/latest/reference/ecr/get-login.html
 echo "Logging into AWS ECR with account ${AWS_ACCOUNT}"
-aws ecr get-login-password --region ${AWS_REGION} | docker login --username AWS --password-stdin ${AWS_ACCOUNT}.dkr.ecr.${AWS_REGION}.amazonaws.com
+aws ${COMMAND} get-login-password --region ${AWS_REGION} | docker login --username AWS --password-stdin ${SERVER}
 
 # writing aws docker creds to desired path
 echo "Writing Docker creds to $1"