fix: replace get_node_text

Author: danilshvalov

lua/orgmode/org/indent.lua

@@ -25,7 +25,7 @@ local get_matches = ts_utils.memoize_by_buf_tick(function(bufnr)
       }
 
       if type == 'headline' then
-        opts.stars = vim.treesitter.query.get_node_text(node:field('stars')[1], bufnr):len()
+        opts.stars = vim.treesitter.get_node_text(node:field('stars')[1], bufnr):len()
         opts.indent = opts.indent + opts.stars + 1
         matches[range.start.line + 1] = opts
       end
@@ -52,7 +52,7 @@ local get_matches = ts_utils.memoize_by_buf_tick(function(bufnr)
         end
         if parent then
           local headline = parent:named_child('headline')
-          local stars = vim.treesitter.query.get_node_text(headline:field('stars')[1], bufnr):len()
+          local stars = vim.treesitter.get_node_text(headline:field('stars')[1], bufnr):len()
           opts.indent = stars + 1
           for i = range.start.line, range['end'].line - 1 do
             matches[i + 1] = opts

lua/orgmode/org/mappings.lua

@@ -599,7 +599,7 @@ function OrgMappings:handle_return(suffix)
       local counter = 1
       while next_sibling do
         local bullet = next_sibling:child(0)
-        local text = vim.treesitter.query.get_node_text(bullet, 0)
+        local text = vim.treesitter.get_node_text(bullet, 0)
         local new_text = tostring(tonumber(text:match('%d+')) + 1) .. closer
 
         if counter == 1 then

lua/orgmode/treesitter/headline.lua

@@ -4,7 +4,7 @@ local tree_utils = require('orgmode.utils.treesitter')
 local Date = require('orgmode.objects.date')
 local Range = require('orgmode.parser.range')
 local config = require('orgmode.config')
-local query = vim.treesitter.query
+local ts = vim.treesitter
 
 ---@class Headline
 ---@field headline userdata
@@ -46,7 +46,7 @@ end
 ---@return number
 function Headline:level()
   local stars = self:stars()
-  return query.get_node_text(stars, 0):len()
+  return ts.get_node_text(stars, 0):len()
 end
 
 function Headline:priority()
@@ -94,7 +94,7 @@ end
 
 function Headline:_handle_promote_demote(recursive, modifier)
   local whole_subtree = function()
-    local text = query.get_node_text(self.headline:parent(), 0)
+    local text = ts.get_node_text(self.headline:parent(), 0)
     local lines = modifier(vim.split(text, '\n', true))
     tree_utils.set_node_lines(self.headline:parent(), lines)
     return self:refresh()
@@ -128,7 +128,7 @@ function Headline:tags()
   local node = self.headline:field('tags')[1]
   local text = ''
   if node then
-    text = query.get_node_text(node, 0)
+    text = ts.get_node_text(node, 0)
   end
   return node, text
 end
@@ -142,7 +142,7 @@ function Headline:set_tags(tags)
     end
   end
 
-  local txt = query.get_node_text(predecessor, 0)
+  local txt = ts.get_node_text(predecessor, 0)
   local pred_end_row, pred_end_col, _ = predecessor:end_()
   local line = vim.fn.getline(pred_end_row + 1)
   local stars = line:match('^%*+%s*')
@@ -190,13 +190,13 @@ function Headline:set_priority(priority)
 
   local todo = self:todo()
   if todo then
-    local text = query.get_node_text(todo, 0)
+    local text = ts.get_node_text(todo, 0)
     tree_utils.set_node_text(todo, ('%s [#%s]'):format(text, priority))
     return
   end
 
   local stars = self:stars()
-  local text = query.get_node_text(stars, 0)
+  local text = ts.get_node_text(stars, 0)
   tree_utils.set_node_text(stars, ('%s [#%s]'):format(text, priority))
 end
 
@@ -209,7 +209,7 @@ function Headline:set_todo(keyword)
   end
 
   local stars = self:stars()
-  local text = query.get_node_text(stars, 0)
+  local text = ts.get_node_text(stars, 0)
   tree_utils.set_node_text(stars, string.format('%s %s', text, keyword))
 end
 
@@ -231,7 +231,7 @@ function Headline:todo()
     return nil
   end
 
-  local text = query.get_node_text(todo_node, 0)
+  local text = ts.get_node_text(todo_node, 0)
   for _, word in ipairs(keywords) do
     -- there may be multiple substitutions necessary
     local escaped_word = vim.pesc(word)
@@ -272,7 +272,7 @@ function Headline:is_done()
 end
 
 function Headline:title()
-  local title = query.get_node_text(self:item(), 0) or ''
+  local title = ts.get_node_text(self:item(), 0) or ''
   local todo, word = self:todo()
   if todo then
     title = title:gsub('^' .. vim.pesc(word) .. '%s*', '')
@@ -334,11 +334,11 @@ function Headline:get_property(property_name)
   for _, node in ipairs(ts_utils.get_named_children(properties)) do
     local name = node:field('name')[1]
     local value = node:field('value')[1]
-    if name and query.get_node_text(name, 0):lower() == property_name:lower() then
+    if name and ts.get_node_text(name, 0):lower() == property_name:lower() then
       return {
         node = node,
         name = name,
-        value = value and query.get_node_text(value, 0),
+        value = value and ts.get_node_text(value, 0),
       }
     end
   end
@@ -369,7 +369,7 @@ function Headline:dates()
   end
 
   for _, node in ipairs(ts_utils.get_named_children(plan)) do
-    local name = query.get_node_text(node:named_child(0), 0)
+    local name = ts.get_node_text(node:named_child(0), 0)
     dates[name] = node
   end
   return dates
@@ -445,7 +445,7 @@ function Headline:update_cookie(list_node)
   local cookie = self:cookie()
   if cookie then
     local new_cookie_val
-    if query.get_node_text(cookie, 0):find('%%') then
+    if ts.get_node_text(cookie, 0):find('%%') then
       new_cookie_val = ('[%d%%]'):format((#checked_boxes / #total_boxes) * 100)
     else
       new_cookie_val = ('[%d/%d]'):format(#checked_boxes, #total_boxes)
@@ -456,7 +456,7 @@ end
 
 function Headline:child_checkboxes(list_node)
   return vim.tbl_map(function(node)
-    local text = query.get_node_text(node, 0)
+    local text = ts.get_node_text(node, 0)
     return text:match('%[.%]')
   end, ts_utils.get_named_children(list_node))
 end
@@ -465,7 +465,7 @@ end
 function Headline:parse(pattern)
   local match = ''
   local matching_nodes = vim.tbl_filter(function(node)
-    local text = query.get_node_text(node, 0) or ''
+    local text = ts.get_node_text(node, 0) or ''
     local m = text:match(pattern)
     if m then
       match = text:match(pattern)
@@ -487,7 +487,7 @@ function Headline:_get_date(type)
   if not timestamp_node then
     return nil
   end
-  local parsed_date = Date.from_org_date(query.get_node_text(timestamp_node, 0), {
+  local parsed_date = Date.from_org_date(ts.get_node_text(timestamp_node, 0), {
     range = Range.from_node(timestamp_node),
   })
   return parsed_date and parsed_date[1] or nil
@@ -522,7 +522,7 @@ function Headline:_add_date(type, date, active)
       break
     end
   end
-  local ptext = query.get_node_text(last_child, 0)
+  local ptext = ts.get_node_text(last_child, 0)
   tree_utils.set_node_text(last_child, ptext .. ' ' .. text)
   return self:refresh()
 end

lua/orgmode/treesitter/listitem.lua

@@ -1,6 +1,6 @@
 local ts_utils = require('nvim-treesitter.ts_utils')
 local tree_utils = require('orgmode.utils.treesitter')
-local query = vim.treesitter.query
+local ts = vim.treesitter
 local Headline = require('orgmode.treesitter.headline')
 
 ---@class Listitem
@@ -36,7 +36,7 @@ function Listitem:checkbox()
   if not checkbox then
     return nil
   end
-  local text = query.get_node_text(checkbox, 0)
+  local text = ts.get_node_text(checkbox, 0)
   return { text = text, range = { checkbox:range() } }
 end
 
@@ -79,7 +79,7 @@ function Listitem:child_checkboxes()
   for _, content in ipairs(contents) do
     if content:type() == 'list' then
       return vim.tbl_map(function(node)
-        local text = query.get_node_text(node, 0)
+        local text = ts.get_node_text(node, 0)
         return text:match('%[.%]')
       end, ts_utils.get_named_children(content))
     end
@@ -94,7 +94,7 @@ function Listitem:cookie()
     return nil
   end
 
-  local text = query.get_node_text(cookie_node, 0)
+  local text = ts.get_node_text(cookie_node, 0)
   if text:match('%[%d*/%d*%]') or text:match('%[%d?%d?%d?%%%]') then
     return cookie_node
   end
@@ -104,7 +104,7 @@ function Listitem:update_cookie(total_child_checkboxes, checked_child_checkboxes
   local cookie = self:cookie()
   if cookie then
     local new_cookie_val
-    if query.get_node_text(cookie, 0):find('%%') then
+    if ts.get_node_text(cookie, 0):find('%%') then
       new_cookie_val = ('[%d%%]'):format((#checked_child_checkboxes / #total_child_checkboxes) * 100)
     else
       new_cookie_val = ('[%d/%d]'):format(#checked_child_checkboxes, #total_child_checkboxes)

lua/orgmode/treesitter/table.lua

@@ -2,7 +2,7 @@ local ts_utils = require('orgmode.utils.treesitter')
 local Table = require('orgmode.parser.table')
 local utils = require('orgmode.utils')
 local config = require('orgmode.config')
-local query = vim.treesitter.query
+local ts = vim.treesitter
 
 ---@class TsTable
 ---@field node userdata
@@ -54,7 +54,7 @@ function TsTable:_parse_data()
           local cell_val = ''
           local cell_content = cell:field('contents')
           if cell_content and #cell_content > 0 then
-            cell_val = query.get_node_text(cell_content[1], 0)
+            cell_val = ts.get_node_text(cell_content[1], 0)
           end
           table.insert(row_data, cell_val)
         end