64 lines
1.6 KiB
Rust
64 lines
1.6 KiB
Rust
use mlua::IntoLua;
|
|
|
|
#[derive(Clone,Copy)]
|
|
pub struct Enum(u32);
|
|
#[derive(Clone,Copy)]
|
|
pub struct EnumItems;
|
|
#[derive(Clone,Copy)]
|
|
pub struct EnumItem<'a>{
|
|
ed:&'a rbx_reflection::EnumDescriptor<'a>,
|
|
}
|
|
|
|
impl Into<rbx_types::Enum> for Enum{
|
|
fn into(self)->rbx_types::Enum{
|
|
rbx_types::Enum::from_u32(self.0)
|
|
}
|
|
}
|
|
|
|
impl<'a> EnumItem<'a>{
|
|
const fn new(ed:&'a rbx_reflection::EnumDescriptor)->Self{
|
|
Self{ed}
|
|
}
|
|
}
|
|
|
|
pub fn set_globals(_lua:&mlua::Lua,globals:&mlua::Table)->Result<(),mlua::Error>{
|
|
globals.set("Enum",EnumItems)
|
|
}
|
|
|
|
impl mlua::UserData for EnumItem<'_>{
|
|
fn add_fields<F:mlua::UserDataFields<Self>>(_fields:&mut F){
|
|
}
|
|
fn add_methods<M:mlua::UserDataMethods<Self>>(methods:&mut M){
|
|
methods.add_meta_function(mlua::MetaMethod::Index,|lua,(this,val):(EnumItem<'_>,mlua::String)|{
|
|
match this.ed.items.get(&*val.to_str()?){
|
|
Some(&id)=>Enum(id).into_lua(lua),
|
|
None=>mlua::Value::Nil.into_lua(lua),
|
|
}
|
|
});
|
|
}
|
|
}
|
|
type_from_lua_userdata_lua_lifetime!(EnumItem);
|
|
|
|
impl mlua::UserData for EnumItems{
|
|
fn add_fields<F:mlua::UserDataFields<Self>>(_fields:&mut F){
|
|
}
|
|
fn add_methods<M:mlua::UserDataMethods<Self>>(methods:&mut M){
|
|
methods.add_meta_function(mlua::MetaMethod::Index,|lua,(_,val):(Self,mlua::String)|{
|
|
let db=rbx_reflection_database::get();
|
|
match db.enums.get(&*val.to_str()?){
|
|
Some(ed)=>EnumItem::new(ed).into_lua(lua),
|
|
None=>mlua::Value::Nil.into_lua(lua),
|
|
}
|
|
});
|
|
}
|
|
}
|
|
type_from_lua_userdata!(EnumItems);
|
|
|
|
impl mlua::UserData for Enum{
|
|
fn add_fields<F:mlua::UserDataFields<Self>>(_fields:&mut F){
|
|
}
|
|
fn add_methods<M:mlua::UserDataMethods<Self>>(_methods:&mut M){
|
|
}
|
|
}
|
|
type_from_lua_userdata!(Enum);
|