Fix visual objects

Adds 'a'/'i' in visual mode
2023-08-17 17:03:54 -06:00 · 2023-08-17 17:03:54 -06:00 · b0ba0f8851
commit b0ba0f8851
parent d4276acab8
4 changed files with 107 additions and 15 deletions
--- a/crates/vim/src/object.rs
+++ b/crates/vim/src/object.rs
@ -84,6 +84,35 @@ impl Object {
            | Object::SquareBrackets => true,
        }
    }
+
+    pub fn always_expands_both_ways(self) -> bool {
+        match self {
+            Object::Word { .. } | Object::Sentence => false,
+            Object::Quotes
+            | Object::BackQuotes
+            | Object::DoubleQuotes
+            | Object::Parentheses
+            | Object::SquareBrackets
+            | Object::CurlyBrackets
+            | Object::AngleBrackets => true,
+        }
+    }
+
+    pub fn target_visual_mode(self, current_mode: Mode) -> Mode {
+        match self {
+            Object::Word { .. } if current_mode == Mode::VisualLine => Mode::Visual,
+            Object::Word { .. } => current_mode,
+            Object::Sentence
+            | Object::Quotes
+            | Object::BackQuotes
+            | Object::DoubleQuotes
+            | Object::Parentheses
+            | Object::SquareBrackets
+            | Object::CurlyBrackets
+            | Object::AngleBrackets => Mode::Visual,
+        }
+    }
+
    pub fn range(
        self,
        map: &DisplaySnapshot,