GitHub/neovim
1
0
Fork 0
mirror of https://github.com/neovim/neovim synced 2025-07-16 01:01:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(treesitter): fix :InspectTree incorrect injections

Browse Source
This commit is contained in:
yuukibarns
2025-03-30 23:06:16 +08:00
committed by Lewis Russell
parent 223ac7782e
commit 284b0e4fa2
1 changed files with 13 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
runtime/lua/vim/treesitter/dev.lua
View File

@ -1,5 +1,7 @@
local api = vim.api
local Range = require('vim.treesitter._range')
local M = {}
---@class (private) vim.treesitter.dev.TSTreeView
@ -96,16 +98,20 @@ function TSTreeView:new(bufnr, lang)
parser:for_each_tree(function(parent_tree, parent_ltree)
local parent = parent_tree:root()
local parent_range = { parent:range() }
for _, child in pairs(parent_ltree:children()) do
for _, tree in pairs(child:trees()) do
local r = tree:root()
local node = assert(parent:named_descendant_for_range(r:range()))
local id = node:id()
if not injections[id] or r:byte_length() > injections[id].root:byte_length() then
injections[id] = {
lang = child:lang(),
root = r,
}
local r_range = { r:range() }
if Range.contains(parent_range, r_range) then
local node = assert(parent:named_descendant_for_range(r:range()))
local id = node:id()
if not injections[id] or r:byte_length() > injections[id].root:byte_length() then
injections[id] = {
lang = child:lang(),
root = r,
}
end
end
end
end