From 2080694e3402980d7b84fa095bfdd084002d64c7 Mon Sep 17 00:00:00 2001
From: Folke Lemaitre <folke.lemaitre@gmail.com>
Date: Wed, 23 Nov 2022 16:08:44 +0100
Subject: [PATCH] feat: load plugin on cmd complete and make completion just
 work

---
 lua/lazy/core/loader.lua | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/lua/lazy/core/loader.lua b/lua/lazy/core/loader.lua
index f8d01edf..48d6e282 100644
--- a/lua/lazy/core/loader.lua
+++ b/lua/lazy/core/loader.lua
@@ -110,23 +110,35 @@ function M.setup()
   -- commands
   Util.track("loader_commands")
   for cmd, plugins in pairs(M.loaders.cmd or {}) do
-    vim.api.nvim_create_user_command(cmd, function(event)
+    local function _load(complete)
       vim.api.nvim_del_user_command(cmd)
-      Util.track("cmd: " .. cmd)
-      M.load(plugins)
+      if complete then
+        Util.track("cmd-complete: " .. cmd)
+      else
+        Util.track("cmd: " .. cmd)
+      end
+      M.load(plugins, { cmd = cmd })
+      Util.track()
+    end
+    vim.api.nvim_create_user_command(cmd, function(event)
+      _load()
       vim.cmd(
         ("%s %s%s%s %s"):format(
           event.mods or "",
           event.line1 == event.line2 and "" or event.line1 .. "," .. event.line2,
           cmd,
           event.bang and "!" or "",
-          event.args
+          event.args or ""
         )
       )
-      Util.track()
     end, {
       bang = true,
       nargs = "*",
+      complete = function()
+        _load(true)
+        -- HACK: trick Neovim to show the newly loaded command completion
+        vim.api.nvim_input("<space><bs><tab>")
+      end,
     })
   end
   Util.track()