80 lines
2.6 KiB
Rust
80 lines
2.6 KiB
Rust
use serde::{Deserialize, Serialize};
|
|
use tree_sitter::Node;
|
|
|
|
use crate::{AbstractTree, Error, Expression, Result, Value, VariableMap};
|
|
|
|
#[derive(Debug, Clone, Serialize, Deserialize, Eq, PartialEq, PartialOrd, Ord)]
|
|
pub struct Logic {
|
|
left: Expression,
|
|
operator: LogicOperator,
|
|
right: Expression,
|
|
}
|
|
|
|
impl AbstractTree for Logic {
|
|
fn from_syntax_node(node: Node, source: &str) -> Result<Self> {
|
|
let left_node = node.child(0).unwrap();
|
|
let left = Expression::from_syntax_node(left_node, source)?;
|
|
|
|
let operator_node = node.child(1).unwrap().child(0).unwrap();
|
|
let operator = match operator_node.kind() {
|
|
"==" => LogicOperator::Equal,
|
|
"&&" => LogicOperator::And,
|
|
"||" => LogicOperator::Or,
|
|
">" => LogicOperator::Greater,
|
|
"<" => LogicOperator::Less,
|
|
">=" => LogicOperator::GreaterOrEqual,
|
|
"<=" => LogicOperator::LessOrEqaul,
|
|
_ => {
|
|
return Err(Error::UnexpectedSyntax {
|
|
expected: "==, && or ||",
|
|
actual: operator_node.kind(),
|
|
location: operator_node.start_position(),
|
|
relevant_source: source[operator_node.byte_range()].to_string(),
|
|
})
|
|
}
|
|
};
|
|
|
|
let right_node = node.child(2).unwrap();
|
|
let right = Expression::from_syntax_node(right_node, source)?;
|
|
|
|
Ok(Logic {
|
|
left,
|
|
operator,
|
|
right,
|
|
})
|
|
}
|
|
|
|
fn run(&self, context: &mut VariableMap) -> Result<Value> {
|
|
let left = self.left.run(context)?;
|
|
let right = self.right.run(context)?;
|
|
let result = match self.operator {
|
|
LogicOperator::Equal => {
|
|
if let (Ok(left_num), Ok(right_num)) = (left.as_number(), right.as_number()) {
|
|
left_num == right_num
|
|
} else {
|
|
left == right
|
|
}
|
|
}
|
|
LogicOperator::And => left.as_boolean()? && right.as_boolean()?,
|
|
LogicOperator::Or => left.as_boolean()? || right.as_boolean()?,
|
|
LogicOperator::Greater => left > right,
|
|
LogicOperator::Less => left < right,
|
|
LogicOperator::GreaterOrEqual => left >= right,
|
|
LogicOperator::LessOrEqaul => left <= right,
|
|
};
|
|
|
|
Ok(Value::Boolean(result))
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, Clone, Serialize, Deserialize, Eq, PartialEq, PartialOrd, Ord)]
|
|
pub enum LogicOperator {
|
|
Equal,
|
|
And,
|
|
Or,
|
|
Greater,
|
|
Less,
|
|
GreaterOrEqual,
|
|
LessOrEqaul,
|
|
}
|