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

feat: add locals to setup procedure

Browse source
This commit is contained in:
Thomas Vigouroux 2020-04-19 14:20:23 +02:00
parent 0d5ef2f980
commit b2b6e51a7c
2 changed files with 19 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -10,9 +10,12 @@ local M = {
locals={}
}
function M.is_supported(lang)
return queries.get_query(lang, "locals") ~= nil
end
function M.collect_locals(bufnr)
local ft = api.nvim_buf_get_option(bufnr, "ft")
local query = queries.get_query(ft, "locals")
if ft then
local query = queries.get_query(ft, 'locals')
@ -33,12 +36,16 @@ function M.collect_locals(bufnr)
end
end
function M.on_lines(_, buf, _, firstline, lastline, new_lastline, _)
function M.on_lines(_, buf, _, firstline, lastline, new_lastline)
M.locals[buf] = M.collect_locals(buf)
end
function M.get_locals(bufnr)
return M.locals[bufnr or api.nvim_get_current_buf()] or {}
end
function M.get_definitions(bufnr)
local locals = M.locals[bufnr]
local locals = M.get_locals(bufnr)
local defs = {}
@ -52,7 +59,7 @@ function M.get_definitions(bufnr)
end
function M.get_scopes(bufnr)
local locals = M.locals[bufnr]
local locals = M.get_locals(bufnr)
local scopes = {}
@ -66,7 +73,7 @@ function M.get_scopes(bufnr)
end
function M.get_references(bufnr)
local locals = M.locals[bufnr or api.nvim_get_current_buf()]
local locals = M.get_locals(bufnr)
local refs = {}