feat: add recursive file fetching

flake.nix

@@ -1,17 +1,12 @@
 {
-  description = "A very basic flake";
+  description = "A declarative wallpaper collection.";
 
-  outputs =
+  outputs = _: {
-    _:
+    wallpapers = (import ./lib).toWallpkgs ./wallpapers [
-    let
+      "png"
-      lib = import ./lib;
+      "jpg"
-    in
+      "jpeg"
-    {
+      "gif"
-      wallpapers = lib.toWallpkgs ./wallpapers [
+    ];
-        "png"
+  };
-        "jpg"
-        "jpeg"
-        "gif"
-      ];
-    };
 }

lib/default.nix

@@ -52,41 +52,55 @@ let
       }) names
     );
 
-  # TODO: This needs to be extensible, possibly in order to allow additional directories.
+  hasSuffix =
-  # In theory, we should only need to handle path*s* instead of a path, and search multiple
+    suffix: content:
-  # paths by extension instead of just once, right?
+    let
+      lenContent = builtins.stringLength content;
+      lenSuffix = builtins.stringLength suffix;
+    in
+    (
+      lenContent >= lenSuffix && builtins.substring (lenContent - lenSuffix) lenContent content == suffix
+    );
+
   toWallpkgs =
     path: extensions:
     let
-      fileExts = extensions;
+
-    in
+      hasValidExtension =
-    builtins.listToAttrs (
+        file: builtins.foldl' (acc: elem: (hasSuffix elem "${file}") || acc) false extensions;
-      map
+
-        (
+      fetchPath =
-          n:
+        path:
-          let
+        filterAttrs (n: t: t == "directory" || (t == "regular" && hasValidExtension n)) (
-            filesByExtension = builtins.filter builtins.pathExists (map (ext: path + /${n}.${ext}) fileExts);
+          builtins.readDir path
-            file =
+        );
-              if filesByExtension == [ ] then
+
-                builtins.throw "Either ${n} is not a file or it does not have the ${builtins.concatStringsSep ", " fileExts} extensions."
+      getFiles =
+        path:
+        builtins.listToAttrs (
+          builtins.attrValues (
+            builtins.mapAttrs (
+              n: v:
+              if v == "directory" then
+                {
+                  name = n;
+                  value = getFiles "${path}/${n}";
+                }
               else
-                builtins.head filesByExtension;
+                {
-          in
+                  name = builtins.head (splitString "." n);
-          {
+                  value = {
-            name = n;
+                    path = "${path}/${n}";
-            value = {
+                    tags = splitString "-" (builtins.head (splitString "." n));
-              path = file;
+                    hash = builtins.hashFile "md5" "${path}/${n}";
-              tags = splitString "-" n;
+                  };
-              hash = builtins.hashFile "md5" file; # in theory, md5 is the fastest option because it produces a 128-bit hash instead of >= 160
+                }
-            };
+            ) (fetchPath path)
-          }
-        )
-        (
-          map (n: builtins.head (splitString "." n)) (
-            builtins.attrNames (filterAttrs (n: _: n != "README.md") (builtins.readDir path))
           )
-        )
+        );
-    );
+
+    in
+    getFiles path;
 in
 {
   # Partial re-implementations of functions from Nixpkgs.