Add option to get ts headline at specific cursor

Author: kristijanhusak

lua/orgmode/treesitter/init.lua

@@ -2,9 +2,10 @@ local tree_utils = require('orgmode.utils.treesitter')
 local Headline = require('orgmode.treesitter.headline')
 local Listitem = require('orgmode.treesitter.listitem')
 
+---@param cursor? Table Cursor position tuple {row, col}
 ---@return Headline
-local function closest_headline()
-  local ts_headline = tree_utils.closest_headline()
+local function closest_headline(cursor)
+  local ts_headline = tree_utils.closest_headline(cursor)
   if not ts_headline then
     error('Unable to locate closest headline')
   end

lua/orgmode/utils/treesitter.lua

@@ -1,11 +1,47 @@
 local ts_utils = require('nvim-treesitter.ts_utils')
+local parsers = require('nvim-treesitter.parsers')
 local M = {}
 
 function M.current_node()
   local window = vim.api.nvim_get_current_win()
   return ts_utils.get_node_at_cursor(window)
 end
 
+---This is a full copy of nvim_treesiter get_node_at_cursor with support for custom cursor position
+---@param cursor? Table Cursor position tuple {row, col}
+---@param winnr? number
+---@param ignore_injected_langs? boolean
+function M.get_node_at_cursor(cursor, winnr, ignore_injected_langs)
+  winnr = winnr or 0
+  cursor = cursor or vim.api.nvim_win_get_cursor(winnr)
+  local cursor_range = { cursor[1] - 1, cursor[2] }
+
+  local buf = vim.api.nvim_win_get_buf(winnr)
+  local root_lang_tree = parsers.get_parser(buf)
+  if not root_lang_tree then
+    return
+  end
+
+  local root
+  if ignore_injected_langs then
+    for _, tree in ipairs(root_lang_tree:trees()) do
+      local tree_root = tree:root()
+      if tree_root and ts_utils.is_in_node_range(tree_root, cursor_range[1], cursor_range[2]) then
+        root = tree_root
+        break
+      end
+    end
+  else
+    root = ts_utils.get_root_for_position(cursor_range[1], cursor_range[2], root_lang_tree)
+  end
+
+  if not root then
+    return
+  end
+
+  return root:named_descendant_for_range(cursor_range[1], cursor_range[2], cursor_range[1], cursor_range[2])
+end
+
 -- walks the tree to find a headline
 function M.find_headline(node)
   if node:type() == 'headline' then
@@ -21,9 +57,9 @@ function M.find_headline(node)
 end
 
 -- returns the nearest headline
-function M.closest_headline()
+function M.closest_headline(cursor)
   vim.treesitter.get_parser(0, 'org', {}):parse()
-  return M.find_headline(M.current_node())
+  return M.find_headline(M.get_node_at_cursor(cursor))
 end
 
 function M.find_parent_type(node, type)