#!/bin/sh

if [ "$#" -ne 1 ] ; then
 	echo "Usage: $0 /dev/disk" >&2
	exit 1
fi

# Unmount mounted partitions
for fs in `grep $1 /proc/mounts|cut -d ' ' -f 1` ; do
	echo Unmounting $fs
	sudo umount $fs
done

sudo sync

echo Overwriting old partition table
sudo dd if=/dev/zero of=$1 bs=1024 count=1024

sudo sync

if [ -e output/images/u-boot.sd ] ; then
	echo Recreating new partition table
	sed 's/#.*//' << EOF | tr -d " \t"  | sudo fdisk $1
		n	# new partition
		p	# primary partition
		1	# number 1
			# default start
		+16M	# 16MB
		t	# New Type field
		53	# OnTrack DM6 Aux3
		n	# new partition
		p	# primary partition
		2	# number 2
			# default start
			# default size
		w	# write changes
EOF

	echo Writing U-Boot bootstream
	sudo dd if=output/images/u-boot.sd of="$1"1
elif [ -d output/images/rpi-firmware ] ; then
	SIZE=`sudo fdisk -l $1 | grep Disk | grep bytes | awk '{print $5}'`	
	echo Disk size: $SIZE bytes
	CYLINDERS=`echo $SIZE/255/63/512 | bc`
	echo Cylinders: $CYLINDERS

	sed 's/#.*//' << EOF | tr -d " \t"  | sudo sfdisk -D -H 255 -S 63 -C $CYLINDERS $1
		,9,0x0C,*	# From http://downloads.angstrom-distribution.org/demo/beaglebone/mkcard.txt
		,,,-		# Found in http://elinux.org/RPi_Advanced_Setup#Advanced_SD_card_setup
EOF

	echo Creaeting boot filesystem
	sudo mkfs.vfat -F 32 -n boot "$1"1 << EOF
y
EOF

	echo Mounting boot filesystem
	sudo mkdir -p /media/boot
	sudo mount "$1"1 /media/boot

	echo Copying bootloader files
	sudo cp output/images/rpi-firmware/* /media/boot/
	sudo cp output/images/*.dtb /media/boot/

	echo Preparing and copying Kernel Image
	sudo output/host/usr/bin/mkknlimg output/images/zImage /media/boot/zImage

	echo Synchronising changes to disk
	sudo sync

	echo Unmounting boot filesystem
	sudo umount /media/boot
	sudo rm -rf /media/boot
else
	echo Could not find a suitable bootstream!
fi

if [ -e output/images/rootfs.tar ] ; then
	echo Creating root filesystem
	sudo mkfs.ext4 "$1"2 -L rootfs << EOF
y
EOF

	echo Mounting root filesystem
	sudo mkdir -p /media/rootfs
	sudo mount "$1"2 /media/rootfs

	echo Copying root filesystem
	sudo tar xfp output/images/rootfs.tar -C /media/rootfs

	echo Synchronising changes to disk
	sudo sync

	echo Unmounting root filesystem
	sudo umount /media/rootfs
	sudo rm -rf /media/rootfs
elif [ -e output/images/rootfs.ext2 ] ; then
	echo Writing ext2 root filesystem
	sudo dd if=output/images/rootfs.ext2 of="$1"2 bs=512

	echo Synchronising changes to disk
	sudo sync
else
	echo Could not find a suitable root filesystem!
fi

sudo eject $1