refactor: save TSNode directly

nvim-treesitter · Sep 7, 2024 · d35952c · d35952c
1 parent 5c6db01
commit d35952c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lua/nvim-treesitter-textobjects/shared.lua b/lua/nvim-treesitter-textobjects/shared.lua
@@ -159,8 +159,8 @@ local function get_capture_ranges_recursively(bufnr, query_string, query_group)
     for _, match in pairs(matches) do
       local found = get_at_path(match, query_string)
       if found then
-        ---@cast found Range6
-        table.insert(ranges, found)
+        ---@cast found TSNode
+        table.insert(ranges, { found:range(true) })
       end
     end
   end)
@@ -194,7 +194,9 @@ function M.find_best_range(bufnr, capture_string, query_group, filter_predicate,
 
   local matches = get_query_matches(bufnr, query_group, root, lang)
   for _, maybe_match in pairs(matches) do
-    local range = get_at_path(maybe_match, capture_string)
+    local node = get_at_path(maybe_match, capture_string)
+    ---@cast node TSNode
+    local range = { node:range(true) }
     ---@cast range Range6
 
     if range and filter_predicate(range) then