file separation
This commit is contained in:
1
.gitignore
vendored
1
.gitignore
vendored
@@ -1 +1,2 @@
|
|||||||
/target
|
/target
|
||||||
|
/.vscode
|
||||||
31
src/context.rs
Normal file
31
src/context.rs
Normal file
@@ -0,0 +1,31 @@
|
|||||||
|
use super::value::Value;
|
||||||
|
use std::collections::HashMap;
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct Context {
|
||||||
|
dict: HashMap<String, Value>,
|
||||||
|
parent: Option<Box<Context>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Context {
|
||||||
|
pub fn new(parent: Option<Box<Context>>) -> Context {
|
||||||
|
Context {
|
||||||
|
dict: HashMap::new(),
|
||||||
|
parent,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn assign(&mut self, key: String, value: Value) {
|
||||||
|
self.dict.insert(key, value);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn value(&self, key: &String) -> Option<&Value> {
|
||||||
|
self.dict
|
||||||
|
.get(key)
|
||||||
|
.or(self.parent.as_ref().and_then(|p| p.value(key)))
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn take(&mut self, key: &String) -> Option<Value> {
|
||||||
|
self.dict.remove(key)
|
||||||
|
}
|
||||||
|
}
|
||||||
728
src/instruction.rs
Normal file
728
src/instruction.rs
Normal file
@@ -0,0 +1,728 @@
|
|||||||
|
use super::{util, Context, Value};
|
||||||
|
use roxmltree::Node;
|
||||||
|
use std::cell::RefCell;
|
||||||
|
use std::error::Error;
|
||||||
|
use std::rc::Rc;
|
||||||
|
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
pub enum Instruction {
|
||||||
|
Value(String),
|
||||||
|
Assign(String, Box<Instruction>),
|
||||||
|
Integer(String),
|
||||||
|
IntegerCast(Box<Instruction>),
|
||||||
|
Float(String),
|
||||||
|
FloatCast(Box<Instruction>),
|
||||||
|
String(String),
|
||||||
|
StringCast(Box<Instruction>),
|
||||||
|
Array(Vec<Instruction>),
|
||||||
|
Add(Vec<Instruction>),
|
||||||
|
Subtract(Vec<Instruction>),
|
||||||
|
Multiply(Vec<Instruction>),
|
||||||
|
Divide(Vec<Instruction>),
|
||||||
|
And(Vec<Instruction>),
|
||||||
|
Or(Vec<Instruction>),
|
||||||
|
Not(Box<Instruction>),
|
||||||
|
Equal(Box<Instruction>, Box<Instruction>),
|
||||||
|
Greater(Box<Instruction>, Box<Instruction>),
|
||||||
|
Lower(Box<Instruction>, Box<Instruction>),
|
||||||
|
Call(Box<Instruction>, Vec<Instruction>),
|
||||||
|
CallNamed(String, Vec<Instruction>),
|
||||||
|
Return(Box<Instruction>),
|
||||||
|
If(Box<Instruction>, Vec<Instruction>),
|
||||||
|
IfElse(Box<Instruction>, Vec<Instruction>, Vec<Instruction>),
|
||||||
|
For {
|
||||||
|
variable: String,
|
||||||
|
from: Box<Instruction>,
|
||||||
|
to: Box<Instruction>,
|
||||||
|
step: Box<Instruction>,
|
||||||
|
body: Vec<Instruction>,
|
||||||
|
},
|
||||||
|
Each(String, Box<Instruction>, Vec<Instruction>),
|
||||||
|
While(Box<Instruction>, Vec<Instruction>),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Instruction {
|
||||||
|
pub fn new(node: Node) -> Result<Instruction, Box<dyn Error>> {
|
||||||
|
Ok(match util::tag_name(&node).as_str() {
|
||||||
|
"value" => Instruction::Value(
|
||||||
|
node.attribute("variable")
|
||||||
|
.and_then(|a| Some(String::from(a)))
|
||||||
|
.ok_or("missing 'variable' attribute on 'value' tag")?,
|
||||||
|
),
|
||||||
|
"assign" => Instruction::Assign(
|
||||||
|
String::from(
|
||||||
|
node.attribute("variable")
|
||||||
|
.ok_or("missing 'variable' attribute on 'assign' tag")?,
|
||||||
|
),
|
||||||
|
Box::new(Instruction::new(
|
||||||
|
node.first_element_child()
|
||||||
|
.ok_or("missing child on 'assign' tag")?,
|
||||||
|
)?),
|
||||||
|
),
|
||||||
|
"integer" => {
|
||||||
|
if let Some(v) = node.attribute("value") {
|
||||||
|
Instruction::Integer(String::from(v))
|
||||||
|
} else if let Some(n) = node.first_element_child() {
|
||||||
|
Instruction::IntegerCast(Box::new(Instruction::new(n)?))
|
||||||
|
} else {
|
||||||
|
Err("missing 'value' attribute or child in 'integer' tag")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"float" => {
|
||||||
|
if let Some(v) = node.attribute("value") {
|
||||||
|
Instruction::Float(String::from(v))
|
||||||
|
} else if let Some(n) = node.first_element_child() {
|
||||||
|
Instruction::FloatCast(Box::new(Instruction::new(n)?))
|
||||||
|
} else {
|
||||||
|
Err("missing 'value' attribute or child in 'float' tag")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"string" => {
|
||||||
|
if let Some(v) = node.attribute("value") {
|
||||||
|
Instruction::String(String::from(v))
|
||||||
|
} else if let Some(n) = node.first_element_child() {
|
||||||
|
Instruction::StringCast(Box::new(Instruction::new(n)?))
|
||||||
|
} else {
|
||||||
|
Err("missing 'value' attribute or child in 'string' tag")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"array" => Instruction::Array(Instruction::from_children(node)?),
|
||||||
|
"add" => Instruction::Add(Instruction::from_children(node)?),
|
||||||
|
"subtract" => Instruction::Subtract(Instruction::from_children(node)?),
|
||||||
|
"multiply" => Instruction::Multiply(Instruction::from_children(node)?),
|
||||||
|
"divide" => Instruction::Divide(Instruction::from_children(node)?),
|
||||||
|
"and" => Instruction::And(Instruction::from_children(node)?),
|
||||||
|
"or" => Instruction::Or(Instruction::from_children(node)?),
|
||||||
|
"not" => Instruction::Not(Box::new(Instruction::new(
|
||||||
|
node.first_element_child()
|
||||||
|
.ok_or("missing value child element in 'not' tag")?,
|
||||||
|
)?)),
|
||||||
|
"equal" => {
|
||||||
|
let children: Vec<Node> = node.children().filter(Node::is_element).collect();
|
||||||
|
if children.len() == 2 {
|
||||||
|
Instruction::Equal(
|
||||||
|
Box::new(Instruction::new(children[0])?),
|
||||||
|
Box::new(Instruction::new(children[1])?),
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
Err("bad child count in 'equal' tag")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"greater" => {
|
||||||
|
let children: Vec<Node> = node.children().filter(Node::is_element).collect();
|
||||||
|
if children.len() == 2 {
|
||||||
|
Instruction::Greater(
|
||||||
|
Box::new(Instruction::new(children[0])?),
|
||||||
|
Box::new(Instruction::new(children[1])?),
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
Err("bad child count in 'greater' tag")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"lower" => {
|
||||||
|
let children: Vec<Node> = node.children().filter(Node::is_element).collect();
|
||||||
|
if children.len() == 2 {
|
||||||
|
Instruction::Lower(
|
||||||
|
Box::new(Instruction::new(children[0])?),
|
||||||
|
Box::new(Instruction::new(children[1])?),
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
Err("bad child count in 'lower' tag")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"call" => {
|
||||||
|
if let Some(function) = node.attribute("function") {
|
||||||
|
Instruction::CallNamed(
|
||||||
|
String::from(function),
|
||||||
|
Instruction::from_children(
|
||||||
|
util::find_node(&node, "arguments")
|
||||||
|
.ok_or("missing 'arguments' block in 'call' tag")?,
|
||||||
|
)?,
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
Instruction::Call(
|
||||||
|
Box::new(Instruction::new(
|
||||||
|
node.first_element_child()
|
||||||
|
.ok_or("missing function child element in 'call' tag")?,
|
||||||
|
)?),
|
||||||
|
Instruction::from_children(
|
||||||
|
util::find_node(&node, "arguments")
|
||||||
|
.ok_or("missing 'arguments' block in 'call' tag")?,
|
||||||
|
)?,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"return" => Instruction::Return(Box::new(Instruction::new(
|
||||||
|
node.first_element_child()
|
||||||
|
.ok_or("missing value child element in 'return' tag")?,
|
||||||
|
)?)),
|
||||||
|
"if" => {
|
||||||
|
if let Some(else_node) = node.children().find(|n| util::tag_name(n) == "else") {
|
||||||
|
Instruction::IfElse(
|
||||||
|
Box::new(Instruction::new(
|
||||||
|
node.first_element_child()
|
||||||
|
.ok_or("missing condition child element in 'if' tag")?,
|
||||||
|
)?),
|
||||||
|
Instruction::from_children(
|
||||||
|
util::find_node(&node, "then")
|
||||||
|
.ok_or("missing 'then' block in 'if' tag")?,
|
||||||
|
)?,
|
||||||
|
Instruction::from_children(else_node)?,
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
Instruction::If(
|
||||||
|
Box::new(Instruction::new(
|
||||||
|
node.first_element_child()
|
||||||
|
.ok_or("missing condition child element in 'if' tag")?,
|
||||||
|
)?),
|
||||||
|
Instruction::from_children(
|
||||||
|
util::find_node(&node, "then")
|
||||||
|
.ok_or("missing 'then' block in 'if' tag")?,
|
||||||
|
)?,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"for" => Instruction::For {
|
||||||
|
variable: String::from(
|
||||||
|
node.attribute("variable")
|
||||||
|
.ok_or("missing 'variable' attribute on 'for' tag")?,
|
||||||
|
),
|
||||||
|
from: Box::new(Instruction::new(
|
||||||
|
util::find_node(&node, "from")
|
||||||
|
.and_then(|n| n.first_element_child())
|
||||||
|
.ok_or("missing 'from' child in 'for' tag")?,
|
||||||
|
)?),
|
||||||
|
to: Box::new(Instruction::new(
|
||||||
|
util::find_node(&node, "to")
|
||||||
|
.and_then(|n| n.first_element_child())
|
||||||
|
.ok_or("missing 'to' child in 'for' tag")?,
|
||||||
|
)?),
|
||||||
|
step: Box::new(Instruction::new(
|
||||||
|
util::find_node(&node, "step")
|
||||||
|
.and_then(|n| n.first_element_child())
|
||||||
|
.ok_or("missing 'step' child in 'for' tag")?,
|
||||||
|
)?),
|
||||||
|
body: Instruction::from_children(
|
||||||
|
util::find_node(&node, "do").ok_or("missing 'do' block in 'for' tag")?,
|
||||||
|
)?,
|
||||||
|
},
|
||||||
|
"each" => Instruction::Each(
|
||||||
|
String::from(
|
||||||
|
node.attribute("variable")
|
||||||
|
.ok_or("missing 'variable' attribute on 'each' tag")?,
|
||||||
|
),
|
||||||
|
Box::new(Instruction::new(
|
||||||
|
node.first_element_child()
|
||||||
|
.ok_or("missing array child element in 'for' tag")?,
|
||||||
|
)?),
|
||||||
|
Instruction::from_children(
|
||||||
|
util::find_node(&node, "do").ok_or("missing 'do' block in 'each' tag")?,
|
||||||
|
)?,
|
||||||
|
),
|
||||||
|
"while" => Instruction::While(
|
||||||
|
Box::new(Instruction::new(
|
||||||
|
node.first_element_child()
|
||||||
|
.ok_or("missing condition child element in 'while' tag")?,
|
||||||
|
)?),
|
||||||
|
Instruction::from_children(
|
||||||
|
util::find_node(&node, "do").ok_or("missing 'do' block in 'while' tag")?,
|
||||||
|
)?,
|
||||||
|
),
|
||||||
|
tag => Err(format!("unknown tag '{}'", tag))?,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn from_children(node: Node) -> Result<Vec<Instruction>, Box<dyn Error>> {
|
||||||
|
node.children()
|
||||||
|
.filter(Node::is_element)
|
||||||
|
.map(Instruction::new)
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
|
|
||||||
|
fn add(vals: Vec<Value>) -> Result<Value, Box<dyn Error>> {
|
||||||
|
if vals.iter().all(|v| matches!(v, Value::Integer(_))) {
|
||||||
|
Ok(Value::Integer(
|
||||||
|
vals.iter()
|
||||||
|
.map(|v| {
|
||||||
|
if let Value::Integer(i) = v {
|
||||||
|
Ok(i)
|
||||||
|
} else {
|
||||||
|
Err("invalid value")?
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.sum::<Result<i64, Box<dyn Error>>>()?,
|
||||||
|
))
|
||||||
|
} else if vals
|
||||||
|
.iter()
|
||||||
|
.all(|v| matches!(v, Value::Integer(_)) || matches!(v, Value::Float(_)))
|
||||||
|
{
|
||||||
|
Ok(Value::Float(
|
||||||
|
vals.iter()
|
||||||
|
.map(|v| {
|
||||||
|
if let Value::Integer(i) = v {
|
||||||
|
Ok(*i as f64)
|
||||||
|
} else if let Value::Float(f) = v {
|
||||||
|
Ok(*f)
|
||||||
|
} else {
|
||||||
|
Err("invalid value")?
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.sum::<Result<f64, Box<dyn Error>>>()?,
|
||||||
|
))
|
||||||
|
} else if vals.iter().all(|v| {
|
||||||
|
matches!(v, Value::Integer(_))
|
||||||
|
|| matches!(v, Value::Float(_))
|
||||||
|
|| matches!(v, Value::String(_))
|
||||||
|
}) {
|
||||||
|
Ok(Value::String(
|
||||||
|
vals.iter()
|
||||||
|
.map(|v| {
|
||||||
|
Ok(if let Value::String(s) = v {
|
||||||
|
s.to_string()
|
||||||
|
} else if let Value::Integer(i) = v {
|
||||||
|
i.to_string()
|
||||||
|
} else if let Value::Float(f) = v {
|
||||||
|
f.to_string()
|
||||||
|
} else {
|
||||||
|
Err("invalid value")?
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.collect::<Result<Vec<String>, Box<dyn Error>>>()?
|
||||||
|
.join(""),
|
||||||
|
))
|
||||||
|
} else {
|
||||||
|
Err("invalid value")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn subtract(vals: Vec<Value>) -> Result<Value, Box<dyn Error>> {
|
||||||
|
Ok(if vals.iter().all(|v| matches!(v, Value::Integer(_))) {
|
||||||
|
let first = match vals.first().ok_or("missing values in 'subtract' tag")? {
|
||||||
|
Value::Integer(i) => i,
|
||||||
|
_ => Err("invalid value")?,
|
||||||
|
};
|
||||||
|
Value::Integer(
|
||||||
|
*first
|
||||||
|
- vals
|
||||||
|
.iter()
|
||||||
|
.skip(1)
|
||||||
|
.map(|v| {
|
||||||
|
if let Value::Integer(i) = v {
|
||||||
|
Ok(i)
|
||||||
|
} else {
|
||||||
|
Err("invalid value")?
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.sum::<Result<i64, Box<dyn Error>>>()?,
|
||||||
|
)
|
||||||
|
} else if vals
|
||||||
|
.iter()
|
||||||
|
.all(|v| matches!(v, Value::Integer(_)) || matches!(v, Value::Float(_)))
|
||||||
|
{
|
||||||
|
let first = match vals.first().ok_or("not enough values in 'subtract' tag")? {
|
||||||
|
Value::Integer(v) => *v as f64,
|
||||||
|
Value::Float(v) => *v,
|
||||||
|
_ => Err("invalid value")?,
|
||||||
|
};
|
||||||
|
Value::Float(
|
||||||
|
first
|
||||||
|
- vals
|
||||||
|
.iter()
|
||||||
|
.skip(1)
|
||||||
|
.map(|val| {
|
||||||
|
Ok(match val {
|
||||||
|
Value::Integer(v) => *v as f64,
|
||||||
|
Value::Float(v) => *v,
|
||||||
|
_ => Err("invalid value")?,
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.sum::<Result<f64, Box<dyn Error>>>()?,
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
Err("invalid value")?
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn multiply(vals: Vec<Value>) -> Result<Value, Box<dyn Error>> {
|
||||||
|
if vals.iter().all(|v| matches!(v, Value::Integer(_))) {
|
||||||
|
Ok(Value::Integer(
|
||||||
|
vals.iter()
|
||||||
|
.map(|v| {
|
||||||
|
if let Value::Integer(i) = v {
|
||||||
|
Ok(i)
|
||||||
|
} else {
|
||||||
|
Err("invalid value")?
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.product::<Result<i64, Box<dyn Error>>>()?,
|
||||||
|
))
|
||||||
|
} else if vals
|
||||||
|
.iter()
|
||||||
|
.all(|v| matches!(v, Value::Integer(_)) || matches!(v, Value::Float(_)))
|
||||||
|
{
|
||||||
|
Ok(Value::Float(
|
||||||
|
vals.iter()
|
||||||
|
.map(|val| {
|
||||||
|
Ok(match val {
|
||||||
|
Value::Integer(v) => *v as f64,
|
||||||
|
Value::Float(v) => *v,
|
||||||
|
_ => Err("invalid value")?,
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.product::<Result<f64, Box<dyn Error>>>()?,
|
||||||
|
))
|
||||||
|
} else {
|
||||||
|
Err("invalid value")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn divide(vals: Vec<Value>) -> Result<Value, Box<dyn Error>> {
|
||||||
|
if vals
|
||||||
|
.iter()
|
||||||
|
.all(|v| matches!(v, Value::Integer(_)) || matches!(v, Value::Float(_)))
|
||||||
|
{
|
||||||
|
let first = match vals.first().ok_or("not enough values in 'divide' tag")? {
|
||||||
|
Value::Integer(v) => *v as f64,
|
||||||
|
Value::Float(v) => *v,
|
||||||
|
_ => Err("invalid value")?,
|
||||||
|
};
|
||||||
|
Ok(Value::Float(
|
||||||
|
first
|
||||||
|
* vals
|
||||||
|
.iter()
|
||||||
|
.skip(1)
|
||||||
|
.map(|val| {
|
||||||
|
Ok(match val {
|
||||||
|
Value::Integer(v) => 1.0 / (*v as f64),
|
||||||
|
Value::Float(v) => 1.0 / *v,
|
||||||
|
_ => Err("invalid value")?,
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.product::<Result<f64, Box<dyn Error>>>()?,
|
||||||
|
))
|
||||||
|
} else {
|
||||||
|
Err("invalid value")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn and(vals: Vec<Value>) -> Value {
|
||||||
|
Value::Integer(if vals.iter().all(Value::to_bool) {
|
||||||
|
1
|
||||||
|
} else {
|
||||||
|
0
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn or(vals: Vec<Value>) -> Value {
|
||||||
|
Value::Integer(if vals.iter().any(Value::to_bool) {
|
||||||
|
1
|
||||||
|
} else {
|
||||||
|
0
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
fn compare(v1: Value, v2: Value) -> Result<i64, Box<dyn Error>> {
|
||||||
|
use std::cmp::Ordering;
|
||||||
|
match v1 {
|
||||||
|
Value::Integer(i1) => match v2 {
|
||||||
|
Value::Integer(i2) => Ok(i1 - i2),
|
||||||
|
Value::Float(f2) => Ok(
|
||||||
|
match (i1 as f64)
|
||||||
|
.partial_cmp(&f2)
|
||||||
|
.ok_or("incompatible comparison values")?
|
||||||
|
{
|
||||||
|
Ordering::Less => -1,
|
||||||
|
Ordering::Equal => 0,
|
||||||
|
Ordering::Greater => 1,
|
||||||
|
},
|
||||||
|
),
|
||||||
|
_ => Err("incompatible comparison values")?,
|
||||||
|
},
|
||||||
|
Value::Float(f1) => match v2 {
|
||||||
|
Value::Integer(i2) => Ok(
|
||||||
|
match f1
|
||||||
|
.partial_cmp(&(i2 as f64))
|
||||||
|
.ok_or("incompatible comparison values")?
|
||||||
|
{
|
||||||
|
Ordering::Less => -1,
|
||||||
|
Ordering::Equal => 0,
|
||||||
|
Ordering::Greater => 1,
|
||||||
|
},
|
||||||
|
),
|
||||||
|
Value::Float(f2) => Ok(
|
||||||
|
match f1
|
||||||
|
.partial_cmp(&f2)
|
||||||
|
.ok_or("incompatible comparison values")?
|
||||||
|
{
|
||||||
|
Ordering::Less => -1,
|
||||||
|
Ordering::Equal => 0,
|
||||||
|
Ordering::Greater => 1,
|
||||||
|
},
|
||||||
|
),
|
||||||
|
_ => Err("incompatible comparison values")?,
|
||||||
|
},
|
||||||
|
Value::String(s1) => {
|
||||||
|
if let Value::String(s2) = v2 {
|
||||||
|
Ok(match s1.cmp(&s2) {
|
||||||
|
Ordering::Less => -1,
|
||||||
|
Ordering::Equal => 0,
|
||||||
|
Ordering::Greater => 1,
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
Err("incompatible comparison values")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
_ => Err("incompatible comparison values")?,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn run_all(
|
||||||
|
ins: &Vec<Instruction>,
|
||||||
|
ctx: &mut Context,
|
||||||
|
) -> Result<Option<Vec<Value>>, Box<dyn Error>> {
|
||||||
|
ins.iter().map(|i| i.run(ctx)).collect()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn run(&self, ctx: &mut Context) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
Ok(if let None = ctx.value(&String::from("__return")) {
|
||||||
|
match self {
|
||||||
|
Instruction::Value(key) => Some(
|
||||||
|
match ctx
|
||||||
|
.value(key)
|
||||||
|
.ok_or(format!("unknown variable '{}'", key))?
|
||||||
|
{
|
||||||
|
Value::Array(vecrc) => Value::Array(Rc::clone(vecrc)),
|
||||||
|
val => val.clone(),
|
||||||
|
},
|
||||||
|
),
|
||||||
|
Instruction::Assign(key, ins) => {
|
||||||
|
let v = ins.run(ctx)?.ok_or("invalid child value in 'assign' tag")?;
|
||||||
|
ctx.assign(key.clone(), v);
|
||||||
|
None
|
||||||
|
}
|
||||||
|
Instruction::Integer(val) => Some(Value::Integer(val.parse()?)),
|
||||||
|
Instruction::IntegerCast(ins) => Some(Value::Integer(
|
||||||
|
match ins.run(ctx)?.ok_or("no value to be cast to 'integer'")? {
|
||||||
|
Value::Integer(i) => i,
|
||||||
|
Value::Float(f) => f as i64,
|
||||||
|
Value::String(s) => s.parse()?,
|
||||||
|
_ => Err("value cannot be cast to 'integer'")?,
|
||||||
|
},
|
||||||
|
)),
|
||||||
|
Instruction::Float(val) => Some(Value::Float(val.parse()?)),
|
||||||
|
Instruction::FloatCast(ins) => Some(Value::Float(
|
||||||
|
match ins.run(ctx)?.ok_or("no value to be cast to 'float'")? {
|
||||||
|
Value::Integer(i) => i as f64,
|
||||||
|
Value::Float(f) => f,
|
||||||
|
Value::String(s) => s.parse()?,
|
||||||
|
_ => Err("value cannot be cast to 'float'")?,
|
||||||
|
},
|
||||||
|
)),
|
||||||
|
Instruction::String(val) => Some(Value::String(val.clone())),
|
||||||
|
Instruction::StringCast(ins) => Some(Value::String(
|
||||||
|
match ins.run(ctx)?.ok_or("no value to be cast to 'string'")? {
|
||||||
|
Value::Integer(i) => i.to_string(),
|
||||||
|
Value::Float(f) => f.to_string(),
|
||||||
|
Value::String(s) => s,
|
||||||
|
_ => Err("value cannot be cast to 'string'")?,
|
||||||
|
},
|
||||||
|
)),
|
||||||
|
Instruction::Array(args) => Some(Value::Array(Rc::new(RefCell::new(
|
||||||
|
Instruction::run_all(args, ctx)?
|
||||||
|
.ok_or("invalid child values in 'array' tag")?,
|
||||||
|
)))),
|
||||||
|
Instruction::Add(args) => {
|
||||||
|
let vals = Instruction::run_all(args, ctx)?
|
||||||
|
.ok_or("invalid child values in 'add' tag")?;
|
||||||
|
Some(Instruction::add(vals)?)
|
||||||
|
}
|
||||||
|
Instruction::Subtract(args) => {
|
||||||
|
let vals = Instruction::run_all(args, ctx)?
|
||||||
|
.ok_or("invalid child values in 'subtract' tag")?;
|
||||||
|
Some(Instruction::subtract(vals)?)
|
||||||
|
}
|
||||||
|
Instruction::Multiply(args) => {
|
||||||
|
let vals = Instruction::run_all(args, ctx)?
|
||||||
|
.ok_or("invalid child values in 'multiply' tag")?;
|
||||||
|
Some(Instruction::multiply(vals)?)
|
||||||
|
}
|
||||||
|
Instruction::Divide(args) => {
|
||||||
|
let vals = Instruction::run_all(args, ctx)?
|
||||||
|
.ok_or("invalid child values in 'divide' tag")?;
|
||||||
|
Some(Instruction::divide(vals)?)
|
||||||
|
}
|
||||||
|
Instruction::And(args) => {
|
||||||
|
let vals = Instruction::run_all(args, ctx)?
|
||||||
|
.ok_or("invalid child values in 'and' tag")?;
|
||||||
|
Some(Instruction::and(vals))
|
||||||
|
}
|
||||||
|
Instruction::Or(args) => {
|
||||||
|
let vals = Instruction::run_all(args, ctx)?
|
||||||
|
.ok_or("invalid child values in 'or' tag")?;
|
||||||
|
Some(Instruction::or(vals))
|
||||||
|
}
|
||||||
|
Instruction::Not(arg) => Some(Value::Integer(
|
||||||
|
if arg
|
||||||
|
.run(ctx)?
|
||||||
|
.ok_or("invalid child value in 'not' tag")?
|
||||||
|
.to_bool()
|
||||||
|
{
|
||||||
|
0
|
||||||
|
} else {
|
||||||
|
1
|
||||||
|
},
|
||||||
|
)),
|
||||||
|
Instruction::Equal(v1, v2) => Some(Value::Integer(
|
||||||
|
if Instruction::compare(
|
||||||
|
v1.run(ctx)?.ok_or("invalid child value in 'equal' tag")?,
|
||||||
|
v2.run(ctx)?.ok_or("invalid child value in 'equal' tag")?,
|
||||||
|
)? == 0
|
||||||
|
{
|
||||||
|
1
|
||||||
|
} else {
|
||||||
|
0
|
||||||
|
},
|
||||||
|
)),
|
||||||
|
Instruction::Greater(v1, v2) => Some(Value::Integer(
|
||||||
|
if Instruction::compare(
|
||||||
|
v1.run(ctx)?.ok_or("invalid child value in 'equal' tag")?,
|
||||||
|
v2.run(ctx)?.ok_or("invalid child value in 'equal' tag")?,
|
||||||
|
)? > 0
|
||||||
|
{
|
||||||
|
1
|
||||||
|
} else {
|
||||||
|
0
|
||||||
|
},
|
||||||
|
)),
|
||||||
|
Instruction::Lower(v1, v2) => Some(Value::Integer(
|
||||||
|
if Instruction::compare(
|
||||||
|
v1.run(ctx)?.ok_or("invalid child value in 'equal' tag")?,
|
||||||
|
v2.run(ctx)?.ok_or("invalid child value in 'equal' tag")?,
|
||||||
|
)? < 0
|
||||||
|
{
|
||||||
|
1
|
||||||
|
} else {
|
||||||
|
0
|
||||||
|
},
|
||||||
|
)),
|
||||||
|
Instruction::Call(fct_ins, args) => {
|
||||||
|
let vals = Instruction::run_all(args, ctx)?
|
||||||
|
.ok_or("invalid argument values in 'call' tag")?;
|
||||||
|
let fct_val = fct_ins
|
||||||
|
.run(ctx)?
|
||||||
|
.ok_or("invalid child function in 'call' tag")?;
|
||||||
|
if let Value::Function(f) = fct_val {
|
||||||
|
f.run(vals, ctx)?
|
||||||
|
} else if let Value::StdFunction(f) = fct_val {
|
||||||
|
f(vals)?
|
||||||
|
} else {
|
||||||
|
Err("invalid function")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Instruction::CallNamed(fct_name, args) => {
|
||||||
|
let vals: Vec<Value> = Instruction::run_all(args, ctx)?
|
||||||
|
.ok_or("invalid argument values in 'or' tag")?;
|
||||||
|
let fct_val = ctx.value(&fct_name).ok_or("unknown function")?;
|
||||||
|
if let Value::Function(f) = fct_val {
|
||||||
|
let mut local = ctx.clone();
|
||||||
|
f.run(vals, &mut local)?
|
||||||
|
} else if let Value::StdFunction(f) = fct_val {
|
||||||
|
f(vals)?
|
||||||
|
} else {
|
||||||
|
Err("invalid function")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Instruction::Return(ins) => {
|
||||||
|
let v = ins.run(ctx)?.ok_or("invalid child value in 'return' tag")?;
|
||||||
|
ctx.assign(String::from("__return"), v);
|
||||||
|
None
|
||||||
|
}
|
||||||
|
Instruction::If(cond, then) => {
|
||||||
|
if cond
|
||||||
|
.run(ctx)?
|
||||||
|
.ok_or("invalid condition value in 'if' tag")?
|
||||||
|
.to_bool()
|
||||||
|
{
|
||||||
|
for i in then {
|
||||||
|
i.run(ctx)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
Instruction::IfElse(cond, then, els) => {
|
||||||
|
if cond
|
||||||
|
.run(ctx)?
|
||||||
|
.ok_or("invalid condition value in 'if' tag")?
|
||||||
|
.to_bool()
|
||||||
|
{
|
||||||
|
for i in then {
|
||||||
|
i.run(ctx)?;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
for i in els {
|
||||||
|
i.run(ctx)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
Instruction::For {
|
||||||
|
variable,
|
||||||
|
from,
|
||||||
|
to,
|
||||||
|
step,
|
||||||
|
body,
|
||||||
|
} => {
|
||||||
|
if let Value::Integer(f) =
|
||||||
|
from.run(ctx)?.ok_or("invalid 'from' value in 'for' tag")?
|
||||||
|
{
|
||||||
|
if let Value::Integer(t) =
|
||||||
|
to.run(ctx)?.ok_or("invalid 'to' value in 'for' tag")?
|
||||||
|
{
|
||||||
|
if let Value::Integer(s) =
|
||||||
|
step.run(ctx)?.ok_or("invalid 'from' value in 'for' tag")?
|
||||||
|
{
|
||||||
|
for i in (f..t).step_by(s.try_into()?) {
|
||||||
|
ctx.assign(variable.clone(), Value::Integer(i));
|
||||||
|
for ins in body {
|
||||||
|
ins.run(ctx)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
Instruction::Each(variable, array_ins, body) => {
|
||||||
|
if let Value::Array(v) = array_ins
|
||||||
|
.run(ctx)?
|
||||||
|
.ok_or("invalid array value in 'each' tag")?
|
||||||
|
{
|
||||||
|
for i in v.borrow().iter() {
|
||||||
|
ctx.assign(variable.clone(), i.clone());
|
||||||
|
for ins in body {
|
||||||
|
ins.run(ctx)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("invalid array")?
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
Instruction::While(cond, body) => {
|
||||||
|
while cond
|
||||||
|
.run(ctx)?
|
||||||
|
.ok_or("invalid condition value in 'while' tag")?
|
||||||
|
.to_bool()
|
||||||
|
{
|
||||||
|
for ins in body {
|
||||||
|
ins.run(ctx)?;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
1024
src/lib.rs
1024
src/lib.rs
File diff suppressed because it is too large
Load Diff
201
src/stl.rs
Normal file
201
src/stl.rs
Normal file
@@ -0,0 +1,201 @@
|
|||||||
|
use super::{Context, Value};
|
||||||
|
use std::cell::RefCell;
|
||||||
|
use std::error::Error;
|
||||||
|
use std::io;
|
||||||
|
use std::rc::Rc;
|
||||||
|
|
||||||
|
pub fn inject_all(ctx: &mut Context) {
|
||||||
|
ctx.assign(String::from("print"), Value::StdFunction(print));
|
||||||
|
ctx.assign(String::from("input"), Value::StdFunction(input));
|
||||||
|
ctx.assign(
|
||||||
|
String::from("string-split"),
|
||||||
|
Value::StdFunction(string_split),
|
||||||
|
);
|
||||||
|
ctx.assign(String::from("array-set"), Value::StdFunction(array_set));
|
||||||
|
ctx.assign(String::from("array-push"), Value::StdFunction(array_push));
|
||||||
|
ctx.assign(String::from("array-pop"), Value::StdFunction(array_pop));
|
||||||
|
ctx.assign(String::from("array-get"), Value::StdFunction(array_get));
|
||||||
|
ctx.assign(
|
||||||
|
String::from("array-length"),
|
||||||
|
Value::StdFunction(array_length),
|
||||||
|
);
|
||||||
|
ctx.assign(String::from("to-ascii"), Value::StdFunction(to_ascii));
|
||||||
|
ctx.assign(String::from("from-ascii"), Value::StdFunction(from_ascii));
|
||||||
|
ctx.assign(String::from("get-args"), Value::StdFunction(get_args));
|
||||||
|
}
|
||||||
|
|
||||||
|
fn print(vals: Vec<Value>) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
if vals.len() == 1 {
|
||||||
|
match &vals[0] {
|
||||||
|
Value::Integer(i) => println!("{}", i),
|
||||||
|
Value::Float(f) => println!("{}", f),
|
||||||
|
Value::String(s) => println!("{}", s),
|
||||||
|
v => println!("{:?}", v), // _ => Err("unprintable value")?,
|
||||||
|
};
|
||||||
|
Ok(Some(vals[0].clone()))
|
||||||
|
} else {
|
||||||
|
Err("bad argument count in call to 'print'")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn input(vals: Vec<Value>) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
if vals.len() == 0 {
|
||||||
|
let mut line = String::new();
|
||||||
|
io::stdin().read_line(&mut line)?;
|
||||||
|
line.pop();
|
||||||
|
Ok(Some(Value::String(line)))
|
||||||
|
} else {
|
||||||
|
Err("bad argument count in call to 'input'")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn string_split(vals: Vec<Value>) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
if vals.len() == 2 {
|
||||||
|
if let Value::String(s) = &vals[0] {
|
||||||
|
if let Value::String(d) = &vals[1] {
|
||||||
|
let mut v = s
|
||||||
|
.split(d)
|
||||||
|
.map(|sub| Value::String(sub.to_string()))
|
||||||
|
.collect::<Vec<Value>>();
|
||||||
|
v.remove(0);
|
||||||
|
v.pop();
|
||||||
|
Ok(Some(Value::Array(Rc::new(RefCell::new(v)))))
|
||||||
|
} else {
|
||||||
|
Err("invalid delimiter string in call to 'string-split'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("invalid target string in call to 'string-split'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("bad argument count in call to 'string-split'")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn array_set(vals: Vec<Value>) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
if vals.len() == 3 {
|
||||||
|
if let Value::Array(v) = &vals[0] {
|
||||||
|
if let Value::Integer(i) = &vals[1] {
|
||||||
|
let index: usize = (*i).try_into()?;
|
||||||
|
if v.borrow().len() > index {
|
||||||
|
v.borrow_mut()[index] = vals[2].clone();
|
||||||
|
Ok(None)
|
||||||
|
} else {
|
||||||
|
Err("index out of array in call to 'array-set'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("invalid index in call to 'array-set'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("invalid array in call to 'array-set'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("bad argument count in call to 'array-set'")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn array_push(vals: Vec<Value>) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
if vals.len() == 2 {
|
||||||
|
if let Value::Array(v) = &vals[0] {
|
||||||
|
v.borrow_mut().push(vals[1].clone());
|
||||||
|
Ok(None)
|
||||||
|
} else {
|
||||||
|
Err("invalid array in call to 'array-push'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("bad argument count in call to 'array-push'")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn array_pop(vals: Vec<Value>) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
if vals.len() == 1 {
|
||||||
|
if let Value::Array(v) = &vals[0] {
|
||||||
|
Ok(Some(
|
||||||
|
v.borrow_mut()
|
||||||
|
.pop()
|
||||||
|
.ok_or("empty array in call to 'array-pop'")?,
|
||||||
|
))
|
||||||
|
} else {
|
||||||
|
Err("invalid array in call to 'array-pop'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("bad argument count in call to 'array-pop'")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn array_get(vals: Vec<Value>) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
if vals.len() == 2 {
|
||||||
|
if let Value::Array(v) = &vals[0] {
|
||||||
|
if let Value::Integer(i) = &vals[1] {
|
||||||
|
let index: usize = (*i).try_into()?;
|
||||||
|
if v.borrow().len() > index {
|
||||||
|
Ok(Some(v.borrow_mut()[index].clone()))
|
||||||
|
} else {
|
||||||
|
Err("index out of array in call to 'array-get'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("invalid index in call to 'array-get'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("invalid array in call to 'array-get'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("bad argument count in call to 'array-get'")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn array_length(vals: Vec<Value>) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
if vals.len() == 1 {
|
||||||
|
if let Value::Array(v) = &vals[0] {
|
||||||
|
Ok(Some(Value::Integer(v.borrow().len() as i64)))
|
||||||
|
} else {
|
||||||
|
Err("invalid array in call to 'array-length'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("bad argument count in call to 'array-length'")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn to_ascii(vals: Vec<Value>) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
if vals.len() == 1 {
|
||||||
|
if let Value::Integer(i) = &vals[0] {
|
||||||
|
if i <= &255 {
|
||||||
|
Ok(Some(Value::String(String::from_utf8(vec![*i as u8])?)))
|
||||||
|
} else {
|
||||||
|
Err("invalid integer in call to 'to-ascii'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("invalid argument in call to 'to-ascii'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("bad argument count in call to 'to-ascii'")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn from_ascii(vals: Vec<Value>) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
if vals.len() == 1 {
|
||||||
|
if let Value::String(s) = &vals[0] {
|
||||||
|
if s.len() == 1 {
|
||||||
|
Ok(Some(Value::Integer(s.as_bytes()[0] as i64)))
|
||||||
|
} else {
|
||||||
|
Err("invalid string in call to 'from-ascii'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("invalid argument in call to 'from-ascii'")?
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
Err("bad argument count in call to 'from-ascii'")?
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn get_args(vals: Vec<Value>) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
if vals.len() == 0 {
|
||||||
|
Ok(Some(Value::Array(Rc::new(RefCell::new(
|
||||||
|
std::env::args()
|
||||||
|
.skip(1)
|
||||||
|
.map(|arg| Value::String(arg))
|
||||||
|
.collect(),
|
||||||
|
)))))
|
||||||
|
} else {
|
||||||
|
Err("bad argument count in call to 'get-args'")?
|
||||||
|
}
|
||||||
|
}
|
||||||
9
src/util.rs
Normal file
9
src/util.rs
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
use roxmltree::Node;
|
||||||
|
|
||||||
|
pub fn tag_name(node: &Node) -> String {
|
||||||
|
node.tag_name().name().to_lowercase()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn find_node<'a>(node: &'a Node<'a, 'a>, tag: &str) -> Option<Node<'a, 'a>> {
|
||||||
|
node.children().find(|n| tag_name(n) == tag)
|
||||||
|
}
|
||||||
52
src/value.rs
Normal file
52
src/value.rs
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
use super::{Context, Instruction};
|
||||||
|
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,
|
||||||
|
) -> Result<Option<Value>, Box<dyn Error>> {
|
||||||
|
if args.len() != self.args.len() {
|
||||||
|
Err("not enough arguments")?;
|
||||||
|
}
|
||||||
|
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)?;
|
||||||
|
}
|
||||||
|
Ok(ctx.take(&String::from("__return")))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[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,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user