luaur_analysis/functions/
create_optional.rs

1use crate::enums::type_type_function_runtime::Type;
2use crate::functions::alloc_type_user_data::alloc_type_user_data;
3use crate::functions::allocate_type_function_type::allocate_type_function_type;
4use crate::functions::get_type_function_runtime_alt_o::get_type_function_type_id;
5use crate::functions::get_type_user_data::get_type_user_data;
6use crate::records::type_function_primitive_type::TypeFunctionPrimitiveType;
7use crate::records::type_function_union_type::TypeFunctionUnionType;
8use crate::type_aliases::lua_state::lua_State;
9use crate::type_aliases::type_function_type_id::TypeFunctionTypeId;
10use crate::type_aliases::type_function_type_variant::TypeFunctionTypeVariant;
11use luaur_vm::functions::lua_gettop::lua_gettop;
12use luaur_vm::functions::lua_l_error_l::lua_l_error_l;
13
14pub unsafe fn create_optional(l: *mut lua_State) -> core::ffi::c_int {
15    let vm_l = l as *mut luaur_vm::records::lua_state::lua_State;
16    let argument_count = lua_gettop(vm_l);
17    if argument_count != 1 {
18        lua_l_error_l(
19            vm_l,
20            c"%s".as_ptr(),
21            core::format_args!(
22                "types.optional: expected 1 argument, but got {}",
23                argument_count
24            ),
25        );
26    }
27
28    let argument: TypeFunctionTypeId = get_type_user_data(l, 1);
29
30    let mut components: alloc::vec::Vec<TypeFunctionTypeId> = alloc::vec::Vec::new();
31
32    let union_ty = get_type_function_type_id::<TypeFunctionUnionType>(argument);
33    if !union_ty.is_null() {
34        components.reserve((*union_ty).components.len() + 1);
35        components.extend((*union_ty).components.iter().copied());
36    } else {
37        components.push(argument);
38    }
39
40    let nil_type = TypeFunctionPrimitiveType::new(Type::NilType);
41    let nil_variant = TypeFunctionTypeVariant::Primitive(nil_type);
42    let nil_id = allocate_type_function_type(l, nil_variant);
43    components.push(nil_id);
44
45    let union_type = TypeFunctionUnionType { components };
46    let union_variant = TypeFunctionTypeVariant::Union(union_type);
47    alloc_type_user_data(l, union_variant, false);
48
49    1
50}
luaur_analysis/functions/create_optional.rs

luaur_analysis/functions/
create_optional.rs
