From 97ad374816726661c3e5447fafd46264ad00409d Mon Sep 17 00:00:00 2001 From: Stephan Seitz Date: Sun, 12 Jul 2020 14:51:11 +0200 Subject: Fix #167: Add custom_captures config key to set highlights for custom queries --- lua/nvim-treesitter/highlight.lua | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'lua/nvim-treesitter/highlight.lua') diff --git a/lua/nvim-treesitter/highlight.lua b/lua/nvim-treesitter/highlight.lua index 1b472243..0d4f3f54 100644 --- a/lua/nvim-treesitter/highlight.lua +++ b/lua/nvim-treesitter/highlight.lua @@ -3,6 +3,7 @@ local ts = vim.treesitter local queries = require'nvim-treesitter.query' local parsers = require'nvim-treesitter.parsers' +local configs = require'nvim-treesitter.configs' local M = { highlighters = {} @@ -54,6 +55,11 @@ hlmap["include"] = "TSInclude" function M.attach(bufnr, lang) local bufnr = bufnr or api.nvim_get_current_buf() local lang = lang or parsers.ft_to_lang(api.nvim_buf_get_option(bufnr, 'ft')) + local config = configs.get_module('highlight') + + 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 -- cgit v1.2.3