nixos-config/lib/trivial.nix

{ pkgs, lib, self }:

rec {
  /* Count the attributes with the given predicate.

     Examples:
       countAttrs (name: value: value) { d = true; f = true; a = false; }
       => 2

       countAttrs (name: value: value.enable) { d = { enable = true; }; f = { enable = false; package = [ ]; }; }
       => 1
  */
  countAttrs = pred: attrs:
    lib.count (attr: pred attr.name attr.value)
      (lib.mapAttrsToList lib.nameValuePair attrs);

  /* Filters and groups the attribute set into two separate attribute where it
     either accepted or denied from a given predicate function.

     Example:
       filterAttrs' (n: v: v == 4) { a = 4; b = 2; c = 6; }
       => { ok = { a = 4; }; notOk = { b = 2; c = 6; }; }
  */
  filterAttrs' = f: attrs:
    lib.foldlAttrs (acc: name: value: let
      isOk = f name value;
    in {
      ok = acc.ok // lib.optionalAttrs isOk { ${name} = value; };
      notOk = acc.notOk // lib.optionalAttrs (!isOk) { ${name} = value; };
    })
    { ok = { }; notOk = { }; }
    attrs;

  /* Convenient function for converting bits to bytes.

     Example:
       bitsToBytes 1600
       => 200
  */
  bitsToBytes = x: x / 8.0;

  /* Gives the exponent with the associated SI prefix.

     Example:
       SIPrefixExponent "M"
       => 6

       SIPrefixExponent "Q"
       => 30
  */
  SIPrefixExponent = c:
    let
      prefixes = {
        Q = 30;
        R = 27;
        Y = 24;
        Z = 21;
        E = 18;
        P = 15;
        T = 12;
        G = 9;
        M = 6;
        k = 3;
        h = 2;
        da = 1;
        d = -1;
        c = -2;
        m = -3;
        "μ" = -6;
        n = -9;
        p = -12;
        f = -15;
        a = -18;
        z = -21;
        y = -24;
        r = -27;
        q = -30;
      };
    in
      prefixes.${c};

  /* Gives the multiplier for the metric units.

     Example:
      metricPrefixMultiplier "M"
      => 1000000

      metricPrefixMultiplier "G"
      => 1000000000
  */
  metricPrefixMultiplier = c:
    self.math.pow 10 (SIPrefixExponent c);

  /* Gives the exponent with the associated binary prefix.

     As an implementation detail, we don't follow the proper IEC unit prefixes
     and instead mapping this to the SI prefix for convenience.

     Example:
       SIPrefixExponent "M"
       => 6

       SIPrefixExponent "Q"
       => 30
  */
  binaryPrefixExponent = c:
    let
      prefixes = {
        Y = 80;
        Z = 70;
        E = 60;
        P = 50;
        T = 40;
        G = 30;
        M = 20;
        K = 10;
      };
    in
      prefixes.${c};

  /* Gives the multiplier for the given byte unit. Essentially returns the
     value in number of bytes.

     Example:
       binaryPrefixMultiplier "M"
       => 1048576

       binaryPrefixMultiplier "G"
       => 1.099511628×10¹²
  */
  binaryPrefixMultiplier = c:
    self.math.pow 2 (binaryPrefixExponent c);
}