#!/bin/bash

kernel_versions=$(ls /boot/vmlinuz-*-xanmod* 2>/dev/null | sed 's|/boot/vmlinuz-||' || true)

if [ -z "$kernel_versions" ]; then
    echo "No xanmod kernels found in /boot, cleanup may be incomplete"
    exit 0
fi

for kernel_version in $kernel_versions; do
    echo "Cleaning up kernel $kernel_version..."

    kernel_path="/boot/vmlinuz-$kernel_version"

    if command -v grubby >/dev/null 2>&1; then
        if grubby --info="$kernel_path" >/dev/null 2>&1; then
            echo "Removing kernel entry using grubby..."
            grubby --remove-kernel="$kernel_path" || true
            echo "Kernel removed from bootloader"
        fi
    fi

    initramfs_img="/boot/initramfs-$kernel_version.img"
    if [ -f "$initramfs_img" ]; then
        echo "Removing initramfs: $initramfs_img"
        rm -f "$initramfs_img"
    fi

    if [ -f "$kernel_path" ]; then
        echo "Removing kernel image: $kernel_path"
        rm -f "$kernel_path"
    fi

    if [ -f "/boot/System.map-$kernel_version" ]; then
        echo "Removing System.map: /boot/System.map-$kernel_version"
        rm -f "/boot/System.map-$kernel_version"
    fi

    if [ -d "/usr/lib/modules/$kernel_version" ]; then
        echo "Removing modules: /usr/lib/modules/$kernel_version"
        rm -rf "/usr/lib/modules/$kernel_version"
    fi

    echo "Kernel $kernel_version has been removed."
done

if command -v update-grub >/dev/null 2>&1; then
    echo "Updating GRUB configuration..."
    update-grub
elif command -v grub2-mkconfig >/dev/null 2>&1 && ! command -v grubby >/dev/null 2>&1; then
    echo "Updating GRUB2 configuration..."
    grub2-mkconfig -o /boot/grub2/grub.cfg
elif command -v grub-mkconfig >/dev/null 2>&1 && ! command -v grubby >/dev/null 2>&1; then
    echo "Updating GRUB configuration..."
    grub-mkconfig -o /boot/grub/grub.cfg
fi

echo ""
echo "All xanmod kernels have been removed."
echo "Please ensure you have another kernel installed before rebooting."