#!/bin/sh -eu

kflavour=
kflavour="${GLOBAL_KFLAVOURS%% *}"

mkdir -p /boot/efi

cp /usr/share/raspberrypi-firmware/*.{bin,dat,elf} /boot/efi/

if [ -f /usr/share/u-boot/rpi_arm64/kernel8.img ]; then
	cp /usr/share/u-boot/rpi_arm64/kernel8.img \
	   /boot/efi/
else
	[ -f /usr/share/u-boot/rpi_3/kernel8.img ] &&
	cp /usr/share/u-boot/rpi_3/kernel8.img \
	   /boot/efi/uboot-rpi_3.bin
	[ -f /usr/share/u-boot/rpi_4/kernel8.img ] &&
	cp /usr/share/u-boot/rpi_4/kernel8.img \
	   /boot/efi/uboot-rpi_4.bin
fi

# copy actualy dtb's for default kernel
if [ -d /boot/dtb/broadcom ]; then
	DTB_DIR=/boot/dtb/broadcom
elif [ -d /boot/dtb ]; then
	DTB_DIR=/boot/dtb
else
	echo "Error: /boot/dtb/broadcom or /boot/dtb is not exists!!!" >&2
	exit 1
fi

cp "$DTB_DIR"/bcm27*-rpi-*.dtb /boot/efi/ || :

[ -f /boot/efi/bcm2710-rpi-3-b.dtb ] ||
	cp "$DTB_DIR"/bcm2837-rpi-3-b.dtb \
	   /boot/efi/bcm2710-rpi-3-b.dtb || :
[ -f /boot/efi/bcm2710-rpi-3-b-plus.dtb ] ||
	cp "$DTB_DIR"/bcm2837-rpi-3-b-plus.dtb \
	   /boot/efi/bcm2710-rpi-3-b-plus.dtb || :
[ -f /boot/efi/bcm2710-rpi-cm3.dtb ] ||
	cp "$DTB_DIR"/bcm2837-rpi-cm3-io3.dtb \
	   /boot/efi/bcm2710-rpi-cm3.dtb || :
[ -f /boot/efi/bcm2711-rpi-cm4.dtb ] ||
	cp "$DTB_DIR"/bcm2711-rpi-cm4-io.dtb \
	   /boot/efi/bcm2711-rpi-cm4.dtb || :
[ -d "$DTB_DIR"/overlays ] &&
	cp -r "$DTB_DIR"/dtb/overlays /boot/efi/ || :

cat >> /boot/efi/config.txt <<EOF
arm_64bit=1
enable_uart=1
disable_overscan=1
dtparam=audio=on
hdmi_ignore_edid_audio=1
EOF

[ -f /boot/efi/kernel8.img ] || cat >> /boot/efi/config.txt <<EOF

[pi3]
kernel=uboot-rpi_3.bin
#dtoverlay=vc4-fkms-v3d

[pi4]
kernel=uboot-rpi_4.bin
#dtoverlay=vc4-fkms-v3d
#max_framebuffers=2

[all]
EOF

# Enable bootloading without u-boot
if [ -n "${GLOBAL_RPI_NOUBOOT-}" ]; then
	rpi4-boot-nouboot $kflavour
	sed -i '/#dtoverlay=vc4-fkms-v3d/s/#//' /boot/efi/config.txt
fi

exit 0
