#!/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 'execute hacks in chroot')"
HELP=""
USAGE="${PROGRAM} [--force]"

Arguments "${@}"

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

Echo_message "Begin executing hacks..."

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

# Checking stage file
Check_stagefile .stage/chroot_hacks

# Checking lock file
Check_lockfile .lock

# Creating lock file
Create_lockfile .lock

case "${LB_PARENT_DISTRIBUTION}" in
	squeeze)
		if echo "${LB_TASKS}" | grep -qs lxde ||
			echo "${LB_TASKS}" | grep -qs xfce
		then
			GDM="gdm"
		else
			GDM="gdm3"
		fi
		;;

	*)
		GDM="gdm3"
		;;
esac

# Handling default desktop configuration
case "${LB_MODE}" in
	debian*)
		# disable kexec-tools
		if [ -e chroot/sbin/kexec ]
		then
			echo "kexec-tools kexec-tools/load_kexec boolean false" > chroot/root/preseed
			Chroot chroot "debconf-set-selections /root/preseed"
			rm -f chroot/root/preseed
			Chroot chroot "dpkg-reconfigure kexec-tools"
		fi
		;;
esac

for TASK in ${LB_TASKS}
do
	case "${TASK}" in
		gnome-desktop)
			# gnome is the only desktop environment in this image
			if [ -z "$(echo ${LB_TASKS} | sed -e 's|gnome-desktop||' -e 's| desktop ||' | grep desktop)" ]
			then
				# set display manager
				echo "${GDM} shared/default-x-display-manager select ${GDM}" > chroot/root/preseed
				Chroot chroot "debconf-set-selections /root/preseed"
				rm -f chroot/root/preseed
				Chroot chroot "dpkg-reconfigure ${GDM}"
				echo "/usr/bin/${GDM}" > chroot/etc/X11/default-display-manager

				# set session manager
				Chroot chroot "update-alternatives --set x-session-manager /usr/bin/gnome-session"
			fi
			;;

		kde-desktop)
			# kde is the only desktop-environment in this image
			if [ -z "$(echo ${LB_TASKS} | sed -e 's|kde-desktop||' -e 's| desktop ||' | grep desktop)" ]
			then
				# set display manager
				echo "kdm shared/default-x-display-manager select kdm" > chroot/root/preseed
				Chroot chroot "debconf-set-selections /root/preseed"
				rm -f chroot/root/preseed
				Chroot chroot "dpkg-reconfigure kdm"
				echo "/usr/bin/kdm" > chroot/etc/X11/default-display-manager

				# set session manager
				Chroot chroot "update-alternatives --set x-session-manager /usr/bin/startkde"

				# get rid of resolvconf
				Chroot chroot "dpkg -P resolvconf"
			fi
			;;

		lxde-desktop)
			# lxde is the only desktop environment in this image
			if [ -z "$(echo ${LB_TASKS} | sed -e 's|lxde-desktop||' -e 's| desktop ||' | grep desktop)" ]
			then
				# set display manager
				echo "${GDM} shared/default-x-display-manager select ${GDM}" > chroot/root/preseed
				Chroot chroot "debconf-set-selections /root/preseed"
				rm -f chroot/root/preseed
				Chroot chroot "dpkg-reconfigure ${GDM}"
				echo "/usr/bin/${GDM}" > chroot/etc/X11/default-display-manager

				# set session manager
				Chroot chroot "update-alternatives --set x-session-manager /usr/bin/startlxde"
			fi
			;;

		xfce-desktop)
			# xfce is the only desktop environment in this image
			if [ -z "$(echo ${LB_TASKS} | sed -e 's|xfce-desktop||' -e 's| desktop ||' | grep desktop)" ]
			then
				# set display manager
				echo "${GDM} shared/default-x-display-manager select ${GDM}" > chroot/root/preseed
				Chroot chroot "debconf-set-selections /root/preseed"
				rm -f chroot/root/preseed
				Chroot chroot "dpkg-reconfigure ${GDM}"
				echo "/usr/bin/${GDM}" > chroot/etc/X11/default-display-manager

				# set session manager
				Chroot chroot "update-alternatives --set x-session-manager /usr/bin/xfce4-session"
			fi
			;;
	esac
done

# Removing openssh-server hostkeys,
# they are regenerated by live-config.
rm -f chroot/etc/ssh/ssh_host_*_key chroot/etc/ssh/ssh_host_*_key.pub

# Removing udev mac caching rule
rm -f chroot/etc/udev/rules.d/*persistent-net.rules

case "${LB_BINARY_IMAGES}" in
	net)
		if [ ! -f chroot/usr/bin/smbmount ]
		then
			Apt chroot install smbfs
		fi

		if [ ! -d chroot/etc/initramfs-tools ]
		then
			mkdir chroot/etc/initramfs-tools
		fi
		if [ ! "$(grep 'MODULES=netboot' chroot/etc/initramfs-tools/initramfs.conf)" ]
		then
			# Configuring initramfs for NFS
cat >> chroot/etc/initramfs-tools/initramfs.conf << EOF
MODULES=netboot
BOOT=nfs
NFSROOT=auto
EOF
		fi
		;;
esac

# Update initramfs (always, because of udev rules in initrd)
case "${LB_INITRAMFS}" in
	casper)
		UPDATE_INITRAMFS_OPTIONS="CASPER_GENERATE_UUID=1"
		;;

	live-boot)
		#UPDATE_INITRAMFS_OPTIONS="LIVE_GENERATE_UUID=1"
		;;
esac

case "${SUBPROJECT}" in
	wubi)
		UPDATE_INITRAMFS_OPTIONS="$UPDATE_INITRAMFS_OPTIONS FRAMEBUFFER=y"
		;;
esac

Chroot chroot "${UPDATE_INITRAMFS_OPTIONS} update-initramfs -k all -t -u"

# 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}"
		done
		;;

	lzma)
		# 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.
		for INITRAMFS in $(find chroot/boot -name 'initrd*'); do
			zcat "${INITRAMFS}" | lzma -c ${LZMA_OPTIONS} > "${INITRAMFS}.new"
			mv "${INITRAMFS}.new" "${INITRAMFS}"
		done
		;;
esac

# Ensure readable permissions on initramfs. loop-aes-utils sets umask to
# protect GPG keys, which live-build does not support.
# Note: Use find rather than chmod on the wildcard, one never knows what
# people might do in local hooks, and there might be no initrds at all.
find chroot/boot -name 'initrd*' -print0 | xargs -r -0 chmod go+r

# Update the mlocate database.  It's convenient for this to be already up to
# date on the live system, and it means that if the live system is later
# installed to a hard disk then less work will be required after
# installation.
if [ -x chroot/usr/bin/updatedb.mlocate ]; then
	Chroot chroot updatedb.mlocate
fi

# Update the Apt Xapian index.  The package would do this itself, but (a) it
# checks policy-rc.d which says it isn't allowed to, and (b) it wants to
# build the index in the background which will be racy in the context of
# live-build.
if [ -x chroot/usr/sbin/update-apt-xapian-index ]; then
	Chroot chroot PYTHONDONTWRITEBYTECODE=1 /usr/sbin/update-apt-xapian-index --force --quiet
fi

# Remove build systems clock drift
echo "0.0 0 0.0" > chroot/etc/adjtime

# Hack to remove some language packs from Ubuntu desktop live images;
# this is 12.04.*-specific and shouldn't be carried forward.
if [ "${LB_ARCHITECTURES}" = "i386" ] && [ "${PROJECT}" = "ubuntu" ]; then
	Chroot chroot "apt-get remove --purge -y language-pack-de-base firefox-locale-de" || Echo_warning "Unable to find German langpack on image so can't remove it"
	Chroot chroot "apt-get remove --purge -y language-pack-pt-base firefox-locale-pt" || Echo_warning "Unable to find Portuguese langpack on image so can't remove it"
fi
if [ "${LB_ARCHITECTURES}" = "amd64" ] && [ "${PROJECT}" = "ubuntu" ]; then
	Chroot chroot "apt-get remove --purge -y language-pack-pt-base firefox-locale-pt" || Echo_warning "Unable to find Portuguese langpack on image so can't remove it"
	Chroot chroot "apt-get remove --purge -y language-pack-es-base firefox-locale-es" || Echo_warning "Unable to find Spanish langpack on image so can't remove it"
fi

# Remove cruft
rm -f chroot/boot/initrd*bak*
rm -f chroot/etc/apt/trusted.gpg~
rm -f chroot/etc/group- chroot/etc/passwd-
rm -f chroot/etc/gshadow- chroot/etc/shadow-
rm -f chroot/var/cache/debconf/*-old
rm -f chroot/var/lib/dpkg/*-old
rm -f chroot/var/lib/dbus/machine-id
rm -f chroot/var/log/apt/term.log

if [ -n "${LB_ROOT_COMMAND}" ]
then
	${LB_ROOT_COMMAND} chown -R --quiet $(whoami):$(whoami) chroot
fi

case "${LB_INITRAMFS}" in
	casper)
		ID="999"
		;;

	live-boot)
		ID="1000"
		;;

	*)
		ID=""
		;;
esac

if [ -d chroot/home/${LB_USERNAME} ] && [ "${ID}" ]
then
	chown -R --quiet ${ID}:${ID} chroot/home/${LB_USERNAME}
fi

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
fi

if [ "${LB_EXPOSED_ROOT}" = "true" ]
then
	# Make sure RW dirs exist so that the initramfs script has
	# a directory in which to bind the tmpfs filesystems
	COW_DIRECTORIES="/home /live /tmp /var/lib/live /var/lock /var/log /var/run /var/tmp /var/spool"

	for DIRECTORY in ${COW_DIRECTORIES}
	do
		mkdir -p chroot/"${DIRECTORY}"
	done

	# Config files which need to be RW
	COW_FILES="/etc/adjtime /etc/fstab /etc/hostname /etc/hosts /etc/live.conf /etc/network/interfaces /etc/resolv.conf /etc/udev/rules.d/*persistent-net.rules /etc/udev/rules.d/*persistent-cd.rules /etc/X11/xorg.conf"

	# Where we will store RW config files
	RW_DIRECTORY="/var/lib/live"

	for FILE in ${COW_FILES}
	do
		DIRECTORY="$(dirname ${FILE})"
		FILE="$(basename ${FILE})"
		RELATIVE_PATH="$(echo ${DIRECTORY} | sed 's|[^/]\+|..|g; s|^/||g')"

		# Touch files in case they don't yet exist
		mkdir -p chroot/${DIRECTORY}
		touch chroot/${DIRECTORY}/${FILE}

		# Move files to the read-write directory
		mkdir -p chroot/${RW_DIRECTORY}/${DIRECTORY}
		mv chroot/${DIRECTORY}/${FILE} chroot/${RW_DIRECTORY}/${DIRECTORY}

		# Create a symbolic link to RW config file
		ln -s ${RELATIVE_PATH}/${RW_DIRECTORY}/${DIRECTORY}/${FILE} chroot/${DIRECTORY}/${FILE}
	done

	# Mount doesn't write to a symlink so use /proc/mounts instead,
	# see debian bug #154438 for more info
	rm -f chroot/etc/mtab
	ln -s /proc/mounts chroot/etc/mtab
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

# Show popular warnings
if [ -e chroot/etc/init.d/resolvconf ]
then
	Echo_warning "Your live system appears to include resolvconf which is by its nature incompatible in most ways with debian-live."
	Echo_warning "Keeping resolvconf installed *can* lead to broken network support in the resulting live system."
	Echo_warning "Unless you really know what you are doing, we recommend that you neither install resolvconf explicitly nor implicitly through another package's relation (e.g. by having automatic installation of recommended packages to true)."
fi

# Creating stage file
Create_stagefile .stage/chroot_hacks