feat(util): added util.on_load to execute code when a plugin loads

LazyVim · Jul 12, 2023 · 9fd8970 · 9fd8970
1 parent b1721bc
commit 9fd8970
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/lua/lazyvim/util/init.lua b/lua/lazyvim/util/init.lua
@@ -256,4 +256,25 @@ function M.lsp_disable(server, cond)
   end)
 end
 
+---@param name string
+---@param fn fun(name:string)
+function M.on_load(name, fn)
+  local Config = require("lazy.core.config")
+  if Config.plugins[name] and Config.plugins[name]._.loaded then
+    vim.schedule(function()
+      fn(name)
+    end)
+  else
+    vim.api.nvim_create_autocmd("User", {
+      pattern = "LazyLoad",
+      callback = function(event)
+        if event.data == name then
+          fn(name)
+          return true
+        end
+      end,
+    })
+  end
+end
+
 return M