#!/usr/bin/env bash

# import gentoo functions
source /lib/gentoo/functions.sh

# generic variables

export local releasename="redcore"
export local releasetarget="desktop"
export local releaseversion="1"
export local kernelver="4.8.5-"$releasename""
export local chrootx64=""$releasename"_"$releaseversion"_core_x64.squashfs"
export local chrootx64md5=""$chrootx64".md5"

# build variables

export local rodir="rodir"
export local rwdir="rwdir"
export local workdir="workdir" # only needed for overlayfs on kernel greather than 3.18
export local overlaydir="overlaydir"
export local chrootuser="root"
export local chroottarget="${@:2}"
export local chrootbuildtarget="emerge -kav "$chroottarget""
export local chrootdevtree="vasile --srcmode"

# iso variables

export local bootcorepath="/tmp/bootcore"
export local bootcorefiles="/tmp/bootcore/cdroot/boot"
export local isouser="root"
export local isousertree="vasile --binmode"
export local isodracutcmd="dracut -N -a dmsquash-live -a pollcdrom -o systemd -o systemd-initrd -o systemd-networkd -o dracut-systemd --force --kver "$kernelver""
export local isodracutrd="initramfs-"$kernelver".img"
export local isoeficmd="grub2-mkimage -d /usr/lib64/grub/x86_64-efi -o bootx64.efi -O x86_64-efi ext2 fat udf btrfs ntfs reiserfs xfs hfsplus lvm ata part_msdos part_gpt part_apple bsd search_fs_uuid normal chain iso9660 configfile help loadenv reboot cat search memdisk tar boot linux chain"
export local isochainloadcmd="grub2-mkimage -d /usr/lib64/grub/i386-pc -o core.img -O i386-pc biosdisk part_msdos fat"
export local isomainarch="x86_64"
export local isorootdir=""$releasename"-"$releasetarget"-"$releaseversion"-"$isomainarch""
export local isokernelname="kernel-genkernel-"$isomainarch"-"$kernelver""
export local isostagingdir="/tmp/"$isorootdir""
export local isosupersquashdir=""$isorootdir"/CDroot"
export local isolivesqfsdir=""$isosupersquashdir"/LiveOS"
export local isoliveosdir=""$isorootdir"/LiveOS"
export local isosyncdir=""$isorootdir"/rootfs"
export local isobootdir=""$isorootdir"/boot/grub"
export local isoefibootdir=""$isorootdir"/efi/boot"

# modechange variables

export local portagedir="/usr/portage"
export local confdir="/etc/portage"
export local gitdir="/opt/redcore-build/conf/intel/portage"
export local clonedir="/opt/redcore-build"

# generic functions (used for both build && iso creation)

kernelconfig () {
	if [[ $(zgrep 'CONFIG_OVERLAY_FS=' /proc/config.gz) && $(zgrep "CONFIG_SQUASHFS=" /proc/config.gz) &&  $(zgrep "CONFIG_BLK_DEV_LOOP=" /proc/config.gz) ]] ; then
		einfo "Kernel configuration seems OK, moving on"
		sleep 1
	else
		eerror "Vasile needs OVERLAYFS && SQUASHFS && LOOP DEVICES to work"
		eerror "Please rebuild the kernel with those activated to use it"
		exit 1
	fi
}

checkroot () {
	if [[ "$(whoami)" != root ]] ; then
		eerror "You're not root?...No cookies for you, go away !!!"
		exit 1
	fi
}

checkiflive () {
	if [[ -L /dev/mapper/live-base ]] ; then
		eerror "Running in Live mode is unsupported"
		exit 1
	fi
}

chrootchecksum () {
	while : true ; do
		if [[ -f "$chrootx64" && -f "$chrootx64md5" ]] ; then
			einfo "Squashed chroot && Checksum file found"
			if [[ "$(md5sum -c "$chrootx64md5")" ]] ; then
				einfo "Squashed chroot checksum passed"
				sleep 1
				break
			else
				eerror "Squashed chroot checksum failed"
				exit 1
			fi
		else
			eerror "Squashed chroot || Checksum file not found"
			exit 1
		fi
	done
}

# build functions

chrootprepare () {
	while : true ; do
		if [[ ! -d "$rodir" && ! -d "$rwdir" && ! -d "$workdir" && ! -d "$overlaydir" ]] ; then
			for i in "$rodir" "$rwdir" "$workdir" "$overlaydir" ; do
				mkdir "$i"
			done
			chrootstart
			break
		elif [[ -d "$rodir" && -d "$rwdir" && -d "$workdir" && -d "$overlaydir" ]] ; then
			chrootstop
			for i in "$rodir" "$rwdir" "$workdir" "$overlaydir" ; do
				rm -rf "$i"
			done
			continue
		fi
	done

}

chrootstop () {
	umount -l "$overlaydir"/proc > /dev/null 2>&1
	umount -l "$overlaydir"/sys > /dev/null 2>&1
	umount -l "$overlaydir"/dev/pts > /dev/null 2>&1
	umount -l "$overlaydir"/dev/shm > /dev/null 2>&1
	umount -l "$overlaydir"/dev > /dev/null 2>&1
	umount -l "$overlaydir"/tmp > /dev/null 2>&1
	umount -l "$overlaydir"/usr/portage/packages > /dev/null 2>&1
	umount -l "$overlaydir"/usr/portage/distfiles > /dev/null 2>&1
	umount -l "$overlaydir" > /dev/null 2>&1
	umount -l "$rodir" > /dev/null 2>&1
}

chrootstart () {
	mount -t squashfs "$chrootx64" "$rodir"
	mount -t overlay -o lowerdir="$rodir",upperdir="$rwdir",workdir="$workdir" overlay "$overlaydir"
	mount -o bind packages "$overlaydir"/usr/portage/packages
	mount -o bind distfiles "$overlaydir"/usr/portage/distfiles
	mount -t proc proc "$overlaydir"/proc
	mount -t sysfs sysfs "$overlaydir"/sys
	mount -t devtmpfs -o relatime,size=3055348k,nr_inodes=763837,mode=755 none "$overlaydir"/dev
	mount -t devpts -o nosuid,noexec,relatime,gid=5,mode=620 none "$overlaydir"/dev/pts
	mount -t tmpfs -o nosuid,nodev none "$overlaydir"/dev/shm
	mount -t tmpfs -o nosuid,nodev,noexec none  "$overlaydir"/tmp
}

chrootdevtree () {
	chroot "$overlaydir" su - "$chrootuser" -c "$chrootdevtree"
}

chrootbuild () {
	einfo "All systems are go!"
	sleep 1
	chroot "$overlaydir" su - "$chrootuser" -c "$chrootbuildtarget"
}

chrootx64 () {
	einfo "DROPPING YOU TO A ROOT SHELL INTO BUILD ENVIRONMENT"
	einfo "FIX FAILURES && ADJUST PORTAGE CONFIGURATION FILES"
	einfo "TEST USEFLAGS && KEYWORDS && MASKS IF REQUIRED"
	einfo "IN THE END : DO NOT FORGET TO COMMIT YOUR CHANGES"
	chroot "$overlaydir" su - "$chrootuser"
}

makepkg () {
	chrootchecksum
	chrootprepare
	chrootdevtree
	chrootbuild
	chrootx64
	chrootstop
}

# iso functions

isostart () {
	mount -o bind packages "$isosyncdir"/usr/portage/packages
	mount -t proc proc "$isosyncdir"/proc
	mount -t sysfs sysfs "$isosyncdir"/sys
	mount -t devtmpfs -o relatime,size=3055348k,nr_inodes=763837,mode=755 none "$isosyncdir"/dev
	mount -t devpts -o nosuid,noexec,relatime,gid=5,mode=620 none "$isosyncdir"/dev/pts
	mount -t tmpfs -o nosuid,nodev none "$isosyncdir"/dev/shm
	mount -t tmpfs -o nosuid,nodev,noexec none  "$isosyncdir"/tmp
}

isostop () {
	umount -l "$isosyncdir"/proc > /dev/null 2>&1
	umount -l "$isosyncdir"/sys > /dev/null 2>&1
	umount -l "$isosyncdir"/dev/pts > /dev/null 2>&1
	umount -l "$isosyncdir"/dev/shm > /dev/null 2>&1
	umount -l "$isosyncdir"/dev > /dev/null 2>&1
	umount -l "$isosyncdir"/tmp > /dev/null 2>&1
	umount -l "$isosyncdir"/usr/portage/packages > /dev/null 2>&1
}

isousertree () {
	chroot "$isosyncdir" su - "$isouser" -c "$isousertree"
}

isoramfs () {
	chroot "$isosyncdir" su - "$isouser" -c "$isodracutcmd"
}

isoefiimg () {
	chroot "$isosyncdir" su - "$isouser" -c "$isoeficmd"
}

isochainload () {
	chroot "$isosyncdir" su - "$isouser" -c "$isochainloadcmd"
}

isoservices () {
	for service in avahi-daemon cups connman dkms dm-event redcorelive lightdm lvm2-lvmetad lvm2-monitor NetworkManager ModemManager virtualbox-guest-additions ; do
		chroot "$isosyncdir" su - "$isouser" -c "systemctl enable "$service""
	done
}

isochroot () {
	einfo "DROPPING YOU TO A ROOT SHELL INTO ISO ENVIRONMENT"
	einfo "INSTALL APPS & TARGETS & DES YOU WANT INSIDE ISO"
	einfo "EXIT WHEN YOU'RE DONE AND I'LL CREATE THE IMAGE"
	chroot "$isosyncdir" su - "$isouser"
}

prepareiso () {
	mkdir -p "$isosyncdir"
	dd if=/dev/zero of=""$isosyncdir".img" bs=50M count=400
	mkfs.ext4 ""$isosyncdir".img"
	mkdir -p "$isostagingdir"
	mkdir -p "$isobootdir"
	mkdir -p "$isoefibootdir"
	mkdir -p "$isoliveosdir"
	mkdir -p "$isosupersquashdir"
	mkdir -p "$isolivesqfsdir"
	mount -t squashfs "$chrootx64" "$isostagingdir"
	mount -t ext4 ""$isosyncdir".img" "$isosyncdir"
	rsync -aHAXr --progress "$isostagingdir/" "$isosyncdir/"
	cp -avx ""$isosyncdir"/boot/"$isokernelname"" ""$isorootdir"/boot/vmlinuz"
	isostart
	isoramfs
	isostop
	mv ""$isosyncdir"/boot/"$isodracutrd"" ""$isorootdir"/boot/initrd"
	isostart
	isoefiimg
	isostop
	mv ""$isosyncdir"/root/bootx64.efi" "$isoefibootdir"
	chmod 755 ""$isoefibootdir"/bootx64.efi"
	isostart
	isochainload
	isostop
	mv ""$isosyncdir"/root/core.img" "$isobootdir"
	cp -avx ""$isosyncdir"/usr/lib64/grub/i386-pc/lnxboot.img" "$isobootdir"
	isostart
	isousertree
	isochroot
	isostop
	isostart
	isoservices
	isostop
	umount -l "$isosyncdir" > /dev/null 2>&1
	mv ""$isosyncdir".img"  "$isolivesqfsdir"
	mksquashfs "$isosupersquashdir" ""$isorootdir"/squashfs.img" -b 1048576 -comp xz -Xdict-size 100%
	mv ""$isorootdir/"squashfs.img" "$isoliveosdir"
}

bootcoreiso () {
	git clone https://gitlab.com/"$releasename"/boot-core.git "$bootcorepath"
	cp -avx "$bootcorefiles" "$isorootdir"
}

cleanupiso () {
	umount "$isostagingdir"
	rm -rf "$isosyncdir"
	rm -rf "$isostagingdir"
	rm -rf "$isosupersquashdir"
	rm -rf "$bootcorepath"
}

makeisoimg () {
	grub2-mkrescue -o ""$releasename"-"$releasetarget"-"$releaseversion"-"$isomainarch".iso" "$isorootdir"
}

makeiso () {
	chrootchecksum
	prepareiso
	bootcoreiso
	cleanupiso
	makeisoimg
}

# modechange functions

removeportagetree() {
	if [ -d ""$portagedir"/.git" ] ; then
		einfo "Removing portage tree"
		find "$portagedir" -mindepth 1 -name "packages" -prune -o -name "distfiles" -prune -o -exec rm -rf {} \; > /dev/null 2>&1
	fi
}

removeoverlays() {
	einfo "Removing "$releasename"-"$releasetarget" overlay"
	layman -d "$releasename"-"$releasetarget"
}

removeportageconfig() {
	einfo "Removing "$releasename" buildsystem"
	rm ""$confdir"/make.conf" > /dev/null 2>&1
	rm ""$confdir"/make.profile" > /dev/null 2>&1
	rm "$confdir" > /dev/null 2>&1
	rm -rf "$clonedir" > /dev/null 2>&1
}

resetmode () {
	removeportagetree
	removeoverlays
	removeportageconfig
}

fetchportageconfig() {
	pushd /opt > /dev/null 2>&1
	einfo "Injecting "$releasename" buildsystem"
	git clone https://gitlab.com/redcore/redcore-build.git
	popd > /dev/null 2>&1
}

fetchfullportagetree() {
	if [[ ! -d "$portagedir"/.git ]] ; then
		einfo "Injecting mixedmode/srcmode (full) portage tree"
		cd "$portagedir" && git init > /dev/null 2>&1
		git remote add origin https://gitlab.com/redcore/portage.git
		git pull --depth=1 origin master
		git branch -u origin/master master
		rm -rf ""$portagedir"/profiles/updates"
	fi
}

fetchminimalportagetree() {
	if [[ ! -d "$portagedir"/.git ]] ; then
		einfo "Injecting binmode (minimal) portage tree"
		cd "$portagedir" && git init > /dev/null 2>&1
		git remote add origin https://gitlab.com/redcore/portage.git
		git config core.sparsecheckout true
		echo "profiles/*" >> .git/info/sparse-checkout
		echo "metadata/*" >> .git/info/sparse-checkout
		echo "eclass/*" >> .git/info/sparse-checkout
		echo ".gitignore" >> .git/info/sparse-checkout
		git pull --depth=1 origin master
		git branch -u origin/master master
		rm -rf ""$portagedir"/profiles/updates"
	fi
}

fetchoverlays () {
	einfo "Injecting "$releasename"-"$releasetarget" overlay"
	layman -f -a "$releasename"-"$releasetarget" -o https://gitlab.com/"$releasename"/"$releasename"-"$releasetarget"/raw/master/overlay.xml
}

setbinmodeconfig() {
	ln -sf "$gitdir" "$confdir"
	ln -sf "$confdir"/make.conf.amd64-binmode "$confdir"/make.conf
	eselect profile set 1
	env-update
	. /etc/profile
}

setmixedmodeconfig() {
	ln -sf "$gitdir" "$confdir"
	ln -sf "$confdir"/make.conf.amd64-mixedmode "$confdir"/make.conf
	eselect profile set 1
	env-update
	. /etc/profile
}

setsrcmodeconfig () {
	ln -sf "$gitdir" "$confdir"
	ln -sf "$confdir"/make.conf.amd64-srcmode "$confdir"/make.conf
	eselect profile set 1
	env-update
	. /etc/profile
}

binmode() {
	resetmode
	fetchoverlays
	fetchportageconfig
	fetchminimalportagetree
	setbinmodeconfig
}

mixedmode() {
	resetmode
	fetchoverlays
	fetchportageconfig
	fetchfullportagetree
	setmixedconfig
}

srcmode() {
	resetmode
	fetchoverlays
	fetchportageconfig
	fetchfullportagetree
	setsrcmodeconfig
}