fix(treesitter-textobjects): use normal ]c,]C,[c,[C when in diff-mode…

… instead of goto class. Fixes #1610
LazyVim · Oct 7, 2023 · 1935486 · 1935486
1 parent 237be9e
commit 1935486
Showing 1 changed file with 18 additions and 8 deletions.
diff --git a/lua/lazyvim/plugins/treesitter.lua b/lua/lazyvim/plugins/treesitter.lua
@@ -11,16 +11,26 @@ return {
       {
         "nvim-treesitter/nvim-treesitter-textobjects",
         config = function()
-          -- Disable class keymaps in diff mode
-          vim.api.nvim_create_autocmd("BufReadPost", {
-            callback = function(event)
-              if vim.wo.diff then
-                for _, key in ipairs({ "[c", "]c", "[C", "]C" }) do
-                  pcall(vim.keymap.del, "n", key, { buffer = event.buf })
+          -- When in diff mode, we want to use the default
+          -- vim text objects c & C instead of the treesitter ones.
+          local move = require("nvim-treesitter.textobjects.move") ---@type table<string,fun(...)>
+          local configs = require("nvim-treesitter.configs")
+          for name, fn in pairs(move) do
+            if name:find("goto") == 1 then
+              move[name] = function(q, ...)
+                if vim.wo.diff then
+                  local config = configs.get_module("textobjects.move")[name] ---@type table<string,string>
+                  for key, query in pairs(config or {}) do
+                    if q == query and key:find("[%]%[][cC]") then
+                      vim.cmd("normal! " .. key)
+                      return
+                    end
+                  end
                 end
+                return fn(q, ...)
               end
-            end,
-          })
+            end
+          end
         end,
       },
     },