From f7d92f663c07b9be35ceea0342c215ca396c48fb Mon Sep 17 00:00:00 2001 From: Thomas Vigouroux Date: Thu, 1 Oct 2020 21:38:38 +0200 Subject: feat(languagetree): implement language tree Allow the LanguageTree to be used as an option for highlighting. Co-authored-by: Santos Gallegos Co-authored-by: Yazdani Kiyan --- lua/nvim-treesitter/highlight.lua | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) (limited to 'lua/nvim-treesitter/highlight.lua') diff --git a/lua/nvim-treesitter/highlight.lua b/lua/nvim-treesitter/highlight.lua index ee68a4ad..7d21bed6 100644 --- a/lua/nvim-treesitter/highlight.lua +++ b/lua/nvim-treesitter/highlight.lua @@ -81,14 +81,19 @@ function M.attach(bufnr, lang) local parser = parsers.get_parser(bufnr, lang) local config = configs.get_module('highlight') - for k, v in pairs(config.custom_captures) do - hlmap[k] = v + if config.use_languagetree then + local ltree = require'nvim-treesitter.languagetree' + ltree.new(bufnr, lang) + else + for k, v in pairs(config.custom_captures) do + hlmap[k] = v + end + + local query = queries.get_query(lang, "highlights") + if not query then return end + + M.highlighters[bufnr] = ts.highlighter.new(parser, query) end - - local query = queries.get_query(lang, "highlights") - if not query then return end - - M.highlighters[bufnr] = ts.highlighter.new(parser, query) end function M.detach(bufnr) -- cgit v1.2.3