#!/bin/bash

cleanup() {
    [ -d "$MNT" ] && umount $MNT && rmdir $MNT
}

exitclean() {
    echo "Cleaning up to exit..."
    cleanup
    exit 1
}

DEV=$(readlink -f "$1")

livecd-iso-to-disk /dev/live $DEV

MNT=$(mktemp -d /media/mnttmp.XXXXXX) 
mount -o loop,rw "$DEV" $MNT || exitclean

cp -r /mnt/live/syslinux $MNT

USBLABEL=$(/sbin/blkid -s UUID -o value $DEV)
USBFS=$(/sbin/blkid -s TYPE -o value $DEV)
sed -i -e "s/root=[^ ]*/root=UUID=$USBLABEL/" -e "s/rootfstype=[^ ]*/rootfstype=$USBFS/" -e "s/overlay=[^ ]*/overlay=$USBLABEL/" $MNT/syslinux/syslinux.cfg

trap exitclean SIGINT SIGTERM

echo "Done."
cleanup