feat!: lazy keymap support

lua/hawtkeys/init.lua

@@ -10,8 +10,8 @@ local M = {}
 ---@field homerow number
 ---@field powerFingers number[]
 ---@field keyboardLayout HawtKeySupportedKeyboardLayouts
----@field keyMapSet { [string] : TSKeyMapArgs | WhichKeyMapargs }
----@field customMaps { [string] : TSKeyMapArgs | WhichKeyMapargs } | nil
+---@field keyMapSet { [string] : TSKeyMapArgs | WhichKeyMapargs | LazyKeyMapArgs }
+---@field customMaps { [string] : TSKeyMapArgs | WhichKeyMapargs | LazyKeyMapArgs } | nil
 ---@field highlights HawtKeyHighlights
 ---@field lhsBlacklist string[]
 
@@ -26,11 +26,11 @@ local M = {}
 ---@field homerow number | nil
 ---@field powerFingers number[] | nil
 ---@field keyboardLayout HawtKeySupportedKeyboardLayouts | nil
----@field customMaps { [string] : TSKeyMapArgs | WhichKeyMapargs } | nil
+---@field customMaps { [string] : TSKeyMapArgs | WhichKeyMapargs | LazyKeyMapArgs } | nil
 ---@field highlights HawtKeyHighlights | nil
 ---@field lhsBlacklist string[] | nil
 
----@type { [string] : TSKeyMapArgs | WhichKeyMapargs }---
+---@type { [string] : TSKeyMapArgs | WhichKeyMapargs | LazyKeyMapArgs }
 local _defaultSet = {
     ["vim.keymap.set"] = {
         modeIndex = 1,

lua/hawtkeys/ts.lua

@@ -9,9 +9,11 @@ local tsQuery = require("nvim-treesitter.query")
 ---@alias VimModes 'n' | 'x' | 'v' | 'i'
 
 ---@alias WhichKeyMethods 'which_key'
----
----@alias TreeSitterMethods 'dot_index_expression' | 'function_call' | 'expression_list'
----
+
+---@alias LazyMethods 'lazy'
+
+---@alias TreeSitterMethods 'dot_index_expression' | 'function_call'
+
 ---@alias SetMethods WhichKeyMethods | TreeSitterMethods
 
 ---@class TSKeyMapArgs
@@ -24,6 +26,9 @@ local tsQuery = require("nvim-treesitter.query")
 ---@class WhichKeyMapargs
 ---@field method WhichKeyMethods
 
+---@class LazyKeyMapArgs
+---@field method LazyMethods
+
 ---@type table<string, boolean>
 local scannedFiles = {}
 
@@ -130,7 +135,7 @@ local function find_maps_in_file(filePath)
     local fileContent = Path:new(filePath):read()
     local parser = vim.treesitter.get_string_parser(fileContent, "lua", {}) -- Get the Lua parser
     local tree = parser:parse()[1]:root()
-    local tsKemaps = {}
+    local tsKeymaps = {}
     -- TODO: This currently doesnt always work, as the options for helper functions are different,
     -- need to use TS to resolve it back to a native keymap
     local dotIndexExpressionQuery = ts.parse_query(
@@ -213,7 +218,7 @@ local function find_maps_in_file(filePath)
                         end
                         map.mode = table.concat(modes, ", ")
                     end
-                    table.insert(tsKemaps, map)
+                    table.insert(tsKeymaps, map)
                 end
             end
         end
@@ -301,7 +306,7 @@ local function find_maps_in_file(filePath)
                         end
                         map.mode = table.concat(modes, ", ")
                     end
-                    table.insert(tsKemaps, map)
+                    table.insert(tsKeymaps, map)
                 end
             end
         end
@@ -342,13 +347,43 @@ local function find_maps_in_file(filePath)
                         rhs = mapping.cmd,
                         from_file = filePath,
                     }
-                    table.insert(tsKemaps, map)
+                    table.insert(tsKeymaps, map)
                 end
             end
         end
     end
 
-    return tsKemaps
+    return tsKeymaps
+end
+
+---@return table
+local function get_keymaps_from_lazy()
+    local lazyKeyMaps = {}
+    for _, args in pairs(hawtkeys.config.keyMapSet) do
+        if args.method == "lazy" then
+            local ok, lazy = pcall(function()
+                return require("lazy").plugins()
+            end)
+            if not ok then
+                error("Lazy Loading requires Lazy")
+                break
+            end
+            for _, v in ipairs(lazy) do
+                if v and v._ and v._.handlers and v._.handlers.keys then
+                    for _, key in pairs(v._.handlers.keys) do
+                        local map = {
+                            lhs = key.lhs,
+                            rhs = key.rhs,
+                            mode = key.mode,
+                            from_file = "Lazy Init:" .. tostring(v[1]),
+                        }
+                        table.insert(lazyKeyMaps, map)
+                    end
+                end
+            end
+        end
+    end
+    return lazyKeyMaps
 end
 
 ---@return table
@@ -420,6 +455,16 @@ function M.get_all_keymaps()
         end
     end
 
+    if
+        hawtkeys.config.keyMapSet.lazy
+        and hawtkeys.config.keyMapSet.lazy.method == "lazy"
+    then
+        local lazyKeyMaps = get_keymaps_from_lazy()
+        for _, keymap in ipairs(lazyKeyMaps) do
+            table.insert(keymaps, keymap)
+        end
+    end
+
     local vimKeymaps = get_keymaps_from_vim()
     returnKeymaps = utils.merge_tables(keymaps, vimKeymaps)
     scannedFiles = {}
@@ -432,5 +477,6 @@ end
 
 M.find_maps_in_file = find_maps_in_file
 M.get_keymaps_from_vim = get_keymaps_from_vim
+M.get_keymaps_from_lazy = get_keymaps_from_lazy
 
 return M

tests/hawtkeys/ts_spec.lua

@@ -95,3 +95,42 @@ describe("Treesitter can extract keymaps", function()
         eq(beforeCount, finalCount)
     end)
 end)
+
+describe("Lazy Managed Plugins", function()
+    before_each(function()
+        require("plenary.reload").reload_module("hawtkeys")
+        hawtkeys.setup({
+        customMaps = {
+            ['lazy']  = {
+            method = 'lazy',
+            },
+        },
+        })
+        require("tests.minimal_init").loadLazy()
+        vim.g.lazy_did_setup = false
+        vim.go.loadplugins = true
+        for modname in pairs(package.loaded) do
+            if modname:find("lazy") == 1 then
+                package.loaded[modname] = nil
+            end
+        end
+        local lazy = require("lazy")
+        lazy.setup({
+            "ellisonleao/nvim-plugin-template",
+            keys = {
+                {
+                    "<leader>1",
+                    ":lua print(1)<CR>",
+                    desc = "Test Lazy Print 1",
+                },
+            },
+        })
+    end)
+    it("extract keys set in a Lazy init", function()
+        local lazyKeymaps = ts.get_keymaps_from_lazy()
+        eq("n", lazyKeymaps[1].mode)
+        eq("<leader>1", lazyKeymaps[1].lhs)
+        eq(':lua print(1)<CR>', lazyKeymaps[1].rhs)
+        eq("Lazy Init:ellisonleao/nvim-plugin-template", lazyKeymaps[1].from_file)
+    end)
+end)

tests/minimal_init.lua

@@ -1,3 +1,21 @@
+local M = {}
+
+function M.loadLazy()
+    local lazy_dir = os.getenv("LAZY_DIR") or "/tmp/lazy.nvim"
+    if vim.fn.isdirectory(lazy_dir) == 0 then
+        vim.fn.system({
+            "git",
+            "clone",
+            "--filter=blob:none",
+            "https://github.com/folke/lazy.nvim.git",
+            "--branch=stable", -- latest stable release
+            lazy_dir,
+        })
+    end
+    vim.opt.rtp:append(lazy_dir)
+print("Installed Lazy to " .. lazy_dir)
+end
+
 local plenary_dir = os.getenv("PLENARY_DIR") or "/tmp/plenary.nvim"
 local treesitter_dir = os.getenv("TREESITTER_DIR") or "/tmp/nvim-treesitter"
 local whichkey_dir = os.getenv("WHICHKEY_DIR") or "/tmp/which-key.nvim"
@@ -39,6 +57,10 @@ vim.opt.rtp:append(plenary_dir)
 vim.opt.rtp:append(treesitter_dir)
 vim.opt.rtp:append(whichkey_dir)
 vim.opt.rtp:append(mini_dir)
+vim.cmd("runtime plugin/which-key.vim")
+require("which-key").setup({})
+require("plenary.busted")
+require("mini.test").setup()
 
 vim.cmd("runtime plugin/plenary.vim")
 vim.cmd("runtime plugin/treesitter.vim")
@@ -48,7 +70,5 @@ require("nvim-treesitter.configs").setup({
         enable = true,
     },
 })
-vim.cmd("runtime plugin/which-key.vim")
-require("which-key").setup({})
-require("plenary.busted")
-require("mini.test").setup()
+
+return M