ui: don't quit if editing a draft

Ask user to save draft or discard it.

1 files changed, 6 insertions, 2 deletions

diff --git a/src/bin.rs b/src/bin.rs
index 35acf63f..d3372223 100644
--- a/src/bin.rs
+++ b/src/bin.rs
@@ -254,8 +254,12 @@ fn main() -> std::result::Result<(), std::io::Error> {
                                 UIMode::Normal => {
                                     match k {
                                         Key::Char('q') | Key::Char('Q') => {
-                                            drop(state);
-                                            break 'main;
+                                            if state.can_quit_cleanly() {
+                                                drop(state);
+                                                break 'main;
+                                            } else {
+                                                state.redraw();
+                                            }
                                         },
                                         Key::Char(' ') => {
                                             state.mode = UIMode::Execute;