From 197fe37daef242596900bcab948d6fc14348f910 Mon Sep 17 00:00:00 2001 From: William Carroll Date: Mon, 29 Aug 2022 17:33:30 -0700 Subject: feat(tvix/eval): Support builtins.length Get the length of a list Change-Id: I41d91e96d833269541a1b3c23b7cc879f96d1e5a Reviewed-on: https://cl.tvl.fyi/c/depot/+/6407 Tested-by: BuildkiteCI Reviewed-by: sterni --- tvix/eval/src/value/list.rs | 4 ++++ tvix/eval/src/value/mod.rs | 11 +++++++++++ 2 files changed, 15 insertions(+) (limited to 'tvix/eval/src/value') diff --git a/tvix/eval/src/value/list.rs b/tvix/eval/src/value/list.rs index b8682a5b3d5a..da86fdda26ec 100644 --- a/tvix/eval/src/value/list.rs +++ b/tvix/eval/src/value/list.rs @@ -28,6 +28,10 @@ impl NixList { lhs } + pub fn len(&self) -> usize { + self.0.len() + } + pub fn construct(count: usize, stack_slice: Vec) -> Self { debug_assert!( count == stack_slice.len(), diff --git a/tvix/eval/src/value/mod.rs b/tvix/eval/src/value/mod.rs index c5565e6176b3..a8bfc164cdf9 100644 --- a/tvix/eval/src/value/mod.rs +++ b/tvix/eval/src/value/mod.rs @@ -103,6 +103,17 @@ impl Value { } } + pub fn as_list(&self) -> EvalResult<&NixList> { + match self { + Value::List(xs) => Ok(xs), + other => Err(ErrorKind::TypeError { + expected: "list", + actual: other.type_of(), + } + .into()), + } + } + pub fn to_string(self) -> EvalResult { match self { Value::String(s) => Ok(s), -- cgit 1.4.1