use crate::{Environment, eval::apply::eval_apply, value::Value};
pub(crate) fn builtin(args: &[Value], env: &Environment) -> Option<(Value, Environment)> {
if let [val] = &args[..] {
match eval_apply(val, env) {
Some((Value::List(elements), new_env)) => Some((Value::Number(elements.len() as f64), new_env)),
Some((Value::String(s), _)) => Some((Value::Number(s.chars().count() as f64), Environment::new())),
Some((e, _)) => {
eprintln!(
concat!(
"Error[ksl::builtin::length]: ",
"Expected a list or a string, but got `{:?}`."
),
e
);
None
}
None => {
eprintln!(
concat!(
"Error[ksl::builtin::length]: ",
"Cannot evaluate expression {:?}."
),
val
);
None
}
}
} else {
eprintln!(
concat!(
"Error[ksl::builtin::length]: ",
"Only accepts 1 argument, but {} were passed."
),
args.len()
);
None
}
}