mago_parser/
lib.rs

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
use mago_ast::sequence::Sequence;
use mago_ast::Program;
use mago_interner::ThreadedInterner;
use mago_lexer::input::Input;
use mago_lexer::Lexer;
use mago_source::Source;

use crate::error::ParseError;
use crate::internal::statement::parse_statement;
use crate::internal::token_stream::TokenStream;

pub mod error;

mod internal;

pub fn parse_source(interner: &ThreadedInterner, source: &Source) -> (Program, Option<ParseError>) {
    let content = interner.lookup(&source.content);
    let lexer = Lexer::new(interner, Input::new(source.identifier, content.as_bytes()));

    construct(interner, lexer)
}

pub fn parse(interner: &ThreadedInterner, input: Input<'_>) -> (Program, Option<ParseError>) {
    let lexer = Lexer::new(interner, input);

    construct(interner, lexer)
}

fn construct<'i>(interner: &'i ThreadedInterner, lexer: Lexer<'_, 'i>) -> (Program, Option<ParseError>) {
    let mut stream = TokenStream::new(interner, lexer);

    let mut error = None;
    let statements = {
        let mut statements = Vec::new();

        loop {
            match stream.has_reached_eof() {
                Ok(false) => match parse_statement(&mut stream) {
                    Ok(statement) => {
                        statements.push(statement);
                    }
                    Err(parse_error) => {
                        error = Some(parse_error);

                        break;
                    }
                },
                Ok(true) => {
                    break;
                }
                Err(syntax_error) => {
                    error = Some(ParseError::from(syntax_error));

                    break;
                }
            }
        }

        statements
    };

    (
        Program {
            source: stream.get_position().source,
            statements: Sequence::new(statements),
            trivia: stream.get_trivia(),
        },
        error,
    )
}