#!/bin/bash

source .config

TMP_DIR="/tmp/driedlamu"
FIRMWARE_PATH="./unpacked"

case $1 in
	fresh)
		SYS_UPGRADE=0
		echo "[i] Installing DriedLamu."
	;;
	upgrade)
		SYS_UPGRADE=1
		echo "[i] Upgrading DriedLamu."
	;;
	*)
		echo "Please specify a valid type of task you wish to do (fresh, upgrade)."
		exit 1
	;;
esac

fastboot oem bldr_spoof off

if [[ $INSTALL_SU == 1 ]]; then
	echo "[i] Root installation enabled."

	if [[ $(fastboot devices | cut -d' ' -f2) == "fastboot" ]]; then
		fastboot flash boot "${FIRMWARE_PATH}/boot_su.img"
		
		if [[ -f "${FIRMWARE_PATH}/init_boot_su.img" ]]; then
			fastboot flash init_boot "${FIRMWARE_PATH}/init_boot_su.img"
		fi
	else
		echo "[i] Already in fastbootd mode. Make sure that patched 'boot.img' was installed."
	fi
fi

if [[ $ENABLE_VBMETA == 1 ]]; then
	echo "[i] Android verified boot enabled."

	if [[ $(fastboot devices | cut -d' ' -f2) == "fastboot" ]]; then
		fastboot flash boot "${STOCK_FIRMWARE_PATH}/boot.img"
		fastboot flash init_boot "${STOCK_FIRMWARE_PATH}/init_boot.img"

		fastboot flash vbmeta "${TMP_DIR}/vbmeta/vbmeta.img"
		fastboot flash vbmeta_system "${TMP_DIR}/vbmeta/vbmeta_system.img"
	else
		echo "[i] Already in fastbootd mode. Make sure that 'vbmeta' was flashed."
	fi
fi

if [[ $(fastboot devices | cut -d' ' -f2) == "fastboot" ]]; then
	echo "[i] Device in fastboot mode. Rebooting to fastbootd."
	sleep 1
	fastboot reboot fastboot
fi

fastboot erase system
fastboot erase system_ext
fastboot erase product

fastboot flash super ./super.img

# fastboot flash system_a "${FIRMWARE_PATH}/system_a.img"
# fastboot flash system_ext_a "${FIRMWARE_PATH}/system_ext_a.img"
# fastboot flash product_a "${FIRMWARE_PATH}/product_a.img"

if [[ $SYS_UPGRADE == 0 ]]; then
	fastboot -w
fi

if [[ $ENABLE_VBMETA == 1 ]]; then
	fastboot reboot-bootloader
	sleep 3
	fastboot oem bldr_spoof on
	fastboot reboot
else
	fastboot reboot
fi

echo "[i] Flashing done!"