#!/bin/sh
set -e

GRUB_D="${GRUB_D:-/etc/grub.d}"

if [ -f "${GRUB_D}/10_linux" ]
then
	sed -i -e "s#GNU/Linux#Linux#g" "${GRUB_D}/10_linux"
	if ! grep -q -e "--class submenu" "${GRUB_D}/10_linux"
	then
		sed -i -e "s/\\(.[\$]menuentry_id_option 'gnulinux-advanced-\\)/--class submenu \\1/" "${GRUB_D}/10_linux"
	fi
fi

if [ -f "${GRUB_D}/30_uefi-firmware" ] && ! grep -q -e "--class efi" "${GRUB_D}/30_uefi-firmware"
then
	sed -i -e "s/\\(menuentry '[\$]LABEL'\\)/\\1 --class efi/" "${GRUB_D}/30_uefi-firmware"
fi
