Add -k flag

2019-06-16 09:09:05 +03:00
parent a834a79f45
commit b6e838bab0
4 changed files with 15 additions and 5 deletions
--- a/src/main.rs
+++ b/src/main.rs
@@ -498,8 +498,8 @@ fn run() -> Result<(), Error> {
        freebsd::audit_packages(&sudo).ok();
    }

-    if env::var("TOPGRADE_KEEP_END").is_ok() {
-        print_info("\nPress R to reboot, S for shell or any other key to continue");
+    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();