#!/bin/bash

# some default settings - gets overidden with the profile config
CDPROFILE="$2"
[[ -z ${CDPROFILE} ]] && CDPROFILE="default"
LIVECDROOT="/mnt/MLIVECDBUILD/${CDPROFILE}"
CDCHROOTDIR="${LIVECDROOT}/chroot"
CDISONAME="livecd-${CDPROFILE}.iso"
CDISOROOT="${LIVECDROOT}/isoroot"
CDKERNELNAME="magelive"
CDHOSTNAME="livecd-${CDPROFILE}"
MAGERC="/etc/mage.rc"
MAGE_PROFILE="$(basename $(readlink /etc/mage-profile))"
#TOOLCHAIN=toolchain
BASESYSTEM=livecdsystem26
CDID="mlcd-$(date +%F-%k%M)"

# global profile pathes
PROFILES_DIR="/usr/lib/mlivecdbuild/profiles"
GLOBAL_PROFILE="${PROFILES_DIR}/global"

die() { echo "ERROR: $@"; exit 1; }

# get_profile $filename
get_profile()
{
	local file="$1"

	if [[ -z ${file} ]] || [[ ! -e ${PROFILES_DIR}/${CDPROFILE}/${file} ]]
	then
		echo "${GLOBAL_PROFILE}/${file}"
		return 0
	fi

	echo "${PROFILES_DIR}/${CDPROFILE}/${file}"
}

bootstrap_system()
{
	mage-bootstrap \
		--root "${CDCHROOTDIR}" \
		--magerc "${MAGERC}" \
		--profile "${MAGE_PROFILE}" \
		--toolchain "${TOOLCHAIN}" \
		--basesystem "${BASESYSTEM}" \
		|| die "bootstrapping toolchain"
}

custom_services()
{
	local cmd="$1"
	local services="$2"

	case ${cmd} in
		add|del|default)
			# add given services from profile
			if [[ -n ${services} ]]
			then
				local i
				for i in ${services}
				do
					MROOT="${CDCHROOTDIR}" rc-config ${cmd} ${i} || die "rc ${cmd} ${i}"
				done
			fi
			;;
		*) die "custom_services: unkown command ${cmd}";;
	esac
}

add_initrc()
{
	local var="$1"

	# sanity checks
	[[ -z ${CDCHROOTDIR} ]] && die "\$CDCHROOTDIR not given."
	echo "${var}" >> ${CDCHROOTDIR}/.installrc || die "add_initrc() adding \$var"
}

custom_packages()
{
	local cmd="$1"
	local packages="$2"

	case ${cmd} in
		install|uninstall)
			if [[ -n ${packages} ]]
			then
				:> ${CDCHROOTDIR}/.installrc

				# respect proxies
				[[ -n ${http_proxy} ]] && add_initrc "export http_proxy=${http_proxy}"
				[[ -n ${ftp_proxy} ]] && add_initrc "export ftp_proxy=${ftp_proxy}"
				[[ -n ${no_proxy} ]] && add_initrc "export no_proxy=${no_proxy}"

				# do not auto-start any services!
				add_initrc "export MAGE_BOOTSTRAP=true"

				local i
				for i in ${packages}
				do
					add_initrc "mage ${cmd} ${i}"
				done
				add_initrc "mage clean"
		
				# now run the .installrc script
				mount -t proc proc ${CDCHROOTDIR}/proc || die "mount proc"
				mount -t sysfs sysfs ${CDCHROOTDIR}/sys || die "mount sys"
				mount -o bind /dev ${CDCHROOTDIR}/dev || die "mount dev"
				chroot ${CDCHROOTDIR} /bin/bash -i /.installrc || die "chr00ting"
				umount ${CDCHROOTDIR}/dev || die "umount dev"
				umount ${CDCHROOTDIR}/proc || die "umount proc"
				umount ${CDCHROOTDIR}/sys || die "umount sys"
				[ -f ${CDCHROOTDIR}/.installrc ] && rm ${CDCHROOTDIR}/.installrc
			fi
			;;
		*) die "custom_packages: unkown command ${cmd}";;
	esac
}

read_config()
{
	local file="$1"
	local pkg

	( cat ${file}; echo ) | while read pkg
	do
		case "${pkg}" in
			\#*|"") continue ;;
		esac
		echo "${pkg}"
	done
}

prepare_iso()
{
	echo Preparing LiveCD ISO Image ...

	# fixes some issues with xfree/xorg xkb
	if [[ -L ${CDCHROOTDIR}/etc/X11/xkb ]] && [[ -d /usr/X11R6/lib/X11/xkb ]]
	then
		rm ${CDCHROOTDIR}/etc/X11/xkb || die
		mv ${CDCHROOTDIR}/usr/X11R6/lib/X11/xkb ${CDCHROOTDIR}/etc/X11 || die
		ln -s ../../../../etc/X11/xkb ${CDCHROOTDIR}/usr/X11R6/lib/X11/xkb || die
	fi

	install -m 0644 $(get_profile inittab) ${CDCHROOTDIR}/etc/inittab || die
	install -m 0644 $(get_profile fstab) ${CDCHROOTDIR}/etc/fstab || die
	install -m 0644 $(get_profile motd) ${CDCHROOTDIR}/etc/motd || die
	install -m 0644 $(get_profile issue) ${CDCHROOTDIR}/etc/issue || die
	install -m 0644 $(get_profile net.eth0) ${CDCHROOTDIR}/etc/conf.d/net.eth0 || die
	echo "${CDHOSTNAME}" > ${CDCHROOTDIR}/etc/hostname || die

	echo Setting up services ...

	# add hardware detection
	MROOT="${CDCHROOTDIR}" rc-config add hwdetect || die "rc add hwdetect"

	# del checkfs
	MROOT="${CDCHROOTDIR}" rc-config del checkfs || die "rc del checkfs"

	# add custom packages
	[[ -n ${ADD_PACKAGES} ]] && custom_packages install "${ADD_PACKAGES}"
	[ -f $(get_profile add_packages) ] && custom_packages install "$(read_config $(get_profile add_packages))"

	# del custom packages
	[[ -n ${DEL_PACKAGES} ]] && custom_packages uninstall "${DEL_PACKAGES}"
	[ -f $(get_profile del_packages) ] && custom_packages uninstall "$(read_config $(get_profile del_packages))"

	# add given services from profile
	[[ -n ${ADD_SERVICES} ]] && custom_services add "${ADD_SERVICES}"
	[ -f $(get_profile add_services) ] && custom_services add "$(read_config $(get_profile add_services))"

	# del given services from profile
	[[ -n ${DEL_SERVICES} ]] && custom_services del "${DEL_SERVICES}"
	[ -f $(get_profile del_services) ] && custom_services del "$(read_config $(get_profile del_services))"

	# setup default runlevel
	[[ -n ${DEFAULT_RUNLEVEL} ]] && custom_services default "${DEFAULT_RUNLEVEL}"

	if [ -f $(get_profile prepare_custom) ]
	then
		echo Running custom user script ...
		source $(get_profile prepare_custom) || die "running custom user script"
	fi

	echo Cleaning unwanted files ...
	:> ${CDCHROOTDIR}/etc/mtab || die "whiping /etc/mtab"
	[ -f ${CDCHROOTDIR}/root/.bash_history ] && rm ${CDCHROOTDIR}/root/.bash_history
}

generate_rootfs()
{
	echo Generating squashfs compressed rootfs loopfile ...
	mksquashfs ${CDCHROOTDIR} ${LIVECDROOT}/livecdrootfs.sqsh #>/dev/null 2>&1

	echo Moving rootfs loopfile to isoroot ...
	install -d ${CDISOROOT}
	mv ${LIVECDROOT}/livecdrootfs.sqsh ${CDISOROOT}
}

install_bootloader()
{
	echo Installing Bootloader ...

	# iso linux binary
	install -d ${CDISOROOT}/isolinux
	install /usr/lib/mkinitrd/isolinux.bin ${CDISOROOT}/isolinux || die

	# kernel
	local kimg="$(basename $(readlink ${CDCHROOTDIR}/boot/vmlinuz))"
	install ${CDCHROOTDIR}/boot/${kimg} ${CDISOROOT}/isolinux/${CDKERNELNAME} || die

	install -m 0644 $(get_profile isolinux.cfg) ${CDISOROOT}/isolinux || die
	install -m 0644 $(get_profile boot.lss) ${CDISOROOT}/isolinux || die
	install -m 0644 $(get_profile boot.msg) ${CDISOROOT}/isolinux || die
	install -m 0644 $(get_profile help.msg) ${CDISOROOT}/isolinux || die

	return 0
}

generate_initrd()
{
	local CHROOTSH="$(mktemp -p ${CDCHROOTDIR})"

	echo Generating initrd image ...
	echo '#!/bin/bash' > ${CHROOTSH} || die
	echo 'kv="$(readlink /boot/vmlinuz)"' >> ${CHROOTSH} || die
	echo 'kv="${kv/kernel-}/"' >> ${CHROOTSH} || die
	echo 'mkinitrd-livecd -f --initramfs /initrd.gz ${kv}' >> ${CHROOTSH} || die
	chmod +x ${CHROOTSH} || die
	chroot ${CDCHROOTDIR} /$(basename ${CHROOTSH}) || die
	[[ -f ${CHROOTSH} ]] && rm ${CHROOTSH} || die

	# move initrd to isoroot
	install -d ${CDISOROOT}/isolinux || die
	mv ${CDCHROOTDIR}/initrd.gz ${CDISOROOT}/isolinux || die
}

generate_iso()
{
	echo Generating ISO Image ...
	install -d ${CDISOROOT}
	pushd ${CDISOROOT} &&
		mkisofs -rock \
			-full-iso9660-filenames \
			-allow-leading-dots \
			-disable-deep-relocation \
			-output ${LIVECDROOT}/${CDISONAME} \
			-eltorito-boot isolinux/isolinux.bin \
			-eltorito-catalog isolinux/boot.cat \
			-no-emul-boot \
			-boot-load-size 4 \
			-boot-info-table \
			-volid "${CDID}" \
			${CDISOROOT} #&> /dev/null
	popd
}

enter_livecd()
{
	mount -t sysfs sysfs ${CDCHROOTDIR}/sys
	mount -t proc proc ${CDCHROOTDIR}/proc
	mount -o bind /dev ${CDCHROOTDIR}/dev

	chroot ${CDCHROOTDIR} \
		/usr/bin/env -i \
			HOME=/root \
			TERM=$TERM PS1='\u:\w\$ ' \
			PATH=/bin:/usr/bin:/sbin:/usr/sbin \
			http_proxy=${http_proxy} \
			ftp_proxy=${ftp_proxy} \
			no_proxy=${no_proxy} \
		/bin/bash -i

	umount ${CDCHROOTDIR}/dev
	umount ${CDCHROOTDIR}/proc
	umount ${CDCHROOTDIR}/sys
}

usage()
{
	echo "Usage: $(basename $0) [command] [profile] ..."
	echo
	echo "Commands:"
	echo "    bootstrap  - bootstraps the rootfs for the livecd"
	echo "    prepare    - prepare the rootfs to run from a livecd"
	echo "    rootfs     - generates the squashfs rootfs image"
	echo "    initrd     - generates a livecd suitable initrd"
	echo "    bootloader - installs the bootloader"
	echo "    isogen     - generate the final livecd iso image"
	echo "    all        - runs all tasks to get a livecd from zero"
	echo "    enter      - enters the rootfs of a livecd"
	echo
	echo "    for example '$(basename $0) all default'"
	echo "    creates a livecd using the default profile."
	echo
}

# source profile config - overrides all other vars
[ -f ${PROFILES_DIR}/${CDPROFILE}/config ] && . ${PROFILES_DIR}/${CDPROFILE}/config

case $1 in
	bootstrap) bootstrap_system;;
	prepare) prepare_iso;;
	rootfs) generate_rootfs;;
	initrd) generate_initrd;;
	bootloader) install_bootloader;;
	isogen) generate_iso;;
	all)
		bootstrap_system
		prepare_iso
		generate_rootfs
		generate_initrd
		install_bootloader
		generate_iso
		;;
	enter) enter_livecd ;;
	*|'') usage;;
esac