#!/bin/sh

processimage()
{
    count1=$(find /boot/dtb-${1}/ -maxdepth 1 -type f -name 'bcm*-rpi-*.dtb' | wc -l)
    count2=$(find /boot/dtb-${1}/overlays/ -maxdepth 1 -type f -name '*.dtbo' | wc -l)
    count2=$((count2 + $(find /boot/dtb-${1}/overlays/ -maxdepth 1 -type f -name '*-overlay.dtb' | wc -l))) # Backward compatibility
    if [ $count1 -lt 1 -o $count2 -lt 1 ]; then
        echo "Aborting due to one or more missing device tree overlays." >&2
        exit 1
    fi

    cp -f /boot/vmlinuz-${1} /boot/kernel.img

    rm -rf /boot/*.dtb /boot/overlays/
    rm -f /boot/kernel7.img
    cp -af /boot/dtb-${1}/* /boot/

    EXIT=$?
    if [ $EXIT -ne 0 ]; then
	echo "Failed to copy device tree overlays." >&2
        exit $EXIT
    fi

    rm -rf /lib/modules/${1}_kernel_*
}

processimage $1 $2
