mdebray/tvl-depot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(tvix/eval): implement builtins.any

Browse source 
Change-Id: I640ee20e7c0a68c4e024a577e429fed9b3a49ece
Reviewed-on: https://cl.tvl.fyi/c/depot/+/6845
Tested-by: BuildkiteCI
Reviewed-by: sterni <sternenseemann@systemli.org>
This commit is contained in:
Vincent Ambo 2022-10-02 20:02:05 +03:00 committed by tazjin
parent 844d288949
commit 0537d88078
3 changed files with 30 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tvix/eval/src/builtins/mod.rs
View file

@ -75,6 +75,20 @@ fn pure_builtins() -> Vec<Builtin> {
Ok(Value::Bool(true))
}),
Builtin::new("any", &[true, true], |args, vm| {
for value in args[1].to_list()?.into_iter() {
let pred_result = {
vm.push(value);
vm.call_value(&args[0])
}?;
if pred_result.force(vm)?.as_bool()? {
return Ok(Value::Bool(true));
}
}
Ok(Value::Bool(false))
}),
Builtin::new("attrNames", &[true], |args, _| {
let xs = args[0].to_attrs()?;
let mut output = Vec::with_capacity(xs.len());

1
tvix/eval/src/tests/tvix_tests/eval-okay-builtins-any.exp Normal file
View file

@ -0,0 +1 @@
[ false true false true true true false true false ]

15
tvix/eval/src/tests/tvix_tests/eval-okay-builtins-any.nix Normal file
View file

@ -0,0 +1,15 @@
[
(builtins.any (x: x) [ ])
(builtins.any (x: x) [ true true true ])
(builtins.any (x: x) [ false false false ])
(builtins.any (x: x) [ true true false ])
(builtins.any (x: x) [ false true true ])
# evaluation should short-circuit
(builtins.any (x: x) [ false true (builtins.abort "should be unreachable") ])
# arbitrary functions supported
(builtins.any (x: x * 2 == 42) [ ])
(builtins.any (x: x * 2 == 42) [ 7 21 42 ])
(builtins.any (x: x * 2 == 42) [ 1 2 3 ])
]