luaur-compiler 0.1.3

Luau source-to-bytecode compiler (Rust).
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
38
39
40

use crate::enums::global::Global;
use crate::records::value_visitor::ValueVisitor;
use luaur_ast::records::ast_expr::AstExpr;
use luaur_ast::records::ast_expr_global::AstExprGlobal;
use luaur_ast::records::ast_expr_local::AstExprLocal;
use luaur_ast::records::ast_node::AstNode;
use luaur_ast::rtti::ast_node_as;

impl ValueVisitor {
    pub fn assign(&mut self, var: *mut AstExpr) {
        if var.is_null() {
            return;
        }

        let var_ptr = var as *mut AstNode;

        let local = unsafe { ast_node_as::<AstExprLocal>(var_ptr) };
        if !local.is_null() {
            let local_ptr = unsafe { (*local).local };
            // C++ `variables[lv->local].written = true`: operator[] creates the entry
            // if it doesn't exist, so use get_or_insert (find_mut would miss new keys,
            // e.g. a numeric for-loop variable assigned inside its body).
            self.variables.get_or_insert(local_ptr).written = true;
            return;
        }

        let global = unsafe { ast_node_as::<AstExprGlobal>(var_ptr) };
        if !global.is_null() {
            let name = unsafe { (*global).name };
            // C++ `globals[gv->name] = Global::Written`: operator[] overwrites; try_insert
            // would leave an existing Default entry unchanged.
            *self.globals.get_or_insert(name) = Global::Written;
            return;
        }

        // C++ `var->visit(this)`: track assignments inside complex lvalues such as
        // `t[function() t = nil end] = 5` by dispatching through the node walker.
        unsafe { luaur_ast::visit::ast_expr_visit(var, self) };
    }
}