##!/usr/bin/env bash
set -euo pipefail

img_amd64=$1
IsPushLatest=$2

img_arm64=$(echo ${img_amd64} | sed 's/-amd64$/-arm64/g')
img_march=${img_amd64%-amd64}
docker pull "$img_amd64"
docker pull --platform linux/arm64 "$img_arm64"
img_amd64_digest=$(docker inspect --format='{{index .RepoDigests 0}}' "$img_amd64")
img_arm64_digest=$(docker inspect --format='{{index .RepoDigests 0}}' "$img_arm64")
echo "sha256 of amd64 is $img_amd64_digest"
echo "sha256 of arm64 is $img_arm64_digest"
docker manifest create "${img_march}" \
    --amend "$img_amd64_digest" \
    --amend "$img_arm64_digest"
docker manifest push "${img_march}"

# PUSH latest if it is a release build
if [ "$IsPushLatest" = "true" ]; then
    img_latest=$(echo "$img_arm64" | cut -d: -f 1):latest
    docker manifest create "${img_latest}" \
        --amend "$img_amd64_digest" \
        --amend "$img_arm64_digest"
    docker manifest push "${img_latest}"
fi