From 939b2194fe6b0159191892d7bfc7ff527c1b0371 Mon Sep 17 00:00:00 2001
From: Vincent Ambo <mail@tazj.in>
Date: Sun, 2 Oct 2022 19:41:39 +0300
Subject: feat(tvix/eval): implement builtins.concatLists

Concatenates (but not flattens) a list of lists.

Change-Id: I692e0b3e7b5a5ff93d5768d3a27849b432ec5747
Reviewed-on: https://cl.tvl.fyi/c/depot/+/6843
Autosubmit: tazjin <tazjin@tvl.su>
Reviewed-by: sterni <sternenseemann@systemli.org>
Tested-by: BuildkiteCI
---
 tvix/eval/src/builtins/mod.rs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'tvix/eval/src/builtins/mod.rs')

diff --git a/tvix/eval/src/builtins/mod.rs b/tvix/eval/src/builtins/mod.rs
index 45630819e1..94a035c2ec 100644
--- a/tvix/eval/src/builtins/mod.rs
+++ b/tvix/eval/src/builtins/mod.rs
@@ -116,6 +116,20 @@ fn pure_builtins() -> Vec<Builtin> {
                 std::cmp::Ordering::Greater => Ok(Value::Integer(1)),
             }
         }),
+        Builtin::new("concatLists", &[true], |args, vm| {
+            let list = args[0].to_list()?;
+            let lists = list
+                .into_iter()
+                .map(|elem| {
+                    let value = elem.force(vm)?;
+                    value.to_list()
+                })
+                .collect::<Result<Vec<NixList>, ErrorKind>>()?;
+
+            Ok(Value::List(NixList::from(
+                lists.into_iter().flatten().collect::<Vec<Value>>(),
+            )))
+        }),
         Builtin::new(
             "div",
             &[false, false],
-- 
cgit 1.4.1