benbot/nvim-treesitter
1
0
Fork 0
mirror of https://github.com/nvim-treesitter/nvim-treesitter.git synced 2026-07-06 21:40:03 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix #167: Add custom_captures config key to set highlights for custom queries

Browse source
This commit is contained in:
Stephan Seitz 2020-07-12 14:51:11 +02:00 • committed by Thomas Vigouroux
parent f803c74ebc
commit 97ad374816
4 changed files with 33 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
lua/nvim-treesitter/configs.lua
View file

@ -18,6 +18,7 @@ local builtin_modules = {
module_path = 'nvim-treesitter.highlight',
enable = false,
disable = {},
custom_captures = {},
is_supported = queries.has_highlights
},
incremental_selection = {
@ -61,6 +62,8 @@ local builtin_modules = {
}
}
local special_config_keys = {'enable', 'disable', 'keymaps'}
local M = {}
-- Resolves a module by requiring the `module_path` or using the module definition.
@ -276,6 +279,12 @@ function M.setup_module(mod, data, mod_name)
end
end
end
for k, v in pairs(data) do
-- Just copy all non-special configuration keys
if not vim.tbl_contains(special_config_keys, k) then
mod[k] = v
end
end
elseif type(data) == 'table' and type(mod) == 'table' then
for key, value in pairs(mod) do
M.setup_module(value, data[key], key)

6
lua/nvim-treesitter/highlight.lua
View file

@ -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