From c3e6369db6c7d091ea3665a458326c7e099c2e47 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Sat, 22 Aug 2020 23:32:35 +0200
Subject: [PATCH] feat(config): add global disable option

---
 README.md                       |  1 +
 doc/nvim-treesitter.txt         |  1 +
 lua/nvim-treesitter/configs.lua | 11 ++++++++---
 3 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index f5689d98f..0e5a7960a 100644
--- a/README.md
+++ b/README.md
@@ -83,6 +83,7 @@ so you'll need to activate them by putting this in your `init.vim` file:
 lua <<EOF
 require'nvim-treesitter.configs'.setup {
   ensure_installed = "all"      -- one of "all", "language", or a list of languages
+  disable = {"markdown"},       -- a list of languages to disable for all modules (only enable manually)
   highlight = {
     enable = true,              -- false will disable the whole extension
     disable = { "c", "rust" },  -- list of language that will be disabled
diff --git a/doc/nvim-treesitter.txt b/doc/nvim-treesitter.txt
index 38c0378e6..c834dafea 100644
--- a/doc/nvim-treesitter.txt
+++ b/doc/nvim-treesitter.txt
@@ -37,6 +37,7 @@ To enable supported features, put this in your `init.vim` file:
   lua <<EOF
   require'nvim-treesitter.configs'.setup {
     ensure_installed = "all"      -- one of "all", "language", or a list of languages
+    disable = {"markdown"},       -- a list of languages to disable for all modules (only enable manually)
     highlight = {
       enable = true,              -- false will disable the whole extension
       disable = { "c", "rust" },  -- list of language that will be disabled
diff --git a/lua/nvim-treesitter/configs.lua b/lua/nvim-treesitter/configs.lua
index 51d982c79..044b3cf86 100644
--- a/lua/nvim-treesitter/configs.lua
+++ b/lua/nvim-treesitter/configs.lua
@@ -300,17 +300,22 @@ function M.setup(user_data)
       config.ensure_installed = data
       require'nvim-treesitter.install'.ensure_installed(data)
     else
+      data.disable = data.disable or {}
+      for _, lang in ipairs(user_data.disable or {}) do
+        if not vim.tbl_contains(data.disable, lang) then
+          table.insert(data.disable, lang)
+        end
+      end
       config.modules[name] = vim.tbl_deep_extend('force', config.modules[name] or {}, data)
 
       recurse_modules(function(_, _, new_path)
         if data.enable then
           enable_all(new_path)
         end
-
-        for _, lang in ipairs(data.disable or {}) do
+        for _, lang in ipairs(config.modules[name].disable) do
           disable_mod_conf_autocmd(new_path, lang)
         end
-      end, config.modules)
+      end, config.modules[name])
     end
   end
 end