mango 0.4.0

The Mango programming language (UNDER CONSTRUCTION)
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153

use ::lazy_static::lazy_static;
use ::regex::Regex;

use crate::lexing::lexer::Lexer;
use crate::lexing::reader::reader::{Reader, ReaderResult};
use crate::token::{EndBlockToken, StartBlockToken, Tokens};

lazy_static! {
    static ref NO_CODE_LINE_RE: Regex = Regex::new(r"^(#|\n)").unwrap();
    static ref INDENT_RE: Regex = Regex::new(r"^(\t| {4})").unwrap();
}

//TODO @mark: should not be called, or should be undone, after continuation (...)
//TODO @mark: should not be called, or should be undone, for empty lines

/// Process the indents at the start of a line, and add the tokens to the Lexer.
pub fn lex_indents(reader: &mut impl Reader, lexer: &mut impl Lexer) {

    // Only try to lex indents if at the start of a line.
    if !lexer.is_at_indentable() {
        return;
    }

    // Skip if this is an empty or comment-only line.
    if let ReaderResult::Match(_) = reader.strip_peek(&*NO_CODE_LINE_RE) {
        return
    }

    // Determine the indent of the line.
    let mut line_indent = 0;
    while let ReaderResult::Match(_) = reader.direct_match(&*INDENT_RE) {
        line_indent += 1;
    }

    // Determine the tokens to create.
    let prev_indent = lexer.get_indent();
    for i in line_indent .. prev_indent {
        lexer.add(Tokens::EndBlock(EndBlockToken::new(true, false)));
    }
    for i in prev_indent .. line_indent {
        lexer.add(Tokens::StartBlock(StartBlockToken::new()));
    }

    lexer.set_at_indentable(false);
    lexer.set_indent(line_indent);
}

#[cfg(test)]
mod indents {
    use crate::io::source::SourceFile;
    use crate::lexing::reader::source_reader::SourceReader;
    use crate::lexing::lexer::{CodeLexer, Lexer};
    use crate::token::{StartBlockToken, Tokens, EndBlockToken};

    use super::lex_indents;
    use crate::lexing::tests::create_lexer;
    use crate::token::collect::token_list::TokenList;

    fn check(initial_indent: u32, input: &str, expected: &[Tokens]) {
        let expected: TokenList = expected.into();
        let (source, mut reader, mut lexer) = create_lexer(input);
        lexer.set_indent(initial_indent);
        lex_indents(&mut reader, &mut lexer);
        assert_eq!(lexer.tokens(), &expected);
    }

    #[test]
    fn increase() {
        check(0,
            "\t    hello",
            &vec![
                Tokens::StartBlock(StartBlockToken::new()),
                Tokens::StartBlock(StartBlockToken::new()),
        ]);
    }

    #[test]
    fn decrease_to_two() {
        check(3,
            "    \thello",
            &vec![
                Tokens::EndBlock(EndBlockToken::new(true, false)),
        ]);
    }

    #[test]
    fn decrease_to_zero() {
        check(2,
            "hello",
            &vec![
                Tokens::EndBlock(EndBlockToken::new(true, false)),
                Tokens::EndBlock(EndBlockToken::new(true, false)),
        ]);
    }

    #[test]
    fn constant_two() {
        check(2,
            "\t    hello",
            &vec![]
        );
    }

    #[test]
    fn constant_zero() {
        check(0,
            "hello",
              &vec![]
        );
    }

    #[test]
    fn direct_comment() {
        check(0,
            "#hello",
              &vec![]
        );
    }

    #[test]
    fn indented_comment() {
        check(0,
            "    \t#hello",
              &vec![]
        );
    }

    #[test]
    fn empty_line() {
        check(0,
            "\n",
              &vec![]
        );
    }

    #[test]
    fn whitespace_line() {
        check(0,
            "\t    \n",
              &vec![]
        );
    }

    #[test]
    fn after_mismatch() {
        check(0,
            "word\t    \n",
              &vec![]
        );
    }

    //TODO @mark: check that the correct characters are stripped, especially for comments
}