feat(lib): add loadfile Function.

shanyouli · Oct 7, 2024 · 0e7ba96 · 0e7ba96
1 parent 9758acc
commit 0e7ba96
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/nix/lib/modules.nix b/nix/lib/modules.nix
@@ -4,9 +4,18 @@
   ...
 }: let
   inherit (builtins) attrValues readDir pathExists concatLists;
-  inherit (lib) id mapAttrsToList filterAttrs hasPrefix hasSuffix nameValuePair removeSuffix mapAttrs';
+  inherit (lib) id mapAttrsToList filterAttrs hasPrefix hasSuffix nameValuePair removeSuffix mapAttrs' pathIsDirectory;
   inherit (attrs) mapFilterAttrs';
 in rec {
+  loadFile = fn: path: let
+    filePath = toString path;
+  in
+    if pathIsDirectory filePath
+    then fn (filePath + "/default.nix")
+    else if pathExists (filePath + ".nix")
+    then fn (filePath + ".nix")
+    else throw "unknown path ${path}";
+
   mapModule = dir: fn: {namefn ? (n: removeSuffix ".nix" n), ...}:
     mapAttrs' (name: _type: {
       name = namefn name;