#!/bin/sh

# lh_binary_rootfs(1) - build rootfs image
# Copyright (C) 2006-2007 Daniel Baumann <daniel@debian.org>
#
# live-helper comes with ABSOLUTELY NO WARRANTY; for details see COPYING.
# This is free software, and you are welcome to redistribute it
# under certain conditions; see COPYING for details.

set -e

# Source common functions
for FUNCTION in /usr/share/live-helper/functions/*.sh
do
	. ${FUNCTION}
done

# Set static variables
DESCRIPTION="build rootfs image"
HELP=""
USAGE="${PROGRAM} [--force]"

Arguments "${@}"

# Reading configuration files
Read_conffile config/common
Read_conffile config/image
Set_defaults

# Requiring stage file
Require_stagefile .stage/bootstrap
Require_stagefile .stage/binary_chroot

# Checking lock file
Check_lockfile .lock

# Creating lock file
Create_lockfile .lock

# Checking stage file
Check_stagefile .stage/binary_rootfs

# Creating directory
if [ ! -d binary/casper ]
then
	mkdir -p binary/casper
fi

case "${LIVE_FILESYSTEM}" in
	ext2)
		if [ ! -f chroot/usr/bin/genext2fs ]
		then
			PACKAGES="${PACKAGES} genext2fs"
		fi

		if [ -n "${PACKAGES}" ]
		then
			# Installing packages
			case "${LH_APT}" in
				apt|apt-get)
					Chroot "apt-get install --yes ${PACKAGES}"
					;;

				aptitude)
					Chroot "aptitude install --assume-yes ${PACKAGES}"
					;;
			esac
		fi

		# Remove old ext2 image
		if [ -f binary/casper/filesystem.ext2 ]
		then
			rm -f binary/casper/filesystem.ext2
		fi

		DU_DIM="`du -ks chroot | cut -f1`"
		REAL_DIM="`expr ${DU_DIM} + ${DU_DIM} / 20`" # Just 5% more to be sure, need something more sophistcated here...

		Chroot "genext2fs --size-in-blocks=${REAL_DIM} --reserved-blocks=0 --root=chroot filesystem.ext2"

		# Move image
		mv chroot/filesystem.ext2 binary/casper
		rm -rf chroot/chroot

		# Removing packages
		if [ -n "${PACKAGES}" ]
		then
			case "${LH_APT}" in
				apt|apt-get)
					Chroot "apt-get remove --purge --yes ${PACKAGES}"
					;;

				aptitude)
					Chroot "aptitude purge --assume-yes ${PACKAGES}"
					;;
			esac
		fi
		;;

	plain)
		if [ -d binary/casper/filesystem.dir ]
		then
			rm -rf binary/casper/filesystem.dir
		fi

		mv chroot/chroot binary/casper/filesystem.dir
		;;

	squashfs)
		if [ ! -f chroot/usr/sbin/mksquashfs ]
		then
			PACKAGES="${PACKAGES} squashfs-tools"
		fi

		if [ -n "${PACKAGES}" ]
		then
			# Installing packages
			case "${LH_APT}" in
				apt|apt-get)
					Chroot "apt-get install --yes ${PACKAGES}"
					;;

				aptitude)
					Chroot "aptitude install --assume-yes ${PACKAGES}"
					;;
			esac
		fi

		# Remove old squashfs image
		if [ -f binary/casper/filesystem.squashfs ]
		then
			rm -f binary/casper/filesystem.squashfs
		fi

		# Create image
		if [ "${LIVE_PACKAGES_LISTS}" != "minimal" ] && [ "${LIVE_PACKAGES_LISTS}" != "mini" ]
		then
			Chroot "mksquashfs chroot filesystem.squashfs"
		else
			Chroot "mksquashfs chroot filesystem.squashfs -e chroot/boot/vmlinuz* chroot/boot/initrd.img* chroot/vmlinuz* chroot/initrd.img* chroot/boot/config-* chroot/boot/System.map-*"
		fi

		# Move image
		mv chroot/filesystem.squashfs binary/casper
		rm -rf chroot/chroot

		# Removing packages
		if [ -n "${PACKAGES}" ]
		then
			case "${LH_APT}" in
				apt|apt-get)
					Chroot "apt-get remove --purge --yes ${PACKAGES}"
					;;

				aptitude)
					Chroot "aptitude purge --assume-yes ${PACKAGES}"
					;;
			esac
		fi
		;;
esac

# Creating stage file
Create_stagefile .stage/binary_rootfs