Add ask_reboot Func

afiniel · Jan 17, 2025 · 04a89b3 · 04a89b3
1 parent e2b6919
commit 04a89b3
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 4 deletions.
diff --git a/install/create_user.sh b/install/create_user.sh
@@ -144,6 +144,7 @@ if [[ ("$UsingSSH" == "yes") ]]; then
     echo
     echo -e "$RED Please reboot the system and log in as$GREEN ${yiimpadmin} $YELLOW and type$GREEN yiimpool$YELLOW to$GREEN continu$YELLOW setup...$NC"
     exit 0
+    ask_reboot
 fi
 
 # New User Password Login Creation
@@ -260,6 +261,7 @@ case $response in
     echo
     echo -e "$RED Please reboot the system and log in as:$GREEN ${yiimpadmin} $YELLOW and type$GREEN yiimpool$YELLOW to$GREEN continu$YELLOW setup...$NC"
     exit 0
+    ask_reboot
     ;;
 
 1)

diff --git a/install/functions.sh b/install/functions.sh
@@ -91,6 +91,14 @@ function install_end_message() {
 
 }
 
+# A function to ask if the user wants to reboot the system
+function ask_reboot() {
+  read -p "Do you want to reboot the system? (y/n): " reboot_choice
+  if [[ "$reboot_choice" == "y" ]]; then
+    sudo reboot
+  fi
+}
+
 function term_art() {
   clear
 

diff --git a/yiimp_single/start.sh b/yiimp_single/start.sh
@@ -111,10 +111,7 @@ print_message() {
 }
 
 term_yiimpool
-
-# Call print_message regardless of UsingDomain
 print_message
-
 exit 0
-
+ask_reboot