#!/bin/bash
set -euxo pipefail

if [[ $# -lt 2 ]]; then
    echo "Usage: $0 <image-name> <version>"
    exit 1
fi

IMAGE="$1"
VERSION="$2"

IMAGE_DIR="images/${IMAGE}"
MANIFEST="${IMAGE_DIR}/manifest.yaml"
ROOTFS="${IMAGE_DIR}/rootfs"
REF="images/${IMAGE}/${VERSION}"

# --- CHECKS ---
if [[ ! -d "$IMAGE_DIR" ]]; then
    echo "Image directory not found: $IMAGE_DIR"
    exit 1
fi

if [[ ! -f "$MANIFEST" ]]; then
    echo "Manifest not found: $MANIFEST"
    exit 1
fi

# --- PREPARE OSTREE REPO ---
mkdir -p repo cache
if [[ ! -f "repo/config" ]]; then
    pushd repo >/dev/null
    ostree init --repo=. --mode=bare-user
    popd >/dev/null
fi

ostree --repo=repo config set core.fsync false

# --- VERSIONING ---
buildid="$(date '+%Y%m%d.0')"
timestamp="$(date --iso-8601=sec)"
echo "${buildid}" > .buildid

echo "Composing ${VERSION}.${buildid} ..."

# --- REPOS IMPORT ---
cp images/shared/*.repo "${IMAGE_DIR}/"
cp "${IMAGE_DIR}"/repos/*.repo "${IMAGE_DIR}/"

# --- MANIFEST FIXUPS ---
sed -i '/^ref:/d' "$MANIFEST"
sed -i '/^releasever:/d' "$MANIFEST"

sed -i "1i releasever: ${VERSION}" "$MANIFEST"
sed -i "1i ref: ${REF}" "$MANIFEST"

# --- OPTIONAL POSTPROCESS ---
POSTPROCESS="$IMAGE_DIR/test.sh"
if [[ -x "$POSTPROCESS" ]]; then
    echo "Running postprocess script: $POSTPROCESS"
    "$POSTPROCESS"
else
    echo "No postprocess.sh found in ${IMAGE_DIR}, skipping."
fi

# --- COMPOSE ROOTFS ---
ARGS=(
    "--cachedir=cache"
)

rm -rf "${ROOTFS}"
mkdir -p "${ROOTFS}"

rpm-ostree compose rootfs \
    "${ARGS[@]}" \
    "$MANIFEST" \
    "$ROOTFS"