#!/bin/bash

#
# SPDX-License-Identifier: Apache-2.0
#

#
# Replicate a dnf.conf and yum.repo.d under a temporary directory and
# then modify the files to point to equivalent repos in the StarlingX mirror.
# This script was originated by Scott Little
#

MAKE_STX_MIRROR_DNF_CONF_DIR="$(dirname "$(readlink -f "${BASH_SOURCE[0]}" )" )"

source "$MAKE_STX_MIRROR_DNF_CONF_DIR/url_utils.sh"

DISTRO="centos"
SUDO=sudo

TEMP_DIR=""
SRC_REPO_DIR="$MAKE_STX_MIRROR_DNF_CONF_DIR/yum.repos.d"
SRC_DNF_CONF="$MAKE_STX_MIRROR_DNF_CONF_DIR/dnf.conf.sample"

RETAIN_REPODIR=0

usage () {
    echo ""
    echo "$0 -d <dest_dir> [-D <distro>] [-y <src_dnf_conf>] [-r <src_repos_dir>] [-R] [-l <layer>] [-u <lower-layer>,<repo_url>]"
    echo ""
    echo "Replicate a dnf.conf and yum.repo.d under a new directory and"
    echo "then modify the files to point to equivalent repos in the StarlingX"
    echo "mirror."
    echo ""
    echo "-d <dest_dir> = Place modified dnf.conf and yum.repo.d into this directory"
    echo "-D <distro>   = Target distro on StarlingX mirror. Default is 'centos'"
    echo "-y <dnf_conf> = Path to dnf.conf file that we will modify.  Default is"
    echo "                'dnf.conf.sample' in same directory as this script"
    echo "-r <repos_dir> = Path to yum.repos.d that we will modify.  Default is"
    echo "                 'yum.repos.d' in same directory as this script"
    echo "-l <layer> = Download only packages required to build a given layer"
    echo "-u <lower-layer>,<build-type>,<repo_url> = Add/change the repo baseurl for a lower layer"
    echo "-n don't use sudo"
}

declare -A layer_urls

set_layer_urls () {
    local option="${1}"
    local layer_and_build_type="${option%,*}"
    local layer="${layer_and_build_type%,*}"
    local build_type="${layer_and_build_type#*,}"
    local layer_url="${option##*,}"

    # Enforce trailing '/'
    if [ "${layer_url:${#layer_url}-1:1}" != "/" ]; then
        layer_url+="/"
    fi

    layer_urls["${layer_and_build_type}"]="${layer_url}"
}


#
# option processing
#
while getopts "D:d:l:nRr:u:y:" o; do
    case "${o}" in
        D)
            DISTRO="${OPTARG}"
            ;;
        d)
            TEMP_DIR="${OPTARG}"
            ;;
        l)
            LAYER="${OPTARG}"
            ;;
        n)
            SUDO=""
            ;;
        r)
            SRC_REPO_DIR="${OPTARG}"
            ;;
        R)
            RETAIN_REPODIR=1
            ;;
        u)
            set_layer_urls "${OPTARG}"
            ;;
        y)
            SRC_DNF_CONF="${OPTARG}"
            ;;
        *)
            usage
            exit 1
            ;;
    esac
done

#
# option validation
#
if [ ! -f $SRC_DNF_CONF ]; then
    echo "Error: dnf.conf not found at '$SRC_DNF_CONF'"
    exit 1
fi

if [ ! -d $SRC_REPO_DIR ]; then
    echo "Error: repo dir not found at '$SRC_REPO_DIR'"
    exit 1
fi

if [ "$TEMP_DIR" == "" ]; then
    echo "Error: working dir not provided"
    usage
    exit 1
fi

if [ ! -d $TEMP_DIR ]; then
    echo "Error: working dir not found at '$TEMP_DIR'"
    exit 1
fi

#
# Get the value of the $releasever variable.
#
# If the source dnf.conf has a releasever= setting, we will honor
# that, even though dnf will not.
#
# Otherwise use dnf to query the host environment (Docker).
# This assumes the host environmnet has the same releasever
# as that which will be used inside StarlingX.
#
# NOTE: In other scripts we will read releasever= out of dnf.conf
# and push it back into dnf via --releasever=<#>.
#
get_releasever () {
    if [ -f $SRC_DNF_CONF ] && grep -q '^releasever=' $SRC_DNF_CONF; then
        grep '^releasever=' $SRC_DNF_CONF | cut -d '=' -f 2
    else
        dnf version nogroups | grep Installed | cut -d ' ' -f 2 | cut -d '/' -f 1
    fi
}

#
# Get the value of the $basearch variable.
#
# Just use dnf to query the host environment (Docker) as we don't support
# cross compiling.
#
get_arch () {
    dnf version nogroups | grep Installed | cut -d ' ' -f 2 | cut -d '/' -f 2
}


#
# Global variables we will use later.
#
CENGN_REPOS_DIR="$TEMP_DIR/yum.repos.d"
CENGN_DNF_CONF="$TEMP_DIR/dnf.conf"
CENGN_DNF_LOG="$TEMP_DIR/dnf.log"
CENGN_DNF_CACHDIR="$TEMP_DIR/cache/dnf/\$basearch/\$releasever"

RELEASEVER="8"
ARCH="x86_64"
CONTENTDIR="centos"

#
# Copy as yet unmodified dnf.conf and yum.repos.d from source to dest.
#
mkdir -p "$CENGN_REPOS_DIR"
echo "\cp -r '$SRC_REPO_DIR/*' '$CENGN_REPOS_DIR/'"
\cp -r "$SRC_REPO_DIR"/* "$CENGN_REPOS_DIR/"
echo "\cp '$SRC_DNF_CONF' '$CENGN_DNF_CONF'"
\cp "$SRC_DNF_CONF" "$CENGN_DNF_CONF"

if [ "$LAYER" != "all" ]; then
    if [ -d ${MAKE_STX_MIRROR_DNF_CONF_DIR}/config/${DISTRO}/${LAYER}/yum.repos.d ]; then
        \cp -f ${MAKE_STX_MIRROR_DNF_CONF_DIR}/config/${DISTRO}/${LAYER}/yum.repos.d/*.repo $CENGN_REPOS_DIR
    fi
fi

#
# Add or modify reposdir= value in our new dnf.conf
#
if grep -q '^reposdir=' $CENGN_DNF_CONF; then
    # reposdir= already exists, modify it
    if [ $RETAIN_REPODIR -eq 1 ]; then
        # Append CENGN_REPOS_DIR
        sed "s#^reposdir=\(.*\)\$#reposdir=\1 $CENGN_REPOS_DIR#" -i $CENGN_DNF_CONF
    else
        # replace with CENGN_REPOS_DIR
        sed "s#^reposdir=.*\$#reposdir=$CENGN_REPOS_DIR#" -i $CENGN_DNF_CONF
    fi
else
    # reposdir= doeas not yet exist, add it
    if [ $RETAIN_REPODIR -eq 1 ]; then
        # Add both SRC_REPO_DIR and CENGN_REPOS_DIR
        echo "reposdir=$SRC_REPO_DIR $CENGN_REPOS_DIR" >> $CENGN_DNF_CONF
    else
        # Add CENGN_REPOS_DIR only
        echo "reposdir=$CENGN_REPOS_DIR" >> $CENGN_DNF_CONF
    fi
fi

#
# modify or add logfile= value in our new dnf.conf
#
if grep -q '^logfile=' $CENGN_DNF_CONF; then
    sed "s#^logfile=.*\$#logfile=$CENGN_DNF_LOG#" -i $CENGN_DNF_CONF
else
    echo "logfile=$CENGN_DNF_LOG" >> $CENGN_DNF_CONF
fi

#
# modify or add cachedir= value in our new dnf.conf
#
if grep -q '^cachedir=' $CENGN_DNF_CONF; then
    sed "s#^cachedir=.*\$#cachedir=$CENGN_DNF_CACHDIR#" -i $CENGN_DNF_CONF
else
    echo "cachedir=$CENGN_DNF_CACHDIR" >> $CENGN_DNF_CONF
fi


#
# Modify all the repo files in our new yum.repos.d
#
for REPO in $(find "$CENGN_REPOS_DIR" -type f -name '*repo'); do
    #
    # Replace mirrorlist with baseurl if required
    #
    if grep -q '^mirrorlist=' "$REPO" ; then
        sed '/^mirrorlist=/d' -i "$REPO"
        sed 's%^#baseurl%baseurl%' -i "$REPO"
    fi

    #
    # Substitute any $releasever or $basearch variables
    #
    sed "s#/[$]releasever/#/$RELEASEVER/#g" -i "$REPO"
    sed "s#/[$]basearch/#/$ARCH/#g" -i "$REPO"
    sed "s#/[$]contentdir/#/$CONTENTDIR/#g" -i "$REPO"

    #
    # Turn off gpgcheck for now.
    # Must revisit this at a later date!
    #
    sed 's#^gpgcheck=1#gpgcheck=0#' -i "$REPO"
    sed '/^gpgkey=/d' -i "$REPO"

    #
    # Convert baseurl(s) to cengn equivalent
    #
    for URL in $(grep '^baseurl=' "$REPO" | sed 's#^baseurl=##'); do
        CENGN_URL="$(url_to_stx_mirror_url "$URL" "$DISTRO")"

        # Test CENGN url
        wget -q --spider $CENGN_URL
        if [ $? -eq 0 ]; then
            # OK, make substitution
            sed "s#^baseurl=$URL\$#baseurl=$CENGN_URL#" -i "$REPO"
        fi
    done

    #
    # Prefix repoid and name with CENGN
    #
    sed "s#^name=\(.*\)#name=CENGN_\1#" -i "$REPO"
    sed "s#^\[\([^]]*\)\]#[CENGN_\1]#" -i "$REPO"
done

for key in "${!layer_urls[@]}"; do
    lower_layer="${key%,*}"
    build_type="${key#*,}"
    REPO="$CENGN_REPOS_DIR/StarlingX_cengn_${lower_layer}_layer.repo"
    if [ -f "$REPO" ]; then
        sed "s#^baseurl=.*/${lower_layer}/.*/${build_type}/\$#baseurl=${layer_urls[${key}]}#" -i "$REPO"
    else
        REPO="$CENGN_REPOS_DIR/StarlingX_local_${lower_layer}_${build_type}_layer.repo"
        (
        echo "[Starlingx-local_${lower_layer}_${build_type}_layer]"
        echo "name=Starlingx-cengn_${lower_layer}_${build_type}_layer"
        echo "baseurl=${layer_urls[${key}]}"
        echo "enabled=1"
        ) > "$REPO"
    fi
done

echo $TEMP_DIR