#!/bin/bash

set -e

DISTROLIST=()
ROOTPATH=
DEVNAME=
KERNELDIR=
DATADIR=

. functions.sh

msg() {
	echo -e "$1" >&2
}

fatalerror() {
	msg "\x1b[1;31m$1\x1b[0m"
	exit 1
}

usage() {
	msg "$0 --root=<rootpath> --dev=<devname> [distro 1] [distro 2] ..."
}

while [[ -n "$1" ]]
do
	case "$1" in
		--root=*)
			ROOTPATH=${1/--root=}
			UUID="$(findmnt -o UUID --raw --noheadings $ROOTPATH \
				|| fatalerror "UUID of $ROOTPATH not found, not a mountpoint?")"
			KERNELDIR="$ROOTPATH/liveusb-kernel"
			DATADIR="$ROOTPATH/liveusb-data"
			install -d "$KERNELDIR" "$DATADIR"
			msg "Files will be copy to $ROOTPATH"
			;;
		--dev=*)
			DEVNAME=${1/--dev=}
			msg "Boot sector will be written to $DEVNAME"
			;;
		*=*)
			usage
			exit 1
			;;
		*)
			if [ -d "distro/$1" ]
			then
				DISTROLIST=(${DISTROLIST[@]} "$1")
			else
				fatalerror "directory distro/$1 not found"
			fi
			;;
	esac
	shift
done

for i in ${DISTROLIST[@]}
do
    process_distro "$i"
    if [ -f "isofiles/$ISOFILE" ]; then
        CKSUM=$(sha256sum "isofiles/$ISOFILE" | cut -d' ' -f1)
        if [[ $CKSUM == $SHA256 ]]; then
            msg "$ISOFILE ok."
        else
            fatalerror "$ISOFILE checksum bad!"
        fi
	else
		fatalerror "File isofiles/$ISOFILE not found!"
	fi

	install_live
done