Add ask_reboot Func

Author: afiniel

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)

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
 

yiimp_single/start.sh

@@ -111,10 +111,7 @@ print_message() {
 }
 
 term_yiimpool
-
-# Call print_message regardless of UsingDomain
 print_message
-
 exit 0
-
+ask_reboot