Implement TSBufToggle.

doc/nvim-treesitter.txt

@@ -229,6 +229,13 @@ A list of modules can be found at |:TSModuleInfo|
 :TSBufDisable {module}~
 
 Disable {module} on the current buffer
+A list of modules can be found at |:TSModuleInfo|
+
+                                                                *:TSBufToggle*
+:TSBufToggle {module}~
+
+Toggle (enable if disabled, disable if enabled) {module} on the current
+buffer.
 A list of modules can be found at |:TSModuleInfo|
 
                                                              *:TSBufEnableAll*

lua/nvim-treesitter/configs.lua

@@ -136,6 +136,21 @@ local function disable_all(mod)
   config_mod.enable = false
 end
 
+-- Toggles a module for a buffer
+-- @param mod path to module
+-- @param bufnr buffer number, defaults to current buffer
+-- @param lang language, defaults to current language
+local function toggle_module(mod, bufnr, lang)
+  local bufnr = bufnr or api.nvim_get_current_buf()
+  local lang = lang or parsers.get_buf_lang(bufnr)
+
+  if attached_buffers_by_module.has(mod, bufnr) then
+    disable_module(mod, bufnr)
+  else
+    enable_module(mod, bufnr, lang)
+  end
+end
+
 -- Recurses through all modules including submodules
 -- @param accumulator function called for each module
 -- @param root root configuration table to start at
@@ -181,6 +196,13 @@ M.commands = {
       "-complete=custom,nvim_treesitter#available_modules",
     },
   },
+  TSBufToggle = {
+    run = toggle_module,
+    args = {
+      "-nargs=1",
+      "-complete=custom,nvim_treesitter#available_modules",
+    },
+  },
   TSEnableAll = {
     run = enable_all,
     args = {