luaur_analysis/functions/
set_function_generics.rs

1use crate::functions::get_generics::get_generics;
2use crate::functions::get_mutable_type_function_runtime_alt_g::get_mutable_type_function_type_id;
3use crate::functions::get_tag::get_tag;
4use crate::functions::get_type_user_data::get_type_user_data;
5use crate::records::type_function_function_type::TypeFunctionFunctionType;
6use crate::type_aliases::lua_state::lua_State;
7use core::ffi::c_int;
8use luaur_vm::functions::lua_gettop::lua_gettop;
9use luaur_vm::functions::lua_l_error_l::lua_l_error_l;
10
11pub unsafe fn set_function_generics(l: *mut lua_State) -> c_int {
12    let vm_l = l as *mut luaur_vm::records::lua_state::lua_State;
13    let self_ty = get_type_user_data(l, 1);
14    let tfft = get_mutable_type_function_type_id::<TypeFunctionFunctionType>(self_ty);
15
16    if tfft.is_null() {
17        lua_l_error_l(
18            vm_l,
19            c"%s".as_ptr(),
20            core::format_args!(
21                "type.setgenerics: expected self to be a function, but got {} instead",
22                get_tag(l, self_ty)
23            ),
24        );
25    }
26
27    if luaur_common::FFlag::LuauTypeFunctionSupportsFrozen.get() && (*self_ty).frozen {
28        lua_l_error_l(
29            vm_l,
30            c"%s".as_ptr(),
31            core::format_args!(
32                "type.setgenerics: cannot be called to mutate a frozen type, use `types.copy` to make a copy"
33            ),
34        );
35    }
36
37    let argument_count = lua_gettop(vm_l);
38
39    if luaur_common::FFlag::LuauTypeFunctionRobustness.get() {
40        if argument_count > 2 {
41            lua_l_error_l(
42                vm_l,
43                c"%s".as_ptr(),
44                core::format_args!(
45                    "type.setgenerics: expected 2 arguments, but got {}",
46                    argument_count
47                ),
48            );
49        }
50    } else if argument_count > 3 {
51        lua_l_error_l(
52            vm_l,
53            c"%s".as_ptr(),
54            core::format_args!(
55                "type.setgenerics: expected 3 arguments, but got {}",
56                argument_count
57            ),
58        );
59    }
60
61    let (generic_types, generic_packs) = get_generics(l, 2, "types.setgenerics");
62
63    (*tfft).generics = generic_types;
64    (*tfft).generic_packs = generic_packs;
65
66    0
67}
luaur_analysis/functions/set_function_generics.rs

luaur_analysis/functions/
set_function_generics.rs
