users/wpcarro/utils/fs.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

{ pkgs, ... }:

# `fs` contains utility functions for working with the filesystem.

let
  inherit (builtins) attrNames hasAttr map readDir;
  inherit (pkgs.lib) filterAttrs;
in {
  # Returns a list of all of the regular files in `dir`.
  files = dir:
    map (name: dir + "/${name}")
      (attrNames
        (filterAttrs (_: type: type == "regular") (readDir dir)));

  # Returns a list of all of the directories in `dir`.
  dirs = dir:
    map (name: dir + "/${name}")
      (attrNames
        (filterAttrs (_: type: type == "directory") (readDir dir)));

  # Returns a list of paths to all of the `name` files starting at `dir`.
  find = name: dir:
    if hasAttr name (readDir dir) then
      [ (dir + name) ] ++ concatMap findAllDefaultNix (dirs dir)
    else
      concatMap findAllDefaultNix (dirs dir);

  # Looks for `name` in `dir`; if it cannot find it, it checks the parent
  # directory.
  resolve = name: dir:
    if hasAttr name (readDir dir) then
      dir + "/${name}"
    else
      # This prevents the function from infinitely recursing and eventually
      # stack overflowing.
      if (dirOf dir) == dir then
        null
      else
        resolve name (dirOf dir);
  };
}