71 lines
1.9 KiB
Rust
71 lines
1.9 KiB
Rust
use super::error::{BadArgumentCount, MissingChild, Unnamed};
|
|
use super::{util, Context, Instruction};
|
|
use roxmltree::Node;
|
|
use std::cell::RefCell;
|
|
use std::error::Error;
|
|
use std::rc::Rc;
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub struct Function {
|
|
pub args: Vec<String>,
|
|
pub ins: Vec<Instruction>,
|
|
}
|
|
|
|
impl Function {
|
|
pub fn run(
|
|
&self,
|
|
args: Vec<Value>,
|
|
ctx: &mut Context,
|
|
globals: &Context,
|
|
) -> Result<Option<Value>, Box<dyn Error>> {
|
|
if args.len() != self.args.len() {
|
|
Err(BadArgumentCount("function", args.len()))?
|
|
}
|
|
self.args
|
|
.iter()
|
|
.zip(args.into_iter())
|
|
.for_each(|(p, a)| ctx.assign(p.clone(), a));
|
|
for i in self.ins.iter() {
|
|
i.run(ctx, globals)?;
|
|
}
|
|
Ok(ctx.take(&String::from("__return")))
|
|
}
|
|
|
|
pub fn from(fun: &Node<'_, '_>) -> Result<Function, Box<dyn Error>> {
|
|
Ok(Function {
|
|
args: util::find_node(&fun, "arguments")
|
|
.ok_or(MissingChild("call", "arguments"))?
|
|
.children()
|
|
.filter(Node::is_element)
|
|
.map(|n| n.attribute("name").and_then(|s| Some(String::from(s))))
|
|
.collect::<Option<Vec<String>>>()
|
|
.ok_or(Unnamed("argument"))?,
|
|
ins: Instruction::from_children(
|
|
util::find_node(&fun, "body").ok_or(MissingChild("call", "body"))?,
|
|
)?,
|
|
})
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub enum Value {
|
|
Integer(i64),
|
|
Float(f64),
|
|
String(String),
|
|
Array(Rc<RefCell<Vec<Value>>>),
|
|
Function(Function),
|
|
StdFunction(fn(Vec<Value>) -> Result<Option<Value>, Box<dyn Error>>),
|
|
}
|
|
|
|
impl Value {
|
|
pub fn to_bool(&self) -> bool {
|
|
match self {
|
|
Value::Integer(i) => *i != 0,
|
|
Value::Float(f) => *f != 0.0,
|
|
Value::String(s) => s.len() != 0,
|
|
Value::Array(v) => v.borrow().len() != 0,
|
|
_ => true,
|
|
}
|
|
}
|
|
}
|