luaur-analysis 0.1.3

Luau type checker and type inference (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

use crate::type_aliases::type_id::TypeId;
use core::ffi::c_void;
use luaur_ast::records::ast_expr::AstExpr;
use luaur_ast::records::ast_expr_table::AstExprTable;
use luaur_ast::records::ast_visitor::AstVisitor;
use luaur_common::records::dense_hash_map::DenseHashMap;
use luaur_common::records::dense_hash_set::DenseHashSet;

#[derive(Debug, Clone)]
pub struct AstExprTableFinder {
    pub result: *mut DenseHashSet<TypeId>,
    pub ast_types: *const DenseHashMap<*const AstExpr, TypeId>,
}

impl AstExprTableFinder {
    pub fn new(
        result: *mut DenseHashSet<TypeId>,
        ast_types: *const DenseHashMap<*const AstExpr, TypeId>,
    ) -> Self {
        Self { result, ast_types }
    }
}

impl AstVisitor for AstExprTableFinder {
    fn visit_expr(&mut self, _node: *mut c_void) -> bool {
        false
    }

    fn visit_expr_table(&mut self, node: *mut c_void) -> bool {
        let tbl = node as *mut AstExprTable;
        unsafe {
            if let Some(ty) = (*self.ast_types).find(&(tbl as *const AstExpr)) {
                (*self.result).insert(*ty);
            }
        }
        true
    }
}