#!/bin/bash

. load-config.sh
test -f $DEVICE_DIR/flash.sh && . $DEVICE_DIR/flash.sh

ADB=${ADB:-adb}
FASTBOOT=${FASTBOOT:-fastboot}
HEIMDALL=${HEIMDALL:-heimdall}
VARIANT=${VARIANT:-eng}
FULLFLASH=false

if [ ! -f "`which \"$ADB\"`" ]; then
	ADB=out/host/`uname -s | tr "[[:upper:]]" "[[:lower:]]"`-x86/bin/adb
fi
if [ ! -f "`which \"$FASTBOOT\"`" ]; then
	FASTBOOT=out/host/`uname -s | tr "[[:upper:]]" "[[:lower:]]"`-x86/bin/fastboot
fi

run_adb()
{
	$ADB $ADB_FLAGS $@
}

run_fastboot()
{
	if [ "$1" = "devices" ]; then
		$FASTBOOT $@
	else
		$FASTBOOT $FASTBOOT_FLAGS $@
	fi
	return $?
}

update_time()
{
	if [ `uname` = Darwin ]; then
		OFFSET=`date +%z`
		OFFSET=${OFFSET:0:3}
		TIMEZONE=`date +%Z$OFFSET|tr +- -+`
	else
		TIMEZONE=`date +%Z%:::z|tr +- -+`
	fi
	echo Attempting to set the time on the device
	run_adb wait-for-device &&
	run_adb shell "toolbox date $(date +%s) ||
	               toolbox date -s $(date +%Y%m%d.%H%M%S)" &&
	run_adb shell setprop persist.sys.timezone $TIMEZONE
}

fastboot_flash_image()
{
	# $1 = {userdata,boot,system,recovery} the image filename
	# $2 = {userdata,boot,system,recovery} the fastboot partition target
	#    if non specified, then $2 = $1
	PARTITION_FILE=$1
	PARTITION_NAME=$2

	if [ -z "${PARTITION_NAME}" ]; then
		PARTITION_NAME=${PARTITION_FILE}
	fi

	if [ "$DEVICE" == "flatfish" ] && [ "${PARTITION_NAME}" == "userdata" ]; then
		PARTITION_NAME="data"
	fi
	imgpath="out/target/product/$DEVICE/${PARTITION_FILE}.img"
	out="$(run_fastboot flash "${PARTITION_NAME}" "$imgpath" 2>&1)"
	rv="$?"
	echo "$out"

	if [[ "$rv" != "0" ]]; then
		# Print a nice error message if we understand what went wrong.
		if grep -q "too large" <(echo "$out"); then
			echo ""
			echo "Flashing $imgpath failed because the image was too large."
			echo "Try re-flashing after running"
			echo "  \$ rm -rf $(dirname "$imgpath")/data && ./build.sh"
		fi
		return $rv
	fi
}

fastboot_flash_image_if_exists()
{
	if [ -e "out/target/product/$DEVICE/$1.img" ]; then
		fastboot_flash_image $1
	fi
}


flash_fastboot()
{
	local lockedness=$1 project=$2
	case $lockedness in
	"unlock"|"nounlock")
		;;
	*)
		echo "$0: $FUNCNAME: Invalid argument: $lockedness"
		return 1
		;;
	esac
	case $project in
	"system"|"boot"|"userdata"|"cache"|"")
		;;
	*)
		echo "$0: Unrecognized project/partition: $project"
		return 1
		;;
	esac

	delete_single_variant_persist

	case $DEVICE in
	"helix")
		run_adb reboot oem-1
		;;
	"flatfish")
		run_adb reboot boot-fastboot
		;;
	*)
		run_adb reboot bootloader
		;;
	esac

	if ! run_fastboot devices; then
		echo Couldn\'t setup fastboot
		return 1
	fi

	case $lockedness in
	"unlock")
		run_fastboot oem unlock || true
		;;
	esac

	case $project in
	"system" | "boot" | "userdata" | "cache")
		fastboot_flash_image $project &&
		run_fastboot reboot
		;;

	"")
		VERB="erase"
		if [ "$DEVICE" == "hammerhead" ] || [ "$DEVICE" == "mako" ] ||
		[ "$DEVICE" == "flo" ]; then
			VERB="format"
		fi
		DATA_PART_NAME="userdata"
		if [ "$DEVICE" == "flatfish" ]; then
			DATA_PART_NAME="data"
		fi
		# helix/dolphin don't support erase command in fastboot mode.
		if [ "$DEVICE" != "helix" -a "$DEVICE_NAME" != "dolphin" ]; then
			run_fastboot $VERB cache &&
			run_fastboot $VERB $DATA_PART_NAME
			if [ $? -ne 0 ]; then
				return $?
			fi
		fi
		case ${DEVICE} in
		"aries"|"leo"|"scorpion"|"sirius"|"honami"|"amami"|"tianchi"|"flamingo"|"eagle"|"seagull")
			fastboot_flash_image recovery FOTAKernel
			;;
		esac
		fastboot_flash_image userdata &&
		fastboot_flash_image_if_exists cache &&
		fastboot_flash_image_if_exists boot &&
		fastboot_flash_image system &&
		run_fastboot reboot &&
		update_time
		;;
	esac
	echo -ne \\a
}

flash_heimdall()
{
	local project=$1
	case $project in
	"system"|"kernel"|"")
		;;
	*)
		echo "$0: Unrecognized project: $project"
		return 1
		;;
	esac

	if [ ! -f "`which \"$HEIMDALL\"`" ]; then
		echo Couldn\'t find heimdall.
		echo Install Heimdall v1.3.1 from http://www.glassechidna.com.au/products/heimdall/
		exit -1
	fi

	delete_single_variant_persist &&
	run_adb reboot download && sleep 8
	if [ $? -ne 0 ]; then
		echo Couldn\'t reboot into download mode. Hope you\'re already in download mode
	fi

	case $project in
	"system")
		$HEIMDALL flash --factoryfs out/target/product/$DEVICE/$project.img
		;;

	"kernel")
		$HEIMDALL flash --kernel device/samsung/$DEVICE/kernel
		;;

	"")
		$HEIMDALL flash --factoryfs out/target/product/$DEVICE/system.img --kernel device/samsung/$DEVICE/kernel &&
		update_time
		;;
	esac

	ret=$?
	echo -ne \\a
	if [ $ret -ne 0 ]; then
		echo Heimdall flashing failed.
		case "`uname`" in
		"Darwin")
			if kextstat | grep com.devguru.driver.Samsung > /dev/null ; then
				echo Kies drivers found.
				echo Uninstall kies completely and restart your system.
			else
				echo Restart your system if you\'ve just installed heimdall.
			fi
			;;
		"Linux")
			echo Make sure you have a line like
			echo SUBSYSTEM==\"usb\", ATTRS{idVendor}==\"04e8\", MODE=\"0666\"
			echo in /etc/udev/rules.d/android.rules
			;;
		esac
		exit -1
	fi

	echo Run \|./flash.sh gaia\| if you wish to install or update gaia.
}

# Delete files in the device's /system/b2g that aren't in
# $GECKO_OBJDIR/dist/b2g.
#
# We do this for general cleanliness, but also because b2g.sh determines
# whether to use DMD by looking for the presence of libdmd.so in /system/b2g.
# If we switch from a DMD to a non-DMD build and then |flash.sh gecko|, we want
# to disable DMD, so we have to delete libdmd.so.
#
# Note that we do not delete *folders* in /system/b2g.  This is intentional,
# because some user data is stored under /system/b2g (e.g. prefs), but it seems
# to be stored only inside directories.
delete_extra_gecko_files_on_device()
{
	files_to_remove="$(cat <(ls "$GECKO_OBJDIR/dist/b2g") <(run_adb shell "ls /system/b2g" | tr -d '\r') | sort | uniq -u)"
	if [[ "$files_to_remove" != "" ]]; then
		# We expect errors from the call to rm below under two circumstances:
		#
		#  - We ask rm to remove a directory (per above, we don't
		#    actually want to remove directories, so rm is doing the
		#    right thing by not removing dirs)
		#
		#  - We ask rm to remove a file which isn't on the device (if
		#    you squint at files_to_remove, you'll see that it will
		#    contain files which are on the host but not on the device;
		#    obviously we can't remove those files from the device).
		for to_remove in $files_to_remove; do
			run_adb shell "rm /system/b2g/$to_remove" > /dev/null
		done
	fi
	return 0
}

delete_single_variant_persist()
{
	run_adb shell 'rm -rf /persist/svoperapps'
}

flash_gecko()
{
	delete_extra_gecko_files_on_device &&
	run_adb push $GECKO_OBJDIR/dist/b2g /system/b2g &&
	return 0
}

flash_gaia()
{
	GAIA_MAKE_FLAGS="ADB=\"$ADB\""
	USER_VARIANTS="user(debug)?"
	# We need to decide where to push the apps here.
	# If the VARIANTS is user or userdebug, send them to /system/b2g.
	# or, we will try to connect the phone and see where Gaia was installed
	# and try not to push to the wrong place.
	if [[ "$VARIANT" =~ $USER_VARIANTS ]]; then
		# Gaia's build takes care of remounting /system for production builds
		echo "Push to /system/b2g ..."
		GAIA_MAKE_FLAGS+=" GAIA_INSTALL_PARENT=/system/b2g"
	else
		echo "Detect GAIA_INSTALL_PARENT ..."
		# This part has been re-implemented in Gaia build script (bug 915484),
		# XXX: Remove this once we no longer support old Gaia branches.
		# Install to /system/b2g if webapps.json does not exist, or
		# points any installed app to /system/b2g.
		run_adb wait-for-device
		if run_adb shell 'cat /data/local/webapps/webapps.json || echo \"basePath\": \"/system\"' | grep -qs '"basePath": "/system' ; then
			echo "Push to /system/b2g ..."
			GAIA_MAKE_FLAGS+=" GAIA_INSTALL_PARENT=/system/b2g"
		else
			echo "Push to /data/local ..."
			GAIA_MAKE_FLAGS+=" GAIA_INSTALL_PARENT=/data/local"
		fi
	fi
	make -C gaia push $GAIA_MAKE_FLAGS

	# For older Gaia without |push| target,
	# run the original |install-gaia| target.
	# XXX: Remove this once we no longer support old Gaia branches.
	if [[ $? -ne 0 ]]; then
		make -C gaia install-gaia $GAIA_MAKE_FLAGS
	fi
	return $?
}

while [ $# -gt 0 ]; do
	case "$1" in
	"-s")
		ADB_FLAGS+="-s $2"
		FASTBOOT_FLAGS+="-s $2"
		shift
		;;
	"-f")
		FULLFLASH=true
		;;
	"-h"|"--help")
		echo "Usage: $0 [-s device] [-f] [project]"
		exit 0
		;;
	"-"*)
		echo "$0: Unrecognized option: $1"
		exit 1
		;;
	*)
		FULLFLASH=true
		PROJECT=$1
		;;
	esac
	shift
done

case "$PROJECT" in
"shallow")
	resp=`run_adb root` || exit $?
	[ "$resp" != "adbd is already running as root" ] && run_adb wait-for-device
	run_adb shell stop b2g &&
	run_adb remount &&
	flash_gecko &&
	flash_gaia &&
	update_time &&
	echo Restarting B2G &&
	run_adb shell start b2g
	exit $?
	;;

"gecko")
	resp=`run_adb root` || exit $?
	[ "$resp" != "adbd is already running as root" ] && run_adb wait-for-device
	run_adb shell stop b2g &&
	run_adb remount &&
	flash_gecko &&
	echo Restarting B2G &&
	run_adb shell start b2g
	exit $?
	;;

"gaia")
	flash_gaia
	exit $?
	;;

"time")
	update_time
	exit $?
	;;
esac

case "$DEVICE" in
"hamachi"|"helix"|"sp6821a_gonk")
	if $FULLFLASH; then
		flash_fastboot nounlock $PROJECT
		exit $?
	else
		run_adb root &&
		run_adb shell stop b2g &&
		run_adb remount &&
		flash_gecko &&
		flash_gaia &&
		update_time &&
		echo Restarting B2G &&
		run_adb shell start b2g
	fi
	exit $?
	;;

"flame"|"otoro"|"unagi"|"keon"|"peak"|"inari"|"wasabi"|"flatfish"|"aries"|"leo"|"scorpion"|"sirius"|"honami"|"amami"|"tianchi"|"flamingo"|"eagle"|"seagull"|"scx15_sp7715"*)
	flash_fastboot nounlock $PROJECT
	;;

"panda"|"maguro"|"crespo"|"crespo4g"|"mako"|"hammerhead"|"flo"|"shamu"|"FP2")
	flash_fastboot unlock $PROJECT
	;;

"galaxys2")
	flash_heimdall $PROJECT
	;;

*)
	if [[ $(type -t flash_${DEVICE}) = function ]]; then
		flash_${DEVICE} $PROJECT
	else
		echo Unsupported device \"$DEVICE\", can\'t flash
		exit -1
	fi
	;;
esac