#!/bin/bash
iso="maverick-desktop-i386"

if [ "$UID" -ne "0" ]
then
echo "This script will only work when run by "root"."
exit 1
fi

# not everyone will have squash tools, install them if not found
if [ ! `which unsquashfs` ]
then
aptitude install squashfs-tools
fi

# not everyone will have genisoimage, install it if not found
if [ ! `which mkisofs` ]
then
aptitude install genisoimage
fi

# not everyone will have xnest, install it if not found
if [ ! `which Xnest` ]
then
aptitude install xnest
fi

# not everyone will have wget, install it if not found
if [ ! `which wget` ]
then
aptitude install wget
fi

START=$(date +%s)

#wget http://cdimage.ubuntu.com/daily-live/current/maverick-desktop-i386.iso
mkdir iso
mount $iso.iso iso/ -o loop

cp -R iso/ image/

echo "Decompressing SquashFS..."
cp iso/casper/filesystem.squashfs ./
unsquashfs filesystem.squashfs

echo "Setting up Live CD chroot..."
mv squashfs-root chroot
cp /etc/resolv.conf chroot/etc/resolv.conf
cp /etc/hosts chroot/etc/
#cp -a /etc/apt/* chroot/etc/apt/

mount --bind /dev/ chroot/dev
chroot chroot/ mount -t proc none /proc
chroot chroot/ mount -t sysfs none /sys
chroot chroot/ mount -t devpts none /dev/pts

mkdir -p chroot/boot/grub
export HOME=/root
export LC_ALL=C

echo "Refreshing Aptitude"
chroot chroot/ aptitude update
chroot chroot/ apt-get autoremove

echo "...Upgrading Packages"
chroot chroot/ aptitude -f -y safe-upgrade

sleep 3

# Please edit for pkg's you want-
echo "Installing Packages you want"
sleep 3
chroot chroot/ aptitude install -y -f menu firefox normalize-audio grub2 plymouth-x11 flashplugin-nonfree flashplugin-nonfree-extrasound lxappearance leafpad pcmanfm gdm gparted casper laptop-detect os-prober ejecter extra-xdg-menus menu-xdg lupin-casper lupin-support lxterminal synaptic ubiquity ubiquity-frontend-gtk discover1 linux-generic gnome-device-manager gnome-menus gnome-settings-daemon wicd gnome-utils xarchiver gftp gnome-format gnome-themes language-selector langupdate xnest zenity zsync usb-creator-gtk e17-svn
chroot chroot/ apt-get check
chroot chroot/ aptitude -y purge linux-image-2.6.32-21-generic
chroot chroot/ apt-get autoremove
chroot chroot/ apt-get clean
chroot chroot/ aptitude clean
chroot chroot/ aptitude autoclean

#sleep 30

# hopefully creating working default user ubiquity, etc desktop
# add "Enlightenment;" to end of ubiquity-gtkui.desktop file!
# cntrl + o + enter saves file, cntrl + x closes Nano
chroot chroot/ nano usr/share/applications/ubiquity-gtkui.desktop

# prevent the installer from changing the apt sources.list (hopefully?)

cp /usr/share/ubiquity/apt-setup chroot/usr/share/ubiquity/apt-setup.saved

chroot chroot/ easy_e17.sh -u

sleep 3

# Here set up the e17 Desktop, note change the sleep ? to whatever
echo "Starting Xnest Session..."
Xnest -ac :3 &
echo "Please set up your root & user Desktop,etc then logout of your Desktop & kill Xnest"
sleep 3
chroot chroot/ env DISPLAY=":3" opt/e17/bin/enlightenment_start &
sleep 300

mkdir e17src
rm -r chroot/usr/share/doc/*
mv chroot/var/cache/e17_src e17src/
chroot chroot/ usr/sbin/update-initramfs -u -v

echo "Creating filesystem.manifest and filesystem.manifest-desktop"
echo "...Copying installed packages text to image/"
sleep 3
chroot chroot/ dpkg-query -W --showformat='${Package} ${Version}\n' > image/casper/filesystem.manifest
cp image/casper/filesystem.manifest image/casper/filesystem.manifest-desktop
sleep 3

#cleanup chroot
echo "Cleaning up chroot..."
chroot chroot/ apt-get check
chroot chroot/ apt-get autoremove
chroot chroot/ apt-get clean
chroot chroot/ aptitude clean
chroot chroot/ aptitude autoclean
chroot chroot/ rm -rf /tmp/*
chroot chroot/ rm /etc/resolv.conf
chroot chroot/ umount -l -f /proc
chroot chroot/ umount -l -f /sys
chroot chroot/ umount /dev/pts
umount chroot/dev

echo "Copying your kernel and initrd for the livecd"
cp chroot/boot/vmlinuz-`uname -r` image/casper/vmlinuz
cp chroot/boot/initrd.img-`uname -r` image/casper/initrd.lz

echo "Removing old SquashFS filesystem..."
rm image/casper/filesystem.squashfs

echo "Creating new SquashFS filesystem...This will take a while"
sleep 1
mksquashfs chroot image/casper/filesystem.squashfs
printf $(sudo du -sx --block-size=1 chroot | cut -f1) > image/casper/filesystem.size

echo "Finding and creating MD5 hash sums of files in image..."
cd image
find . -type f -print0 | xargs -0 md5sum > md5sum.txt

sleep 3

echo "Creating new image..."
cd ..
genisoimage -r -V "Ubuntu 10.04 i386" -cache-inodes -J -l \
-b isolinux/isolinux.bin -c isolinux/boot.cat -no-emul-boot \
-boot-load-size 4 -boot-info-table -o Phalanx2-e17-10.04-i386-alpha1.iso image/

#cleanup working directory
echo "Cleaning up working directory..."
umount iso/

# In case you want to chroot back into squashfs-root or fix something in image/ , deleting squashfs-root and image folders commented out.
#rm -rf squashfs-root
#rm -rf image
rm -rf iso
rm filesystem.squashfs

echo "Getting MD5 and SHA1 sum of image..."
echo "MD5: " > Phalanx2-e17-10.04-i386-alpha1.sums
md5sum Phalanx2-e17-10.04-i386-alpha1.iso >> Phalanx2-e17-10.04-i386-alpha1.iso.md5sums
echo "SHA1: " >> Phalanx2-e17-10.04-i386-alpha1.sums
sha1sum Phalanx2-e17-10.04-i386-alpha1.iso >> Phalanx2-e17-10.04-i386-alpha1.iso.sha1sums

END=$(date +%s)

echo "Done at `date`. The whole process took $(($END - $START)) seconds!"
exit