#!/bin/bash

if [ ${DIB_DEBUG_TRACE:-0} -gt 0 ]; then
    set -x
fi
set -eu
set -o pipefail

[ -n "$ARCH" ]
[ -n "$TARGET_ROOT" ]

if [[ "${DIB_RELEASE}" = 7 ]]; then
    if [[ "amd64 x86_64" =~ "$ARCH" ]]; then
        ARCH="x86_64"
        DIB_CLOUD_IMAGES=${DIB_CLOUD_IMAGES:-https://cloud.centos.org/centos/${DIB_RELEASE}/images}
    elif [[ "arm64 aarch64" =~ "$ARCH" ]]; then
        ARCH="aarch64"
        DIB_CLOUD_IMAGES=${DIB_CLOUD_IMAGES:-https://cloud.centos.org/altarch/${DIB_RELEASE}/images/aarch64}
    elif [[ "ppc64le" =~ "$ARCH" ]]; then
        DIB_CLOUD_IMAGES=${DIB_CLOUD_IMAGES:-https://cloud.centos.org/altarch/${DIB_RELEASE}/images/ppc64le}
    else
        echo 'centos root element only support the x86_64, aarch64 and ppc64le values for $ARCH'
        exit 1
    fi
else
    if [[ "amd64 x86_64 arm64 aarch64 ppc64le" =~ "$ARCH" ]]; then
        if [[ "amd64" =~ "$ARCH" ]]; then
            ARCH="x86_64"
        elif [[ "arm64" =~ "$ARCH" ]]; then
            ARCH="aarch64"
        fi
        if [[ "${DIB_RELEASE}" == "9" ]]; then
            dib_release_path=9-stream
        else
            dib_release_path=${DIB_RELEASE}
        fi
        DIB_CLOUD_IMAGES=${DIB_CLOUD_IMAGES:-https://cloud.centos.org/centos/${dib_release_path}/${ARCH}/images}
    else
        echo 'centos root element only support the x86_64, aarch64 and ppc64le values for $ARCH'
        exit 1
    fi

fi

DIB_LOCAL_IMAGE=${DIB_LOCAL_IMAGE:-}

if [ -n "$DIB_LOCAL_IMAGE" ]; then
    IMAGE_LOCATION=$DIB_LOCAL_IMAGE
    # No need to copy a local image into the cache directory, so just specify
    # the cached path as the original path.
    CACHED_IMAGE=$IMAGE_LOCATION
    BASE_IMAGE_FILE=$(basename $DIB_LOCAL_IMAGE)
    BASE_IMAGE_TAR=$BASE_IMAGE_FILE.tgz
else
    DIB_FLAVOR=${DIB_FLAVOR:-GenericCloud}
    if [[ "${DIB_RELEASE}" = 7 ]]; then
        BASE_IMAGE_FILE=${BASE_IMAGE_FILE:-CentOS-${DIB_RELEASE}-${ARCH}-${DIB_FLAVOR}.qcow2.xz}
    else
        if [[ "${DIB_RELEASE}" == "9" ]]; then
            dib_release_path=9-stream
        else
            dib_release_path=${DIB_RELEASE}
        fi
        BASE_IMAGE_FILE=${BASE_IMAGE_FILE:-$(curl -s https://cloud.centos.org/centos/${dib_release_path}/${ARCH}/images/ | grep -o "CentOS-.[^>]*${DIB_FLAVOR}-.[^>]*.qcow2" | sort -r | head -1)}
    fi
    BASE_IMAGE_TAR=$BASE_IMAGE_FILE.tgz
    IMAGE_LOCATION=$DIB_CLOUD_IMAGES/$BASE_IMAGE_FILE
    CACHED_IMAGE=$DIB_IMAGE_CACHE/$BASE_IMAGE_FILE
fi

$TMP_HOOKS_PATH/bin/extract-image $BASE_IMAGE_FILE $BASE_IMAGE_TAR $IMAGE_LOCATION $CACHED_IMAGE