#!/bin/bash

echo ""

if [[ $EUID -ne 0 ]]; then
   sudo $0
   exit 0
fi

DIR=$PWD
SRCDIR=${DIR}
DSTDIR=/media/rootfs

dd if=/dev/zero of=linux.img bs=64k count=6000

echo "Formatting Linux partition..."
mkfs.ext4 -L rootfs linux.img || exit 0
echo ""

echo "Mounting Linux partition..."
if [ ! -d ${DSTDIR} ]; then
   mkdir -p ${DSTDIR} || exit 0
fi
mount linux.img ${DSTDIR} || exit 0

echo "Copying main rootfs files..."
tar xfp ${SRCDIR}/rootfs.tar.bz2 --warning=no-timestamp -C ${DSTDIR} || exit 0

echo "Copying kernel modules rootfs files..."
tar xfp ${SRCDIR}/modules.tar.gz --strip-components=2 --warning=no-timestamp -C ${DSTDIR}/lib || exit 0

echo "Copying devices firmwares..."
mkdir -p ${DSTDIR}/lib/firmware || exit 0
tar xfp ${SRCDIR}/firmware.tar.gz --warning=no-timestamp -C ${DSTDIR}/lib/firmware || exit 0

echo "Copying additional modifications..."
if [ -d ${SRCDIR}/.addon ]; then
   [ -f ${SRCDIR}/addon.tar ] && rm -f ${SRCDIR}/addon.tar
          tar cf ${SRCDIR}/addon.tar -C ${SRCDIR}/.addon .
fi
tar xfp ${SRCDIR}/addon.tar --warning=no-timestamp -C ${DSTDIR} || exit 0
mkdir -p ${DSTDIR}/media/fat || exit 0
sed 's/getty/agetty\ \-\-nohostname/g' -i ${DSTDIR}/etc/inittab
sed 's/115200//g' -i ${DSTDIR}/etc/inittab
sed '/::sysinit:\/bin\/mount \-a/a ::sysinit:\/etc\/resync\ \&' -i ${DSTDIR}/etc/inittab
sed '/::sysinit:\/bin\/mount \-a/a ::sysinit:\/media\/fat\/MiSTer\ \&' -i ${DSTDIR}/etc/inittab
sed '/::sysinit:\/etc\/init.d\/rcS/a ::sysinit:\/bin\/setfont' -i ${DSTDIR}/etc/inittab
sed '/::sysinit:\/etc\/init.d\/rcS/a ::sysinit:\/bin\/loadkeys\ \/etc\/kbd.map' -i ${DSTDIR}/etc/inittab
sed '/GENERIC_SERIAL/a ::sysinit:\/sbin\/gpm\ \-m\ \/dev\/input\/mice\ \-t\ imps2' -i ${DSTDIR}/etc/inittab
sed '/GENERIC_SERIAL/a console::respawn:\/sbin\/agetty\ \-\-nohostname\ \-L\ tty1\ linux' -i ${DSTDIR}/etc/inittab

echo "tmpfs		/var/lib/samba	tmpfs	mode=1777	0	0" >>${DSTDIR}/etc/fstab
echo "tmpfs		/var/db/dhcpcd	tmpfs	mode=0750	0	0" >>${DSTDIR}/etc/fstab
sed 's/rw,noauto/rw,noauto,noatime,nodiratime/g' -i ${DSTDIR}/etc/fstab
sed 's/ext2/ext4/g' -i ${DSTDIR}/etc/fstab

sed 's/\/sh/\/bash/g' -i ${DSTDIR}/etc/passwd
rm ${DSTDIR}/etc/init.d/*sysctl
mkdir -p ${DSTDIR}/media/rootfs || exit 0
sed '/PATH/ s/$/:\/media\/fat\/linux:\/media\/fat\/Scripts:\./' -i ${DSTDIR}/etc/profile
sed s/\'\#\ \'/\'\$\(pwd\)\#\ \'/g -i ${DSTDIR}/etc/profile
sed s/\'\$\ \'/\'\$\(pwd\)\$\ \'/g -i ${DSTDIR}/etc/profile

cat >> ${DSTDIR}/etc/profile <<- __EOF__

export LC_ALL=en_US.UTF-8
resize >/dev/null
mount -o remount,rw /

__EOF__
echo -n $(date +%y%m%d) > ${DSTDIR}/MiSTer.version
sed 's/#hostname/hostname/g' -i ${DSTDIR}/etc/dhcpcd.conf
sed -e 's/^#clientid/clientid/' -e 's/^duid/#duid/' -i ${DSTDIR}/etc/dhcpcd.conf

echo "Fixing permissions..."
chown -R root:root ${DSTDIR} || exit 0
sync
sleep 3

echo "Unmounting Linux partition..."
umount ${DSTDIR} || exit 0
sleep 3

echo "Done!"
echo ""