#!/bin/bash -ex

IMAGE_NAME=$1

if [ -z "$1" ]; then
    ONLY_IF_DATASET_MODIF=false
else
    ONLY_IF_DATASET_MODIF=true
fi

if [ -z "$IMAGE_NAME" ]; then
    echo "PAS DE NOM DE MODULE"
    exit 1
fi

# root dir configuration
RISOTTO_DIR="/var/lib/risotto"
RISOTTO_IMAGE_DIR="$RISOTTO_DIR/images"
# image configuration
IMAGE_BASE_RISOTTO_BASE_DIR="$RISOTTO_IMAGE_DIR/image_bases"
IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP="$RISOTTO_IMAGE_DIR/tmp/$IMAGE_NAME"
IMAGE_NAME_RISOTTO_IMAGE_DIR="$RISOTTO_IMAGE_DIR/$IMAGE_NAME"
IMAGE_DIR_RECIPIENT_IMAGE="$RISOTTO_DIR/images_files/$IMAGE_NAME"


rm -f /var/log/risotto/build_image.log
mkdir -p "$RISOTTO_IMAGE_DIR" "$RISOTTO_IMAGE_DIR/tmp/"
PKG=""
BASE_DIR=""
for script in $(ls "$IMAGE_DIR_RECIPIENT_IMAGE"/preinstall/*.sh 2> /dev/null); do
    . "$script"
done

if [ -z "$OS_NAME" ]; then
    echo "NO OS NAME DEFINED"
    exit 1
fi
if [ -z "$RELEASEVER" ]; then
    echo "NO RELEASEVER DEFINED"
    exit 1
fi
if [ -z "$INSTALL_TOOL" ]; then
    echo "NO INSTALL TOOL DEFINED"
    exit 1
fi
BASE_NAME="$OS_NAME-$RELEASEVER"
BASE_DIR="$IMAGE_BASE_RISOTTO_BASE_DIR/$BASE_NAME"
TMP_BASE_DIR="$IMAGE_BASE_RISOTTO_BASE_DIR/tmp/$BASE_NAME"
BASE_PKGS_FILE="$IMAGE_BASE_RISOTTO_BASE_DIR-$BASE_NAME.pkgs"
BASE_LOCK="$IMAGE_BASE_RISOTTO_BASE_DIR-$BASE_NAME.build"


function dnf_opt_base() {
    INSTALL_DIR=$1
    echo "--setopt=install_weak_deps=False --setopt=fastestmirror=True --nodocs --noplugins --installroot=$INSTALL_DIR --releasever $RELEASEVER"
}

function dnf_opt() {
    INSTALL_DIR=$1
    INSTALL_PKG=$2
    OPT=$(dnf_opt_base "$INSTALL_DIR")
    echo "$OPT install $INSTALL_PKG"
}
function new_package_base() {
    if [ "$INSTALL_TOOL" = "dnf" ]; then
        OPT=$(dnf_opt "$TMP_BASE_DIR" "$BASE_PKG")
        dnf --assumeno $OPT | grep ^" " > "$BASE_PKGS_FILE".new
    else
        debootstrap --include="$BASE_PKG" --variant=minbase "$RELEASEVER" "$TMP_BASE_DIR" >> /var/log/risotto/build_image.log
        chroot "$TMP_BASE_DIR" dpkg-query -f '${binary:Package} ${source:Version}\n' -W > "$BASE_PKGS_FILE".new
    fi
}
function install_base() {
    if [ "$INSTALL_TOOL" = "dnf" ]; then
        OPT=$(dnf_opt "$TMP_BASE_DIR" "$BASE_PKG")
        dnf --assumeyes $OPT
    fi
}
function new_package() {
    if [ "$INSTALL_TOOL" = "dnf" ]; then
        OPT=$(dnf_opt_base "$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP")
        set +e
        dnf --assumeno $OPT update >> /var/log/risotto/build_image.log
        set -e
        OPT=$(dnf_opt "$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP" "$PKG")
        dnf --assumeno $OPT | grep ^" " > "$IMAGE_NAME_RISOTTO_IMAGE_DIR".pkgs.new
    else
        chroot "$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP" apt update >> /var/log/risotto/build_image.log 2>&1
        chroot "$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP" apt install --no-install-recommends --yes $PKG -s 2>/dev/null|grep ^"Inst " > "$IMAGE_NAME_RISOTTO_IMAGE_DIR".pkgs.new
    fi
}
function install_pkg() {
    if [ "$INSTALL_TOOL" = "dnf" ]; then
        OPT=$(dnf_opt "$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP" "$PKG")
        dnf --assumeyes $OPT
    else
        if [ "$ONLY_IF_DATASET_MODIF" = true ]; then
            chroot "$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP" apt update
        fi
        chroot "$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP" bash -c "export DEBIAN_FRONTEND=noninteractive; apt install --no-install-recommends --yes $PKG"
    fi
}

if [ ! -f "$BASE_LOCK" ] || [ ! -d "$BASE_DIR" ]; then
    echo "  - reinstallation de l'image de base"
    new_package_base
    diff -u "$BASE_PKGS_FILE" "$BASE_PKGS_FILE".new &> /dev/null && NEW_BASE=false || NEW_BASE=true
    if [ ! -d "$BASE_DIR" ] || [ "$NEW_BASE" = true ]; then
        mkdir -p "$IMAGE_BASE_RISOTTO_BASE_DIR"
        rm -rf "$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP"
        install_base
        if [ -f "$BASE_PKGS_FILE" ]; then
            mv "$BASE_PKGS_FILE" "$BASE_PKGS_FILE".old
        fi
        mv "$BASE_PKGS_FILE".new "$BASE_PKGS_FILE"
        rm -rf "$BASE_DIR"
        mv "$TMP_BASE_DIR" "$BASE_DIR"
    fi
    touch "$BASE_LOCK"
fi

rm -rf "$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP"
cp --reflink=auto -a "$BASE_DIR/" "$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP"
if [ -n "$COPR" ]; then
    #FIXME signature...
    mkdir -p "$REPO_DIR"
    cd "$REPO_DIR"
    wget -q "$COPR"
    cd - > /dev/null
fi
if [ "$FUSION" = true ]; then
    dnf -y install "https://download1.rpmfusion.org/free/fedora/rpmfusion-free-release-$RELEASEVER.noarch.rpm" --installroot="$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP" >> /var/log/risotto/build_image.log
fi

if [ -f "$IMAGE_NAME_RISOTTO_IMAGE_DIR".base.pkgs ] && [ -f "$IMAGE_NAME_RISOTTO_IMAGE_DIR".pkgs ]; then
    echo "  - différence(s) avec les paquets de base"
    diff -u "$IMAGE_NAME_RISOTTO_IMAGE_DIR".base.pkgs "$BASE_PKGS_FILE" && INSTALL=false || INSTALL=true
    [ ! -d "$IMAGE_NAME_RISOTTO_IMAGE_DIR" ] && INSTALL=true
else
    INSTALL=true
fi

if [ "$ONLY_IF_DATASET_MODIF" = false ] || [ ! -f "$IMAGE_NAME_RISOTTO_IMAGE_DIR".pkgs ]; then
    new_package
else
    cp --reflink=auto "$IMAGE_NAME_RISOTTO_IMAGE_DIR".pkgs "$IMAGE_NAME_RISOTTO_IMAGE_DIR".pkgs.new
fi
if [ "$INSTALL" = false ]; then
    echo "  - différence(s) avec les paquets de l'image"
    diff -u "$IMAGE_NAME_RISOTTO_IMAGE_DIR".pkgs "$IMAGE_NAME_RISOTTO_IMAGE_DIR".pkgs.new && INSTALL=false || INSTALL=true
fi
find "$IMAGE_DIR_RECIPIENT_IMAGE" -type f -exec md5sum '{}' \; > "$IMAGE_NAME_RISOTTO_IMAGE_DIR".md5sum.new
if [ "$INSTALL" = false ]; then
    echo "  - différence(s) du dataset"
    diff -u "$IMAGE_NAME_RISOTTO_IMAGE_DIR".md5sum "$IMAGE_NAME_RISOTTO_IMAGE_DIR".md5sum.new && INSTALL=false || INSTALL=true
fi
if [ "$INSTALL" = true ]; then
    echo "  - installation"
    if [ -f "$IMAGE_NAME_RISOTTO_IMAGE_DIR".version ]; then
        VERSION=$(cat "$IMAGE_NAME_RISOTTO_IMAGE_DIR".version)
    else
        VERSION=0
    fi
    if [ -d "$IMAGE_NAME_RISOTTO_IMAGE_DIR" ]; then
        cd "$IMAGE_NAME_RISOTTO_IMAGE_DIR"
        make_changelog "$IMAGE_NAME" "$VERSION" "$OS_NAME" "$RELEASEVER" > "$IMAGE_NAME_RISOTTO_IMAGE_DIR"_"$RELEASEVER"_"$VERSION"_changelog.md
        cd - > /dev/null
    fi
    install_pkg
    sleep 2
    
    for script in $(ls $IMAGE_DIR_RECIPIENT_IMAGE/postinstall/*.sh 2> /dev/null); do
        . "$script"
    done
    
    ROOT=$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP make_volatile /etc
    if [ ! "$?" = 0 ]; then
        echo "make_volatile failed"
        exit 1
    fi
    rm -rf "$IMAGE_NAME_RISOTTO_IMAGE_DIR"
    mv "$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP" "$IMAGE_NAME_RISOTTO_IMAGE_DIR"
    cp --reflink=auto -f "$BASE_PKGS_FILE" "$IMAGE_NAME_RISOTTO_IMAGE_DIR".base.pkgs
    mv -f "$IMAGE_NAME_RISOTTO_IMAGE_DIR".pkgs.new "$IMAGE_NAME_RISOTTO_IMAGE_DIR".pkgs
    mv -f "$IMAGE_NAME_RISOTTO_IMAGE_DIR".md5sum.new "$IMAGE_NAME_RISOTTO_IMAGE_DIR".md5sum
    VERSION=$((VERSION + 1)) 
    echo "$VERSION" > "$IMAGE_NAME_RISOTTO_IMAGE_DIR".version
fi
rm -rf "$IMAGE_NAME_RISOTTO_IMAGE_DIR_TMP"

echo "  => OK"
exit 0