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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

use crate::enums::polarity::Polarity;
use crate::records::blocked_type_pack::BlockedTypePack;
use crate::records::constraint_generator::ConstraintGenerator;
use crate::records::generic_type_pack::GenericTypePack;
use crate::records::generic_type_pack_definition::GenericTypePackDefinition;
use crate::records::scope::Scope;
use crate::type_aliases::name_type::Name;
use crate::type_aliases::scope_ptr_constraint_generator::ScopePtr;
use crate::type_aliases::type_pack_id::TypePackId;
use alloc::vec::Vec;
use luaur_ast::records::ast_array::AstArray;
use luaur_ast::records::ast_generic_type_pack::AstGenericTypePack;

impl ConstraintGenerator {
    pub fn create_generic_packs(
        &mut self,
        scope: &ScopePtr,
        generics: AstArray<*mut AstGenericTypePack>,
        use_cache: bool,
        add_types: bool,
    ) -> Vec<(Name, GenericTypePackDefinition)> {
        let mut result: Vec<(Name, GenericTypePackDefinition)> = Vec::new();
        let scope_ptr = scope.as_ref() as *const Scope as *mut Scope;

        for i in 0..generics.size {
            let generic = unsafe { *generics.data.add(i) };

            let generic_name_ptr = unsafe { (*generic).name.value };
            let generic_name_str = unsafe {
                core::ffi::CStr::from_ptr(generic_name_ptr)
                    .to_string_lossy()
                    .into_owned()
            };

            let mut generic_ty: Option<TypePackId> = None;

            if use_cache {
                if let Some(parent_scope) = scope.parent.as_ref() {
                    if let Some(type_pack_id) = parent_scope
                        .type_alias_type_pack_parameters
                        .get(&generic_name_str)
                    {
                        generic_ty = Some(*type_pack_id);
                    }
                }
            }

            if generic_ty.is_none() {
                let mut gtp = GenericTypePack {
                    index: 0,
                    level: Default::default(),
                    scope: core::ptr::null_mut(),
                    name: Default::default(),
                    explicitName: false,
                    polarity: Polarity::None,
                };
                gtp.generic_type_pack_scope_name_polarity(
                    scope_ptr,
                    generic_name_str.clone(),
                    Polarity::None,
                );
                generic_ty = Some(unsafe { (*self.arena).add_type_pack_t(gtp) });

                if let Some(parent_scope) = scope.parent.as_ref() {
                    let parent_scope_ptr = parent_scope.as_ref() as *const Scope as *mut Scope;
                    unsafe {
                        (*parent_scope_ptr)
                            .type_alias_type_pack_parameters
                            .insert(generic_name_str.clone(), generic_ty.unwrap());
                    }
                }
            }

            let default_ty: Option<TypePackId> = if unsafe { (*generic).default_value }.is_null() {
                None
            } else {
                let mut btp = BlockedTypePack {
                    index: 0,
                    owner: core::ptr::null_mut(),
                };
                btp.blocked_type_pack_blocked_type_pack();
                Some(unsafe { (*self.arena).add_type_pack_t(btp) })
            };

            if add_types {
                unsafe {
                    (*scope_ptr)
                        .private_type_pack_bindings
                        .insert(generic_name_str.clone(), generic_ty.unwrap());
                }
            }

            result.push((
                generic_name_str,
                GenericTypePackDefinition {
                    tp: generic_ty.unwrap(),
                    defaultValue: default_ty,
                },
            ));
        }

        result
    }
}