From 96cd08ef455b8b813f3cc89420339afe70cce895 Mon Sep 17 00:00:00 2001
From: Leon Heidelbach <leon.heidelbach@hhu.de>
Date: Sat, 14 May 2022 01:31:13 +0200
Subject: [PATCH] fix: default mappings can now be properly disabled,
 independent of load order

---
 lua/core/utils.lua | 59 +++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 51 insertions(+), 8 deletions(-)

diff --git a/lua/core/utils.lua b/lua/core/utils.lua
index 10965b08..57d4ad72 100644
--- a/lua/core/utils.lua
+++ b/lua/core/utils.lua
@@ -31,6 +31,7 @@ nvchad.load_config = function()
       -- merge user config if it exists and is a table; otherwise display an error
       local user_config = require "custom.chadrc"
       if type(user_config) == "table" then
+         conf.mappings = nvchad.prune_key_map(conf.mappings, user_config.mappings, { "mode_opts" })
          conf = vim.tbl_deep_extend("force", conf, user_config)
       else
          error "User config (chadrc.lua) *must* return a table!"
@@ -40,6 +41,46 @@ nvchad.load_config = function()
    return conf
 end
 
+-- reduces a given keymap to a table of modes each containing a list of key maps for that mode
+nvchad.reduce_key_map = function(key_map, ignore_modes)
+   local prune_keys = {}
+   for _, modes in pairs(key_map) do
+      for mode, mappings in pairs(modes) do
+         if not vim.tbl_contains(ignore_modes, mode) then
+            prune_keys[mode] = prune_keys[mode] and prune_keys[mode] or {}
+            prune_keys[mode] = vim.list_extend(prune_keys[mode], vim.tbl_keys(mappings))
+         end
+      end
+   end
+   return prune_keys
+end
+
+-- prune keys from a key map table by matching against another key map table
+nvchad.prune_key_map = function(key_map, prune_map, ignore_modes)
+   if not prune_map then return key_map end
+   if not key_map then return prune_map end
+   local prune_keys = nvchad.reduce_key_map(prune_map, ignore_modes)
+
+   for ext, modes in pairs(key_map) do
+      for mode, mappings in pairs(modes) do
+         if not vim.tbl_contains(ignore_modes, mode) then
+            if prune_keys[mode] then
+               -- filter mappings table so that only keys that are not in user_mappings are left
+               local filtered_mappings = {}
+               for k, v in pairs(mappings) do
+                  if not vim.tbl_contains(prune_keys[mode], k) then
+                     filtered_mappings[k] = v
+                  end
+               end
+               key_map[ext][mode] = filtered_mappings
+            end
+         end
+      end
+   end
+
+   return key_map
+end
+
 nvchad.map = function(mode, keys, command, opt)
    local options = { silent = true }
 
@@ -58,16 +99,18 @@ nvchad.map = function(mode, keys, command, opt)
 end
 
 -- For those who disabled whichkey
-
 nvchad.no_WhichKey_map = function()
    local mappings = nvchad.load_config().mappings
-
-   for key, _ in pairs(mappings) do
-      for mode, _ in pairs(mappings[key]) do
-         for keybind, cmd in pairs(mappings[key][mode]) do
-            -- disabled keys will not have cmd set
-            if cmd ~= "" then
-               nvchad.map(mode, keybind, cmd[1])
+   local ignore_modes = { "mode_opts" }
+
+   for _, value in pairs(mappings) do
+      for mode, keymap in pairs(value) do
+         if not vim.tbl_contains(ignore_modes, mode) then
+            for keybind, cmd in pairs(keymap) do
+               -- disabled keys will not have cmd set
+               if cmd ~= "" then
+                  nvchad.map(mode, keybind, cmd[1])
+               end
             end
          end
       end