Extend textobject selections in select mode

helix-editor · Jul 30, 2022 · 9c5f716 · 9c5f716
1 parent 73ecf6c
commit 9c5f716
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/helix-term/src/commands.rs b/helix-term/src/commands.rs
@@ -4226,16 +4226,24 @@ fn goto_ts_object_impl(cx: &mut Context, object: &'static str, direction: Direct
             let root = syntax.tree().root_node();
 
             let selection = doc.selection(view.id).clone().transform(|range| {
-                movement::goto_treesitter_object(
+                let new_range = movement::goto_treesitter_object(
                     text,
                     range,
                     object,
                     direction,
                     root,
                     lang_config,
                     count,
-                )
-                .with_direction(direction)
+                );
+
+                if doc.mode == Mode::Select {
+                    Range::new(
+                        range.anchor,
+                        new_range.with_direction(range.direction()).head,
+                    )
+                } else {
+                    new_range.with_direction(direction)
+                }
             });
 
             doc.set_selection(view.id, selection);