dust-lang 0.4.2

General purpose programming language
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

use serde::{Deserialize, Serialize};

use crate::{
    error::{RuntimeError, SyntaxError, ValidationError},
    AbstractTree, Context, Expression, Format, MathOperator, SourcePosition, SyntaxNode, Type,
    Value,
};

/// Abstract representation of a math operation.
///
/// Dust currently supports the four basic operations and the modulo (or
/// remainder) operator.
#[derive(Debug, Clone, Serialize, Deserialize, Eq, PartialEq, PartialOrd, Ord)]
pub struct Math {
    left: Expression,
    operator: MathOperator,
    right: Expression,
    position: SourcePosition,
}

impl AbstractTree for Math {
    fn from_syntax(node: SyntaxNode, source: &str, context: &Context) -> Result<Self, SyntaxError> {
        SyntaxError::expect_syntax_node("math", node)?;

        let left_node = node.child(0).unwrap();
        let left = Expression::from_syntax(left_node, source, context)?;

        let operator_node = node.child(1).unwrap();
        let operator = MathOperator::from_syntax(operator_node, source, context)?;

        let right_node = node.child(2).unwrap();
        let right = Expression::from_syntax(right_node, source, context)?;

        Ok(Math {
            left,
            operator,
            right,
            position: node.range().into(),
        })
    }

    fn expected_type(&self, context: &Context) -> Result<Type, ValidationError> {
        self.left.expected_type(context)
    }

    fn validate(&self, _source: &str, _context: &Context) -> Result<(), ValidationError> {
        self.left.validate(_source, _context)?;
        self.right.validate(_source, _context)
    }

    fn run(&self, source: &str, context: &Context) -> Result<Value, RuntimeError> {
        let left = self.left.run(source, context)?;
        let right = self.right.run(source, context)?;
        let value = match self.operator {
            MathOperator::Add => left.add(right, self.position)?,
            MathOperator::Subtract => left.subtract(right, self.position)?,
            MathOperator::Multiply => left.multiply(right, self.position)?,
            MathOperator::Divide => left.divide(right, self.position)?,
            MathOperator::Modulo => left.modulo(right, self.position)?,
        };

        Ok(value)
    }
}

impl Format for Math {
    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);
    }
}