fix(treesitter): fixup for InspectTree

Fixes #25120
This commit is contained in:
Lewis Russell 2023-09-12 15:26:57 +01:00 committed by Lewis Russell
parent bc67cf3ccd
commit 1f551e068f

View File

@ -101,16 +101,18 @@ function TSTreeView:new(bufnr, lang)
-- the root in the child tree to the {injections} table.
local root = parser:parse(true)[1]:root()
local injections = {} ---@type table<integer,table>
parser:for_each_tree(function(tree, ltree)
local r = tree:root()
local node = root:named_descendant_for_range(r:range())
if node then
injections[node:id()] = {
lang = ltree:lang(),
root = r,
}
end
end)
for _, child in pairs(parser:children()) do
child:for_each_tree(function(tree, ltree)
local r = tree:root()
local node = root:named_descendant_for_range(r:range())
if node then
injections[node:id()] = {
lang = ltree:lang(),
root = r,
}
end
end)
end
local nodes = traverse(root, 0, parser:lang(), injections, {})