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

@@ -116,18 +116,41 @@ enable()                                      *nvim-treesitter-context-enable()*
 
 go_to_context({depth})                 *nvim-treesitter-context-go_to_context()*
 
-    Jump to the context at {depth}.
+    Jump to parent scope within the context
+    window at {depth}.
+
+    A depth of 1 implies nearest, 2 second
+    nearest and so on. Set to `vim.v.count1`
+    to support motions with counts as depth.
 
     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 parent scope at {depth}.
+
+    A depth of 1 implies nearest, 2 second
+    nearest and so on. Set to `vim.v.count1`
+    to support motions with counts as 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,31 +315,59 @@ 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 ranges Range4[]
+local function go_to(depth, ranges)
   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 {}
+  local range = nil
 
-  for idx = #contexts, 1, -1 do
-    local c = contexts[idx]
+  for idx = #ranges, 1, -1 do
+    local c = ranges[idx]
     if depth == 0 then
       break
     end
     if c[1] + 1 < line then
-      context = c
+      range = c
       depth = depth - 1
     end
   end
 
-  if context == nil then
+  if range == nil then
     return
   end
 
-  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! m']]) -- add current cursor position to the jump list
+  api.nvim_win_set_cursor(0, { range[1] + 1, range[2] })
+  vim.cmd([[normal! ^]]) -- place cursor at start of text
+end
+
+--- Jump to parent scope at depth.
+---
+--- A depth of 1 implies nearest, 2 second
+--- nearest and so on. Set to `vim.v.count1`
+--- to support motions with counts as depth.
+---
+--- @param depth integer? default 1
+function M.go_to_parent(depth)
+  depth = depth or 1
+  local bufnr = api.nvim_get_current_buf()
+  local ranges = all_parent_ranges[bufnr] or {}
+  go_to(depth, ranges)
+end
+
+--- Jump to parent scope within the context
+--- window at depth.
+---
+--- A depth of 1 implies nearest, 2 second
+--- nearest and so on. Set to `vim.v.count1`
+--- to support motions with counts as depth.
+---
+--- @param depth integer? default 1
+function M.go_to_context(depth)
+  depth = depth or 1
+  local bufnr = api.nvim_get_current_buf()
+  local ranges = all_contexts[bufnr] or {}
+  go_to(depth, ranges)
 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