luaur-analysis 0.1.0

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 crate::functions::alloc_type_user_data::alloc_type_user_data;
use crate::functions::get_tag::get_tag;
use crate::functions::get_type_function_runtime_alt_n::get_type_function_type_pack_id;
use crate::functions::get_type_function_runtime_alt_o::get_type_function_type_id;
use crate::functions::get_type_user_data::get_type_user_data;
use crate::records::type_function_function_type::TypeFunctionFunctionType;
use crate::records::type_function_generic_type::TypeFunctionGenericType;
use crate::records::type_function_generic_type_pack::TypeFunctionGenericTypePack;
use crate::type_aliases::lua_state::lua_State;
use crate::type_aliases::type_function_type_variant::TypeFunctionTypeVariant;
use luaur_common::macros::luau_assert::LUAU_ASSERT;
use luaur_vm::functions::lua_createtable::lua_createtable;
use luaur_vm::functions::lua_l_error_l::lua_l_error_l;
use luaur_vm::functions::lua_rawseti::lua_rawseti;

pub unsafe fn get_function_generics(l: *mut lua_State) -> core::ffi::c_int {
    let vm_l = l as *mut luaur_vm::records::lua_state::lua_State;
    let self_ty = get_type_user_data(l, 1);

    let tfft = get_type_function_type_id::<TypeFunctionFunctionType>(self_ty);
    if tfft.is_null() {
        lua_l_error_l(
            vm_l,
            c"%s".as_ptr(),
            core::format_args!(
                "type.generics: expected self to be a function, but got {} instead",
                get_tag(l, self_ty)
            ),
        );
    }

    lua_createtable(
        vm_l,
        ((*tfft).generics.len() + (*tfft).generic_packs.len()) as i32,
        0,
    );

    let mut pos: i32 = 1;

    for el in &(*tfft).generics {
        alloc_type_user_data(l, (*(*el)).type_variant.clone(), false);
        lua_rawseti(vm_l, -2, pos);
        pos += 1;
    }

    for el in &(*tfft).generic_packs {
        let gty = get_type_function_type_pack_id::<TypeFunctionGenericTypePack>(*el);
        LUAU_ASSERT!(!gty.is_null());
        alloc_type_user_data(
            l,
            TypeFunctionTypeVariant::Generic(TypeFunctionGenericType {
                is_named: (*gty).is_named,
                is_pack: true,
                name: (*gty).name.clone(),
            }),
            false,
        );
        lua_rawseti(vm_l, -2, pos);
        pos += 1;
    }

    1
}