aiscript-vm 0.2.0

AIScript programming language interpreter
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148

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>,
    args: Vec<Value<'gc>>,
) -> Result<Value<'gc>, VmError> {
    if args.len() != 2 {
        return Err(VmError::RuntimeError(
            "filter() takes exactly 2 arguments.".into(),
        ));
    }

    // Get the iterable
    let vec = match &args[0] {
        Value::List(list) => &list.borrow().data,
        _ => {
            return Err(VmError::RuntimeError(
                "filter() 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(
                "filter() doesn't support native functions yet.".into(),
            ));
        }
        _ => {
            return Err(VmError::RuntimeError(
                "filter() second argument must be a function.".into(),
            ));
        }
    };

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

    // Filter elements based on predicate function
    for value in vec.iter() {
        // Prepare arguments for the function call
        let call_args = vec![*value];

        // Call function and check if it returns true
        if state.eval_function(function, &call_args)?.is_true() {
            result.push(*value);
        }
    }

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

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

    // Get all arrays and validate input
    let mut arrays = Vec::new();
    for arg in &args {
        match arg {
            Value::List(list) => arrays.push(list),
            _ => {
                return Err(VmError::RuntimeError(
                    "zip() arguments must be arrays.".into(),
                ));
            }
        }
    }

    // Find the length of the shortest array
    let min_len = arrays
        .iter()
        .map(|arr| arr.borrow().data.len())
        .min()
        .unwrap_or(0);

    // Create result array
    let mut result = Vec::new();

    // Zip the arrays together
    for i in 0..min_len {
        let mut tuple = Vec::new();
        for arr in arrays.iter() {
            tuple.push(arr.borrow().data[i]);
        }
        result.push(Value::array(state, tuple));
    }

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