luaur-ast 0.1.2

Lexer, parser, and AST for Luau (faithful Rust port).
Documentation 
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

//! `std::pair<AstName, Lexeme::Type> Lexer::read_name()` — Ast/src/Lexer.cpp:705.

use crate::functions::is_alpha::is_alpha;
use crate::functions::is_digit_lexer::is_digit;
use crate::records::ast_name::AstName;
use crate::records::ast_name_table::AstNameTable;
use crate::records::lexeme::Type;
use crate::records::lexer::Lexer;
use luaur_common::LUAU_ASSERT;

impl Lexer {
    pub(crate) fn read_name(&mut self) -> (AstName, Type) {
        LUAU_ASSERT!(is_alpha(self.peekch()) || self.peekch() == '_' || self.peekch() == '@');

        let start_offset = self.offset;

        loop {
            self.consume();
            if !(is_alpha(self.peekch()) || is_digit(self.peekch()) || self.peekch() == '_') {
                break;
            }
        }

        let read_names = self.read_names;
        let data = unsafe { self.buffer.add(start_offset as usize) };
        let length = (self.offset - start_offset) as usize;
        let names: &mut AstNameTable = unsafe { &mut *self.names };

        if read_names {
            names.get_or_add_with_type(data, length)
        } else {
            names.get_with_type(data, length)
        }
    }
}