lib/plugins/mkVimPlugin: Support lazyloading

Signed-off-by: saygo-png <[email protected]>

Author: saygo-png

lib/plugins/mk-vim-plugin.nix

@@ -59,26 +59,86 @@ let
       options = lib.setAttrByPath loc (
         {
           enable = lib.mkEnableOption name;
+          lazyLoad = lib.nixvim.mkLazyLoadOption name;
           autoLoad = lib.nixvim.mkAutoLoadOption cfg name;
         }
         // settingsOption
         // extraOptions
       );
 
       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)
         ]
       );