benbot/nvim-treesitter
1
0
Fork 0
mirror of https://github.com/nvim-treesitter/nvim-treesitter.git synced 2026-07-04 04:20:09 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(refactor): highlight def perf issue fix

Browse source
This commit is contained in:
Steven Sojka 2020-07-07 07:51:19 -05:00 • committed by Thomas Vigouroux
parent 884080f89a
commit 26b3b3311e
2 changed files with 27 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

32
lua/nvim-treesitter/locals.lua
View file

@ -23,6 +23,24 @@ setmetatable(query_cache, default_dict)
local M = {}
function M.collect_locals(bufnr, query_kind)
local locals = {}
for prepared_match in M.iter_locals(bufnr, nil, query_kind) do
table.insert(locals, prepared_match)
end
return locals
end
local function update_cached_locals(bufnr, changed_tick, query_kind)
query_cache[query_kind][bufnr] = {tick=changed_tick, cache=( M.collect_locals(bufnr, query_kind) or {} )}
end
-- Iterates matches from a locals query file.
-- @param bufnr the buffer
-- @param root the root node
-- @param query_kind the query file to use
function M.iter_locals(bufnr, root, query_kind)
query_kind = query_kind or 'locals'
local lang = parsers.ft_to_lang(api.nvim_buf_get_option(bufnr, "ft"))
@ -34,20 +52,10 @@ function M.collect_locals(bufnr, query_kind)
local parser = parsers.get_parser(bufnr, lang)
if not parser then return end
local root = parser:parse():root()
local root = root or parser:parse():root()
local start_row, _, end_row, _ = root:range()
local locals = {}
for prepared_match in queries.iter_prepared_matches(query, root, bufnr, start_row, end_row) do
table.insert(locals, prepared_match)
end
return locals
end
local function update_cached_locals(bufnr, changed_tick, query_kind)
query_cache[query_kind][bufnr] = {tick=changed_tick, cache=( M.collect_locals(bufnr, query_kind) or {} )}
return queries.iter_prepared_matches(query, root, bufnr, start_row, end_row)
end
function M.get_locals(bufnr, query_kind)