#! /bin/bash
bootloader=${1:-grub}   # grub
bootloaderentry=${2:-2} # third entry
needreboot=${3:-yes}    # need reboot

echo "Usage: $(basename $0) <bootloader (grub or lilo)> <entry number or name> <need reboot (yes or no)>"

if [ "${bootloader}" == "grub" ]; then
    if ! echo "savedefault --default=${bootloaderentry} --once \nquit" | /sbin/grub --no-floppy --batch; then
        echo "failed to set up grub with \"savedefault --default=${bootloaderentry} --once\""
        exit 1
    fi
elif [ "${bootloader}" == "lilo" ]; then
    if ! /sbin/lilo -R ${bootloaderentry}; then
        echo "failed to set up lilo with \"-R ${bootloaderentry}\""
        exit 1
    fi
else
    echo "Bootloader \"${bootloader}\" not supported"
    exit 1
fi

#echo "Press Enter to Reboot/Shutdown"
#read ENTER

if [ "${needreboot}" == "yes" ]; then
    /sbin/shutdown -r now
elif [ "${needreboot}" == "no" ]; then
    /sbin/shutdown -h now
else
    echo "needreboot \"${needreboot}\" should be \"yes\" or \"no\""
    exit 1
fi