rework end-screen

airootfs/root/install.sh

@@ -450,17 +450,18 @@ EOT
   # remove shell histories
   rm -f /root/.zsh_history
 
-  dialog --stdout --clear --cr-wrap --no-collapse --yes-label "Reboot" --no-label "Alpine shell" --yesno "\n             Installation finished" 7 50
-  case $? in
-  0) reboot ;;
-  1)
-    clear
-    exit
-    ;;
-  255)
-    clear
-    exit
-    ;;
+  selected_option=$(dialog --output-fd 1 --menu "What would you like to do?" 10 70 5 reboot "Reboot into your newly installed system" poweroff "Shut down the current live system" alpine-shell "Switch to an interactive shell")
+  case "$selected_option" in
+    "reboot")
+        reboot
+        ;;
+    "poweroff")
+        poweroff
+        ;;
+    "alpine-shell")
+        clear
+        exit
+        ;;
   esac
 }