compose-image-yml.sh

#!/bin/bash

set -e

yq() {
  docker run -i --rm -v "${PWD}/":/workdir -w /workdir mikefarah/yq:4.40.5 "$@"
}

process-image-template() {
    local out_templ_path="$1"
    local eve_version="$2"

    local flags
    local -a bits

    # Drop everything before the git hashcode (including the hash)
    flags="$(sed -r 's/.*[0-9a-fA-F]{8}(.*)/\1/p' <<< "${eve_version}")"
    # Drop dirty flag
    flags="$(sed -r 's/-dirty[0-9.\-]{18}//g' <<< "${flags}")"
    IFS='-' read -r -a bits <<< "${flags}"

    local dev=0
    local k=0
    for bit in "${bits[@]}"; do
        case "${bit}" in
            dev)
                dev=1
                ;;
            k)
                k=1
                ;;
        esac
    done

    local pillar_tag="PILLAR_TAG"
    if [[ $dev -eq 1 && $k -eq 1 ]]; then
      pillar_tag="PILLAR_K_DEV_TAG"
    elif [[ $dev -eq 1 ]]; then
      pillar_tag="PILLAR_DEV_TAG"
    elif [[ $k -eq 1 ]]; then
      pillar_tag="PILLAR_K_TAG"
    fi

    # shellcheck disable=SC2094
    yq '(.onboot[] | select(.image == "PILLAR_TAG").image) |= "'"$pillar_tag"'"' < "${out_templ_path}" | spongefile "${out_templ_path}"
    # shellcheck disable=SC2094
    yq '(.services[] | select(.image == "PILLAR_TAG").image) |= "'"$pillar_tag"'"' < "${out_templ_path}" | spongefile "${out_templ_path}"
}

patch_hv() {
    # note that we have to do that careful shell substitution, because yq 4
    # doesn't support passing in a variable as the value to --arg, requiring
    # setting it as an env var; which is more difficult to do when passed to docker
    # shellcheck disable=SC2016,SC2094
    yq '(.files[] | select(.contents == "EVE_HV")).contents |= "'"$1"'"' < "$2" | spongefile "$2"
}

patch_platform() {
    # shellcheck disable=SC2016,SC2094
    yq '(.files[] | select(.contents == "EVE_PLATFORM")).contents |= "'"$1"'"' < "$2" | spongefile "$2"
}

# because sponge doesn't exist everywhere, and this one uses a tmpfile
spongefile() {
    local tmp=""
    tmp=$(mktemp)
    cat > "$tmp"
    cat "$tmp" > "$1"
    rm "$tmp"
}

main() {
    local base_templ_path=""
    local out_templ_path=""
    local eve_version=""
    local eve_hv=""
    local eve_platform=""

    while getopts "b:o:v:h:p:" opt; do
        case ${opt} in
            b )
                base_templ_path=$OPTARG
                ;;
            o )
                out_templ_path=$OPTARG
                ;;
            v )
                eve_version=$OPTARG
                ;;
            h )
                eve_hv=$OPTARG
                ;;
            p )
                eve_platform=$OPTARG
                ;;
            \? )
                echo "Invalid option: -$OPTARG" 1>&2
                exit 1
                ;;
            : )
                echo "Invalid option: -$OPTARG requires an argument" 1>&2
                exit 1
                ;;
        esac
    done
    shift $((OPTIND -1))
    if [ -z "$out_templ_path" ] || [ -z "$eve_version" ] || [ -z "$eve_hv" ] || [ -z "$base_templ_path" ] || [ -z "$eve_platform" ]; then
        usage
    fi
    local modifiers="$*"

    cp "${base_templ_path}" "${out_templ_path}"

    for modifier in ${modifiers}; do
        if [ ! -f "${modifier}" ]; then
            continue
        fi
        # shellcheck disable=SC2094
        yq --from-file "${modifier}" < "${out_templ_path}" | spongefile "${out_templ_path}"|| exit 1
    done

    patch_hv "${eve_hv}" "${out_templ_path}"
    patch_platform "${eve_platform}" "${out_templ_path}"

    process-image-template "${out_templ_path}" "${eve_version}"
}

usage() {
    echo "Usage: $0 -b <base template> -o <output template> -v <eve version> -h <eve hv> -p <platform> <modifiers>"
    exit 1
}

main "$@"