#!/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 "${@}"

Echo_debug "Init ${PROGRAM}"

# Reading configuration files
Read_conffile config/common
Read_conffile config/bootstrap
Read_conffile config/chroot
Read_conffile config/binary
Read_conffile config/source
Set_defaults

Breakpoint "binary_rootfs: Init"

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

# Checking stage file
Check_stagefile .stage/binary_rootfs

# Checking lock file
Check_lockfile .lock

# Creating lock file
Create_lockfile .lock

case "${LH_INITRAMFS}" in
	casper)
		INITFS="casper"
		;;

	live-initramfs)
		INITFS="live"
		;;
esac

# Creating directory
if [ ! -d binary/${INITFS} ]
then
	mkdir -p binary/${INITFS}
fi

case "${LIVE_CHROOT_FILESYSTEM}" in
	ext2)
		# Checking depends
		Check_package chroot/usr/bin/genext2fs genext2fs

		# Installing depends
		Install_package

		# Remove old ext2 image
		if [ -f binary/${INITFS}/filesystem.ext2 ]
		then
			rm -f binary/${INITFS}/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/${INITFS}
		rm -rf chroot/chroot

		# Removing depends
		Remove_package
		;;

	plain)
		if [ -d binary/${INITFS}/filesystem.dir ]
		then
			rm -rf binary/${INITFS}/filesystem.dir
		fi

		mv chroot/chroot binary/${INITFS}/filesystem.dir
		;;

	squashfs)
		# Checking depends
		Check_package chroot/usr/sbin/mksquashfs squashfs-tools

		# Installing depends
		Install_package

		# Remove old squashfs image
		if [ -f binary/${INITFS}/filesystem.squashfs ]
		then
			rm -f binary/${INITFS}/filesystem.squashfs
		fi

		if [ "${LH_QUIET}" = "enabled" ] && [ "${LIVE_DISTRIBUTION}" != "etch" ] && [ "${LIVE_DISTRIBUTION}" != "testing" ]
		then
			MKSQUASHFS_OPTIONS="${MKSQUASHFS_OPTIONS} -no-progress"
		fi

		if [ "${LH_VERBOSE}" = "enabled" ]
		then
			MKSQUASHFS_OPTIONS="${MKSQUASHFS_OPTIONS} -info"
		fi

		if [ "${LIVE_PACKAGES_LISTS}" != "minimal" ] && [ "${LIVE_PACKAGES_LISTS}" != "mini" ]
		then
			MKSQUASHFS_OPTIONS="${MKSQUASHFS_OPTIONS} -e chroot/boot/vmlinuz* chroot/boot/initrd.img* chroot/vmlinuz* chroot/initrd.img* chroot/boot/config-* chroot/boot/System.map-*"
		fi

		# Create image
		Chroot "mksquashfs chroot filesystem.squashfs ${MKSQUASHFS_OPTIONS}"

		# Move image
		${LH_ROOT_COMMAND} mv chroot/filesystem.squashfs binary/${INITFS}
		${LH_ROOT_COMMAND} rm -rf chroot/chroot

		if [ -n "${LH_ROOT_COMMAND}" ]
		then
			${LH_ROOT_COMMAND} chown -R `whoami`:`whoami` binary/${INITFS}
		fi

		# Removing depends
		Remove_package
		;;
esac

# Creating stage file
Create_stagefile .stage/binary_rootfs