#!/bin/sh

# This script set a specific grub entry (winLOOSE…) for the next boot.
# Then reboot the host

# Vars {{{
readonly PROGNAME=$(basename "${0}")
readonly PROGDIR=$(readlink -m $(dirname "${0}"))
readonly ARGS="${*}"
readonly NBARGS="${#}"

## Debug mode to print specific messages
DEBUG=1

## Grub specific vars
readonly GRUB_DEFAULT_PATH='/etc/default/grub'
readonly GRUB_DEFAULT_DIR='/etc/default/grub.d'
readonly WIN_GRUB="2"

## By default, the host should reboot at the end of the script
REBOOT=0
HALT=1

## Colors
readonly PURPLE='\033[1;35m'
readonly RED='\033[0;31m'
readonly RESET='\033[0m'
readonly COLOR_DEBUG="${PURPLE}"

# }}}

usage() {                                                       # {{{

	cat <<- EOF
usage: $PROGNAME [-d,--debug|-h,--help|--halt,--poweroff|--reboot]

Try to set a specific grub's entry for the next boot and
reboot the host (default behaviour).

EXAMPLES :
    - Set the grub entry ${WIN_GRUB} (Windows by default) and reboot
        ${PROGNAME}

    - Set the grub entry ${WIN_GRUB} (Windows by default) and shutdown
        ${PROGNAME} --halt

OPTIONS :
    -d,--debug
        Enable debug messages.

    -h,--help
        Print this help message.

    --halt,--poweroff
        Ask to shutdown at the end of the script.

    --reboot
        Ask to reboot at the end of the script (default behaviour).
EOF

}
# }}}
debug_message() {                                               # {{{

	local_debug_message="${1}"

	## Print message if DEBUG is enable (=0)
	[ "${DEBUG}" -eq "0" ] && printf '\e[1;35m%-6b\e[m\n' "DEBUG − ${PROGNAME} : ${local_debug_message}"

}
# }}}
main() {                                                        # {{{

	# First check in Grub default dir if saved mode is set
	if grep -q -E -R -- "^GRUB_DEFAULT=saved" "${GRUB_DEFAULT_DIR}"
	then
		printf '%b' "Reboot to windaube partition\\n"
		sudo grub-reboot "${WIN_GRUB}"
		[ "${REBOOT}" -eq "0" ] && sudo systemctl reboot
		[ "${HALT}" -eq "0" ] && sudo systemctl poweroff
	fi

	# Otherwise check the default grub file
	if grep -q -E -- "^GRUB_DEFAULT=saved" "${GRUB_DEFAULT_PATH}"
	then
		printf '%b' "Reboot to windaube partition\\n"
		sudo grub-reboot "${WIN_GRUB}"
		[ "${REBOOT}" -eq "0" ] && sudo systemctl reboot
		[ "${HALT}" -eq "0" ] && sudo systemctl poweroff
	else

		printf '%b' "GRUB_DEFAULT is not set in 'saved' mode\\n"
		sudo sed -i 's/\(^GRUB_DEFAULT.*\)/#\1\nGRUB_DEFAULT=saved/' "${GRUB_DEFAULT_PATH}"
		GRUB_DEFAULT_ENTRY=$(grep -E -- "#GRUB_DEFAULT=" "${GRUB_DEFAULT_PATH}" | cut -d"=" -f2)
		sudo grub-set-default "${GRUB_DEFAULT_ENTRY}"
		sudo update-grub
		printf '%b' "Please launch this script (${PROGDIR}/${PROGNAME} ${ARGS}) once again.\\n"
	fi
}
# }}}

# Manage arguments                                                # {{{
# This code can't be in a function due to arguments

if [ ! "${NBARGS}" -eq "0" ]; then

	manage_arg="0"

	# Parse all options (start with a "-") one by one
	while printf -- '%s' "${1}" | grep -q -E -- "^-+"; do

	case "${1}" in
		-d|--debug )        ## debug
			DEBUG=0
			;;
		-h|--help )         ## help
			usage
			## Exit after help informations
			exit 0
			;;
		--halt|--poweroff ) ## A poweroff was requested
			REBOOT=1
			HALT=0
			;;
		--reboot )          ## A reboot was requested (default behaviour)
			REBOOT=0
			HALT=1
			;;
		-- )                ## End of options list
			## End the while loop
			break
			;;
		* )                 ## unknow option
			printf '%b\n' "${RED}Invalid option: ${1}${RESET}"
			printf '%b\n' "---"
			usage
			exit 1
			;;
	esac

	debug_message "Arguments management − \
${RED}${1}${COLOR_DEBUG} option managed."

	## Move to the next argument
	shift
	manage_arg=$((manage_arg+1))

	done

	debug_message "Arguments management − \
${RED}${manage_arg}${COLOR_DEBUG} argument(s) successfully managed."
else
	debug_message "Arguments management − \
No arguments/options to manage."
fi

# }}}

main

exit 0