#!/bin/sh

## live-build(7) - System Build Scripts
## Copyright (C) 2006-2011 Daniel Baumann <daniel@debian.org>
##
## live-build 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

# Including common functions
. "${LB_BASE:-/usr/share/live/build}"/scripts/build.sh

# Setting static variables
DESCRIPTION="$(Echo 'build source image')"
HELP=""
USAGE="${PROGRAM} [--force]"

Arguments "${@}"

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

if [ "${LB_SOURCE}" != "true" ]
then
	exit 0
fi

if ! In_list virtual-hdd "${LB_SOURCE_IMAGES}"
then
	exit 0
fi

Echo_message "Begin building source virtual-hdd image..."

# Requiring stage file
Require_stagefile .stage/config .stage/source_debian

# Checking stage file
Check_stagefile .stage/source_virtual-hdd

# Checking lock file
Check_lockfile .lock

# Creating lock file
Create_lockfile .lock

if [ "${LB_BUILD_WITH_CHROOT}" = "true" ]
then
	case "${LB_BINARY_FILESYSTEM}" in
		ext2|ext3)
			Check_package chroot/sbin/mkfs.ext2 e2fsprogs
			;;
	esac
fi

# Installing depends
Install_package

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

# Everything which comes here needs to be cleaned up,
DU_DIM="$(du -ms source | cut -f1)"
REAL_DIM="$(Calculate_partition_size ${DU_DIM} ${LB_BINARY_FILESYSTEM})"
dd if=/dev/zero of=source-virtual.img bs=1024k count=0 seek=${REAL_DIM}

if [ "${LB_BUILD_WITH_CHROOT}" = "true" ]
then
	mv source-virtual.img chroot

	case "${LB_BINARY_FILESYSTEM}" in
		ext2)
			Chroot chroot "mkfs.ext2 -F -m 0 source-virtual.img"
			;;

		ext3)
			Chroot chroot "mkfs.ext3 -F source-virtual.img"
			;;
	esac

	mv chroot/source-virtual.img ./
else
	case "${LB_BINARY_FILESYSTEM}" in
		ext2)
			mkfs.ext2 -F -m 0 source-virtual.img
			;;

		ext3)
			mkfs.ext3 -F -m 0 source-virtual.img
			;;
	esac
fi

mkdir -p source.tmp
${LB_ROOT_COMMAND} mount -o loop source-virtual.img source.tmp
cp -r source/* source.tmp

${LB_ROOT_COMMAND} umount source.tmp
rmdir source.tmp

# Saving cache
Save_cache cache/packages_binary

# Removing depends
Remove_package

# Creating stage file
Create_stagefile .stage/source_virtual-hdd