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

Add condition function to module config

Browse source 
The function is called with the language and bufnr, if it returns false,
the module is disabled for that buffer.

This gives the user more fine-grained control over whether a module is
started.
This commit is contained in:
patrick96 2021-11-09 18:24:35 +01:00 • committed by Stephan Seitz
parent 6888242b98
commit e3dabec462
2 changed files with 12 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -335,8 +335,9 @@ M.commands = {
}
-- @param mod: module (string)
-- @param ft: filetype (string)
function M.is_enabled(mod, lang)
-- @param lang: the language of the buffer (string)
-- @param bufnr: the bufnr (number)
function M.is_enabled(mod, lang, bufnr)
if not parsers.list[lang] or not parsers.has_parser(lang) then
return false
end
@ -350,6 +351,10 @@ function M.is_enabled(mod, lang)
return false
end
if module_config.cond and not module_config.cond(lang, bufnr) then
return false
end
for _, parser in pairs(module_config.disable) do
if lang == parser then
return false
@ -445,7 +450,7 @@ function M.attach_module(mod_name, bufnr, lang)
local lang = lang or parsers.get_buf_lang(bufnr)
local resolved_mod = resolve_module(mod_name)
if resolved_mod and not attached_buffers_by_module.has(mod_name, bufnr) and M.is_enabled(mod_name, lang) then
if resolved_mod and not attached_buffers_by_module.has(mod_name, bufnr) and M.is_enabled(mod_name, lang, bufnr) then
attached_buffers_by_module.set(mod_name, bufnr, true)
resolved_mod.attach(bufnr, lang)
end