lazy: add setupOpts support

lib/types/plugins.nix

@@ -65,10 +65,27 @@
       #   type=  str;
       # }
 
+      # Non-lz.n options
+
       package = mkOption {
         type = pluginType;
+        description = "Plugin package";
       };
 
+      setupModule = mkOption {
+        type = nullOr str;
+        description = "Lua module to run setup function on.";
+        default = null;
+      };
+
+      setupOpts = mkOption {
+        type = submodule {freeformType = attrsOf anything;};
+        description = "Options to pass to the setup function";
+        default = {};
+      };
+
+      # lz.n options
+
       before = mkOption {
         type = nullOr luaInline;
         description = "Code to run before plugin is loaded";

modules/wrapper/lazy/config.nix

@@ -3,15 +3,29 @@
   config,
   ...
 }: let
+  inherit (builtins) toJSON;
   inherit (lib.modules) mkIf;
   inherit (lib.attrsets) mapAttrsToList;
+  inherit (lib.generators) mkLuaInline;
+  inherit (lib.strings) optionalString;
   inherit (lib.nvim.lua) toLuaObject;
   inherit (lib.nvim.dag) entryAnywhere;
   cfg = config.vim.lazy;
 
-  toLuaLznSpec = name: plugin:
+  toLuaLznSpec = name: spec:
-    (removeAttrs plugin ["package"])
+    (removeAttrs spec ["package" "setupModule" "setupOpts"])
-    // {"@1" = name;};
+    // {
+      "@1" = name;
+      after = mkLuaInline ''
+        function()
+          ${
+          optionalString (spec.setupModule != null)
+          "require(${toJSON spec.setupModule}).setup(${toLuaObject spec.setupOpts})"
+        }
+          ${optionalString (spec.after != null) spec.after}
+        end
+      '';
+    };
   lznSpecs = mapAttrsToList toLuaLznSpec cfg.plugins;
 in {
   config.vim = mkIf cfg.enable {