Len function for strings, arrays and dictionaries

2 files changed, 15 insertions, 0 deletions

diff --git a/src/library/basic.rs b/src/library/basic.rs
index a094f039..333416b4 100644
--- a/src/library/basic.rs
+++ b/src/library/basic.rs
@@ -31,6 +31,20 @@ pub fn repr(ctx: &mut EvalContext, args: &mut FuncArgs) -> Value {
     }
 }
 
+/// `len`: The length of a string, an array or a dictionary.
+pub fn len(ctx: &mut EvalContext, args: &mut FuncArgs) -> Value {
+    match args.eat_expect::<Spanned<Value>>(ctx, "collection") {
+        Some(Spanned { v: Value::Str(v), .. }) => Value::Int(v.len() as i64),
+        Some(Spanned { v: Value::Array(v), .. }) => Value::Int(v.len() as i64),
+        Some(Spanned { v: Value::Dict(v), .. }) => Value::Int(v.len() as i64),
+        Some(other) if other.v != Value::Error => {
+            ctx.diag(error!(other.span, "expected string, array or dictionary"));
+            Value::Error
+        }
+        _ => Value::Error,
+    }
+}
+
 /// `rgb`: Create an RGB(A) color.
 ///
 /// # Positional parameters
diff --git a/src/library/mod.rs b/src/library/mod.rs
index 0536eaa1..6c736081 100644
--- a/src/library/mod.rs
+++ b/src/library/mod.rs
@@ -56,6 +56,7 @@ pub fn new() -> Scope {
     std.def_func("h", h);
     std.def_func("image", image);
     std.def_func("lang", lang);
+    std.def_func("len", len);
     std.def_func("max", max);
     std.def_func("min", min);
     std.def_func("overline", overline);