feat: restore buffer local keymaps

Author: yungthai

lua/diffview/vcs/file.lua

@@ -15,6 +15,9 @@ local pl = lazy.access(utils, "path") ---@type PathLib
 local api = vim.api
 local M = {}
 
+-- keymaps to restore
+local R = {}
+
 local HAS_NVIM_0_10 = vim.fn.has("nvim-0.10") == 1
 
 ---@alias git.FileDataProducer fun(kind: vcs.FileKind, path: string, pos: "left"|"right"): string[]
@@ -350,8 +353,56 @@ function File:attach_buffer(force, opt)
       state.keymaps = config.extend_keymaps(conf.keymaps.view, state.keymaps)
       local default_map_opt = { silent = true, nowait = true, buffer = self.bufnr }
 
+      R = {}
+
       for _, mapping in ipairs(state.keymaps) do
         local map_opt = vim.tbl_extend("force", default_map_opt, mapping[4] or {}, { buffer = self.bufnr })
+
+        local mode = mapping[1] -- string
+        local name_lhs = mapping[2] -- string
+
+        local lhs_pat = string.format("%s", name_lhs)
+        -- escape special characters for search pattern
+        lhs_pat = string.gsub(lhs_pat, "%[", "%%%[")
+        lhs_pat = string.gsub(lhs_pat, "%]", "%%%]")
+
+        local buf_mappings = api.nvim_buf_get_keymap(self.bufnr, mode)
+
+        for _, buf_km_dict in pairs(buf_mappings) do
+          if buf_km_dict["lhs"] ~= nil then
+            local result = string.find(buf_km_dict["lhs"], lhs_pat)
+
+            if result ~= nil and result ~= "" then
+              local rhs = ""
+              if buf_km_dict.rhs ~= nil then
+                rhs = buf_km_dict.rhs
+              else
+                rhs = buf_km_dict.callback
+              end
+              -- save buffer keymap
+              if buf_km_dict.buffer == self.bufnr then
+                local obj = {
+                  bufnr = self.bufnr,
+                  mode = mode,
+                  lhs = name_lhs,
+                  rhs = rhs,
+                  opts = {
+                    buffer = self.bufnr,
+                    desc = buf_km_dict["desc"],
+                    silent = buf_km_dict["silent"],
+                    noremap = buf_km_dict["noremap"],
+                  },
+                }
+                table.insert(R, obj)
+              end
+              -- found buffer keymap, so stop searching
+              do
+                break
+              end
+            end
+          end
+        end
+
         vim.keymap.set(mapping[1], mapping[2], mapping[3], map_opt)
       end
 
@@ -387,6 +438,13 @@ function File:detach_buffer()
         end
       end
 
+      -- restore buffer keymaps
+      for _, dict in pairs(R) do
+        if dict.bufnr == self.bufnr then
+          vim.keymap.set(dict.mode, dict.lhs, dict.rhs, dict.opts)
+        end
+      end
+
       -- Diagnostics
       if state.disable_diagnostics then
         if HAS_NVIM_0_10 then