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

use crate::functions::get_mutable_type_function_runtime_alt_g::get_mutable_type_function_type_id;
use crate::functions::get_tag::get_tag;
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_property::TypeFunctionProperty;
use crate::records::type_function_singleton_type::TypeFunctionSingletonType;
use crate::records::type_function_table_type::TypeFunctionTableType;
use crate::type_aliases::lua_state::lua_State;
use luaur_vm::functions::lua_gettop::lua_gettop;
use luaur_vm::functions::lua_l_error_l::lua_l_error_l;
use luaur_vm::macros::lua_isnil::lua_isnil;

pub unsafe fn set_write_table_prop(l: *mut lua_State) -> i32 {
    let vm_l = l as *mut luaur_vm::records::lua_state::lua_State;
    let argument_count = lua_gettop(vm_l);
    if argument_count < 2 || argument_count > 3 {
        lua_l_error_l(
            vm_l,
            c"%s".as_ptr(),
            core::format_args!(
                "type.setwriteproperty: expected 2-3 arguments, but got {}",
                argument_count
            ),
        );
    }

    let self_ty = get_type_user_data(l, 1);
    let tftt = get_mutable_type_function_type_id::<TypeFunctionTableType>(self_ty);
    if tftt.is_null() {
        lua_l_error_l(
            vm_l,
            c"%s".as_ptr(),
            core::format_args!(
                "type.setwriteproperty: expected self to be a table, but got {} instead",
                get_tag(l, self_ty)
            ),
        );
    }

    if luaur_common::FFlag::LuauTypeFunctionSupportsFrozen.get() && (*self_ty).frozen {
        lua_l_error_l(
            vm_l,
            c"%s".as_ptr(),
            core::format_args!(
                "type.setwriteproperty: cannot be called to mutate a frozen type, use `types.copy` to make a copy"
            ),
        );
    }

    let key = get_type_user_data(l, 2);
    let tfst = get_type_function_type_id::<TypeFunctionSingletonType>(key);
    if tfst.is_null() {
        lua_l_error_l(
            vm_l,
            c"%s".as_ptr(),
            core::format_args!(
                "type.setwriteproperty: expected to be given a singleton type, but got {} instead",
                get_tag(l, key)
            ),
        );
    }

    let tfsst = (*tfst).variant.get_if_1();
    if tfsst.is_none() {
        lua_l_error_l(
            vm_l,
            c"%s".as_ptr(),
            core::format_args!(
                "type.setwriteproperty: expected to be given a string singleton type, but got {} instead",
                get_tag(l, key)
            ),
        );
    }

    let key_name = tfsst.unwrap().value.clone();

    if argument_count == 2 || lua_isnil!(vm_l, 3) {
        if let Some(existing) = (*tftt).props.get(&key_name) {
            if existing.is_write_only() {
                (*tftt).props.remove(&key_name);
            } else if let Some(prop) = (*tftt).props.get_mut(&key_name) {
                prop.write_ty = None;
            }
        }

        return 0;
    }

    let value = get_type_user_data(l, 3);
    if let Some(prop) = (*tftt).props.get_mut(&key_name) {
        prop.write_ty = Some(value);
    } else {
        (*tftt)
            .props
            .insert(key_name, TypeFunctionProperty::writeonly(value));
    }

    0
}