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
/*
 * smartcalc v1.0.7
 * Copyright (c) Erhan BARIS (Ruslan Ognyanov Asenov)
 * Licensed under the GNU General Public License v2.0.
 */

use crate::types::*;
use crate::syntax::{SyntaxParser, SyntaxParserTrait};
use crate::syntax::unary::UnaryParser;
use alloc::rc::Rc;

pub struct ModuloParser;
pub struct MultiplyDivideParser;
pub struct AddSubtractParser;

impl SyntaxParserTrait for ModuloParser {
    fn parse(parser: &mut SyntaxParser) -> AstResult {
        parse_binary::<MultiplyDivideParser>(parser, &['%'])
    }
}

impl SyntaxParserTrait for MultiplyDivideParser {
    fn parse(parser: &mut SyntaxParser) -> AstResult {
        parse_binary::<UnaryParser>(parser, &['*', '/'])
    }
}

impl SyntaxParserTrait for AddSubtractParser {
    fn parse(parser: &mut SyntaxParser) -> AstResult {
        parse_binary::<ModuloParser>(parser, &['+', '-'])
    }
}

pub fn parse_binary<T: SyntaxParserTrait>(parser: &mut SyntaxParser, operators: &[char]) -> AstResult {
    let mut left_expr = T::parse(parser)?;
    
    if let SmartCalcAstType::None = left_expr {
        return Ok(left_expr)
    }

    loop {
        let index_backup = parser.get_index();

        if let Some(operator) = parser.match_operator(operators) {
            loop {
                let right_expr = T::parse(parser);
                match right_expr {
                    Ok(SmartCalcAstType::None) => (),
                    Ok(_) => {
                        left_expr = SmartCalcAstType::Binary {
                            left: Rc::new(left_expr),
                            operator,
                            right: Rc::new(right_expr.unwrap())
                        };
                        break;
                    },
                    Err(_) => return right_expr
                };
            }
        }
        else {
            parser.set_index(index_backup);
            break;
        }
    }

    Ok(left_expr)
}