#!/bin/sh
set -e

flash_fastboot()
{
	fastboot flash BOOT arch/arm/boot/boot.img
	sleep 1
	fastboot flash RECOVERY arch/arm/boot/boot.img
	sleep 1
	fastboot reboot
	echo "OK"
}

flash_heimdall()
{
	heimdall flash \
		--BOOT arch/arm/boot/boot.img \
		--RECOVERY arch/arm/boot/boot.img
	echo "OK"
}

usage()
{
	echo "Usage: $0 <fastboot|heimdall>"
	exit 1
}

if [ $# -ne 1 ] ; then
	usage
elif [ "$1" = "fastboot" ] ; then
	flash_fastboot
elif [ "$1" = "heimdall" ] ; then
	flash_heimdall
else
	usage
fi