#!/bin/bash

source ./envsetup

if [[ -z "${DEV}" ]]; then
	DEV=sdb1
fi

UDISK=$WS/u-disk

OUT_ZSBL=$WS_LINUX/out-zsbl
OUT_SBI=$WS_LINUX/out-sbi
OUT_LINUX=$WS_LINUX/out-linux

sudo umount $UDISK
sudo mount /dev/$DEV $UDISK
if [ $? -ne 0 ]; then
	echo "ERROR: mount failed"
	exit 1
fi

if [ ! -f $UDISK/FLAG_MINIMAL ]; then
	echo "ERROR: not mininal sdcard, pleaes double-check!"
	exit 1
fi

if [ "$1" == "-a" ]; then
	echo "Removing zsbl and sbi ......"
	sudo rm $UDISK/zsbl.bin
	sudo rm $UDISK/riscv64/fw_dynamic.bin
	echo "Removing Done!"

        echo "Copying zsbl and sbi to sd-card"
        sudo cp $OUT_ZSBL/zsbl.bin $UDISK/zsbl.bin
        sudo cp $OUT_SBI/platform/generic/firmware/fw_dynamic.bin $UDISK/riscv64/fw_dynamic.bin
        echo "Copying Done!"
fi

echo "Removing dtb and kernel ......"
sudo rm $UDISK/riscv64/mango-milkv-pioneer.dtb
sudo rm $UDISK/riscv64/riscv64_Image
echo "Removing Done!"

echo "Copying dtb and kernel to sd-card"
sudo cp $OUT_LINUX/arch/riscv/boot/dts/sophgo/sg2042-milkv-pioneer.dtb $UDISK/riscv64/mango-milkv-pioneer.dtb
sudo cp $OUT_LINUX/arch/riscv/boot/Image $UDISK/riscv64/riscv64_Image
echo "Copying Done!"

sudo umount $UDISK

echo "Done!"