Add mkEnv parameter libOverlay to extend lib

Feature discussed in #20

Author: arnarg

flake.nix

@@ -27,9 +27,10 @@
           modules ? [],
           extraSpecialArgs ? {},
           charts ? {},
+          libOverlay ? null,
         }:
           import ./modules {
-            inherit pkgs lib extraSpecialArgs kubenix;
+            inherit pkgs lib extraSpecialArgs kubenix libOverlay;
             kubelib = nix-kube-generators;
             modules =
               modules
@@ -47,11 +48,12 @@
           extraSpecialArgs ? {},
           envs ? {},
           charts ? {},
+          libOverlay ? null,
         }:
           lib.mapAttrs (
             env: conf:
               mkEnv {
-                inherit pkgs lib charts;
+                inherit pkgs lib charts libOverlay;
                 extraSpecialArgs = extraSpecialArgs // (conf.extraSpecialArgs or {});
                 modules =
                   [{nixidy.target.rootPath = lib.mkDefault "./manifests/${env}";}]

modules/default.nix

@@ -5,6 +5,7 @@
   kubenix,
   lib ? pkgs.lib,
   extraSpecialArgs ? {},
+  libOverlay ? null,
 }: let
   extendedLib = import ../lib {inherit pkgs kubelib;};
 
@@ -25,7 +26,10 @@
     specialArgs =
       {
         inherit pkgs;
-        lib = extendedLib;
+        lib =
+          if builtins.isFunction libOverlay
+          then extendedLib.extend libOverlay
+          else extendedLib;
       }
       // extraSpecialArgs;
   };