#!/bin/sh

## live-build(7) - System Build Scripts
## Copyright (C) 2016-2020 The Debian Live team
## Copyright (C) 2006-2015 Daniel Baumann <mail@daniel-baumann.ch>
##
## This program 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
[ -e "${LIVE_BUILD}/scripts/build.sh" ] && . "${LIVE_BUILD}/scripts/build.sh" || . /usr/lib/live/build.sh

# Setting static variables
DESCRIPTION="Execute hacks in chroot"
USAGE="${PROGRAM} [--force]"

# Processing arguments and configuration files
Init_config_data "${@}"

Echo_message "Begin executing hacks..."

# Requiring stage file
Require_stagefiles config bootstrap

# Checking stage file
Check_stagefile

# Acquire lock file
Acquire_lockfile

case "${LB_IMAGE_TYPE}" in
	netboot)
		if [ ! -f chroot/sbin/mount.cifs ]
		then
			Apt chroot install cifs-utils
		fi

		if [ ! -d chroot/etc/initramfs-tools ]
		then
			mkdir chroot/etc/initramfs-tools
		fi

		# Configuring initramfs for NFS
		if ! grep -qs "BOOT=nfs" chroot/etc/initramfs-tools/initramfs.conf
		then
			echo "BOOT=nfs" >> chroot/etc/initramfs-tools/initramfs.conf
		fi

		if ! grep -qs "NFSROOT=auto" chroot/etc/initramfs-tools/initramfs.conf
		then
			echo "NFSROOT=auto" >> chroot/etc/initramfs-tools/initramfs.conf
		fi
		# initrd.img must be regenerated with the settings above
		rm -f chroot/initrd.img_has_already_been_regenerated
		;;
esac

# Now that all packages are installed, remove the os-release divert as it's safe to do so
if Chroot chroot dpkg-divert --list /etc/os-release | grep -q os-release.debootstrap; then
	Chroot chroot dpkg-divert --quiet --local --remove --no-rename /etc/os-release
fi

# Update initramfs (always, because of udev rules in initrd)
case "${LB_INITRAMFS}" in
	live-boot)
		if [ ! -e chroot/initrd.img_has_already_been_regenerated ]; then
			Chroot chroot "${UPDATE_INITRAMFS_OPTIONS} update-initramfs -k all -t -u"
		fi
		;;
	dracut-live)
		# Enable the live module (with a unique name)
		if [ "${LB_IMAGE_TYPE}" = 'netboot' ]; then
			echo "add_dracutmodules+=\" livenet nfs \"" > chroot/etc/dracut.conf.d/20-live-module-${SOURCE_DATE_EPOCH}.conf
		else
			echo "add_dracutmodules+=\" dmsquash-live \"" > chroot/etc/dracut.conf.d/20-live-module-${SOURCE_DATE_EPOCH}.conf
		fi
		# This ensures that dracut uses the same filename as initramfs-tools
		Chroot chroot dpkg-reconfigure dracut
		rm -f chroot/etc/dracut.conf.d/20-live-module-${SOURCE_DATE_EPOCH}.conf
		;;
esac
rm -f chroot/vmlinuz.old
rm -f chroot/initrd.img.old
rm -f chroot/initrd.img_has_already_been_regenerated

# We probably ought to use COMPRESS= in a temporary file in
# /etc/initramfs-tools/conf.d/ instead, but it's hard to pass options that
# way.
case "${LB_INITRAMFS_COMPRESSION}" in
	gzip)
		;;

	bzip2)
		for INITRAMFS in $(find chroot/boot -name 'initrd*'); do
			zcat "${INITRAMFS}" | bzip2 -c ${BZIP2_OPTIONS} > "${INITRAMFS}.new"
			mv "${INITRAMFS}.new" "${INITRAMFS}"
			echo "$(sha1sum ${INITRAMFS} | awk '{ print $1 }')  /boot/$(basename ${INITRAMFS})" > chroot/var/lib/initramfs-tools/$(basename ${INITRAMFS} | sed -e 's|initrd.img-||')
		done
		;;

	lzma)
		for INITRAMFS in $(find chroot/boot -name 'initrd*'); do
			zcat "${INITRAMFS}" | lzma -c ${LZMA_OPTIONS} > "${INITRAMFS}.new"
			mv "${INITRAMFS}.new" "${INITRAMFS}"
			echo "$(sha1sum ${INITRAMFS} | awk '{ print $1 }')  /boot/$(basename ${INITRAMFS})" > chroot/var/lib/initramfs-tools/$(basename ${INITRAMFS} | sed -e 's|initrd.img-||')
		done
		;;
esac

rm -f chroot/etc/hosts

if [ "${LB_DEBIAN_INSTALLER}" = "live" ]
then
	# This is a temporary hack to get rid of fstab;
	# needs cleanup in live-boot first to proper fix.
	rm -f chroot/etc/fstab
	touch chroot/etc/fstab
fi

if [ "${LB_SWAP_FILE_PATH}" ]; then
	dd if=/dev/zero of="chroot/${LB_SWAP_FILE_PATH}" bs=1024k count="${LB_SWAP_FILE_SIZE}"
	mkswap "chroot/${LB_SWAP_FILE_PATH}"
fi

# Creating stage file
Create_stagefile
