diff options
| author | Stephan Seitz <stephan.seitz@fau.de> | 2020-07-12 14:51:11 +0200 |
|---|---|---|
| committer | Thomas Vigouroux <39092278+vigoux@users.noreply.github.com> | 2020-07-13 22:04:24 +0200 |
| commit | 97ad374816726661c3e5447fafd46264ad00409d (patch) | |
| tree | 24cb5de90e068c260994e10caaece5a849aa8207 /lua/nvim-treesitter/highlight.lua | |
| parent | f803c74ebc1e30ae8d047c087d9b094134a667d5 (diff) | |
Fix #167: Add custom_captures config key to set highlights for custom queries
Diffstat (limited to 'lua/nvim-treesitter/highlight.lua')
| -rw-r--r-- | lua/nvim-treesitter/highlight.lua | 6 |
1 files changed, 6 insertions, 0 deletions
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 |