about summary refs log tree commit diff
path: root/tvix/eval/src/builtins
diff options
context:
space:
mode:
Diffstat (limited to 'tvix/eval/src/builtins')
-rw-r--r--tvix/eval/src/builtins/mod.rs15
1 files changed, 15 insertions, 0 deletions
diff --git a/tvix/eval/src/builtins/mod.rs b/tvix/eval/src/builtins/mod.rs
index 050481eb7ba3..d767d09b99d5 100644
--- a/tvix/eval/src/builtins/mod.rs
+++ b/tvix/eval/src/builtins/mod.rs
@@ -194,6 +194,21 @@ fn pure_builtins() -> Vec<Builtin> {
             &[false, false],
             |args: Vec<Value>, vm: &mut VM| arithmetic_op!(&*args[0].force(vm)?, &*args[1].force(vm)?, /),
         ),
+        Builtin::new("dirOf", &[true], |args: Vec<Value>, vm: &mut VM| {
+            let s = args[0].coerce_to_string(CoercionKind::Weak, vm)?;
+            let result = s
+                .rsplit_once('/')
+                .map(|(x, _)| match x {
+                    "" => "/",
+                    _ => x,
+                })
+                .unwrap_or(".");
+            if args[0].is_path() {
+                Ok(Value::Path(result.into()))
+            } else {
+                Ok(result.into())
+            }
+        }),
         Builtin::new("elem", &[true, true], |args: Vec<Value>, vm: &mut VM| {
             for val in args[1].to_list()? {
                 if val.nix_eq(&args[0], vm)? {