From 058e8d2296515041be982c6f23c119ec6b6d1ba9 Mon Sep 17 00:00:00 2001 From: Steven Sojka Date: Thu, 25 Jun 2020 13:26:31 -0500 Subject: feat(refactor): highlight usages module --- lua/nvim-treesitter/locals.lua | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'lua/nvim-treesitter/locals.lua') diff --git a/lua/nvim-treesitter/locals.lua b/lua/nvim-treesitter/locals.lua index 8c7d400e..fae8195e 100644 --- a/lua/nvim-treesitter/locals.lua +++ b/lua/nvim-treesitter/locals.lua @@ -11,7 +11,7 @@ local M = { locals = {} } -function M.collect_locals(bufnr) +function M.collect_locals(bufnr, root) local lang = parsers.ft_to_lang(api.nvim_buf_get_option(bufnr, "ft")) if not lang then return end @@ -21,7 +21,7 @@ function M.collect_locals(bufnr) 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 = {} -- cgit v1.2.3