fix(treesitter): fixup for InspectTree

Fixes #25120
This commit is contained in:
Lewis Russell
2023-09-12 15:26:57 +01:00
parent bc67cf3ccd
commit 040cba1faa

View File

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