2009-02-09 16:51:03 +00:00
|
|
|
|
# Operations on attribute sets.
|
|
|
|
|
|
2009-03-06 23:21:14 +00:00
|
|
|
|
with {
|
2014-12-30 23:07:29 +00:00
|
|
|
|
inherit (builtins) head tail length;
|
2009-11-19 17:19:39 +00:00
|
|
|
|
inherit (import ./trivial.nix) or;
|
2009-03-06 23:21:14 +00:00
|
|
|
|
inherit (import ./default.nix) fold;
|
2009-03-10 15:18:38 +00:00
|
|
|
|
inherit (import ./strings.nix) concatStringsSep;
|
2013-02-01 05:39:26 +00:00
|
|
|
|
inherit (import ./lists.nix) concatMap concatLists all deepSeqList;
|
2009-03-06 23:21:14 +00:00
|
|
|
|
};
|
2009-02-09 16:51:03 +00:00
|
|
|
|
|
|
|
|
|
rec {
|
2012-08-13 04:08:21 +00:00
|
|
|
|
inherit (builtins) attrNames listToAttrs hasAttr isAttrs getAttr;
|
2009-02-09 16:51:03 +00:00
|
|
|
|
|
|
|
|
|
|
2009-03-10 15:18:38 +00:00
|
|
|
|
/* Return an attribute from nested attribute sets. For instance
|
|
|
|
|
["x" "y"] applied to some set e returns e.x.y, if it exists. The
|
2009-11-19 16:43:58 +00:00
|
|
|
|
default value is returned otherwise. */
|
2009-05-24 10:57:41 +00:00
|
|
|
|
attrByPath = attrPath: default: e:
|
2009-02-09 16:51:03 +00:00
|
|
|
|
let attr = head attrPath;
|
|
|
|
|
in
|
|
|
|
|
if attrPath == [] then e
|
2014-10-04 22:03:52 +00:00
|
|
|
|
else if e ? ${attr}
|
|
|
|
|
then attrByPath (tail attrPath) default e.${attr}
|
2009-02-09 16:51:03 +00:00
|
|
|
|
else default;
|
|
|
|
|
|
2012-10-23 13:35:48 +00:00
|
|
|
|
|
2009-09-28 18:22:44 +00:00
|
|
|
|
/* Return nested attribute set in which an attribute is set. For instance
|
|
|
|
|
["x" "y"] applied with some value v returns `x.y = v;' */
|
|
|
|
|
setAttrByPath = attrPath: value:
|
|
|
|
|
if attrPath == [] then value
|
2013-10-28 06:51:46 +00:00
|
|
|
|
else listToAttrs
|
|
|
|
|
[ { name = head attrPath; value = setAttrByPath (tail attrPath) value; } ];
|
2009-09-28 18:22:44 +00:00
|
|
|
|
|
2009-03-06 23:21:14 +00:00
|
|
|
|
|
2009-03-10 15:18:38 +00:00
|
|
|
|
getAttrFromPath = attrPath: set:
|
|
|
|
|
let errorMsg = "cannot find attribute `" + concatStringsSep "." attrPath + "'";
|
2009-05-24 10:57:41 +00:00
|
|
|
|
in attrByPath attrPath (abort errorMsg) set;
|
2012-10-23 13:35:48 +00:00
|
|
|
|
|
2009-03-06 23:21:14 +00:00
|
|
|
|
|
2009-03-10 15:18:38 +00:00
|
|
|
|
/* Return the specified attributes from a set.
|
2009-03-06 23:21:14 +00:00
|
|
|
|
|
2009-03-10 15:18:38 +00:00
|
|
|
|
Example:
|
|
|
|
|
attrVals ["a" "b" "c"] as
|
|
|
|
|
=> [as.a as.b as.c]
|
|
|
|
|
*/
|
2014-10-04 22:03:52 +00:00
|
|
|
|
attrVals = nameList: set: map (x: set.${x}) nameList;
|
2009-02-09 16:51:03 +00:00
|
|
|
|
|
2009-03-10 15:18:38 +00:00
|
|
|
|
|
|
|
|
|
/* Return the values of all attributes in the given set, sorted by
|
|
|
|
|
attribute name.
|
|
|
|
|
|
|
|
|
|
Example:
|
|
|
|
|
attrValues {c = 3; a = 1; b = 2;}
|
|
|
|
|
=> [1 2 3]
|
|
|
|
|
*/
|
2014-10-04 16:30:35 +00:00
|
|
|
|
attrValues = builtins.attrValues or (attrs: attrVals (attrNames attrs) attrs);
|
2009-03-10 15:18:38 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Collect each attribute named `attr' from a list of attribute
|
|
|
|
|
sets. Sets that don't contain the named attribute are ignored.
|
|
|
|
|
|
|
|
|
|
Example:
|
|
|
|
|
catAttrs "a" [{a = 1;} {b = 0;} {a = 2;}]
|
|
|
|
|
=> [1 2]
|
|
|
|
|
*/
|
2014-10-04 16:30:35 +00:00
|
|
|
|
catAttrs = builtins.catAttrs or
|
2014-10-04 22:03:52 +00:00
|
|
|
|
(attr: l: concatLists (map (s: if s ? ${attr} then [s.${attr}] else []) l));
|
2009-03-10 15:18:38 +00:00
|
|
|
|
|
|
|
|
|
|
2012-04-05 15:37:52 +00:00
|
|
|
|
/* Filter an attribute set by removing all attributes for which the
|
|
|
|
|
given predicate return false.
|
|
|
|
|
|
|
|
|
|
Example:
|
|
|
|
|
filterAttrs (n: v: n == "foo") { foo = 1; bar = 2; }
|
|
|
|
|
=> { foo = 1; }
|
|
|
|
|
*/
|
|
|
|
|
filterAttrs = pred: set:
|
2015-07-23 14:12:25 +00:00
|
|
|
|
listToAttrs (concatMap (name: let v = set.${name}; in if pred name v then [(nameValuePair name v)] else []) (attrNames set));
|
2012-04-05 15:37:52 +00:00
|
|
|
|
|
2012-10-23 13:35:48 +00:00
|
|
|
|
|
2012-08-28 12:40:24 +00:00
|
|
|
|
/* foldAttrs: apply fold functions to values grouped by key. Eg accumulate values as list:
|
|
|
|
|
foldAttrs (n: a: [n] ++ a) [] [{ a = 2; } { a = 3; }]
|
|
|
|
|
=> { a = [ 2 3 ]; }
|
|
|
|
|
*/
|
|
|
|
|
foldAttrs = op: nul: list_of_attrs:
|
|
|
|
|
fold (n: a:
|
|
|
|
|
fold (name: o:
|
2015-07-23 15:41:35 +00:00
|
|
|
|
o // (listToAttrs [{inherit name; value = op n.${name} (a.${name} or nul); }])
|
2012-08-28 12:40:24 +00:00
|
|
|
|
) a (attrNames n)
|
|
|
|
|
) {} list_of_attrs;
|
2012-04-05 15:37:52 +00:00
|
|
|
|
|
2012-10-23 13:35:48 +00:00
|
|
|
|
|
2009-06-11 16:03:33 +00:00
|
|
|
|
/* Recursively collect sets that verify a given predicate named `pred'
|
|
|
|
|
from the set `attrs'. The recursion is stopped when the predicate is
|
|
|
|
|
verified.
|
|
|
|
|
|
|
|
|
|
Type:
|
|
|
|
|
collect ::
|
|
|
|
|
(AttrSet -> Bool) -> AttrSet -> AttrSet
|
|
|
|
|
|
|
|
|
|
Example:
|
2013-11-12 12:48:19 +00:00
|
|
|
|
collect isList { a = { b = ["b"]; }; c = [1]; }
|
2009-09-29 15:34:19 +00:00
|
|
|
|
=> [["b"] [1]]
|
2009-06-11 16:03:33 +00:00
|
|
|
|
|
|
|
|
|
collect (x: x ? outPath)
|
|
|
|
|
{ a = { outPath = "a/"; }; b = { outPath = "b/"; }; }
|
|
|
|
|
=> [{ outPath = "a/"; } { outPath = "b/"; }]
|
|
|
|
|
*/
|
|
|
|
|
collect = pred: attrs:
|
|
|
|
|
if pred attrs then
|
|
|
|
|
[ attrs ]
|
2013-11-12 12:50:45 +00:00
|
|
|
|
else if isAttrs attrs then
|
2009-06-11 16:03:33 +00:00
|
|
|
|
concatMap (collect pred) (attrValues attrs)
|
|
|
|
|
else
|
|
|
|
|
[];
|
|
|
|
|
|
|
|
|
|
|
2009-03-10 15:18:38 +00:00
|
|
|
|
/* Utility function that creates a {name, value} pair as expected by
|
|
|
|
|
builtins.listToAttrs. */
|
|
|
|
|
nameValuePair = name: value: { inherit name value; };
|
|
|
|
|
|
2012-10-23 13:35:48 +00:00
|
|
|
|
|
2009-03-10 15:18:38 +00:00
|
|
|
|
/* Apply a function to each element in an attribute set. The
|
|
|
|
|
function takes two arguments --- the attribute name and its value
|
|
|
|
|
--- and returns the new value for the attribute. The result is a
|
|
|
|
|
new attribute set.
|
|
|
|
|
|
|
|
|
|
Example:
|
|
|
|
|
mapAttrs (name: value: name + "-" + value)
|
2012-05-25 17:01:58 +00:00
|
|
|
|
{ x = "foo"; y = "bar"; }
|
|
|
|
|
=> { x = "x-foo"; y = "y-bar"; }
|
2009-03-10 15:18:38 +00:00
|
|
|
|
*/
|
|
|
|
|
mapAttrs = f: set:
|
2014-10-04 22:03:52 +00:00
|
|
|
|
listToAttrs (map (attr: { name = attr; value = f attr set.${attr}; }) (attrNames set));
|
2012-05-25 17:01:58 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Like `mapAttrs', but allows the name of each attribute to be
|
|
|
|
|
changed in addition to the value. The applied function should
|
|
|
|
|
return both the new name and value as a `nameValuePair'.
|
2012-10-23 13:35:48 +00:00
|
|
|
|
|
2012-05-25 17:01:58 +00:00
|
|
|
|
Example:
|
|
|
|
|
mapAttrs' (name: value: nameValuePair ("foo_" + name) ("bar-" + value))
|
|
|
|
|
{ x = "a"; y = "b"; }
|
|
|
|
|
=> { foo_x = "bar-a"; foo_y = "bar-b"; }
|
|
|
|
|
*/
|
|
|
|
|
mapAttrs' = f: set:
|
2014-10-04 22:03:52 +00:00
|
|
|
|
listToAttrs (map (attr: f attr set.${attr}) (attrNames set));
|
2012-10-23 13:35:48 +00:00
|
|
|
|
|
2009-03-10 15:18:38 +00:00
|
|
|
|
|
2012-06-14 19:07:01 +00:00
|
|
|
|
/* Call a function for each attribute in the given set and return
|
|
|
|
|
the result in a list.
|
2012-10-23 13:35:48 +00:00
|
|
|
|
|
2012-06-14 19:07:01 +00:00
|
|
|
|
Example:
|
|
|
|
|
mapAttrsToList (name: value: name + value)
|
|
|
|
|
{ x = "a"; y = "b"; }
|
|
|
|
|
=> [ "xa" "yb" ]
|
|
|
|
|
*/
|
|
|
|
|
mapAttrsToList = f: attrs:
|
2014-10-04 22:03:52 +00:00
|
|
|
|
map (name: f name attrs.${name}) (attrNames attrs);
|
2012-10-23 13:35:48 +00:00
|
|
|
|
|
2012-06-14 19:07:01 +00:00
|
|
|
|
|
2009-03-10 15:18:38 +00:00
|
|
|
|
/* Like `mapAttrs', except that it recursively applies itself to
|
2009-03-30 13:19:57 +00:00
|
|
|
|
attribute sets. Also, the first argument of the argument
|
|
|
|
|
function is a *list* of the names of the containing attributes.
|
|
|
|
|
|
|
|
|
|
Type:
|
|
|
|
|
mapAttrsRecursive ::
|
|
|
|
|
([String] -> a -> b) -> AttrSet -> AttrSet
|
2009-03-10 15:18:38 +00:00
|
|
|
|
|
|
|
|
|
Example:
|
|
|
|
|
mapAttrsRecursive (path: value: concatStringsSep "-" (path ++ [value]))
|
|
|
|
|
{ n = { a = "A"; m = { b = "B"; c = "C"; }; }; d = "D"; }
|
|
|
|
|
=> { n = { a = "n-a-A"; m = { b = "n-m-b-B"; c = "n-m-c-C"; }; }; d = "d-D"; }
|
|
|
|
|
*/
|
2009-03-30 13:19:57 +00:00
|
|
|
|
mapAttrsRecursive = mapAttrsRecursiveCond (as: true);
|
|
|
|
|
|
2012-10-23 13:35:48 +00:00
|
|
|
|
|
2009-03-30 13:19:57 +00:00
|
|
|
|
/* Like `mapAttrsRecursive', but it takes an additional predicate
|
|
|
|
|
function that tells it whether to recursive into an attribute
|
|
|
|
|
set. If it returns false, `mapAttrsRecursiveCond' does not
|
|
|
|
|
recurse, but does apply the map function. It is returns true, it
|
|
|
|
|
does recurse, and does not apply the map function.
|
|
|
|
|
|
|
|
|
|
Type:
|
|
|
|
|
mapAttrsRecursiveCond ::
|
|
|
|
|
(AttrSet -> Bool) -> ([String] -> a -> b) -> AttrSet -> AttrSet
|
|
|
|
|
|
|
|
|
|
Example:
|
|
|
|
|
# To prevent recursing into derivations (which are attribute
|
|
|
|
|
# sets with the attribute "type" equal to "derivation"):
|
|
|
|
|
mapAttrsRecursiveCond
|
|
|
|
|
(as: !(as ? "type" && as.type == "derivation"))
|
|
|
|
|
(x: ... do something ...)
|
|
|
|
|
attrs
|
2013-03-06 15:33:01 +00:00
|
|
|
|
*/
|
2009-03-30 13:19:57 +00:00
|
|
|
|
mapAttrsRecursiveCond = cond: f: set:
|
2009-03-10 15:18:38 +00:00
|
|
|
|
let
|
2009-03-30 13:19:57 +00:00
|
|
|
|
recurse = path: set:
|
2009-03-10 15:18:38 +00:00
|
|
|
|
let
|
|
|
|
|
g =
|
|
|
|
|
name: value:
|
2009-03-30 13:19:57 +00:00
|
|
|
|
if isAttrs value && cond value
|
|
|
|
|
then recurse (path ++ [name]) value
|
2009-03-10 15:18:38 +00:00
|
|
|
|
else f (path ++ [name]) value;
|
|
|
|
|
in mapAttrs g set;
|
2009-03-30 13:19:57 +00:00
|
|
|
|
in recurse [] set;
|
2009-04-25 14:08:29 +00:00
|
|
|
|
|
|
|
|
|
|
2013-03-06 15:33:01 +00:00
|
|
|
|
/* Generate an attribute set by mapping a function over a list of
|
|
|
|
|
attribute names.
|
|
|
|
|
|
|
|
|
|
Example:
|
|
|
|
|
genAttrs [ "foo" "bar" ] (name: "x_" + name)
|
|
|
|
|
=> { foo = "x_foo"; bar = "x_bar"; }
|
|
|
|
|
*/
|
|
|
|
|
genAttrs = names: f:
|
|
|
|
|
listToAttrs (map (n: nameValuePair n (f n)) names);
|
|
|
|
|
|
|
|
|
|
|
2009-04-25 14:08:29 +00:00
|
|
|
|
/* Check whether the argument is a derivation. */
|
|
|
|
|
isDerivation = x: isAttrs x && x ? type && x.type == "derivation";
|
|
|
|
|
|
2009-09-10 10:52:51 +00:00
|
|
|
|
|
2015-08-06 17:55:42 +00:00
|
|
|
|
/* Convert a store path to a fake derivation. */
|
|
|
|
|
toDerivation = path:
|
|
|
|
|
let path' = builtins.storePath path; in
|
|
|
|
|
{ type = "derivation";
|
|
|
|
|
name = builtins.unsafeDiscardStringContext (builtins.substring 33 (-1) (baseNameOf path'));
|
|
|
|
|
outPath = path';
|
|
|
|
|
outputs = [ "out" ];
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
2009-09-10 10:52:51 +00:00
|
|
|
|
/* If the Boolean `cond' is true, return the attribute set `as',
|
|
|
|
|
otherwise an empty attribute set. */
|
|
|
|
|
optionalAttrs = cond: as: if cond then as else {};
|
2009-09-29 14:57:00 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Merge sets of attributes and use the function f to merge attributes
|
|
|
|
|
values. */
|
|
|
|
|
zipAttrsWithNames = names: f: sets:
|
|
|
|
|
listToAttrs (map (name: {
|
|
|
|
|
inherit name;
|
|
|
|
|
value = f name (catAttrs name sets);
|
|
|
|
|
}) names);
|
|
|
|
|
|
|
|
|
|
# implentation note: Common names appear multiple times in the list of
|
|
|
|
|
# names, hopefully this does not affect the system because the maximal
|
|
|
|
|
# laziness avoid computing twice the same expression and listToAttrs does
|
|
|
|
|
# not care about duplicated attribute names.
|
2013-10-28 03:46:36 +00:00
|
|
|
|
zipAttrsWith = f: sets: zipAttrsWithNames (concatMap attrNames sets) f sets;
|
2009-09-29 14:57:00 +00:00
|
|
|
|
|
|
|
|
|
zipAttrs = zipAttrsWith (name: values: values);
|
|
|
|
|
|
|
|
|
|
/* backward compatibility */
|
|
|
|
|
zipWithNames = zipAttrsWithNames;
|
2013-08-22 07:06:43 +00:00
|
|
|
|
zip = builtins.trace "lib.zip is deprecated, use lib.zipAttrsWith instead" zipAttrsWith;
|
2009-09-29 14:57:00 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* Does the same as the update operator '//' except that attributes are
|
|
|
|
|
merged until the given pedicate is verified. The predicate should
|
2009-11-19 16:07:47 +00:00
|
|
|
|
accept 3 arguments which are the path to reach the attribute, a part of
|
2009-09-29 14:57:00 +00:00
|
|
|
|
the first attribute set and a part of the second attribute set. When
|
|
|
|
|
the predicate is verified, the value of the first attribute set is
|
|
|
|
|
replaced by the value of the second attribute set.
|
|
|
|
|
|
|
|
|
|
Example:
|
|
|
|
|
recursiveUpdateUntil (path: l: r: path == ["foo"]) {
|
|
|
|
|
# first attribute set
|
|
|
|
|
foo.bar = 1;
|
|
|
|
|
foo.baz = 2;
|
|
|
|
|
bar = 3;
|
|
|
|
|
} {
|
|
|
|
|
#second attribute set
|
|
|
|
|
foo.bar = 1;
|
|
|
|
|
foo.quz = 2;
|
|
|
|
|
baz = 4;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
returns: {
|
|
|
|
|
foo.bar = 1; # 'foo.*' from the second set
|
2012-10-23 13:35:48 +00:00
|
|
|
|
foo.quz = 2; #
|
2009-09-29 14:57:00 +00:00
|
|
|
|
bar = 3; # 'bar' from the first set
|
|
|
|
|
baz = 4; # 'baz' from the second set
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
*/
|
|
|
|
|
recursiveUpdateUntil = pred: lhs: rhs:
|
|
|
|
|
let f = attrPath:
|
|
|
|
|
zipAttrsWith (n: values:
|
|
|
|
|
if tail values == []
|
|
|
|
|
|| pred attrPath (head (tail values)) (head values) then
|
|
|
|
|
head values
|
|
|
|
|
else
|
|
|
|
|
f (attrPath ++ [n]) values
|
|
|
|
|
);
|
|
|
|
|
in f [] [rhs lhs];
|
|
|
|
|
|
2012-11-14 10:38:47 +00:00
|
|
|
|
/* A recursive variant of the update operator ‘//’. The recusion
|
|
|
|
|
stops when one of the attribute values is not an attribute set,
|
|
|
|
|
in which case the right hand side value takes precedence over the
|
2009-09-29 14:57:00 +00:00
|
|
|
|
left hand side value.
|
|
|
|
|
|
|
|
|
|
Example:
|
|
|
|
|
recursiveUpdate {
|
|
|
|
|
boot.loader.grub.enable = true;
|
|
|
|
|
boot.loader.grub.device = "/dev/hda";
|
|
|
|
|
} {
|
|
|
|
|
boot.loader.grub.device = "";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
returns: {
|
|
|
|
|
boot.loader.grub.enable = true;
|
|
|
|
|
boot.loader.grub.device = "";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
*/
|
|
|
|
|
recursiveUpdate = lhs: rhs:
|
|
|
|
|
recursiveUpdateUntil (path: lhs: rhs:
|
|
|
|
|
!(isAttrs lhs && isAttrs rhs)
|
|
|
|
|
) lhs rhs;
|
|
|
|
|
|
2009-11-19 17:19:39 +00:00
|
|
|
|
matchAttrs = pattern: attrs:
|
|
|
|
|
fold or false (attrValues (zipAttrsWithNames (attrNames pattern) (n: values:
|
|
|
|
|
let pat = head values; val = head (tail values); in
|
2012-08-13 22:08:35 +00:00
|
|
|
|
if length values == 1 then false
|
2009-11-19 17:19:39 +00:00
|
|
|
|
else if isAttrs pat then isAttrs val && matchAttrs head values
|
2012-08-13 22:08:35 +00:00
|
|
|
|
else pat == val
|
2009-11-19 17:19:39 +00:00
|
|
|
|
) [pattern attrs]));
|
|
|
|
|
|
2010-07-08 15:31:59 +00:00
|
|
|
|
# override only the attributes that are already present in the old set
|
|
|
|
|
# useful for deep-overriding
|
|
|
|
|
overrideExisting = old: new:
|
2014-10-04 22:03:52 +00:00
|
|
|
|
old // listToAttrs (map (attr: nameValuePair attr (attrByPath [attr] old.${attr} new)) (attrNames old));
|
2010-07-08 15:31:59 +00:00
|
|
|
|
|
2013-02-01 05:39:26 +00:00
|
|
|
|
deepSeqAttrs = x: y: deepSeqList (attrValues x) y;
|
2009-03-10 15:18:38 +00:00
|
|
|
|
}
|