From 282e33ad9c96a44a092ab8a356bba7626c838b68 Mon Sep 17 00:00:00 2001 From: Steven Sojka Date: Wed, 12 Aug 2020 07:38:15 -0500 Subject: fix(definitions): optimize and fix definition highlighting --- lua/nvim-treesitter/query.lua | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'lua/nvim-treesitter/query.lua') diff --git a/lua/nvim-treesitter/query.lua b/lua/nvim-treesitter/query.lua index cd2495af..33637be3 100644 --- a/lua/nvim-treesitter/query.lua +++ b/lua/nvim-treesitter/query.lua @@ -203,7 +203,6 @@ end -- @param query_group the query file to use -- @param root the root node function M.iter_group_results(bufnr, query_group, root) - local lang = parsers.get_buf_lang(bufnr) if not lang then return end @@ -216,7 +215,8 @@ function M.iter_group_results(bufnr, query_group, root) local root = root or parser:parse():root() local start_row, _, end_row, _ = root:range() - return M.iter_prepared_matches(query, root, bufnr, start_row, end_row) + -- The end row is exclusive so we need to add 1 to it. + return M.iter_prepared_matches(query, root, bufnr, start_row, end_row + 1) end function M.collect_group_results(bufnr, query_group, root) -- cgit v1.2.3