Register readable prefix for grouped keymap settings

lua/keymap/bind.lua

@@ -156,10 +156,18 @@ function bind.nvim_load_mapping(mapping)
 				local rhs = value.cmd
 				local options = value.options
 				local buf = value.buffer
+				local M = require("modules.utils.keymap")
 				if buf and type(buf) == "number" then
 					vim.api.nvim_buf_set_keymap(buf, mode, keymap, rhs, options)
+					--- if begin with <leader>
+					if keymap:sub(1, #"<leader>") == "<leader>" and #keymap > #"<leader>x" then
+						M.insert_queue(keymap:sub(1, #"<leader>x"), mode, buf)
+					end
 				else
 					vim.api.nvim_set_keymap(mode, keymap, rhs, options)
+					if keymap:sub(1, #"<leader>") == "<leader>" and #keymap > #"<leader>x" then
+						M.insert_queue(keymap:sub(1, #"<leader>x"), mode, nil)
+					end
 				end
 			end
 		end

lua/keymap/prefix.lua

@@ -0,0 +1,10 @@
+local icons = {
+	ui_sep = require("modules.utils.icons").get("ui", true),
+}
+
+--- Need to expand by hand
+local prefix_desc = {
+	["<leader>b"] = icons.ui_sep.Buffer .. "Buffer",
+}
+
+return prefix_desc

lua/modules/configs/tool/which-key.lua

@@ -20,7 +20,7 @@ return function()
 		icons = {
 			breadcrumb = icons.ui.Separator,
 			separator = icons.misc.Vbar,
-			group = icons.misc.Add,
+			group = "",
 		},
 
 		window = {
@@ -31,4 +31,6 @@ return function()
 			winblend = 0,
 		},
 	})
+
+	require("modules.utils.keymap").which_key_register()
 end

lua/modules/utils/icons.lua

@@ -79,6 +79,7 @@ local data = {
 		BigCircle = "",
 		BigUnfilledCircle = "",
 		BookMark = "󰃃",
+		Buffer = "󰓩",
 		Bug = "",
 		Calendar = "",
 		Check = "󰄳",

lua/modules/utils/keymap.lua

@@ -202,4 +202,39 @@ function M.replace(mapping)
 	end
 end
 
+--- Register queued which-key mappings
+function M.which_key_register()
+	if M.which_key_queue then
+		local wk_avail, wk = pcall(require, "which-key")
+		if wk_avail then
+			for registration, opts in pairs(M.which_key_queue) do
+				wk.register(registration, opts)
+			end
+			M.which_key_queue = nil
+		end
+	end
+end
+
+--- Insert a prefix keymap into register queue.
+function M.insert_queue(prefix_keymap, mode, buffer)
+	local registration = {}
+	local options = {
+		mode = mode,
+		buffer = buffer,
+	}
+	local P = require("keymap.prefix")
+	if not P[prefix_keymap] then
+		return
+	end
+	--- set name with icon
+	registration[prefix_keymap] = {}
+	registration[prefix_keymap]["name"] = P[prefix_keymap:sub(1, #"<leader>x")]
+	if not M.which_key_queue then
+		M.which_key_queue = {}
+	end
+	if not M.which_key_queue[registration] then
+		M.which_key_queue[registration] = options
+	end
+end
+
 return M