quickfall/compiler/ast_parser/src/control/for_loop.rs at 1c2f11daa104fc2417b51ce453c5b17358af5325 · Quickfall/quickfall · GitHub

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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
use diagnostics::DiagnosticResult;
use lexer::token::{LexerToken, LexerTokenType};

use ast::tree::{ASTTreeNode, ASTTreeNodeKind};

use crate::{
    functions::parse_node_body, parser::parse_ast_node_in_body, ranges::parse_value_range,
    value::parse_ast_value, variables::decl::parse_variable_declaration,
};

pub fn parse_for_loop(
    tokens: &Vec<LexerToken>,
    ind: &mut usize,
) -> DiagnosticResult<Box<ASTTreeNode>> {
    let start = tokens[*ind].pos.clone();

    if tokens[*ind + 1].tok_type != LexerTokenType::ParenOpen {
        return parse_for_ranged_loop(tokens, ind);
    }

    *ind += 1;

    tokens[*ind].expects(LexerTokenType::ParenOpen)?;
    *ind += 1;

    tokens[*ind].expects(LexerTokenType::Var)?;

    let initial = parse_variable_declaration(tokens, ind, true)?;

    tokens[*ind].expects(LexerTokenType::Comma)?;

    *ind += 1;
    let cond = parse_ast_value(tokens, ind)?;

    tokens[*ind].expects(LexerTokenType::Comma)?;
    *ind += 1;

    let increment = parse_ast_node_in_body(tokens, ind)?;

    tokens[*ind].expects(LexerTokenType::ParenClose)?;
    *ind += 1;

    tokens[*ind].expects(LexerTokenType::BracketOpen)?;

    let body = parse_node_body(tokens, ind)?;

    let end = tokens[*ind - 1].get_end_pos();

    return Ok(Box::new(ASTTreeNode::new(
        ASTTreeNodeKind::ForBlock {
            initial_state: initial,
            cond,
            increment,
            body,
        },
        start,
        end,
    )));
}

pub fn parse_for_ranged_loop(
    tokens: &Vec<LexerToken>,
    ind: &mut usize,
) -> DiagnosticResult<Box<ASTTreeNode>> {
    let start = tokens[*ind].pos.clone();

    let var = parse_variable_declaration(tokens, ind, false)?;

    tokens[*ind].expects(LexerTokenType::EqualSign)?;
    *ind += 1;

    tokens[*ind].expects(LexerTokenType::AngelBracketClose)?;
    *ind += 1;

    let range = parse_value_range(tokens, ind)?;

    tokens[*ind].expects(LexerTokenType::BracketOpen)?;

    let body = parse_node_body(tokens, ind)?;

    let end: compiler_utils::Position = tokens[*ind - 1].get_end_pos();

    Ok(Box::new(ASTTreeNode::new(
        ASTTreeNodeKind::RangedForBlock { var, range, body },
        start,
        end,
    )))
}