feat(tvix/eval): Implement builtins.intersectAttrs r/5137

Change-Id: Iaba9bcfa19f283cd0c1931be2f211e2528a1a940
Reviewed-on: https://cl.tvl.fyi/c/depot/+/6998
Tested-by: BuildkiteCI
Reviewed-by: tazjin <tazjin@tvl.su>
Reviewed-by: kanepyork <rikingcoding@gmail.com>

1 files changed, 15 insertions, 0 deletions

diff --git a/tvix/eval/src/builtins/mod.rs b/tvix/eval/src/builtins/mod.rs
index 0082d36d5735..03466af65f89 100644
--- a/tvix/eval/src/builtins/mod.rs
+++ b/tvix/eval/src/builtins/mod.rs
@@ -309,6 +309,21 @@ fn pure_builtins() -> Vec<Builtin> {
                 None => Err(ErrorKind::IndexOutOfBounds { index: 0 }),
             }
         }),
+        Builtin::new(
+            "intersectAttrs",
+            &[true, true],
+            |args: Vec<Value>, _: &mut VM| {
+                let mut res = BTreeMap::new();
+                let attrs1 = args[0].to_attrs()?;
+                let attrs2 = args[1].to_attrs()?;
+                for (k, v) in attrs2.iter() {
+                    if attrs1.contains(k) {
+                        res.insert(k.clone(), v.clone());
+                    }
+                }
+                Ok(Value::attrs(NixAttrs::from_map(res)))
+            },
+        ),
         // For `is*` predicates we force manually, as Value::force also unwraps any Thunks
         Builtin::new("isAttrs", &[false], |args: Vec<Value>, vm: &mut VM| {
             let value = args[0].force(vm)?;