#!/usr/bin/env bash
set -euo pipefail
DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

source "$DIR/argparse.bash" || exit 1
argparse "$@" <<EOF || exit 1

parser.add_argument('dest', help="The destination folder")
parser.add_argument('-g', '--get_image', action='store_true',
                    help='Pick a number [default %(default)s]')
parser.add_argument('-v', '--variant', action='store',
                    choices=['raspios_lite_armhf', 'raspios_lite_arm64', 'raspios_armhf', 'raspios_arm64', 'raspios_full_armhf', 'raspios_full_arm64'],
                    default='raspios_lite_armhf',
                    help='Which variant to use [default: %(default)s]')
EOF

case $DEST in
    *"-"*)
        echo "Error, destination folder cannot contain a dash"
        exit 1
        ;;
    *" "*)
        echo "Error, destination folder cannot contain a space"
        exit 1
        ;;
    *)
        ;;
esac

echo Settings:
echo "making dstro in ${DEST}"
echo "variant: ${VARIANT}"

for a in "${MULTIPLE[@]}"; do
    echo "  $a"
done

if [ -d "${DEST}" ]; then
    echo "Error, folder already exists: ${DEST}"
    exit 1
fi

if [ -f "${DEST}" ]; then
    echo "Error, file already exists: ${DEST}"
    exit 1
fi

cp -a "${DIR}/dist_generators/dist_example" "${DEST}"
chown -R "${USER}":"$(id -gn "${USER}")" "${DEST}"

"${DIR}/dist_generators/dist_example_script" "${DEST}"

"$DIR/update-custompios-paths" "${DEST}/src"

if [ "$GET_IMAGE" ]; then
    echo -n "Downloading latest Raspbian image"

    CURRENT_RASPBIAN=$(curl -s "https://downloads.raspberrypi.org/${VARIANT}/images/" | grep raspios | tail -n 1 | awk -F "href=\"" '{print $2}' | awk -F "/" '{print $1}')
    if [ $? -ne 0 ]; then
        echo -e "\nerror getting date"
        exit 1
    fi
    CURRENT_RASPBIAN_FILE="$(curl -s "http://downloads.raspberrypi.org/${VARIANT}/images/${CURRENT_RASPBIAN}"/ | grep .xz | head -n 1 | awk -F "href=\"" '{print $2}' | awk -F "\">" '{print $1}')"
    if [ $? -ne 0 ]; then
        echo -e "\nerror getting file name"
        exit 1
    fi
    CURRENT_RASPBIAN_URL="https://downloads.raspberrypi.org/${VARIANT}/images/${CURRENT_RASPBIAN}/${CURRENT_RASPBIAN_FILE}"
    echo " from ${CURRENT_RASPBIAN_URL}"
    curl -L -o "${DEST}/src/image/${CURRENT_RASPBIAN_FILE}" "${CURRENT_RASPBIAN_URL}"
fi