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

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