From 271fb2e6e899b1dbae0092ac337dcc7d2eedbe8e Mon Sep 17 00:00:00 2001
From: AuxXxilium <info@auxxxilium.tech>
Date: Sun, 25 Aug 2024 10:30:08 +0200
Subject: [PATCH] arc-functions: ask for reboot after loader update

Signed-off-by: AuxXxilium <info@auxxxilium.tech>
---
 files/initrd/opt/arc/arc-functions.sh | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/files/initrd/opt/arc/arc-functions.sh b/files/initrd/opt/arc/arc-functions.sh
index f52caf61..c62fb2ba 100755
--- a/files/initrd/opt/arc/arc-functions.sh
+++ b/files/initrd/opt/arc/arc-functions.sh
@@ -933,6 +933,19 @@ function updateMenu() {
         updateLoader "${TAG}"
         writeConfigKey "arc.builddone" "false" "${USER_CONFIG_FILE}"
         BUILDDONE="$(readConfigKey "arc.builddone" "${USER_CONFIG_FILE}")"
+        # Ask for Reboot
+        dialog --clear --backtitle "$(backtitle)" --title "Update done"\
+          --no-cancel --menu "Reboot now?" 7 40 0 \
+          1 "Yes - Reboot Arc Loader now" \
+          2 "No - I want to update more" \
+        2>"${TMP_PATH}/resp"
+        resp=$(cat ${TMP_PATH}/resp)
+        [ -z "${resp}" ] && return 1
+        if [ ${resp} -eq 1 ]; then
+          rebootTo config
+        elif [ ${resp} -eq 2 ]; then
+          return 0
+        fi
         ;;
       4)
         # Ask for Tag