#!/bin/bash
# This script automatically configures your Live-USB-Stick to be persistent
# 
# Written by Andreas Loibl <andreas@andreas-loibl.de>
KDIALOG="$(which kdialog)"                      || KDIALOG="/usr/bin/kdialog"
ZENITY="$(which zenity)"                        || ZENITY="/usr/bin/zenity"

partition=3
filesystem=ext4

. /etc/default/distro
isodev="$(awk '{if($2=="/live/image"||$2=="/lib/live/mount/medium"){print $1;}}' /proc/mounts)"
case $isodev in
/dev/sd*) ;;
*) isodev= ;;
esac
case "$( ls /grub.cmdline 2>/dev/null)" in
/grub.cmdline)
if [ -x "$KDIALOG" ]; then
    $KDIALOG --error "Persistent mode seems to be enabled already!" --title "Persistent Live-USB-Stick"
elif [ -x "$ZENITY" ]; then
    $ZENITY --error --text "Persistent mode seems to be enabled already!" --title "Persistent Live-USB-Stick"
fi
exit 1
;;
esac

if [ "$FLL_DISTRO_MODE" != "live" -o -z "$isodev" ]; then
if [ -x "$KDIALOG" ]; then
    $KDIALOG --error "This script can only be used from a Live USB-Stick!" --title "Persistent Live-USB-Stick"
elif [ -x "$ZENITY" ]; then
        $ZENITY --error --text "This script can only be used from a Live USB-Stick!" --title "Persistent Live-USB-Stick"
fi
exit 1
fi

if ((UID)); then
if [ -x "$KDIALOG" ]; then
$KDIALOG --warningcontinuecancel "This script automatically configures your Live-USB-Stick to be persistent:

* it adds a new partition into the unused space of your stick
* it formats, mounts and copies the contents of the current ramdisk-overlay to the new partiton
* it redirects the overlay to the stick
* it restarts the system services to complete the overlay-redirection

The graphical desktop environment and all running programs will be restarted!
===============================================================
CLOSE ALL RUNNING PROGRAMS NOW OR MAKE SURE YOU HAVE NO UNSAVED CHANGES!
===============================================================

Press \"Continue\" when you are ready to start.

" --title "Persistent Live-USB-Stick" || exit 0
elif [ -x "$ZENITY" ]; then

$ZENITY --question --text "This script automatically configures your Live-USB-Stick to be persistent:

* it adds a new partition into the unused space of your stick
* it formats, mounts and copies the contents of the current ramdisk-overlay to the new partiton
* it redirects the overlay to the stick
* it restarts the system services to complete the overlay-redirection

The graphical desktop environment and all running programs will be restarted!
===============================================================
CLOSE ALL RUNNING PROGRAMS NOW OR MAKE SURE YOU HAVE NO UNSAVED CHANGES!
===============================================================

Press \"Continue\" when you are ready to start.

" --title "Persistent Live-USB-Stick" || exit 0
fi
cp "$0" "/tmp/$(basename "$0")"
chmod +x "/tmp/$(basename "$0")"
if [ -x "$KDIALOG" ]; then
$KDIALOG --progressbar "Please wait..." --title "Persistent Live-USB-Stick" 0
sudo "/tmp/$(basename "$0")" "$@"
elif [ -x "$ZENITY" ]; then
sudo "/tmp/$(basename "$0")" "$@" | $ZENITY --progress --pulsate --text "Please wait..." --title "Persistent Live-USB-Stick" 0
fi

exit $?
fi

exec >/tmp/persistent.log 2>&1
set -x

. /usr/share/acritoxinstaller/modules/partitions
isodisk="$(get_disk $isodev)"
is_removeable "$isodisk" || exit 1

nop() { return 0; }
trap nop TERM KILL HUP

# Debug output
set -x

if [ ! -e "$isodisk"$partition ]; then
	# Add partition
	echo "n|p|$partition|||w" | tr '|' '\n' | fdisk "$isodisk" 
	partprobe "$isodisk"
	while [ ! -e "$isodisk"$partition ]; do sleep 1; done
	mkfs.$filesystem -L "persistence" "$isodisk"$partition
fi

if [ -d /lib/live/mount ]; then
    mkdir -p /lib/live/mount/persistence
    mount "$isodisk"$partition /lib/live/mount/persistence
    for i in /etc/rc1.d/K* /etc/init.d/dbus; do $i stop; done
    fuser -km -TERM /lib/live/mount/overlay
    rsync --exclude=lost+found -aHSx --devices --specials --delete-before /lib/live/mount/overlay/ /lib/live/mount/persistence/
    set -- $(cat /proc/cmdline)
    shift
    echo "set persistence_cmdline='$@ persistence'" > /lib/live/mount/persistence/grub.cmdline
    echo "/ union" > /lib/live/mount/persistence/persistence.conf
    #echo "/ union" > /lib/live/mount/persistence/live-persistence.conf # old file 
    mount -v -t aufs -o remount,prepend:/lib/live/mount/persistence,mod:/lib/live/mount/overlay=ro,del:/lib/live/mount/overlay none /
    fuser -km -TERM /lib/live/mount/overlay

    # Debug output:
    fuser -vm /lib/live/mount/overlay
    ps aux

    umount -l /lib/live/mount/overlay
    mount --move /lib/live/mount/persistence /lib/live/mount/overlay
    rmdir /lib/live/mount/persistence
elif [ -d /live ]; then
    mkdir -p /live/persistence
    mount "$isodisk"$partition /live/persistence
    for i in /etc/rc1.d/K* /etc/init.d/dbus; do $i stop; done
    fuser -km -TERM /live/overlay
    rsync --exclude=lost+found -aHSx --devices --specials --delete-before /live/overlay/ /live/persistence/
    set -- $(cat /proc/cmdline)
    shift
    echo "set persistence_cmdline='$@ persistence'" > /live/persistence/grub.cmdline
    echo "/ union" > /live/persistence/persistence.conf
    #echo "/ union" > /live/persistence/live-persistence.conf # old file
    mount -v -t aufs -o remount,prepend:/live/persistence,mod:/live/overlay=ro,del:/live/overlay none /
    fuser -km -TERM /live/overlay

    # Debug output:
    fuser -vm /live/overlay
    ps aux

    umount -l /live/overlay
    mount --move /live/persistence /live/overlay
    rmdir /live/persistence
fi

for i in /etc/rc5.d/S*; do $i start; done
#killall -9 kdialog
kill $(ps ax | grep kdialog | grep Live-USB | cut -c2-5)
exit 0