about summary refs log tree commit diff
path: root/tvix/eval/src/builtins
diff options
context:
space:
mode:
authorGriffin Smith <root@gws.fyi>2022-10-23T16·28-0400
committergrfn <grfn@gws.fyi>2022-10-24T13·44+0000
commit7b3bda9e089fe9de2d28a28f64ef8532abb3ac9c (patch)
tree14b0a70c2977daec417a958e78547a39094d5ab7 /tvix/eval/src/builtins
parent8497b83569761e6cb67b355f395d45ef8b02ee80 (diff)
feat(tvix/eval): Implement builtins.mapAttrs r/5192
I played around a little bit with doing this in-place, but ended up
going with this perhaps slightly clone-heavy approach for now because
ideally most clones on Value are cheap - but later we should benchmark
alternate approaches that get to reuse allocations better if necessary
or possible.

Change-Id: If998eb2056cedefdf2fb480b0568ac8329ccfc44
Reviewed-on: https://cl.tvl.fyi/c/depot/+/7068
Autosubmit: grfn <grfn@gws.fyi>
Reviewed-by: tazjin <tazjin@tvl.su>
Tested-by: BuildkiteCI
Diffstat (limited to 'tvix/eval/src/builtins')
-rw-r--r--tvix/eval/src/builtins/mod.rs13
1 files changed, 13 insertions, 0 deletions
diff --git a/tvix/eval/src/builtins/mod.rs b/tvix/eval/src/builtins/mod.rs
index d6b52e20a2a3..e7eb4e8bf953 100644
--- a/tvix/eval/src/builtins/mod.rs
+++ b/tvix/eval/src/builtins/mod.rs
@@ -436,6 +436,19 @@ fn pure_builtins() -> Vec<Builtin> {
                 .map_err(Into::into)
         }),
         Builtin::new(
+            "mapAttrs",
+            &[true, true],
+            |args: Vec<Value>, vm: &mut VM| {
+                let attrs = args[1].to_attrs()?;
+                let mut res = BTreeMap::new();
+                for (key, value) in attrs.as_ref() {
+                    let value = vm.call_with(&args[0], [key.clone().into(), value.clone()])?;
+                    res.insert(key.clone(), value);
+                }
+                Ok(Value::attrs(NixAttrs::from_map(res)))
+            },
+        ),
+        Builtin::new(
             "match",
             &[true, true],
             |mut args: Vec<Value>, _: &mut VM| {