#!/bin/bash

# create_initrd_loop(size)
create_initrd_loop() {
	[ "$#" -ne "1" ] && gen_die "invalid use of create_initrd_loop"
	mkdir -p ${TEMP}/initrd-temp || gen_die "could not create loopback dir"
	dd if=/dev/zero of=${TEMP}/initrd-loop bs=1k count=${1} >> "${DEBUGFILE}" 2>&1 || gen_die "could not zero initrd-loop"
	mke2fs -F -q -N${1} "${TEMP}/initrd-loop" >> "${DEBUGFILE}" 2>&1 || gen_die "could not format initrd-loop"
	mount -t ext2 -o loop "${TEMP}/initrd-loop" "${TEMP}/initrd-temp" >> "${DEBUGFILE}" 2>&1 || gen_die "could not mount initrd filesystem"
}

create_initrd_unmount_loop()
{
	cd ${TEMP}
	umount "${TEMP}/initrd-temp" || gen_die "could not unmount initrd system"
}

create_base_initrd_sys() {
	mkdir -p ${TEMP}/initrd-temp/dev
	mkdir -p ${TEMP}/initrd-temp/bin
	mkdir -p ${TEMP}/initrd-temp/etc
	mkdir -p ${TEMP}/initrd-temp/usr
	mkdir -p ${TEMP}/initrd-temp/proc
	mkdir -p ${TEMP}/initrd-temp/temp
	mkdir -p ${TEMP}/initrd-temp/.initrd
	mkdir -p ${TEMP}/initrd-temp/new_root
	mkdir -p ${TEMP}/initrd-temp/keymaps
	ln -s bin ${TEMP}/initrd-temp/sbin
	ln -s ../bin ${TEMP}/initrd-temp/usr/bin
	ln -s ../bin ${TEMP}/initrd-temp/usr/sbin
	echo "/dev/ram0     /           ext2    defaults" > ${TEMP}/initrd-temp/etc/fstab
	echo "proc          /proc       proc    defaults    0 0" >> ${TEMP}/initrd-temp/etc/fstab

	cd ${TEMP}/initrd-temp/dev
	MAKEDEV generic-i386
	MAKEDEV scd

	cp "${BUSYBOX_BINCACHE}" "${TEMP}/initrd-temp/bin/busybox.bz2" || gen_die "could not copy busybox from bincache"
	bunzip2 "${TEMP}/initrd-temp/bin/busybox.bz2" || gen_die "could not uncompress busybox"

	if [ "${PAT}" -gt "4" ]
	then
		cp "${MODULE_INIT_TOOLS_BINCACHE}" "${TEMP}/initrd-temp/bin/insmod.static.bz2" || gen_die "could not copy insmod.static from bincache"
	else
		cp "${MODUTILS_BINCACHE}" "${TEMP}/initrd-temp/bin/insmod.static.bz2" || gen_die "could not copy insmod.static from bincache"
	fi

	bunzip2 "${TEMP}/initrd-temp/bin/insmod.static.bz2" || gen_die "could not uncompress insmod.static"

	for i in '[' ash basename cat chroot clear cp dirname echo env false find \
	grep gunzip gzip insmod ln ls loadkmap losetup lsmod mkdir mknod modprobe more mount mv \
	pivot_root ps awk pwd rm rmdir rmmod sh sleep tar test touch true umount uname \
	xargs yes zcat chmod chown; do
		rm -f ${TEMP}/initrd-temp/bin/$i > /dev/null
		ln  ${TEMP}/initrd-temp/bin/busybox ${TEMP}/initrd-temp/bin/$i || gen_die "could not link ${i}"
	done
}

create_initrd_modules() {
	if [ "${PAT}" -gt "4" ]
	then
		MOD_EXT=".ko"
	else
		MOD_EXT=".o"
	fi
	local modc i mods mymod
	for modc in storage firewire ataraid pcmcia usb
	do
		mkdir -p ${TEMP}/initrd-temp/lib/modules/${modc}
		mods=`echo $modc | tr [:lower:] [:upper:]`_MODULES
		eval mymods=\$$mods
		for i in ${mymods}
		do
			print_info 2 "$i : module searching" 1 0
			mymod=`find /lib/modules/${KV} -name "${i}${MOD_EXT}"`
			if [ -z "${mymod}" ]
			then
				print_info 2 "Warning : ${i}${MOD_EXT} not found; skipping..."
				continue;
			fi
			print_info 2 "copying ${mymod} to initrd"
			cp -ax --parents "${mymod}" "${TEMP}/initrd-temp"
		done
	done

	cp -ax --parents /lib/modules/${KV}/modules* ${TEMP}/initrd-temp
	cat ${GK_SHARE}/${ARCH}/linuxrc | sed 	-e "s/%%STORAGE_MODULES%%/${STORAGE_MODULES}/" \
						-e "s/%%FIREWIRE_MODULES%%/${FIREWIRE_MODULES}/" \
						-e "s/%%ATARAID_MODULES%%/${ATARAID_MODULES}/" \
						-e "s/%%PCMCIA_MODULES%%/${PCMCIA_MODULES}/" \
						-e "s/%%USB_MODULES%%/${USB_MODULES}/" \
						> ${TEMP}/initrd-temp/linuxrc
	chmod +x ${TEMP}/initrd-temp/linuxrc
}

create_initrd() {
	local MOD_EXT
	print_info 1 "initrd: creating loopback filesystem"
	create_initrd_loop 5000

	print_info 1 "initrd: creating base system"
	create_base_initrd_sys

	print_info 1 "initrd: copying modules"
	create_initrd_modules

	print_info 1 "initrd: cleaning up and compressing initrd"
	create_initrd_unmount_loop

	if [ "${COMPRESS_INITRD}" ]
	then
		gzip -f -9 ${TEMP}/initrd-loop
		mv ${TEMP}/initrd-loop.gz ${TEMP}/initrd-loop
	fi

	if [ "${BOOTSPLASH}" -eq "1" ]
	then
		print_info 1 "initrd: copying bootsplash"
		/sbin/splash -s -f /etc/bootsplash/gentoo/config/bootsplash-800x600.cfg >> ${TEMP}/initrd-loop || gen_die "could not copy 800x600 bootsplash"
		/sbin/splash -s -f /etc/bootsplash/gentoo/config/bootsplash-1024x768.cfg >> ${TEMP}/initrd-loop || gen_die "could not copy 1024x768 bootsplash"
		/sbin/splash -s -f /etc/bootsplash/gentoo/config/bootsplash-1280x1024.cfg >> ${TEMP}/initrd-loop || gen_die "could not copy 1280x1024 bootsplash"
		/sbin/splash -s -f /etc/bootsplash/gentoo/config/bootsplash-1600x1200.cfg >> ${TEMP}/initrd-loop || gen_die "could not copy 1600x1200 bootsplash"
	fi
	cp ${TEMP}/initrd-loop /boot/initrd-${KV} || gen_die "could not copy initrd to boot"
}