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
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
71

extern crate alloc;

use crate::functions::is_identifier::is_identifier;
use alloc::vec::Vec;
use luaur_ast::records::ast_node::AstNode;
use luaur_ast::records::ast_stat_block::AstStatBlock;
use luaur_ast::records::ast_stat_error::AstStatError;
use luaur_ast::rtti::AstNodeClass;

pub fn extract_stat<T: AstNodeClass>(ancestry: &Vec<*mut AstNode>) -> *mut T {
    let node = if ancestry.len() >= 1 {
        ancestry[ancestry.len() - 1]
    } else {
        core::ptr::null_mut()
    };

    if node.is_null() {
        return core::ptr::null_mut();
    }

    let t = unsafe { (*node).as_item_mut::<T>() };
    if !t.is_null() {
        return t;
    }

    let parent = if ancestry.len() >= 2 {
        ancestry[ancestry.len() - 2]
    } else {
        core::ptr::null_mut()
    };

    if parent.is_null() {
        return core::ptr::null_mut();
    }

    let grand_parent = if ancestry.len() >= 3 {
        ancestry[ancestry.len() - 3]
    } else {
        core::ptr::null_mut()
    };

    let great_grand_parent = if ancestry.len() >= 4 {
        ancestry[ancestry.len() - 4]
    } else {
        core::ptr::null_mut()
    };

    if grand_parent.is_null() {
        return core::ptr::null_mut();
    }

    let t_parent = unsafe { (*parent).as_item_mut::<T>() };
    if !t_parent.is_null() && unsafe { (*grand_parent).is::<AstStatBlock>() } {
        return t_parent;
    }

    if great_grand_parent.is_null() {
        return core::ptr::null_mut();
    }

    let t_great = unsafe { (*great_grand_parent).as_item_mut::<T>() };
    if !t_great.is_null()
        && unsafe { (*grand_parent).is::<AstStatBlock>() }
        && unsafe { (*parent).is::<AstStatError>() }
        && is_identifier(node)
    {
        return t_great;
    }

    core::ptr::null_mut()
}