From 6b4126ea701c0a07918d70207b9269d91e90bd22 Mon Sep 17 00:00:00 2001
From: crides <zhuhaoqing@live.cn>
Date: Mon, 1 Jan 2024 21:46:02 -0800
Subject: [PATCH] incomplete key sequences should be completed first

---
 src/edit_mode/vi/mod.rs | 56 +++++++++++++++++++++++++++++++++++++----
 1 file changed, 51 insertions(+), 5 deletions(-)

diff --git a/src/edit_mode/vi/mod.rs b/src/edit_mode/vi/mod.rs
index c687a86b..f8567a4c 100644
--- a/src/edit_mode/vi/mod.rs
+++ b/src/edit_mode/vi/mod.rs
@@ -30,6 +30,7 @@ pub struct Vi {
     previous: Option<ReedlineEvent>,
     // last f, F, t, T motion for ; and ,
     last_char_search: Option<ViCharSearch>,
+    seq_completed: bool,
 }
 
 impl Default for Vi {
@@ -41,6 +42,7 @@ impl Default for Vi {
             mode: ViMode::Insert,
             previous: None,
             last_char_search: None,
+            seq_completed: true,
         }
     }
 }
@@ -65,12 +67,13 @@ impl EditMode for Vi {
                 (ViMode::Normal, modifier, KeyCode::Char(c)) => {
                     let c = c.to_ascii_lowercase();
 
-                    if let Some(event) = self
+                    let binding = self
                         .normal_keybindings
-                        .find_binding(modifiers, KeyCode::Char(c))
+                        .find_binding(modifiers, KeyCode::Char(c));
+                    if !self.seq_completed
+                        || binding.is_none()
+                            && (modifier == KeyModifiers::NONE || modifier == KeyModifiers::SHIFT)
                     {
-                        event
-                    } else if modifier == KeyModifiers::NONE || modifier == KeyModifiers::SHIFT {
                         self.cache.push(if modifier == KeyModifiers::SHIFT {
                             c.to_ascii_uppercase()
                         } else {
@@ -78,7 +81,7 @@ impl EditMode for Vi {
                         });
 
                         let res = parse(&mut self.cache.iter().peekable());
-
+                        self.seq_completed = res.is_complete();
                         if !res.is_valid() {
                             self.cache.clear();
                             ReedlineEvent::None
@@ -93,6 +96,8 @@ impl EditMode for Vi {
                         } else {
                             ReedlineEvent::None
                         }
+                    } else if let Some(event) = binding {
+                        event
                     } else {
                         ReedlineEvent::None
                     }
@@ -277,4 +282,45 @@ mod test {
 
         assert_eq!(result, ReedlineEvent::None);
     }
+
+    #[test]
+    fn find_custom_keybinding_test() {
+        let mut keybindings = default_vi_normal_keybindings();
+        keybindings.add_binding(
+            KeyModifiers::SHIFT,
+            KeyCode::Char('B'),
+            ReedlineEvent::Edit(vec![EditCommand::MoveBigWordLeft { select: false }]),
+        );
+        let mut vi = Vi {
+            insert_keybindings: default_vi_insert_keybindings(),
+            normal_keybindings: keybindings,
+            mode: ViMode::Normal,
+            ..Default::default()
+        };
+
+        let _ = vi.parse_event(
+            ReedlineRawEvent::convert_from(Event::Key(KeyEvent::new(
+                KeyCode::Char('f'),
+                KeyModifiers::NONE,
+            )))
+            .unwrap(),
+        );
+        let res = vi.parse_event(
+            ReedlineRawEvent::convert_from(Event::Key(KeyEvent::new(
+                KeyCode::Char('B'),
+                KeyModifiers::SHIFT,
+            )))
+            .unwrap(),
+        );
+
+        assert_eq!(
+            res,
+            ReedlineEvent::Multiple(vec![ReedlineEvent::Edit(vec![
+                EditCommand::MoveRightUntil {
+                    c: 'B',
+                    select: false
+                }
+            ])])
+        );
+    }
 }