luaur-analysis 0.1.1

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

use alloc::vec::Vec;
use luaur_ast::records::ast_node::AstNode;
use luaur_ast::records::ast_stat_function::AstStatFunction;
use luaur_ast::records::ast_visitor::AstVisitor;
use luaur_ast::records::position::Position;
use luaur_ast::visit::ast_expr_visit;

#[derive(Debug, Clone)]
pub struct FindFullAncestry {
    pub(crate) nodes: Vec<*mut AstNode>,
    pub(crate) pos: Position,
    pub(crate) document_end: Position,
    pub(crate) include_types: bool,
}

impl AstVisitor for FindFullAncestry {
    fn visit_node(&mut self, node: *mut core::ffi::c_void) -> bool {
        let node = node as *mut AstNode;
        let node_ref = unsafe { &*node };

        if node_ref.location.contains(self.pos) {
            self.nodes.push(node);
            return true;
        }

        if node_ref.location.end == self.document_end && self.pos >= self.document_end {
            self.nodes.push(node);
            return true;
        }

        false
    }

    fn visit_type(&mut self, node: *mut core::ffi::c_void) -> bool {
        if self.include_types {
            self.visit_node(node)
        } else {
            false
        }
    }

    fn visit_stat_function(&mut self, node: *mut core::ffi::c_void) -> bool {
        let node = node as *mut AstStatFunction;
        let node_ref = unsafe { &*node };

        self.visit_node(node as *mut AstNode as *mut core::ffi::c_void);

        if unsafe { (*node_ref.name).base.location.contains(self.pos) } {
            unsafe {
                ast_expr_visit(node_ref.name, self);
            }
        } else if unsafe { (*node_ref.func).base.base.location.contains(self.pos) } {
            unsafe {
                ast_expr_visit(
                    node_ref.func as *mut luaur_ast::records::ast_expr::AstExpr,
                    self,
                );
            }
        }

        false
    }
}