mirrors/neovim
1
0
Fork 0
mirror of https://github.com/neovim/neovim.git synced 2026-08-31 03:13:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

treesitter: default start and end row when omitted

Browse Source 
Add support for default start and end row when omitted in the
query:iter_captures and query:iter_matches functions.

When the start and end row values are omitted, the values of the given
node is used. The end row value is incremented by 1 to include the node end
row in the match.

Updated tests and docs accordingly.
This commit is contained in:
“jvgrootveld”
2021-01-15 21:44:40 +01:00
parent 0fad952a2b
commit a2818819bb
3 changed files with 43 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
runtime/lua/vim/treesitter/query.lua
View File

@@ -353,6 +353,12 @@ function Query:iter_captures(node, source, start, stop)
if type(source) == "number" and source == 0 then
source = vim.api.nvim_get_current_buf()
end
if start == nil and stop == nil then
start, _, stop, _ = node:range()
stop = stop + 1 -- Make stop inclusive
end
local raw_iter = node:_rawquery(self.query, true, start, stop)
local function iter()
local capture, captured_node, match = raw_iter()
@@ -385,6 +391,12 @@ function Query:iter_matches(node, source, start, stop)
if type(source) == "number" and source == 0 then
source = vim.api.nvim_get_current_buf()
end
if start == nil and stop == nil then
start, _, stop, _ = node:range()
stop = stop + 1 -- Make stop inclusive
end
local raw_iter = node:_rawquery(self.query, false, start, stop)
local function iter()
local pattern, match = raw_iter()