diff options
Diffstat (limited to 'tvix/eval/src/value')
-rw-r--r-- | tvix/eval/src/value/list.rs | 8 | ||||
-rw-r--r-- | tvix/eval/src/value/mod.rs | 15 |
2 files changed, 23 insertions, 0 deletions
diff --git a/tvix/eval/src/value/list.rs b/tvix/eval/src/value/list.rs index d13e220bbe64..66f7eb8108e9 100644 --- a/tvix/eval/src/value/list.rs +++ b/tvix/eval/src/value/list.rs @@ -49,6 +49,14 @@ mod arbitrary { } impl NixList { + pub fn new() -> Self { + Self(vec![]) + } + + pub fn push(&mut self, val: Value) { + self.0.push(val) + } + pub fn concat(&self, other: &Self) -> Self { let mut ret = self.clone(); ret.0.extend_from_slice(&other.0); diff --git a/tvix/eval/src/value/mod.rs b/tvix/eval/src/value/mod.rs index c065d5c1bd6d..14b1a5c6122e 100644 --- a/tvix/eval/src/value/mod.rs +++ b/tvix/eval/src/value/mod.rs @@ -79,6 +79,19 @@ macro_rules! gen_cast { }; } +/// Generate an `as_*_mut/to_*_mut` accessor method that returns either the +/// expected type, or a type error. +macro_rules! gen_cast_mut { + ( $name:ident, $type:ty, $expected:expr, $variant:ident) => { + pub fn $name(&mut self) -> Result<&mut $type, ErrorKind> { + match self { + Value::$variant(x) => Ok(x), + other => Err(type_error($expected, &other)), + } + } + }; +} + /// Generate an `is_*` type-checking method. macro_rules! gen_is { ( $name:ident, $variant:pat ) => { @@ -284,6 +297,8 @@ impl Value { gen_cast!(to_list, NixList, "list", Value::List(l), l.clone()); gen_cast!(to_closure, Closure, "lambda", Value::Closure(c), c.clone()); + gen_cast_mut!(as_list_mut, NixList, "list", List); + gen_is!(is_path, Value::Path(_)); gen_is!(is_number, Value::Integer(_) | Value::Float(_)); gen_is!(is_bool, Value::Bool(_)); |