#!/bin/bash

if [ "$#" -ne 1 ]; then
    echo "Usage: $0 IMAGE_NAME"
    exit 1
fi

image_name="$1"

declare -A enforced_labels=( ["com.jfrog.artifactory.retention.maxCount"]="10" ["com.jfrog.artifactory.retention.maxDays"]="7")

if ! docker inspect --type=image "$1" > /dev/null ; then 
    echo "Error: image does not exist"
    exit 0
fi

for label in "${!enforced_labels[@]}" ; do     
    if ! docker inspect -f "{{json .Config.Labels }}" "$image_name" | jq -e '.['\"$label\"']' > /dev/null ; then 
        echo "Key not found" 
        echo "$label = ${enforced_labels[$label]}"
        label_args+=" --label $label=\"${enforced_labels[$label]}\""
    fi    
done

if [ -v label_args ]; then 
    run_command="echo \"FROM $image_name\" | docker build $label_args -t $image_name -"
    eval "$run_command"
fi