Function map

aiscript-vm/src/builtins/function.rs:3–51

(
    state: &mut State<'gc>,
    args: Vec<Value<'gc>>,
)

use crate::{Value, VmError, vm::State};

pub(super) fn map<'gc>(
    state: &mut State<'gc>,
    args: Vec<Value<'gc>>,
) -> Result<Value<'gc>, VmError> {
    if args.len() != 2 {
        return Err(VmError::RuntimeError(
            "map() takes exactly 2 arguments.".into(),
        ));
    }

    // Get the iterable
    let vec = match &args[0] {
        Value::List(list) => &list.borrow().data,
        _ => {
            return Err(VmError::RuntimeError(
                "map() first argument must be an array.".into(),
            ));
        }
    };

    // Get the function id
    let function = match args[1] {
        Value::Closure(ref f) => f.function,
        Value::NativeFunction(_) => {
            return Err(VmError::RuntimeError(
                "map() doesn't support native functions yet.".into(),
            ));
        }
        _ => {
            return Err(VmError::RuntimeError(
                "map() second argument must be a function.".into(),
            ));
        }
    };

    // Create result array
    let mut result = Vec::with_capacity(vec.len());

    // Apply function to each element
    for value in vec.iter() {
        // Prepare arguments for the function call
        let call_args = vec![*value];

        // Call the function and convert result
        result.push(state.eval_function(function, &call_args)?);
    }

    Ok(Value::array(state, result))
}

pub(super) fn filter<'gc>(
    state: &mut State<'gc>,