feat(enhancement): support nearest parent

Fixes #453

Author: isakbm

README.md

@@ -404,9 +404,13 @@ hi TreesitterContextLineNumberBottom gui=underline guisp=Grey
 ## Jumping to context (upwards)
 
 ```lua
-vim.keymap.set("n", "[c", function()
+vim.keymap.set("n", "[C", function()
   require("treesitter-context").go_to_context(vim.v.count1)
 end, { silent = true })
+
+vim.keymap.set("n", "[c", function()
+  require("treesitter-context").go_to_parent(vim.v.count1)
+end, { silent = true })
 ```
 
 ## Adding support for other languages

doc/nvim-treesitter-context.txt

@@ -120,14 +120,28 @@ go_to_context({depth})                 *nvim-treesitter-context-go_to_context()*
 
     Example use in a keymap:
 >lua
-        vim.keymap.set("n", "[c", function()
+        vim.keymap.set("n", "[C", function()
           require("treesitter-context").go_to_context(vim.v.count1)
         end, { silent = true })
 <
 
     Parameters: ~
         {depth} (`integer`, default: `1`) Depth to jump to.
 
+go_to_parent({depth})                 *nvim-treesitter-context-go_to_parent()*
+
+    Jump to the parent at {depth}.
+
+    Example use in a keymap:
+>lua
+        vim.keymap.set("n", "[c", function()
+          require("treesitter-context").go_to_parent(vim.v.count1)
+        end, { silent = true })
+<
+
+    Parameters: ~
+        {depth} (`integer`, default: `1`) Depth to jump to.
+
 
 setup({config})                                *nvim-treesitter-context-setup()*
 

lua/treesitter-context.lua

@@ -26,6 +26,9 @@ local Render = defer_require('treesitter-context.render')
 --- @type table<integer, Range4[]>
 local all_contexts = {}
 
+--- @type table<integer, Range4[]>
+local all_parent_ranges = {}
+
 --- Schedule a function to run on the next event loop iteration.
 --- If the function is called again within 150ms, it will be scheduled
 --- again to run on the next event loop iteration. This means that
@@ -115,8 +118,10 @@ local update_single_context = throttle_by_id(function(winid)
     return
   end
 
-  local context_ranges, context_lines = require('treesitter-context.context').get(bufnr, winid)
+  local context_ranges, context_lines, parent_ranges =
+    require('treesitter-context.context').get(bufnr, winid)
   all_contexts[bufnr] = context_ranges
+  all_parent_ranges[bufnr] = parent_ranges
 
   if not context_ranges or #context_ranges == 0 then
     Render.close(winid)
@@ -310,13 +315,11 @@ function M.setup(options)
   end
 end
 
---- @param depth integer? default 1
-function M.go_to_context(depth)
-  depth = depth or 1
+--- @param depth integer
+--- @param contexts Range4[]
+function go_to(depth, contexts)
   local line = api.nvim_win_get_cursor(0)[1]
   local context = nil
-  local bufnr = api.nvim_get_current_buf()
-  local contexts = all_contexts[bufnr] or {}
 
   for idx = #contexts, 1, -1 do
     local c = contexts[idx]
@@ -333,8 +336,29 @@ function M.go_to_context(depth)
     return
   end
 
-  vim.cmd([[ normal! m' ]]) -- add current cursor position to the jump list
+  vim.cmd([[normal! m']]) -- add current cursor position to the jump list
   api.nvim_win_set_cursor(0, { context[1] + 1, context[2] })
+  vim.cmd([[normal! ^]]) -- place cursor at start of text
+end
+
+--- jump to the nearest parent
+---
+--- @param depth integer? default 1
+function M.go_to_parent(depth)
+  depth = depth or 1
+  local bufnr = api.nvim_get_current_buf()
+  local contexts = all_parent_ranges[bufnr] or {}
+  go_to(depth, contexts)
+end
+
+--- jump to the nearest context parent
+---
+--- @param depth integer? default 1
+function M.go_to_context(depth)
+  depth = depth or 1
+  local bufnr = api.nvim_get_current_buf()
+  local contexts = all_contexts[bufnr] or {}
+  go_to(depth, contexts)
 end
 
 return M

lua/treesitter-context/context.lua

@@ -316,7 +316,7 @@ end
 
 --- @param bufnr integer
 --- @param winid integer
---- @return Range4[]?, string[]?
+--- @return Range4[]?, string[]?, Range4[]?
 function M.get(bufnr, winid)
   -- vim.treesitter.get_parser() calls bufload(), but we don't actually want to load the buffer:
   -- this method is called during plugin init, before other plugins or the user's config
@@ -345,6 +345,7 @@ function M.get(bufnr, winid)
   end
 
   local context_ranges = {} --- @type Range4[]
+  local parent_ranges = {} --- @type Range4[]
   local context_lines = {} --- @type string[][]
   local contexts_height = 0
 
@@ -354,6 +355,7 @@ function M.get(bufnr, winid)
     local line_range = { node_row, col0, node_row, col0 + 1 }
 
     context_ranges = {}
+    parent_ranges = {}
     context_lines = {}
     contexts_height = 0
 
@@ -366,7 +368,7 @@ function M.get(bufnr, winid)
 
         local contexts_end_row = top_row + separator_offset + math.min(max_lines, contexts_height)
 
-        -- Only process the parent if it is not in view.
+        -- context_ranges: Only process the parent if it is not in view.
         if parent_start_row < contexts_end_row then
           local range0 = context_range(parent, bufnr, query)
           if range0 and range_is_valid(range0) then
@@ -386,6 +388,15 @@ function M.get(bufnr, winid)
             end
           end
         end
+
+        -- parent_ranges: Process all
+        local range0 = context_range(parent, bufnr, query)
+        if range0 and range_is_valid(range0) then
+          local range, _lines = get_text_for_range(range0, bufnr)
+          if range_is_valid(range) then
+            parent_ranges[#parent_ranges + 1] = range
+          end
+        end
       end
     end
 
@@ -401,7 +412,7 @@ function M.get(bufnr, winid)
     trim_contexts(context_ranges, context_lines, trim, config.trim_scope == 'outer')
   end
 
-  return context_ranges, tbl_flatten(context_lines)
+  return context_ranges, tbl_flatten(context_lines), parent_ranges
 end
 
 return M