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/query.lua | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'lua/nvim-treesitter/query.lua') diff --git a/lua/nvim-treesitter/query.lua b/lua/nvim-treesitter/query.lua index 42e83582..d04e368a 100644 --- a/lua/nvim-treesitter/query.lua +++ b/lua/nvim-treesitter/query.lua @@ -13,6 +13,12 @@ local function read_query_files(filenames) return table.concat(contents, '\n') end +local function get_query_gaurd(query) + return function(lang) + return M.get_query(lang, query) ~= nil + end +end + -- Some treesitter grammars extend others. -- We can use that to import the queries of the base language M.base_language_map = { @@ -21,6 +27,9 @@ M.base_language_map = { tsx = {'typescript', 'javascript'}, } +M.has_locals = get_query_gaurd('locals') +M.has_highlights = get_query_gaurd('highlights') + function M.get_query(lang, query_name) local query_files = api.nvim_get_runtime_file(string.format('queries/%s/%s.scm', lang, query_name), true) local query_string = '' -- cgit v1.2.3