Extend textobject selections in select mode

Author: the-mikedavis

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);