use crate::eval::coercion::to_number;
use crate::eval::functions::check_arity;
use crate::types::{ErrorKind, Value};
pub fn sum_fn(args: &[Value]) -> Value {
if let Some(err) = check_arity(args, 1, 255) {
return err;
}
let mut sum = 0.0_f64;
for arg in args {
match sum_value(arg) {
Err(e) => return e,
Ok(n) => sum += n,
}
}
if !sum.is_finite() {
return Value::Error(ErrorKind::Num);
}
Value::Number(sum)
}
fn sum_value(v: &Value) -> Result<f64, Value> {
match v {
Value::Array(elems) => {
let mut total = 0.0_f64;
for elem in elems {
total += sum_value(elem)?;
}
Ok(total)
}
other => to_number(other.clone()),
}
}
#[cfg(test)]
mod tests;