Skip to main content

luaur_analysis/functions/
create_singleton.rs

1use crate::enums::type_type_function_runtime::Type;
2use crate::functions::alloc_type_user_data::alloc_type_user_data;
3use crate::records::type_function_boolean_singleton::TypeFunctionBooleanSingleton;
4use crate::records::type_function_primitive_type::TypeFunctionPrimitiveType;
5use crate::records::type_function_singleton_type::TypeFunctionSingletonType;
6use crate::records::type_function_string_singleton::TypeFunctionStringSingleton;
7use crate::type_aliases::lua_state::lua_State;
8use crate::type_aliases::type_function_singleton_variant::TypeFunctionSingletonVariant;
9use crate::type_aliases::type_function_type_variant::TypeFunctionTypeVariant;
10use core::ffi::c_int;
11use core::ffi::CStr;
12use luaur_vm::enums::lua_type::lua_Type;
13use luaur_vm::functions::lua_l_checkboolean::lua_l_checkboolean;
14use luaur_vm::functions::lua_l_error_l::lua_l_error_l;
15use luaur_vm::functions::lua_type::lua_type;
16use luaur_vm::functions::lua_typename::lua_typename;
17use luaur_vm::macros::lua_isboolean::lua_isboolean;
18use luaur_vm::macros::lua_isnil::lua_isnil;
19use luaur_vm::macros::lua_l_checkstring::luaL_checkstring;
20
21pub unsafe fn create_singleton(l: *mut lua_State) -> c_int {
22    let vm_l = l as *mut luaur_vm::records::lua_state::lua_State;
23
24    if lua_isboolean!(vm_l, 1) {
25        let value = lua_l_checkboolean(vm_l, 1) != 0;
26        alloc_type_user_data(
27            l,
28            TypeFunctionTypeVariant::Singleton(TypeFunctionSingletonType {
29                variant: TypeFunctionSingletonVariant::V0(TypeFunctionBooleanSingleton { value }),
30            }),
31            false,
32        );
33
34        return 1;
35    }
36
37    if lua_type(vm_l, 1) == lua_Type::LUA_TSTRING as i32 {
38        let value = luaL_checkstring!(vm_l, 1);
39        alloc_type_user_data(
40            l,
41            TypeFunctionTypeVariant::Singleton(TypeFunctionSingletonType {
42                variant: TypeFunctionSingletonVariant::V1(TypeFunctionStringSingleton {
43                    value: CStr::from_ptr(value).to_string_lossy().into_owned(),
44                }),
45            }),
46            false,
47        );
48
49        return 1;
50    }
51
52    if lua_isnil!(vm_l, 1) {
53        alloc_type_user_data(
54            l,
55            TypeFunctionTypeVariant::Primitive(TypeFunctionPrimitiveType::new(Type::NilType)),
56            false,
57        );
58
59        return 1;
60    }
61
62    let type_name = CStr::from_ptr(lua_typename(vm_l, 1)).to_string_lossy();
63    lua_l_error_l(
64        vm_l,
65        c"%s".as_ptr(),
66        core::format_args!(
67            "types.singleton: can't create singleton from `{}` type",
68            type_name
69        ),
70    );
71    0
72}