use std::any::Any; use std::cell::{Cell, Ref, RefCell, RefMut}; use std::marker::PhantomData; use std::mem; use std::os::raw::c_void; use std::rc::Rc; #[cfg(feature = "serialize")] use serde::Serialize; use crate::error::{Error, Result}; use crate::ffi; use crate::function::Function; use crate::lua::Lua; use crate::types::{Callback, LuaRef, MaybeSend, UserDataCell}; use crate::userdata::{ AnyUserData, MetaMethod, UserData, UserDataFields, UserDataMethods, UserDataWrapped, }; use crate::util::{ assert_stack, init_userdata_metatable, protect_lua_closure, push_string, push_userdata, take_userdata, StackGuard, }; use crate::value::{FromLua, FromLuaMulti, MultiValue, ToLua, ToLuaMulti, Value}; #[cfg(feature = "async")] use { crate::types::AsyncCallback, futures_core::future::{Future, LocalBoxFuture}, futures_util::future::{self, TryFutureExt}, }; /// Constructed by the [`Lua::scope`] method, allows temporarily creating Lua userdata and /// callbacks that are not required to be Send or 'static. /// /// See [`Lua::scope`] for more details. /// /// [`Lua::scope`]: struct.Lua.html#method.scope #[allow(clippy::type_complexity)] pub struct Scope<'lua, 'scope> { lua: &'lua Lua, destructors: RefCell, fn(LuaRef<'lua>) -> Vec>)>>, _scope_invariant: PhantomData>, } impl<'lua, 'scope> Scope<'lua, 'scope> { pub(crate) fn new(lua: &'lua Lua) -> Scope<'lua, 'scope> { Scope { lua, destructors: RefCell::new(Vec::new()), _scope_invariant: PhantomData, } } /// Wraps a Rust function or closure, creating a callable Lua function handle to it. /// /// This is a version of [`Lua::create_function`] that creates a callback which expires on /// scope drop. See [`Lua::scope`] for more details. /// /// [`Lua::create_function`]: struct.Lua.html#method.create_function /// [`Lua::scope`]: struct.Lua.html#method.scope pub fn create_function<'callback, A, R, F>(&'callback self, func: F) -> Result> where A: FromLuaMulti<'callback>, R: ToLuaMulti<'callback>, F: 'scope + Fn(&'callback Lua, A) -> Result, { // Safe, because 'scope must outlive 'callback (due to Self containing 'scope), however the // callback itself must be 'scope lifetime, so the function should not be able to capture // anything of 'callback lifetime. 'scope can't be shortened due to being invariant, and // the 'callback lifetime here can't be enlarged due to coming from a universal // quantification in Lua::scope. // // I hope I got this explanation right, but in any case this is tested with compiletest_rs // to make sure callbacks can't capture handles with lifetime outside the scope, inside the // scope, and owned inside the callback itself. unsafe { self.create_callback(Box::new(move |lua, args| { func(lua, A::from_lua_multi(args, lua)?)?.to_lua_multi(lua) })) } } /// Wraps a Rust mutable closure, creating a callable Lua function handle to it. /// /// This is a version of [`Lua::create_function_mut`] that creates a callback which expires /// on scope drop. See [`Lua::scope`] and [`Scope::create_function`] for more details. /// /// [`Lua::create_function_mut`]: struct.Lua.html#method.create_function_mut /// [`Lua::scope`]: struct.Lua.html#method.scope /// [`Scope::create_function`]: #method.create_function pub fn create_function_mut<'callback, A, R, F>( &'callback self, func: F, ) -> Result> where A: FromLuaMulti<'callback>, R: ToLuaMulti<'callback>, F: 'scope + FnMut(&'callback Lua, A) -> Result, { let func = RefCell::new(func); self.create_function(move |lua, args| { (&mut *func .try_borrow_mut() .map_err(|_| Error::RecursiveMutCallback)?)(lua, args) }) } /// Wraps a Rust async function or closure, creating a callable Lua function handle to it. /// /// This is a version of [`Lua::create_async_function`] that creates a callback which expires on /// scope drop. See [`Lua::scope`] and [`Lua::async_scope`] for more details. /// /// Requires `feature = "async"` /// /// [`Lua::create_async_function`]: struct.Lua.html#method.create_async_function /// [`Lua::scope`]: struct.Lua.html#method.scope /// [`Lua::async_scope`]: struct.Lua.html#method.async_scope #[cfg(feature = "async")] #[cfg_attr(docsrs, doc(cfg(feature = "async")))] pub fn create_async_function<'callback, A, R, F, FR>( &'callback self, func: F, ) -> Result> where A: FromLuaMulti<'callback>, R: ToLuaMulti<'callback>, F: 'scope + Fn(&'callback Lua, A) -> FR, FR: 'callback + Future>, { unsafe { self.create_async_callback(Box::new(move |lua, args| { let args = match A::from_lua_multi(args, lua) { Ok(args) => args, Err(e) => return Box::pin(future::err(e)), }; Box::pin(func(lua, args).and_then(move |ret| future::ready(ret.to_lua_multi(lua)))) })) } } /// Create a Lua userdata object from a custom userdata type. /// /// This is a version of [`Lua::create_userdata`] that creates a userdata which expires on /// scope drop, and does not require that the userdata type be Send (but still requires that the /// UserData be 'static). /// See [`Lua::scope`] for more details. /// /// [`Lua::create_userdata`]: struct.Lua.html#method.create_userdata /// [`Lua::scope`]: struct.Lua.html#method.scope pub fn create_userdata(&self, data: T) -> Result> where T: 'static + UserData, { self.create_userdata_inner(UserDataWrapped::new(data)) } /// Create a Lua userdata object from a custom serializable userdata type. /// /// This is a version of [`Lua::create_ser_userdata`] that creates a userdata which expires on /// scope drop, and does not require that the userdata type be Send (but still requires that the /// UserData be 'static). /// See [`Lua::scope`] for more details. /// /// Requires `feature = "serialize"` /// /// [`Lua::create_ser_userdata`]: struct.Lua.html#method.create_ser_userdata /// [`Lua::scope`]: struct.Lua.html#method.scope #[cfg(feature = "serialize")] #[cfg_attr(docsrs, doc(cfg(feature = "serialize")))] pub fn create_ser_userdata(&self, data: T) -> Result> where T: 'static + UserData + Serialize, { self.create_userdata_inner(UserDataWrapped::new_ser(data)) } fn create_userdata_inner(&self, data: UserDataWrapped) -> Result> where T: 'static + UserData, { // Safe even though T may not be Send, because the parent Lua cannot be sent to another // thread while the Scope is alive (or the returned AnyUserData handle even). unsafe { let u = self.lua.make_userdata(data)?; self.destructors.borrow_mut().push((u.0.clone(), |u| { let state = u.lua.state; assert_stack(state, 2); u.lua.push_ref(&u); // Clear uservalue #[cfg(any(feature = "lua54", feature = "lua53", feature = "lua52"))] ffi::lua_pushnil(state); #[cfg(any(feature = "lua51", feature = "luajit"))] ffi::lua_newtable(state); ffi::lua_setuservalue(state, -2); // We know the destructor has not run yet because we hold a reference to the // userdata. vec![Box::new(take_userdata::>(state))] })); Ok(u) } } /// Create a Lua userdata object from a custom userdata type. /// /// This is a version of [`Lua::create_userdata`] that creates a userdata which expires on /// scope drop, and does not require that the userdata type be Send or 'static. See /// [`Lua::scope`] for more details. /// /// Lifting the requirement that the UserData type be 'static comes with some important /// limitations, so if you only need to eliminate the Send requirement, it is probably better to /// use [`Scope::create_userdata`] instead. /// /// The main limitation that comes from using non-'static userdata is that the produced userdata /// will no longer have a `TypeId` associated with it, becuase `TypeId` can only work for /// 'static types. This means that it is impossible, once the userdata is created, to get a /// reference to it back *out* of an `AnyUserData` handle. This also implies that the /// "function" type methods that can be added via [`UserDataMethods`] (the ones that accept /// `AnyUserData` as a first parameter) are vastly less useful. Also, there is no way to re-use /// a single metatable for multiple non-'static types, so there is a higher cost associated with /// creating the userdata metatable each time a new userdata is created. /// /// [`Scope::create_userdata`]: #method.create_userdata /// [`Lua::create_userdata`]: struct.Lua.html#method.create_userdata /// [`Lua::scope`]: struct.Lua.html#method.scope /// [`UserDataMethods`]: trait.UserDataMethods.html pub fn create_nonstatic_userdata(&self, data: T) -> Result> where T: 'scope + UserData, { let data = Rc::new(RefCell::new(UserDataWrapped::new(data))); // 'callback outliving 'scope is a lie to make the types work out, required due to the // inability to work with the more correct callback type that is universally quantified over // 'lua. This is safe though, because `UserData::add_methods` does not get to pick the 'lua // lifetime, so none of the static methods UserData types can add can possibly capture // parameters. fn wrap_method<'scope, 'lua, 'callback: 'scope, T: 'scope>( scope: &Scope<'lua, 'scope>, data: Rc>, method: NonStaticMethod<'callback, T>, ) -> Result> { // On methods that actually receive the userdata, we fake a type check on the passed in // userdata, where we pretend there is a unique type per call to // `Scope::create_nonstatic_userdata`. You can grab a method from a userdata and call // it on a mismatched userdata type, which when using normal 'static userdata will fail // with a type mismatch, but here without this check would proceed as though you had // called the method on the original value (since we otherwise completely ignore the // first argument). let check_data = data.clone(); let check_ud_type = move |lua: &'callback Lua, value| { if let Some(Value::UserData(ud)) = value { unsafe { let _sg = StackGuard::new(lua.state); assert_stack(lua.state, 3); lua.push_ref(&ud.0); if ffi::lua_getmetatable(lua.state, -1) == 0 { return Err(Error::UserDataTypeMismatch); } ffi::lua_pushstring(lua.state, cstr!("__mlua")); if ffi::lua_rawget(lua.state, -2) == ffi::LUA_TLIGHTUSERDATA { let ud_ptr = ffi::lua_touserdata(lua.state, -1); if ud_ptr == check_data.as_ptr() as *mut c_void { return Ok(()); } } } }; Err(Error::UserDataTypeMismatch) }; match method { NonStaticMethod::Method(method) => { let f = Box::new(move |lua, mut args: MultiValue<'callback>| { check_ud_type(lua, args.pop_front())?; let data = data .try_borrow() .map(|cell| Ref::map(cell, AsRef::as_ref)) .map_err(|_| Error::UserDataBorrowError)?; method(lua, &*data, args) }); unsafe { scope.create_callback(f) } } NonStaticMethod::MethodMut(method) => { let method = RefCell::new(method); let f = Box::new(move |lua, mut args: MultiValue<'callback>| { check_ud_type(lua, args.pop_front())?; let mut method = method .try_borrow_mut() .map_err(|_| Error::RecursiveMutCallback)?; let mut data = data .try_borrow_mut() .map(|cell| RefMut::map(cell, AsMut::as_mut)) .map_err(|_| Error::UserDataBorrowMutError)?; (&mut *method)(lua, &mut *data, args) }); unsafe { scope.create_callback(f) } } NonStaticMethod::Function(function) => unsafe { scope.create_callback(function) }, NonStaticMethod::FunctionMut(function) => { let function = RefCell::new(function); let f = Box::new(move |lua, args| { (&mut *function .try_borrow_mut() .map_err(|_| Error::RecursiveMutCallback)?)( lua, args ) }); unsafe { scope.create_callback(f) } } } } let mut ud_fields = NonStaticUserDataFields::default(); let mut ud_methods = NonStaticUserDataMethods::default(); T::add_fields(&mut ud_fields); T::add_methods(&mut ud_methods); unsafe { let lua = self.lua; let _sg = StackGuard::new(lua.state); assert_stack(lua.state, 13); // We need to wrap dummy userdata because their memory can be accessed by serializer push_userdata(lua.state, UserDataCell::new(UserDataWrapped::new(())))?; // Prepare metatable, add meta methods first and then meta fields protect_lua_closure(lua.state, 0, 1, |state| { ffi::lua_newtable(state); // Add internal metamethod to store reference to the data ffi::lua_pushstring(state, cstr!("__mlua")); ffi::lua_pushlightuserdata(lua.state, data.as_ptr() as *mut c_void); ffi::lua_rawset(state, -3); })?; for (k, m) in ud_methods.meta_methods { push_string(lua.state, k.validate()?.name())?; lua.push_value(Value::Function(wrap_method(self, data.clone(), m)?))?; protect_lua_closure(lua.state, 3, 1, |state| { ffi::lua_rawset(state, -3); })?; } for (k, f) in ud_fields.meta_fields { push_string(lua.state, k.validate()?.name())?; lua.push_value(f(mem::transmute(lua))?)?; protect_lua_closure(lua.state, 3, 1, |state| { ffi::lua_rawset(state, -3); })?; } let metatable_index = ffi::lua_absindex(lua.state, -1); let mut field_getters_index = None; if ud_fields.field_getters.len() > 0 { protect_lua_closure(lua.state, 0, 1, |state| { ffi::lua_newtable(state); })?; for (k, m) in ud_fields.field_getters { push_string(lua.state, &k)?; lua.push_value(Value::Function(wrap_method(self, data.clone(), m)?))?; protect_lua_closure(lua.state, 3, 1, |state| { ffi::lua_rawset(state, -3); })?; } field_getters_index = Some(ffi::lua_absindex(lua.state, -1)); } let mut field_setters_index = None; if ud_fields.field_setters.len() > 0 { protect_lua_closure(lua.state, 0, 1, |state| { ffi::lua_newtable(state); })?; for (k, m) in ud_fields.field_setters { push_string(lua.state, &k)?; lua.push_value(Value::Function(wrap_method(self, data.clone(), m)?))?; protect_lua_closure(lua.state, 3, 1, |state| { ffi::lua_rawset(state, -3); })?; } field_setters_index = Some(ffi::lua_absindex(lua.state, -1)); } let mut methods_index = None; if ud_methods.methods.len() > 0 { // Create table used for methods lookup protect_lua_closure(lua.state, 0, 1, |state| { ffi::lua_newtable(state); })?; for (k, m) in ud_methods.methods { push_string(lua.state, &k)?; lua.push_value(Value::Function(wrap_method(self, data.clone(), m)?))?; protect_lua_closure(lua.state, 3, 1, |state| { ffi::lua_rawset(state, -3); })?; } methods_index = Some(ffi::lua_absindex(lua.state, -1)); } init_userdata_metatable::<()>( lua.state, metatable_index, field_getters_index, field_setters_index, methods_index, )?; let count = field_getters_index.map(|_| 1).unwrap_or(0) + field_setters_index.map(|_| 1).unwrap_or(0) + methods_index.map(|_| 1).unwrap_or(0); ffi::lua_pop(lua.state, count); let mt_id = ffi::lua_topointer(lua.state, -1); ffi::lua_setmetatable(lua.state, -2); let ud = AnyUserData(lua.pop_ref()); lua.register_userdata_metatable(mt_id as isize); self.destructors.borrow_mut().push((ud.0.clone(), |ud| { // We know the destructor has not run yet because we hold a reference to the userdata. let state = ud.lua.state; assert_stack(state, 2); ud.lua.push_ref(&ud); // Deregister metatable ffi::lua_getmetatable(state, -1); let mt_id = ffi::lua_topointer(state, -1); ffi::lua_pop(state, 1); ud.lua.deregister_userdata_metatable(mt_id as isize); // Clear uservalue #[cfg(any(feature = "lua54", feature = "lua53", feature = "lua52"))] ffi::lua_pushnil(state); #[cfg(any(feature = "lua51", feature = "luajit"))] ffi::lua_newtable(state); ffi::lua_setuservalue(state, -2); vec![Box::new(take_userdata::>(state))] })); Ok(ud) } } // Unsafe, because the callback can improperly capture any value with 'callback scope, such as // improperly capturing an argument. Since the 'callback lifetime is chosen by the user and the // lifetime of the callback itself is 'scope (non-'static), the borrow checker will happily pick // a 'callback that outlives 'scope to allow this. In order for this to be safe, the callback // must NOT capture any parameters. unsafe fn create_callback<'callback>( &self, f: Callback<'callback, 'scope>, ) -> Result> { let f = mem::transmute::, Callback<'lua, 'static>>(f); let f = self.lua.create_callback(f)?; let mut destructors = self.destructors.borrow_mut(); destructors.push((f.0.clone(), |f| { let state = f.lua.state; assert_stack(state, 3); f.lua.push_ref(&f); // We know the destructor has not run yet because we hold a reference to the callback. ffi::lua_getupvalue(state, -1, 1); let ud1 = take_userdata::(state); ffi::lua_pushnil(state); ffi::lua_setupvalue(state, -2, 1); ffi::lua_getupvalue(state, -1, 2); let ud2 = take_userdata::(state); ffi::lua_pushnil(state); ffi::lua_setupvalue(state, -2, 2); ffi::lua_pop(state, 1); vec![Box::new(ud1), Box::new(ud2)] })); Ok(f) } #[cfg(feature = "async")] unsafe fn create_async_callback<'callback>( &self, f: AsyncCallback<'callback, 'scope>, ) -> Result> { let f = mem::transmute::, AsyncCallback<'lua, 'static>>(f); let f = self.lua.create_async_callback(f)?; let mut destructors = self.destructors.borrow_mut(); destructors.push((f.0.clone(), |f| { let state = f.lua.state; assert_stack(state, 4); f.lua.push_ref(&f); // We know the destructor has not run yet because we hold a reference to the callback. // First, get the environment table #[cfg(any(feature = "lua54", feature = "lua53", feature = "lua52"))] ffi::lua_getupvalue(state, -1, 1); #[cfg(any(feature = "lua51", feature = "luajit"))] ffi::lua_getfenv(state, -1); // Second, get the `get_poll()` closure using the corresponding key ffi::lua_pushstring(state, cstr!("get_poll")); ffi::lua_rawget(state, -2); // Destroy all upvalues ffi::lua_getupvalue(state, -1, 1); let ud1 = take_userdata::(state); ffi::lua_pushnil(state); ffi::lua_setupvalue(state, -2, 1); ffi::lua_getupvalue(state, -1, 2); let ud2 = take_userdata::(state); ffi::lua_pushnil(state); ffi::lua_setupvalue(state, -2, 2); ffi::lua_pop(state, 1); let mut data: Vec> = vec![Box::new(ud1), Box::new(ud2)]; // Finally, get polled future and destroy it ffi::lua_pushstring(state, cstr!("poll")); if ffi::lua_rawget(state, -2) == ffi::LUA_TFUNCTION { ffi::lua_getupvalue(state, -1, 1); let ud3 = take_userdata::>>(state); ffi::lua_pushnil(state); ffi::lua_setupvalue(state, -2, 1); data.push(Box::new(ud3)); ffi::lua_getupvalue(state, -1, 2); let ud4 = take_userdata::(state); ffi::lua_pushnil(state); ffi::lua_setupvalue(state, -2, 2); data.push(Box::new(ud4)); } data })); Ok(f) } } impl<'lua, 'scope> Drop for Scope<'lua, 'scope> { fn drop(&mut self) { // We separate the action of invalidating the userdata in Lua and actually dropping the // userdata type into two phases. This is so that, in the event a userdata drop panics, we // can be sure that all of the userdata in Lua is actually invalidated. // All destructors are non-panicking, so this is fine let to_drop = self .destructors .get_mut() .drain(..) .flat_map(|(r, dest)| dest(r)) .collect::>(); drop(to_drop); } } enum NonStaticMethod<'lua, T> { Method(Box) -> Result>>), MethodMut(Box) -> Result>>), Function(Box) -> Result>>), FunctionMut(Box) -> Result>>), } struct NonStaticUserDataMethods<'lua, T: UserData> { methods: Vec<(Vec, NonStaticMethod<'lua, T>)>, meta_methods: Vec<(MetaMethod, NonStaticMethod<'lua, T>)>, } impl<'lua, T: UserData> Default for NonStaticUserDataMethods<'lua, T> { fn default() -> NonStaticUserDataMethods<'lua, T> { NonStaticUserDataMethods { methods: Vec::new(), meta_methods: Vec::new(), } } } impl<'lua, T: UserData> UserDataMethods<'lua, T> for NonStaticUserDataMethods<'lua, T> { fn add_method(&mut self, name: &S, method: M) where S: AsRef<[u8]> + ?Sized, A: FromLuaMulti<'lua>, R: ToLuaMulti<'lua>, M: 'static + MaybeSend + Fn(&'lua Lua, &T, A) -> Result, { self.methods.push(( name.as_ref().to_vec(), NonStaticMethod::Method(Box::new(move |lua, ud, args| { method(lua, ud, A::from_lua_multi(args, lua)?)?.to_lua_multi(lua) })), )); } fn add_method_mut(&mut self, name: &S, mut method: M) where S: AsRef<[u8]> + ?Sized, A: FromLuaMulti<'lua>, R: ToLuaMulti<'lua>, M: 'static + MaybeSend + FnMut(&'lua Lua, &mut T, A) -> Result, { self.methods.push(( name.as_ref().to_vec(), NonStaticMethod::MethodMut(Box::new(move |lua, ud, args| { method(lua, ud, A::from_lua_multi(args, lua)?)?.to_lua_multi(lua) })), )); } #[cfg(feature = "async")] fn add_async_method(&mut self, _name: &S, _method: M) where T: Clone, S: AsRef<[u8]> + ?Sized, A: FromLuaMulti<'lua>, R: ToLuaMulti<'lua>, M: 'static + MaybeSend + Fn(&'lua Lua, T, A) -> MR, MR: 'lua + Future>, { // The panic should never happen as async non-static code wouldn't compile // Non-static lifetime must be bounded to 'lua lifetime mlua_panic!("asynchronous methods are not supported for non-static userdata") } fn add_function(&mut self, name: &S, function: F) where S: AsRef<[u8]> + ?Sized, A: FromLuaMulti<'lua>, R: ToLuaMulti<'lua>, F: 'static + MaybeSend + Fn(&'lua Lua, A) -> Result, { self.methods.push(( name.as_ref().to_vec(), NonStaticMethod::Function(Box::new(move |lua, args| { function(lua, A::from_lua_multi(args, lua)?)?.to_lua_multi(lua) })), )); } fn add_function_mut(&mut self, name: &S, mut function: F) where S: AsRef<[u8]> + ?Sized, A: FromLuaMulti<'lua>, R: ToLuaMulti<'lua>, F: 'static + MaybeSend + FnMut(&'lua Lua, A) -> Result, { self.methods.push(( name.as_ref().to_vec(), NonStaticMethod::FunctionMut(Box::new(move |lua, args| { function(lua, A::from_lua_multi(args, lua)?)?.to_lua_multi(lua) })), )); } #[cfg(feature = "async")] fn add_async_function(&mut self, _name: &S, _function: F) where T: Clone, S: AsRef<[u8]> + ?Sized, A: FromLuaMulti<'lua>, R: ToLuaMulti<'lua>, F: 'static + MaybeSend + Fn(&'lua Lua, A) -> FR, FR: 'lua + Future>, { // The panic should never happen as async non-static code wouldn't compile // Non-static lifetime must be bounded to 'lua lifetime mlua_panic!("asynchronous functions are not supported for non-static userdata") } fn add_meta_method(&mut self, meta: S, method: M) where S: Into, A: FromLuaMulti<'lua>, R: ToLuaMulti<'lua>, M: 'static + MaybeSend + Fn(&'lua Lua, &T, A) -> Result, { self.meta_methods.push(( meta.into(), NonStaticMethod::Method(Box::new(move |lua, ud, args| { method(lua, ud, A::from_lua_multi(args, lua)?)?.to_lua_multi(lua) })), )); } fn add_meta_method_mut(&mut self, meta: S, mut method: M) where S: Into, A: FromLuaMulti<'lua>, R: ToLuaMulti<'lua>, M: 'static + MaybeSend + FnMut(&'lua Lua, &mut T, A) -> Result, { self.meta_methods.push(( meta.into(), NonStaticMethod::MethodMut(Box::new(move |lua, ud, args| { method(lua, ud, A::from_lua_multi(args, lua)?)?.to_lua_multi(lua) })), )); } fn add_meta_function(&mut self, meta: S, function: F) where S: Into, A: FromLuaMulti<'lua>, R: ToLuaMulti<'lua>, F: 'static + MaybeSend + Fn(&'lua Lua, A) -> Result, { self.meta_methods.push(( meta.into(), NonStaticMethod::Function(Box::new(move |lua, args| { function(lua, A::from_lua_multi(args, lua)?)?.to_lua_multi(lua) })), )); } fn add_meta_function_mut(&mut self, meta: S, mut function: F) where S: Into, A: FromLuaMulti<'lua>, R: ToLuaMulti<'lua>, F: 'static + MaybeSend + FnMut(&'lua Lua, A) -> Result, { self.meta_methods.push(( meta.into(), NonStaticMethod::FunctionMut(Box::new(move |lua, args| { function(lua, A::from_lua_multi(args, lua)?)?.to_lua_multi(lua) })), )); } } struct NonStaticUserDataFields<'lua, T: UserData> { field_getters: Vec<(Vec, NonStaticMethod<'lua, T>)>, field_setters: Vec<(Vec, NonStaticMethod<'lua, T>)>, meta_fields: Vec<(MetaMethod, Box Result>>)>, } impl<'lua, T: UserData> Default for NonStaticUserDataFields<'lua, T> { fn default() -> NonStaticUserDataFields<'lua, T> { NonStaticUserDataFields { field_getters: Vec::new(), field_setters: Vec::new(), meta_fields: Vec::new(), } } } impl<'lua, T: UserData> UserDataFields<'lua, T> for NonStaticUserDataFields<'lua, T> { fn add_field_method_get(&mut self, name: &S, method: M) where S: AsRef<[u8]> + ?Sized, R: ToLua<'lua>, M: 'static + MaybeSend + Fn(&'lua Lua, &T) -> Result, { self.field_getters.push(( name.as_ref().to_vec(), NonStaticMethod::Method(Box::new(move |lua, ud, _| { method(lua, ud)?.to_lua_multi(lua) })), )); } fn add_field_method_set(&mut self, name: &S, mut method: M) where S: AsRef<[u8]> + ?Sized, A: FromLua<'lua>, M: 'static + MaybeSend + FnMut(&'lua Lua, &mut T, A) -> Result<()>, { self.field_setters.push(( name.as_ref().to_vec(), NonStaticMethod::MethodMut(Box::new(move |lua, ud, args| { method(lua, ud, A::from_lua_multi(args, lua)?)?.to_lua_multi(lua) })), )); } fn add_field_function_get(&mut self, name: &S, function: F) where S: AsRef<[u8]> + ?Sized, R: ToLua<'lua>, F: 'static + MaybeSend + Fn(&'lua Lua, AnyUserData<'lua>) -> Result, { self.field_getters.push(( name.as_ref().to_vec(), NonStaticMethod::Function(Box::new(move |lua, args| { function(lua, AnyUserData::from_lua_multi(args, lua)?)?.to_lua_multi(lua) })), )); } fn add_field_function_set(&mut self, name: &S, mut function: F) where S: AsRef<[u8]> + ?Sized, A: FromLua<'lua>, F: 'static + MaybeSend + FnMut(&'lua Lua, AnyUserData<'lua>, A) -> Result<()>, { self.field_setters.push(( name.as_ref().to_vec(), NonStaticMethod::FunctionMut(Box::new(move |lua, args| { let (ud, val) = <_>::from_lua_multi(args, lua)?; function(lua, ud, val)?.to_lua_multi(lua) })), )); } fn add_meta_field_with(&mut self, meta: S, f: F) where S: Into, F: 'static + MaybeSend + Fn(&'lua Lua) -> Result, R: ToLua<'lua>, { let meta = meta.into(); self.meta_fields.push(( meta.clone(), Box::new(move |lua| { let value = f(lua)?.to_lua(lua)?; if meta == MetaMethod::Index || meta == MetaMethod::NewIndex { match value { Value::Nil | Value::Table(_) | Value::Function(_) => {} _ => { return Err(Error::MetaMethodTypeError { method: meta.to_string(), type_name: value.type_name(), message: Some("expected nil, table or function".to_string()), }) } } } Ok(value) }), )); } }