84 lines
2.8 KiB
Rust
84 lines
2.8 KiB
Rust
use serde::{Deserialize, Serialize};
|
|
use tree_sitter::Node;
|
|
|
|
use crate::{AbstractTree, Error, Expression, Format, LogicOperator, Map, Result, Type, Value};
|
|
|
|
/// Abstract representation of a logic expression.
|
|
#[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(source: &str, node: Node, context: &Map) -> Result<Self> {
|
|
Error::expect_syntax_node(source, "logic", node)?;
|
|
|
|
let first_node = node.child(0).unwrap();
|
|
let (left_node, operator_node, right_node) = {
|
|
if first_node.is_named() {
|
|
(first_node, node.child(1).unwrap(), node.child(2).unwrap())
|
|
} else {
|
|
(
|
|
node.child(1).unwrap(),
|
|
node.child(2).unwrap(),
|
|
node.child(3).unwrap(),
|
|
)
|
|
}
|
|
};
|
|
let left = Expression::from_syntax_node(source, left_node, context)?;
|
|
let operator = LogicOperator::from_syntax_node(source, operator_node, context)?;
|
|
let right = Expression::from_syntax_node(source, right_node, context)?;
|
|
|
|
Ok(Logic {
|
|
left,
|
|
operator,
|
|
right,
|
|
})
|
|
}
|
|
|
|
fn run(&self, source: &str, context: &Map) -> Result<Value> {
|
|
let left = self.left.run(source, context)?;
|
|
let right = self.right.run(source, 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::NotEqual => {
|
|
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::LessOrEqual => left <= right,
|
|
};
|
|
|
|
Ok(Value::Boolean(result))
|
|
}
|
|
|
|
fn expected_type(&self, _context: &Map) -> Result<Type> {
|
|
Ok(Type::Boolean)
|
|
}
|
|
}
|
|
|
|
impl Format for Logic {
|
|
fn format(&self, output: &mut String, indent_level: u8) {
|
|
self.left.format(output, indent_level);
|
|
output.push(' ');
|
|
self.operator.format(output, indent_level);
|
|
output.push(' ');
|
|
self.right.format(output, indent_level);
|
|
}
|
|
}
|