From 56890ce5f439e9bbf275ed5ec2573b4e29371bb5 Mon Sep 17 00:00:00 2001
From: Folke Lemaitre <folke.lemaitre@gmail.com>
Date: Thu, 22 Dec 2022 17:49:32 +0100
Subject: [PATCH] fix: remove lazy keymaps with the correct mode. Fixes #97

---
 lua/lazy/core/handler/keys.lua | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/lua/lazy/core/handler/keys.lua b/lua/lazy/core/handler/keys.lua
index e65120d..3b6b431 100644
--- a/lua/lazy/core/handler/keys.lua
+++ b/lua/lazy/core/handler/keys.lua
@@ -28,7 +28,10 @@ end
 
 ---@param value string|LazyKeys
 function M.parse(value)
-  return (type(value) == "string" and { value } or value) --[[@as LazyKeys]]
+  local ret = vim.deepcopy(value)
+  ret = (type(ret) == "string" and { ret } or ret) --[[@as LazyKeys]]
+  ret.mode = ret.mode or "n"
+  return ret
 end
 
 function M.opts(keys)
@@ -45,21 +48,24 @@ end
 function M:_add(value)
   local keys = M.parse(value)
   local lhs = keys[1]
-  vim.keymap.set(keys.mode or "n", lhs, function()
+  local opts = M.opts(keys)
+  opts.noremap = false
+  vim.keymap.set(keys.mode, lhs, function()
     Util.track({ keys = lhs })
     self:_del(value)
     Loader.load(self.active[value], { keys = lhs })
     M.retrigger(lhs)
     Util.track()
-  end, M.opts(keys))
+  end, opts)
 end
 
 ---@param value string|LazyKeys
 function M:_del(value)
   local keys = M.parse(value)
-  pcall(vim.keymap.del, "n", keys[1])
+  vim.pretty_print(keys)
+  pcall(vim.keymap.del, keys.mode, keys[1])
   if keys[2] then
-    vim.keymap.set(keys.mode or "n", keys[1], keys[2], M.opts(keys))
+    vim.keymap.set(keys.mode, keys[1], keys[2], M.opts(keys))
   end
 end