#!/bin/bash

DISTRO="voyage"
MOUNT_PROC_SH=/usr/local/sbin/mount-proc.sh

# define squashfs options here
export MKSQUASHFS_OPTIONS="-b 1048576 -comp xz"
#export MKSQUASHFS_OPTIONS="-comp xz"

# define linux packages here for different editions
#VOYAGE_LINUX_PACKAGES="linux-image-3.16.7-ckt9"
VOYAGE_LINUX_PACKAGES="linux-image-4.1.6"
ONE_LINUX_PACKAGES="linux-image-3.14.12 dahdi-modules-3.14.12"
#MPD_LINUX_PACKAGES="linux-image-3.14.12"
MPD_LINUX_PACKAGES="linux-image-4.1.6"

if [ $(uname -m) == "x86_64" ] ; then
	ARCH="_amd64"
	lb config -a amd64
fi


Chroot ()
{
    # Execute commands in chroot
    chroot "${1}" /usr/bin/env -i HOME="/root" DEBIAN_FRONTEND="noninteractive" \
        TERM="${TERM}" PATH="/usr/sbin:/usr/local/sbin:/usr/bin:/sbin:/bin" \
        ftp_proxy="${LIVE_FTPPROXY}" http_proxy="${LIVE_HTTPPROXY}" ${2}
}

Chroot_MountProc ()
{
    if [ -f ${1}/${MOUNT_PROC_SH} ] ;
    then
        # Execute commands in chroot
        chroot "${1}" /usr/bin/env -i HOME="/root" DEBIAN_FRONTEND="noninteractive" \
			 TERM="${TERM}" PATH="/usr/sbin:/usr/local/sbin:/usr/bin:/sbin:/bin" \
            ftp_proxy="${LIVE_FTPPROXY}" http_proxy="${LIVE_HTTPPROXY}" ${MOUNT_PROC_SH} ${2}
    else
        echo "No ${MOUNT_PROC_SH}, call Chroot() instead"
        Chroot "${1}" "${2}"
    fi
}

BuildTar()
{
	lb clean
	lb config -b tar --chroot-filesystem plain --linux-packages="$VOYAGE_LINUX_PACKAGES"
	lb build

	Chroot_MountProc binary/live/filesystem.dir "apt-get -y remove --purge busybox live-initramfs"
	Chroot_MountProc binary/live/filesystem.dir "apt-get -y autoremove --purge"
	rm -f binary/live/filesystem.dir/boot/*.bak
	
	if [ -d binary/live/filesystem.dir ] ; then
		mv binary/live/filesystem.dir binary/live/$DISTRO-current$ARCH
		#tar -jcf $DISTRO-current$ARCH.tar.bz2 -C binary/live/	$DISTRO-current$ARCH/. 
		tar -Jcf $DISTRO-current$ARCH.tar.xz -C binary/live/	$DISTRO-current$ARCH/. 
		mv binary/live/$DISTRO-current$ARCH binary/live/filesystem.dir
	else
		echo "binary/live/filesystem.dir not found!"
	fi
}

BuildImg()
{
	lb clean
	lb config -b usb-hdd --binary-filesystem fat16 --chroot-filesystem squashfs --linux-packages="$VOYAGE_LINUX_PACKAGES"
	lb build

	if [ -f binary.img ] ; then
		mv binary.img $DISTRO-current$ARCH.img
	else
		echo "binary.img not found!"
	fi
}

BuildISO()
{
	lb clean
	lb config -b iso-hybrid --chroot-filesystem squashfs --linux-packages="$VOYAGE_LINUX_PACKAGES"
	lb build
	
	if [ -f binary.iso ] ; then
		mv binary.iso $DISTRO-current$ARCH.iso
	elif [ -f binary.hybrid.iso ] ; then
            mv binary.hybrid.iso $DISTRO-current$ARCH.iso
	elif [ -f live-image*.hybrid.iso ] ; then
            mv live-image*.hybrid.iso $DISTRO-current$ARCH.iso
        else
            echo "binary.iso not found!"
	fi
}

BuildSDK()
{
	lb clean
	lb config -b iso-hybrid --chroot-filesystem squashfs --linux-packages="$VOYAGE_LINUX_PACKAGES"
	lb build
	
	if [ -f binary.iso ] ; then
		mv binary.iso $DISTRO-sdk-current$ARCH.iso
        elif [ -f binary.hybrid.iso ] ; then
        	mv binary.hybrid.iso $DISTRO-sdk-current$ARCH.iso
        elif [ -f live-image*.hybrid.iso ] ; then
        	mv live-image*hybrid.iso $DISTRO-sdk-current$ARCH.iso
	else
		echo "binary.iso not found!"
	fi
}

#
# $1 - package list name (e.g. voyage-one-cd)
# $2 - distro name (e.g. one, mpd)
# $3 - linux packages 
#
BuildDistro()
{
	lb clean
	lb config -b tar --chroot-filesystem plain --linux-packages="$3"
	lb build

	Chroot_MountProc binary/live/filesystem.dir "apt-get -y remove --purge busybox live-initramfs"
	Chroot_MountProc binary/live/filesystem.dir "apt-get -y autoremove --purge"
	rm binary/live/filesystem.dir/boot/*.bak
	
	if [ -d binary/live/filesystem.dir ] ; then
		mv binary/live/filesystem.dir binary/live/$DISTRO-$2-current$ARCH
		#tar -jcf $DISTRO-$2-current$ARCH.tar.bz2 -C binary/live/	$DISTRO-$2-current$ARCH/. 
		tar -Jcf $DISTRO-$2-current$ARCH.tar.xz -C binary/live/	$DISTRO-$2-current$ARCH/. 
		mv binary/live/$DISTRO-$2-current$ARCH binary/live/filesystem.dir
	else
		echo "binary/live/filesystem.dir not found!"
	fi
}

#
# $1 - package list name (e.g. voyage-one-cd)
# $2 - distro name (e.g. one, mpd)
# $3 - linux packages
#
BuildHybrid()
{
	lb clean
	lb config -b iso-hybrid --chroot-filesystem squashfs  --linux-packages="$3"
	lb build
	
	if [ -f binary.iso ] ; then
		mv binary.iso $DISTRO-$2-current$ARCH.iso
	elif [ -f binary.hybrid.iso ] ; then
		mv binary.hybrid.iso $DISTRO-$2-current$ARCH.iso
	elif [ -f live-image*.hybrid.iso ] ; then
		mv live-image*.hybrid.iso $DISTRO-$2-current$ARCH.iso
	else
		echo "binary.iso not found!"
	fi
}

#
# $1 - package list name (e.g. voyage-one-cd)
# $2 - distro name (e.g. one, mpd)
# $3 - linux packages
#
BuildCD()
{
	lb clean
	lb config -b iso --chroot-filesystem squashfs --linux-packages="$3"
	lb build
	
	if [ -f binary.iso ] ; then
		mv binary.iso $DISTRO-$2-current$ARCH.iso
	else
		echo "binary.iso not found!"
	fi
}

Banner()
{
	echo "########################################################"
	echo "# Build $1"
	echo "########################################################"
}

#
# $1 - local package list to use by the distro
#
PreparePackageList()
{
	rm -f config/chroot_local-packageslists/*.list
	rm -f config/package-lists/*.list.chroot

	LISTS=`echo "$1" | sed -e "s/ /\n/g"`
	for LIST in $LISTS ; do
		#echo "cp -p config/chroot_local-packageslists/$LIST config/chroot_local-packageslists/$LIST.list"
		#cp -p config/chroot_local-packageslists/$LIST config/chroot_local-packageslists/$LIST.list

		echo "cp -p config/package-lists/$LIST config/package-lists/$LIST.list.chroot"
		cp -p config/package-lists/$LIST config/package-lists/$LIST.list.chroot
	done
	
}

if [ -n "$http_proxy" ] ; then
	lb config --apt-http-proxy "$http_proxy"
	export http_proxy
	Banner "Setting HTTP_PROXY to $http_proxy"
else
	lb config --apt-http-proxy ""
fi

## Set default kernel flavour to "voyage"
lb config -k "voyage"

for TYPE in $1; do
	case "$TYPE" in
		img)
			Banner "Voyage Linux Image"
			PreparePackageList "voyage voyage-cd"
			BuildImg
		;;
		tar)
			Banner "Voyage Linux Tarball"
			PreparePackageList "voyage"
			BuildTar
		;;
		iso)
			Banner "Voyage Linux Live CD"
			PreparePackageList "voyage voyage-cd"
			BuildISO
		;;
		sdk)
			Banner "Voyage SDK Live CD"
			PreparePackageList "voyage voyage-cd voyage-sdk"
			BuildSDK
		;;
		onecd)
			Banner "Voyage ONE Live CD"
			PreparePackageList "voyage voyage-cd one"
			BuildHybrid voyage-one-cd one "$ONE_LINUX_PACKAGES"
		;;
		one)
			Banner "Voyage ONE Tarball"
			PreparePackageList "voyage one"
			BuildDistro voyage-one one "$ONE_LINUX_PACKAGES"
		;;
		mpdcd)
			Banner "Voyage MPD Live CD"
			PreparePackageList "voyage voyage-cd mpd"
			lb config -k "voyage-rt"
			BuildHybrid voyage-mpd-cd mpd "$MPD_LINUX_PACKAGES"
		;;
		mpd)
			Banner "Voyage MPD Tarball"
			PreparePackageList "voyage mpd"
			lb config -k "voyage-rt"
			BuildDistro voyage-mpd mpd "$MPD_LINUX_PACKAGES"
		;;
#		mpdse2012)
#			Banner "Voyage MPD SE2012 Tarball"
#			PreparePackageList "voyage mpdse2012"
#			BuildDistro voyage-mpdse2012 mpdse2012 "$MPD_LINUX_PACKAGES"
#		;;
		test)
			Chroot_MountProc binary/live/filesystem.dir "apt-get -y remove busybox live-initramfs"
			Chroot_MountProc binary/live/filesystem.dir "apt-get -y autoremove"
		;;
		clean)
			lb clean --purge
		;;
		*)
			echo "unknown build type $TYPE"
		;;
	esac
done