Use Q for quitting instead of any key

2019-08-04 09:25:35 +03:00
parent 15510b7280
commit 0993e3845d
1 changed files with 14 additions and 8 deletions
--- a/src/main.rs
+++ b/src/main.rs
@@ -511,15 +511,21 @@ fn run() -> Result<(), Error> {
    }

    if config.keep_at_end() {
-        print_info("\n(R)eboot\n(S)hell\n\nPress any other key to continue");
-        match get_char() {
-            's' | 'S' => {
-                run_shell();
+        print_info("\n(R)eboot\n(S)hell\n(Q)uit");
+        loop {
+            match get_char() {
+                's' | 'S' => {
+                    run_shell();
+                }
+                'r' | 'R' => {
+                    reboot();
+                }
+                'q' | 'Q' => (),
+                _ => {
+                    continue;
+                }
            }
-            'r' | 'R' => {
-                reboot();
-            }
-            _ => (),
+            break;
        }
    }