mirror of
https://github.com/nvim-treesitter/nvim-treesitter.git
synced 2026-07-01 19:17:02 -04:00
45dcebb15f
features:
- node_movement is moving between scopes.
- add selection initialization from normal mode
- add a decremental selection
improvements:
- attach to buffer to run tree parsing on change
- run state update on CursorMoved
- the buffer state is:
```
{
cursor_pos = { row=row, col=col },
current_node = node_under_cursor,
selection = {
range = nil, -- activates when starting a selection
nodes = {} -- filling up when starting an incremental selection
},
parser = parser, -- parser for current buffer
}
```
- refacto all the modules reliant on parsing the tree, update the current nodes, get the current nodes...
fixes:
- fix has_parser to look for .so libraries
- fix should select the whole file when selection root in selection
92 lines
2 KiB
Lua
92 lines
2 KiB
Lua
-- Functions to handle locals
|
|
-- Locals are a generalization of definition and scopes
|
|
-- its the way nvim-treesitter uses to "understand" the code
|
|
local api = vim.api
|
|
local ts = vim.treesitter
|
|
|
|
local queries = require'nvim-treesitter.query'
|
|
local utils = require'nvim-treesitter.utils'
|
|
|
|
local M = {
|
|
locals = {}
|
|
}
|
|
|
|
function M.collect_locals(bufnr)
|
|
local ft = api.nvim_buf_get_option(bufnr, "ft")
|
|
if not ft then return end
|
|
|
|
local query = queries.get_query(ft, 'locals')
|
|
if not query then return end
|
|
|
|
local parser = utils.get_parser(bufnr, ft)
|
|
if not parser then return end
|
|
|
|
local root = 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)
|
|
M.locals[bufnr] = {tick=changed_tick, cache=( M.collect_locals(bufnr) or {} )}
|
|
end
|
|
|
|
function M.get_locals(bufnr)
|
|
local bufnr = bufnr or api.nvim_get_current_buf()
|
|
local cached_local = M.locals[bufnr]
|
|
if not cached_local or api.nvim_buf_get_changedtick(bufnr) < cached_local.tick then
|
|
update_cached_locals(bufnr,api.nvim_buf_get_changedtick(bufnr))
|
|
end
|
|
|
|
return M.locals[bufnr].cache
|
|
end
|
|
|
|
function M.get_definitions(bufnr)
|
|
local locals = M.get_locals(bufnr)
|
|
|
|
local defs = {}
|
|
|
|
for _, loc in ipairs(locals) do
|
|
if loc.definition then
|
|
table.insert(defs, loc.definition)
|
|
end
|
|
end
|
|
|
|
return defs
|
|
end
|
|
|
|
function M.get_scopes(bufnr)
|
|
local locals = M.get_locals(bufnr)
|
|
|
|
local scopes = {}
|
|
|
|
for _, loc in ipairs(locals) do
|
|
if loc.scope and loc.scope.node then
|
|
table.insert(scopes, loc.scope.node)
|
|
end
|
|
end
|
|
|
|
return scopes
|
|
end
|
|
|
|
function M.get_references(bufnr)
|
|
local locals = M.get_locals(bufnr)
|
|
|
|
local refs = {}
|
|
|
|
for _, loc in ipairs(locals) do
|
|
if loc.reference and loc.reference.node then
|
|
table.insert(refs, loc.reference.node)
|
|
end
|
|
end
|
|
|
|
return refs
|
|
end
|
|
|
|
return M
|