From 75c0bd1dd3fcb1e7a8b3ba3e567422148a184a15 Mon Sep 17 00:00:00 2001
From: saygo-png <saygo.mail@proton.me>
Date: Mon, 10 Nov 2025 08:26:14 +0100
Subject: [PATCH] lib/plugins/mkVimPlugin: Support lazyloading

Signed-off-by: saygo-png <saygo.mail@proton.me>
---
 lib/plugins/mk-vim-plugin.nix | 68 ++++++++++++++++++++++++++++++++---
 1 file changed, 64 insertions(+), 4 deletions(-)

diff --git a/lib/plugins/mk-vim-plugin.nix b/lib/plugins/mk-vim-plugin.nix
index 9f7ea25af2..34d12ca998 100644
--- a/lib/plugins/mk-vim-plugin.nix
+++ b/lib/plugins/mk-vim-plugin.nix
@@ -59,6 +59,7 @@ let
       options = lib.setAttrByPath loc (
         {
           enable = lib.mkEnableOption name;
+          lazyLoad = lib.nixvim.mkLazyLoadOption name;
           autoLoad = lib.nixvim.mkAutoLoadOption cfg name;
         }
         // settingsOption
@@ -66,19 +67,78 @@ let
       );
 
       config = lib.mkIf cfg.enable (
+        let
+          globalsConfig = lib.nixvim.applyPrefixToAttrs globalPrefix (cfg.settings or { });
+        in
         lib.mkMerge [
-          {
-            inherit extraPackages extraPlugins;
-            globals = lib.nixvim.applyPrefixToAttrs globalPrefix (cfg.settings or { });
-          }
+          { inherit extraPackages extraPlugins; }
+
+          (lib.mkIf cfg.lazyLoad.enable {
+            assertions = [
+              {
+                assertion = (isColorscheme && colorscheme != null) || cfg.lazyLoad.settings != { };
+                message = "You have enabled lazy loading for ${name} but have not provided any configuration.";
+              }
+            ];
+
+            plugins.lz-n = {
+              plugins = [
+                (
+                  {
+                    # The packpath name is always the derivation name
+                    __unkeyed-1 = lib.getName cfg.package;
+                    # Use provided before, otherwise fallback to normal function wrapped globals config
+                    before =
+                      let
+                        before = cfg.lazyLoad.settings.before or null;
+                        default =
+                          lib.mkIf
+                            (
+                              (lib.filterAttrs (
+                                _: v:
+                                !builtins.elem v [
+                                  [ ]
+                                  null
+                                  { }
+                                ]
+                              ) globalsConfig) != { }
+                            )
+                            ''
+                              function()
+                                local globals = ${lib.nixvim.toLuaObject globalsConfig}
+
+                                for k,v in pairs(globals) do
+                                  vim.g[k] = v
+                                end
+                              end
+                            '';
+                      in
+                      if (lib.isString before || lib.types.rawLua.check before) then before else default;
+                    colorscheme = lib.mkIf isColorscheme (cfg.lazyLoad.settings.colorscheme or colorscheme);
+                  }
+                  // lib.removeAttrs cfg.lazyLoad.settings [
+                    "before"
+                    "colorscheme"
+                  ]
+                )
+              ];
+            };
+          })
+
+          (lib.mkIf (!cfg.lazyLoad.enable) {
+            globals = globalsConfig;
+          })
+
           (lib.optionalAttrs (isColorscheme && colorscheme != null) {
             colorscheme = lib.mkDefault colorscheme;
           })
+
           (lib.optionalAttrs (args ? extraConfig) (
             lib.nixvim.plugins.utils.applyExtraConfig {
               inherit extraConfig cfg opts;
             }
           ))
+
           (lib.nixvim.plugins.utils.enableDependencies dependencies)
         ]
       );