#!/bin/bash
# Convert src:<package> to src:<package>-cuda
#
# Based on Mo Zhou's solution for pytorch-cuda and tensorpipe-cuda, with minor
# modifications to maintain a fuller changelog.
set -e

# Prerequisites
NEEDED_FILES="\
control
libggml0-backend-cuda.install
not-installed
rules
tests/control"
for debfile in $NEEDED_FILES; do
    if ! [ -f "debian/alt-builds/cuda/${debfile}" ]; then
        echo "debian/alt-builds/cuda/${debfile} not found." >&2
        exit 1
    fi
done

if [ -z "$DEBFULLNAME" ] || [ -z "$DEBEMAIL" ]; then
    echo "DEBFULLNAME and DEBEMAIL envvars must be set." >&2
fi

origname="$(dpkg-parsechangelog -SSource)"
cudaname="$origname-cuda"

echo "Converting src:$origname to src:$cudaname"

echo "1. Preparing debian/changelog ..."

head -n1 debian/changelog >debian/changelog.cuda
cat >>debian/changelog.cuda <<EOF

  * Rebuild src:$origname as src:$cudaname with CUDA (non-free) enabled
    - Automatically generated by src:$origname :: debian/switch-ggml-cuda.sh
    - Please observe README.Source when making changes
    - Changelog entries below are from src:$origname
EOF
tail -n +2 debian/changelog >>debian/changelog.cuda
sed -i -e "s/^$origname/$cudaname/" debian/changelog.cuda
mv debian/changelog debian/changelog.orig
mv debian/changelog.cuda debian/changelog

echo "2. Switching debian/{rules,control} to the CUDA version ..."
mv debian/control debian/control.orig
mv debian/alt-builds/cuda/control debian/control
sed -i -e "s/@GGML_DEBIAN_VERSION@/$(dpkg-parsechangelog -SVersion)/" debian/control
mv debian/rules debian/rules.orig
mv debian/alt-builds/cuda/rules debian/rules

echo "3. Switching debian/tests/control to the CUDA version ..."
mv debian/tests/control debian/tests/control.orig
mv debian/alt-builds/cuda/tests/control debian/tests/control

echo "4. Switching debian/not-installed to the CUDA version ..."
mv debian/alt-builds/cuda/not-installed debian/not-installed

echo "5. Moving *.install file(s) for the CUDA version ..."
mv debian/alt-builds/cuda/*.install debian/

echo "6. Removing irrelevant files from the original src:ggml ..."
rm -rf debian/libggml0-backend-blas.* \
    debian/libggml0-backend-hip.* \
    debian/libggml0-backend-rpc.* \
    debian/libggml0-backend-vulkan.* \
    debian/libggml0.* \
    debian/libggml-dev.* \
    debian/switch-* \
	debian/*.orig \
	debian/alt-builds

echo "Conversion finished."
