feat(toggle): add keymap to toggle treesitter context (#1711)

* feat: add keymap to toggle treesitter context * fix: get correct enabled state --------- Co-authored-by: Folke Lemaitre <[email protected]>
LazyVim · Oct 14, 2023 · 5e1a86d · 5e1a86d
1 parent f724eae
commit 5e1a86d
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/lua/lazyvim/plugins/formatting.lua b/lua/lazyvim/plugins/formatting.lua
@@ -6,7 +6,7 @@ local M = {}
 local format_opts = {}
 
 ---@param opts ConformOpts
-function M.setup(plugin, opts)
+function M.setup(_, opts)
   local util = require("conform.util")
   opts.formatters = opts.formatters or {}
   for name, formatter in pairs(opts.formatters) do

diff --git a/lua/lazyvim/plugins/treesitter.lua b/lua/lazyvim/plugins/treesitter.lua
@@ -110,6 +110,22 @@ return {
     event = "LazyFile",
     enabled = true,
     opts = { mode = "cursor" },
+    keys = {
+      {
+        "<leader>ut",
+        function()
+          local Util = require("lazyvim.util")
+          local tsc = require("treesitter-context")
+          tsc.toggle()
+          if Util.inject.get_upvalue(tsc.toggle, "enabled") then
+            Util.info("Enabled Treesitter Context", { title = "Option" })
+          else
+            Util.warn("Disabled Treesitter Context", { title = "Option" })
+          end
+        end,
+        desc = "Toggle Treesitter Context",
+      },
+    },
   },
 
   -- Automatically add closing tags for HTML and JSX