[chris@localhost ~]$ ls -l /sbin/init
-rwxr-xr-x 1 root root 35264 Jul  2  2006 /sbin/init*
[chris@localhost ~]$

#!/bin/bash

# Graphical remaster script
# (C) 2007 HighKing & Chris and mandrivaclub.nl
# License GPL 2.0 or later

# Set script title
TITLE="MCNLive remaster-on-the-fly"

# Check if kdialog exists, if not, use gdialog (wrapper for zenity) for some dialogs.
# Note: Most dialogs are pure zenity, but zenity does not have the yes/no function :(
if [ -f /usr/bin/kdialog ]; then
    DIALOG=kdialog
else
    DIALOG=gdialog
fi

# Check if user is logged in as root
if [ "$UID" != "0" ]; then
    zenity --error --title="$TITLE" --text="Please run this script as root!"
    exit 0
fi

# Check wether or not we're running in live mode.
if [ -d "/union" ]; then
    echo "Running in live mode..."
    ROOT=/union
else
    echo "Running in installed mode..."
    ROOT=/
fi

# First question: Are you sure?
zenity --width=400 --height=400 --title "$TITLE" --question --text "You're about to create a remaster of this live cd.\n
* A compressed image of the running live system will be created on a chosen partition.\n
* Requirements: min. 512 MB RAM, min. 2 GB free space on the chosen partition.\n
* A Windows FAT filesystem can't be used. A Windows NTFS partition must be mounted read-write which is not the default! (Start MCNLive with 'livecd ntfsrw')\n
* If you are unsure read on in the HowTo page in your browser, and press 'Cancel'."
if [ "$?" == "1" ]; then
    zenity --info --text="You chose not to continue ..." --title="$TITLE"
    exit 0
fi

$DIALOG --title $TITLE --yesno "Would you like to provide some advanced options?

For Example: --keyb=fr"
if [ "$?" == "0" ]; then
    # Ask for custom boot screens
    $DIALOG --title $TITLE --yesno "Would you like to use your own customized boot screens?"
    if [ "$?" == "0" ]; then
        BOOTSCR=$(zenity --title="Please select your splash message" --file-selection --filename=/home/guest/splash.msg);
        if [ "$?" == "1" ]; then
            BOOTSCR=/home/guest/splash.msg
        fi

        BOOTF1=$(zenity --title="Please select your F1 bootscreen" --file-selection --filename=/home/guest/myboot.msg);
        if [ "$?" == "1" ]; then
            BOOTF1=/home/guest/myboot.msg
        fi

        BOOTF2=$(zenity --title="Please select your F2 bootscreen" --file-selection --filename=/home/guest/help2.msg);
        if [ "$?" == "1" ]; then
            BOOTF2=/home/guest/help2.msg
        fi
    else
        BOOTSCR=/home/guest/splash.msg
        BOOTF1=/home/guest/myboot.msg
        BOOTF2=/home/guest/help2.msg
    fi
    
    # Ask for special mklivecd options
    OPTIONS=$(zenity --title="$TITLE" --entry --text="Insert your mklivecd options here:
Note: any wrong option will break the command!")
else
    BOOTSCR=/home/guest/splash.msg
    BOOTF1=/home/guest/myboot.msg
    BOOTF2=/home/guest/help2.msg
    OPTIONS=""
fi

# Set the directory for the temporary files
if [ "$ROOT" == "/" ]; then
    TMPPART=/
else
    # Ask for a place for the temporary files
    TMPDEV=$(df -h --exclude-type=vfat --exclude-type=ntfs | grep 'hd\|sd' |
    zenity --list --title="$TITLE" --text="Select a partition for the temporary files." --width=400 --height=200 --column=Partitie)
        # If a user pressed the cancel button, just quit
        if [ "$?" == "1" ]; then
            exit 0
        fi
    TMPDEV=$(echo "$TMPDEV" | cut -b1-9)
    TMPPART=$(mount | grep $TMPDEV | cut -f 3 -d ' ')

    # Set needed space on the temporary partition...
    ROOTSPACE=$(df -h / | grep / | cut -b28-34 | sed s/G/*1024/\;s/M//\;s/,/./ | bc)
    UNIONSPACE=$(df -h /union | grep /union | cut -b28-34 | sed s/G/*1024/\;s/M//\;s/,/./ | bc)
    NEEDSPACE=$(echo "{($ROOTSPACE+$UNIONSPACE*2.1)/1}"|bc)
    echo "Needed space on temporary partition: "$NEEDSPACE"M"

    # Check if TMPPART has enough space to do the remaster (at least the value of NEEDSPACE)
    FREESPACE=$(df -h $TMPPART | grep $TMPPART | cut -b35-40 | sed s/G/*1024/\;s/M//\;s/,/./ | bc)
    FREESPACE=$(echo "{$FREESPACE/1}" | bc)
    echo -n "Chosen partition has "$FREESPACE"Mb of free space... "

    if [ "$FREESPACE" -gt "$NEEDSPACE" ]; then
            echo "that should be enough."
    else
            echo "that is NOT enough."
            zenity --error --title="$TITLE" --text="Chosen partition has not enough free space left\nYou will need at least $NEEDSPACE MB free!"
            exit 1
    fi
    # Check if /tmp directory exist on device
    if [ -d "$TMPPART/tmp" ]; then
            TMPDIR="$TMPPART/tmp"
    else
            TMPDIR="$TMPPART"
    fi
fi

# Remove any pre-existing mkremaster temporary directories
testmklivecd=$(echo $TMPDIR"mklivecd.*");
rm -rf $testmklivecd

# Ask where to put the remaster
ISOFILE=$(zenity --title="$TITLE" --file-selection --save --filename=$TMPPART/remaster.iso);
if [ "$?" == "1" ]; then
    echo "Cancelled!"
    exit 1
fi

# Create the remaster
echo "Running mklivecd to create the remaster..."
konsole -T 'Creating your personal remaster...' --nomenubar --notabbar -e mklivecd --verbose --root=$ROOT --tmp=$TMPDIR --bootmsg $BOOTSCR --bootkey F1=$BOOTF1 --bootkey F2=$BOOTF2 $OPTIONS $ISOFILE >/dev/null 2>&1

# Check if the mklivecd temporary files are removed, if not
# something must have gone wrong! :-(
testmklivecd=$( ls -d $tmpdir* | grep mklivecd. );
if [ "$testmklivecd" == "" ]; then
    echo "mklivecd temp dir does not exist anymore... (this is actually a good thing)"
else
    zenity --title="$TITLE" --error --text="Something went wrong! :-("
    exit 1
fi

# Check if the remaster iso actually exists, and if yes, calculate the md5-sum.
if [ -f $ISOFILE ]; then
    # I stole this line from the Make MD5 script on http://g-scripts.sf.net
    # It doesn't quite work though, at least I don't get to see any progres....
    ( md5sum $ISOFILE > $ISOFILE.md5.txt ) 2>&1 | zenity --progress --title="$TITLE" --text="Creating MD5SUM check file" --pulsate --auto-close
else
    zenity --title="$TITLE" --error --text="Something went wrong! :-("
    exit 1
fi

# Popup notification, tell the user that we're done
ISOMD5=$(cat $ISOFILE.md5.txt | cut -b1-32);
zenity --info --title="$TITLE" --text="The remaster was created successfully\!\n\nIt is created in $ISOFILE.\nThe MD5 sum of the file is:\n$ISOMD5"

# The end! :-)
exit 0