#!/bin/sh

if [ -e /lib/firmware/vendor ]; then
	echo "Vendor firmware was loaded by the bootloader"
	return 1
fi

if [ ! -e /proc/device-tree/chosen/asahi,efi-system-partition ]; then
	echo "Missing asahi,efi-system-partition variable, firmware will not be loaded!"
	return 1
fi

echo "Load NVME modules"
modprobe apple-mailbox
modprobe nvme-apple
modprobe xhci-plat-hcd

for i in $(seq 0 50); do
[ -e /sys/bus/platform/drivers/nvme-apple/*.nvme/nvme/nvme*/nvme*n1/ ] && break
	sleep 0.1
done

if [ ! -e /sys/bus/platform/drivers/nvme-apple/*.nvme/nvme/nvme*/nvme*n1/ ]; then
echo "Timed out waiting for NVMe device"
	return 1
fi

# If the above exists, hopefully the /dev device exists and this will work
VENDORFW="/run/.system-efi/vendorfw/"

(
	. /usr/share/asahi-scripts/functions.sh
	mount_sys_esp /run/.system-efi
)

if [ ! -e "$VENDORFW/firmware.cpio" ]; then
	echo "Vendor firmware not available in ESP!"
	umount /run/.system-efi
	return 1
fi


( cd /; cpio -i < "$VENDORFW/firmware.cpio" )
umount /run/.system-efi

echo "Asahi: Vendor firmware unpacked successfully"

mkdir -p /lib/firmware/vendor
mount -t tmpfs -o mode=0755 vendorfw /lib/firmware/vendor
cp -a /vendorfw/* /vendorfw/.vendorfw.manifest /lib/firmware/vendor

echo "Asahi: Installed vendorfw"

modprobe appledrm

echo "Asahi: loaded DRM driver"
