#!/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

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

echo Recreating new partition table
sudo fdisk $1 << EOF
n
p
1

+16MB
t
53
n
p
2


w
EOF

if [ -e output/images/imx23_olinuxino_dev_linux.sb ] ; then
	echo Writing mxs-bootlet bootstream
	sudo dd if=output/images/imx23_olinuxino_dev_linux.sb bs=512 of="$1"1 seek=4
elif [ -e output/images/u-boot.sd ] ; then
	echo Writing U-Boot bootstream
	sudo dd if=output/images/u-boot.sd of="$1"1
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

	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
else
	echo Writing root filesystem
	sudo dd if=output/images/rootfs.ext2 of="$1"2 bs=512

	echo Synchronising changes to disk
	sudo sync
fi