fix(swap): preserve swapped nodes' range metadata (#550)

Author: ribru17

lua/nvim-treesitter/textobjects/shared.lua

@@ -332,7 +332,7 @@ function M.next_textobject(node, query_string, query_group, same_parent, overlap
 
   local next_node = queries.find_best_match(bufnr, query_string, query_group, filter_function, scoring_function)
 
-  return next_node and next_node.node
+  return next_node and next_node.node, next_node.metadata
 end
 
 function M.previous_textobject(node, query_string, query_group, same_parent, overlapping_range_ok, bufnr)
@@ -367,7 +367,7 @@ function M.previous_textobject(node, query_string, query_group, same_parent, ove
 
   local previous_node = queries.find_best_match(bufnr, query_string, query_group, filter_function, scoring_function)
 
-  return previous_node and previous_node.node
+  return previous_node and previous_node.node, previous_node.metadata
 end
 
 return M

lua/nvim-treesitter/textobjects/swap.lua

@@ -27,9 +27,9 @@ local function swap_textobject(query_strings_regex, query_group, direction)
   local same_parent = true
   for _ = 1, math.abs(direction), step do
     local forward = direction > 0
-    local adjacent =
+    local adjacent, metadata =
       shared.get_adjacent(forward, node, query_string, query_group, same_parent, overlapping_range_ok, bufnr)
-    ts_utils.swap_nodes(textobject_range, adjacent, bufnr, "yes, set cursor!")
+    ts_utils.swap_nodes(textobject_range, metadata and metadata.range or adjacent, bufnr, "yes, set cursor!")
   end
 end