#!/bin/sh

# lh_binary_usb-hdd(1) - build binary 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 binary 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_usb-hdd: Init"

for IMAGE in ${LIVE_BINARY_IMAGE}
do
	if [ "${IMAGE}" = "usb-hdd" ]
	then
		# Requiring stage file
		Require_stagefile .stage/bootstrap
		Require_stagefile .stage/chroot_proc

		# Checking stage file
		Check_stagefile .stage/binary_usb-hdd

		# Checking lock file
		Check_lockfile .lock

		# Creating lock file
		Create_lockfile .lock

		# Checking depends
		Check_package chroot/sbin/mkdosfs dosfstools
		Check_package chroot/usr/sbin/mtools mtools
		Check_package chroot/sbin/parted parted

		case "${LIVE_BOOTLOADER}" in
			grub)
				Check_package chroot/usr/sbin/grub grub
				;;

			syslinux)
				Check_package chroot/usr/bin/syslinux syslinux
				;;
		esac

		# Installing depends
		Install_package

		# Remove old binary
		if [ -f binary.img ]
		then
			rm -f binary.img
		fi

		# Everything which comes here needs to be cleaned up,
		DU_DIM="`du -ms binary | cut -f1`"
		REAL_DIM="`expr ${DU_DIM} + ${DU_DIM} / 50`" # Just 2% more to be sure, need something more sophistcated here...
		dd if=/dev/zero of=binary.img bs=1024k count=${REAL_DIM}
		FREELO="`${LH_LOSETUP} -f`"
		if [ ! -b chroot/${FREELO} ]
		then
			MAKEDEV="true"

			mv chroot/dev chroot/dev.tmp
			find /dev | cpio -dmpu chroot
		fi

		echo "!!! The following error/warning messages can be ignored !!!"
		lh_losetup $FREELO binary.img 0
		Chroot "parted -s ${FREELO} mklabel msdos" || true
		Chroot "parted -s ${FREELO} mkpartfs primary fat16 0.0 100%" || true
		Chroot "parted -s ${FREELO} set 1 boot on" || true
		Chroot "parted -s ${FREELO} set 1 lba off" || true

		case "${LIVE_BOOTLOADER}" in
			grub)
				#echo "(hd0) ${FREELO}" > binary/boot/grub/device.map
#grub --batch --no-floppy --device-map=binary/boot/grub/device.map << EOF
#setup --stage1=binary/boot/grub/stage1 --stage2=binary/boot/grub/stage2 --prefix=/boot/grub (hd0)
#
#quit
#EOF
				#rm -f binary/boot/grub/device.map 

				#Chroot "grub-install --no-floppy --root-directory=/ (hd0)"
				;;

			syslinux)
				cat chroot/usr/lib/syslinux/mbr.bin > ${FREELO}
				;;
		esac

		${LH_LOSETUP} -d ${FREELO}

		FREELO="`${LH_LOSETUP} -f`"
		lh_losetup $FREELO binary.img 1
		Chroot " mkfs.vfat -F 16 -n DEBIAN_LIVE ${FREELO}"
		mkdir -p binary.tmp
		${LH_ROOT_COMMAND} mount ${FREELO} binary.tmp
		cp -r binary/* binary.tmp
		${LH_ROOT_COMMAND} umount binary.tmp
		rmdir binary.tmp

		if [ "${LIVE_BOOTLOADER}" = "syslinux" ]
		then
			Chroot "syslinux ${FREELO}"
		fi

		${LH_LOSETUP} -d ${FREELO}

		#if [ "${LIVE_BOOTLOADER}" = "grub" ]
		#then
		#	FREELO="`${LH_LOSETUP} -f`"
		#	lh_losetup $FREELO binary.img 0

		#	dd if=binary/boot/grub/stage1 of=${FREELO} bs=512 count=1
		#	dd if=binary/boot/grub/stage2 of=${FREELO} bs=512 seek=1

		#	${LH_LOSETUP} -d ${FREELO}
		#fi
		echo "!!! The above error/warning messages can be ignored !!!"

		if [ -n "${MAKEDEV}" ]
		then
			rm -f chroot/dev
			mv chroot/dev.tmp chroot/dev
		fi

		# Removing depends
		Remove_package

		# Creating stage file
		Create_stagefile .stage/binary_usb-hdd
	fi
done