luaur-ast 0.1.0

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
36
37

use crate::records::ast_local::AstLocal;
use crate::records::binding::Binding;
use crate::records::parser::Parser;

impl Parser {
    #[allow(non_snake_case)]
    pub fn push_local(&mut self, binding: &Binding) -> *mut AstLocal {
        let name = binding.name;

        // C++: `AstLocal*& local = local_map[name.name];` — operator[] inserts a
        // null slot when absent, and the prior value becomes the new local's
        // shadow before the slot is reassigned to the freshly allocated local.
        let shadow = *self.local_map.get_or_insert(name.name);

        let function_depth = self.function_stack.len() - 1;
        let loop_depth = self.function_stack.last().unwrap().loop_depth as usize;

        let new_local = AstLocal {
            name: name.name,
            location: name.location,
            shadow,
            function_depth,
            loop_depth,
            is_const: binding.is_const,
            is_exported: false,
            annotation: binding.annotation,
        };

        // Safety: self.allocator is a raw pointer to the arena Allocator.
        let local = unsafe { (*self.allocator).alloc(new_local) };

        *self.local_map.get_or_insert(name.name) = local;
        self.local_stack.push(local);

        local
    }
}