fix: add desc node for child_containing_descendant

nvim-treesitter · Oct 2, 2024 · ddbc09b · ddbc09b
1 parent 17d9c1a
commit ddbc09b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lua/treesitter-context/context.lua b/lua/treesitter-context/context.lua
@@ -27,11 +27,12 @@ local function get_parent_nodes(langtree, range)
   local ret = {} --- @type TSNode[]
 
   if root.child_containing_descendant ~= nil then
-    local p = root
+    local p = root --- @type TSNode?
     while p do
       table.insert(ret, 1, p)
       p = p:child_containing_descendant(n)
     end
+    table.insert(ret, 1, n)
   else
     while n do
       ret[#ret + 1] = n