Experimental: Automatically restore mistyped words

huytd · Jul 8, 2023 · 36f5b34 · 36f5b34
1 parent 4d5cebc
commit 36f5b34
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/src/input.rs b/src/input.rs
@@ -18,7 +18,7 @@ use crate::{
 // be around 10 to 12.
 const MAX_POSSIBLE_WORD_LENGTH: usize = 10;
 const MAX_DUPLICATE_LENGTH: usize = 4;
-const TONE_DUPLICATE_PATTERNS: [&str; 5] = ["ss", "ff", "jj", "rr", "xx"];
+const TONE_DUPLICATE_PATTERNS: [&str; 6] = ["ss", "ff", "jj", "rr", "xx", "ww"];
 
 pub static mut INPUT_STATE: Lazy<InputState> = Lazy::new(InputState::new);
 
@@ -184,6 +184,14 @@ impl InputState {
         self.should_track = true;
     }
 
+    pub fn get_typing_buffer(&self) -> &str {
+        &self.buffer
+    }
+
+    pub fn get_displaying_word(&self) -> &str {
+        &self.display_buffer
+    }
+
     pub fn stop_tracking(&mut self) {
         self.clear();
         self.should_track = false;

diff --git a/src/main.rs b/src/main.rs
@@ -37,6 +37,18 @@ fn do_transform_keys(handle: Handle, is_delete: bool) -> bool {
     false
 }
 
+fn do_restore_word(handle: Handle) {
+    unsafe {
+        let backspace_count = INPUT_STATE.get_backspace_count(true);
+        debug!("Backspace count: {}", backspace_count);
+        _ = send_backspace(handle, backspace_count);
+        let typing_buffer = INPUT_STATE.get_typing_buffer();
+        _ = send_string(handle, typing_buffer);
+        debug!("Sent: {:?}", typing_buffer);
+        INPUT_STATE.replace(typing_buffer.to_owned());
+    }
+}
+
 fn event_handler(handle: Handle, keycode: Option<char>, modifiers: KeyModifier) -> bool {
     unsafe {
         match keycode {
@@ -54,6 +66,9 @@ fn event_handler(handle: Handle, keycode: Option<char>, modifiers: KeyModifier)
                 if INPUT_STATE.is_enabled() {
                     match keycode {
                         KEY_ENTER | KEY_TAB | KEY_SPACE | KEY_ESCAPE => {
+                            if !vi::validation::is_valid_word(INPUT_STATE.get_displaying_word()) {
+                                do_restore_word(handle);
+                            }
                             INPUT_STATE.new_word();
                         }
                         KEY_DELETE => {