2020-05-06 20:18:56 -04:00
|
|
|
use std::any::Any;
|
|
|
|
use std::cell::{Cell, RefCell};
|
|
|
|
use std::marker::PhantomData;
|
|
|
|
use std::mem;
|
|
|
|
use std::os::raw::c_void;
|
|
|
|
use std::rc::Rc;
|
|
|
|
|
|
|
|
use crate::error::{Error, Result};
|
|
|
|
use crate::ffi;
|
|
|
|
use crate::function::Function;
|
|
|
|
use crate::lua::Lua;
|
|
|
|
use crate::types::{Callback, LuaRef, MaybeSend};
|
|
|
|
use crate::userdata::{AnyUserData, MetaMethod, UserData, UserDataMethods};
|
|
|
|
use crate::util::{
|
|
|
|
assert_stack, init_userdata_metatable, protect_lua_closure, push_string, push_userdata,
|
|
|
|
take_userdata, StackGuard,
|
|
|
|
};
|
|
|
|
use crate::value::{FromLuaMulti, MultiValue, ToLuaMulti, Value};
|
|
|
|
|
|
|
|
#[cfg(feature = "async")]
|
|
|
|
use {
|
|
|
|
crate::types::AsyncCallback,
|
|
|
|
futures_core::future::Future,
|
|
|
|
futures_util::future::{self, TryFutureExt},
|
|
|
|
std::os::raw::c_char,
|
|
|
|
};
|
|
|
|
|
|
|
|
/// 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
|
|
|
|
pub struct Scope<'lua, 'scope> {
|
|
|
|
lua: &'lua Lua,
|
|
|
|
destructors: RefCell<Vec<(LuaRef<'lua>, fn(LuaRef<'lua>) -> Vec<Box<dyn Any>>)>>,
|
|
|
|
_scope_invariant: PhantomData<Cell<&'scope ()>>,
|
|
|
|
}
|
|
|
|
|
|
|
|
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<Function<'lua>>
|
|
|
|
where
|
|
|
|
A: FromLuaMulti<'callback>,
|
|
|
|
R: ToLuaMulti<'callback>,
|
|
|
|
F: 'scope + Fn(&'callback Lua, A) -> Result<R>,
|
|
|
|
{
|
|
|
|
// 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<Function<'lua>>
|
|
|
|
where
|
|
|
|
A: FromLuaMulti<'callback>,
|
|
|
|
R: ToLuaMulti<'callback>,
|
|
|
|
F: 'scope + FnMut(&'callback Lua, A) -> Result<R>,
|
|
|
|
{
|
|
|
|
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.
|
|
|
|
///
|
|
|
|
/// [`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")]
|
|
|
|
pub fn create_async_function<'callback, A, R, F, FR>(
|
|
|
|
&'callback self,
|
|
|
|
func: F,
|
|
|
|
) -> Result<Function<'lua>>
|
|
|
|
where
|
|
|
|
A: FromLuaMulti<'callback>,
|
|
|
|
R: ToLuaMulti<'callback>,
|
|
|
|
F: 'scope + Fn(&'callback Lua, A) -> FR,
|
|
|
|
FR: 'callback + Future<Output = Result<R>>,
|
|
|
|
{
|
|
|
|
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<T>(&self, data: T) -> Result<AnyUserData<'lua>>
|
|
|
|
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);
|
|
|
|
// We know the destructor has not run yet because we hold a reference to the
|
|
|
|
// userdata.
|
|
|
|
vec![Box::new(take_userdata::<RefCell<T>>(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<T>(&self, data: T) -> Result<AnyUserData<'lua>>
|
|
|
|
where
|
|
|
|
T: 'scope + UserData,
|
|
|
|
{
|
|
|
|
let data = Rc::new(RefCell::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<RefCell<T>>,
|
|
|
|
method: NonStaticMethod<'callback, T>,
|
|
|
|
) -> Result<Function<'lua>> {
|
|
|
|
// 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) = value {
|
|
|
|
if let Value::UserData(u) = value {
|
|
|
|
unsafe {
|
|
|
|
assert_stack(lua.state, 1);
|
|
|
|
lua.push_ref(&u.0);
|
|
|
|
ffi::lua_getuservalue(lua.state, -1);
|
|
|
|
#[cfg(any(feature = "lua52", feature = "lua51", feature = "luajit"))]
|
|
|
|
{
|
|
|
|
ffi::lua_rawgeti(lua.state, -1, 1);
|
|
|
|
ffi::lua_remove(lua.state, -2);
|
|
|
|
}
|
|
|
|
return ffi::lua_touserdata(lua.state, -1)
|
|
|
|
== check_data.as_ptr() as *mut c_void;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
false
|
|
|
|
};
|
|
|
|
|
|
|
|
match method {
|
|
|
|
NonStaticMethod::Method(method) => {
|
|
|
|
let method_data = data.clone();
|
|
|
|
let f = Box::new(move |lua, mut args: MultiValue<'callback>| {
|
|
|
|
if !check_ud_type(lua, args.pop_front()) {
|
|
|
|
return Err(Error::UserDataTypeMismatch);
|
|
|
|
}
|
|
|
|
let data = method_data
|
|
|
|
.try_borrow()
|
|
|
|
.map_err(|_| Error::UserDataBorrowError)?;
|
|
|
|
method(lua, &*data, args)
|
|
|
|
});
|
|
|
|
unsafe { scope.create_callback(f) }
|
|
|
|
}
|
|
|
|
NonStaticMethod::MethodMut(method) => {
|
|
|
|
let method = RefCell::new(method);
|
|
|
|
let method_data = data.clone();
|
|
|
|
let f = Box::new(move |lua, mut args: MultiValue<'callback>| {
|
|
|
|
if !check_ud_type(lua, args.pop_front()) {
|
|
|
|
return Err(Error::UserDataTypeMismatch);
|
|
|
|
}
|
|
|
|
let mut method = method
|
|
|
|
.try_borrow_mut()
|
|
|
|
.map_err(|_| Error::RecursiveMutCallback)?;
|
|
|
|
let mut data = method_data
|
|
|
|
.try_borrow_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_methods = NonStaticUserDataMethods::default();
|
|
|
|
T::add_methods(&mut ud_methods);
|
|
|
|
|
|
|
|
unsafe {
|
|
|
|
let lua = self.lua;
|
|
|
|
let _sg = StackGuard::new(lua.state);
|
|
|
|
assert_stack(lua.state, 6);
|
|
|
|
|
|
|
|
push_userdata(lua.state, ())?;
|
2020-05-08 07:42:40 -04:00
|
|
|
#[cfg(any(feature = "lua54", feature = "lua53"))]
|
2020-05-06 20:18:56 -04:00
|
|
|
ffi::lua_pushlightuserdata(lua.state, data.as_ptr() as *mut c_void);
|
|
|
|
#[cfg(any(feature = "lua52", feature = "lua51", feature = "luajit"))]
|
|
|
|
protect_lua_closure(lua.state, 0, 1, |state| {
|
|
|
|
// Lua 5.2/5.1 allows to store only table. Then we will wrap the value.
|
|
|
|
ffi::lua_createtable(state, 1, 0);
|
|
|
|
ffi::lua_pushlightuserdata(state, data.as_ptr() as *mut c_void);
|
|
|
|
ffi::lua_rawseti(state, -2, 1);
|
|
|
|
})?;
|
|
|
|
ffi::lua_setuservalue(lua.state, -2);
|
|
|
|
|
|
|
|
protect_lua_closure(lua.state, 0, 1, move |state| {
|
|
|
|
ffi::lua_newtable(state);
|
|
|
|
})?;
|
|
|
|
|
|
|
|
for (k, m) in ud_methods.meta_methods {
|
|
|
|
push_string(lua.state, k.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);
|
|
|
|
})?;
|
|
|
|
}
|
|
|
|
|
|
|
|
if ud_methods.methods.is_empty() {
|
|
|
|
init_userdata_metatable::<()>(lua.state, -1, None)?;
|
|
|
|
} else {
|
|
|
|
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);
|
|
|
|
})?;
|
|
|
|
}
|
|
|
|
|
|
|
|
init_userdata_metatable::<()>(lua.state, -2, Some(-1))?;
|
|
|
|
ffi::lua_pop(lua.state, 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
ffi::lua_setmetatable(lua.state, -2);
|
|
|
|
|
|
|
|
Ok(AnyUserData(lua.pop_ref()))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// 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<Function<'lua>> {
|
|
|
|
let f = mem::transmute::<Callback<'callback, 'scope>, 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::<Callback>(state);
|
|
|
|
ffi::lua_pushnil(state);
|
|
|
|
ffi::lua_setupvalue(state, -2, 1);
|
|
|
|
|
|
|
|
ffi::lua_getupvalue(state, -1, 2);
|
|
|
|
let ud2 = take_userdata::<Lua>(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<Function<'lua>> {
|
|
|
|
let f = mem::transmute::<AsyncCallback<'callback, 'scope>, 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
|
2020-05-08 07:42:40 -04:00
|
|
|
#[cfg(any(feature = "lua54", feature = "lua53", feature = "lua52"))]
|
2020-05-06 20:18:56 -04:00
|
|
|
ffi::lua_getupvalue(state, -1, 1);
|
|
|
|
#[cfg(any(feature = "lua51", feature = "luajit"))]
|
|
|
|
ffi::lua_getfenv(state, -1);
|
|
|
|
|
|
|
|
// Then, get the get_poll() closure using the corresponding key
|
|
|
|
let key = "get_poll";
|
|
|
|
ffi::lua_pushlstring(state, key.as_ptr() as *const c_char, key.len());
|
|
|
|
ffi::lua_rawget(state, -2);
|
|
|
|
|
|
|
|
// Finally, destroy all upvalues
|
|
|
|
ffi::lua_getupvalue(state, -1, 1);
|
|
|
|
let ud1 = take_userdata::<AsyncCallback>(state);
|
|
|
|
ffi::lua_pushnil(state);
|
|
|
|
ffi::lua_setupvalue(state, -2, 1);
|
|
|
|
|
|
|
|
ffi::lua_getupvalue(state, -1, 2);
|
|
|
|
let ud2 = take_userdata::<Lua>(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)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
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::<Vec<_>>();
|
|
|
|
|
|
|
|
drop(to_drop);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
enum NonStaticMethod<'lua, T> {
|
|
|
|
Method(Box<dyn Fn(&'lua Lua, &T, MultiValue<'lua>) -> Result<MultiValue<'lua>>>),
|
|
|
|
MethodMut(Box<dyn FnMut(&'lua Lua, &mut T, MultiValue<'lua>) -> Result<MultiValue<'lua>>>),
|
|
|
|
Function(Box<dyn Fn(&'lua Lua, MultiValue<'lua>) -> Result<MultiValue<'lua>>>),
|
|
|
|
FunctionMut(Box<dyn FnMut(&'lua Lua, MultiValue<'lua>) -> Result<MultiValue<'lua>>>),
|
|
|
|
}
|
|
|
|
|
|
|
|
struct NonStaticUserDataMethods<'lua, T: UserData> {
|
|
|
|
methods: Vec<(Vec<u8>, 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<S, A, R, M>(&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<R>,
|
|
|
|
{
|
|
|
|
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<S, A, R, M>(&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<R>,
|
|
|
|
{
|
|
|
|
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<S, A, R, M, MR>(&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<Output = Result<R>>,
|
|
|
|
{
|
|
|
|
// 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<S, A, R, F>(&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<R>,
|
|
|
|
{
|
|
|
|
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<S, A, R, F>(&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<R>,
|
|
|
|
{
|
|
|
|
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<S, A, R, F, FR>(&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<Output = Result<R>>,
|
|
|
|
{
|
|
|
|
// 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<A, R, M>(&mut self, meta: MetaMethod, method: M)
|
|
|
|
where
|
|
|
|
A: FromLuaMulti<'lua>,
|
|
|
|
R: ToLuaMulti<'lua>,
|
|
|
|
M: 'static + MaybeSend + Fn(&'lua Lua, &T, A) -> Result<R>,
|
|
|
|
{
|
|
|
|
self.meta_methods.push((
|
|
|
|
meta,
|
|
|
|
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<A, R, M>(&mut self, meta: MetaMethod, mut method: M)
|
|
|
|
where
|
|
|
|
A: FromLuaMulti<'lua>,
|
|
|
|
R: ToLuaMulti<'lua>,
|
|
|
|
M: 'static + MaybeSend + FnMut(&'lua Lua, &mut T, A) -> Result<R>,
|
|
|
|
{
|
|
|
|
self.meta_methods.push((
|
|
|
|
meta,
|
|
|
|
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<A, R, F>(&mut self, meta: MetaMethod, function: F)
|
|
|
|
where
|
|
|
|
A: FromLuaMulti<'lua>,
|
|
|
|
R: ToLuaMulti<'lua>,
|
|
|
|
F: 'static + MaybeSend + Fn(&'lua Lua, A) -> Result<R>,
|
|
|
|
{
|
|
|
|
self.meta_methods.push((
|
|
|
|
meta,
|
|
|
|
NonStaticMethod::Function(Box::new(move |lua, args| {
|
|
|
|
function(lua, A::from_lua_multi(args, lua)?)?.to_lua_multi(lua)
|
|
|
|
})),
|
|
|
|
));
|
|
|
|
}
|
|
|
|
|
|
|
|
fn add_meta_function_mut<A, R, F>(&mut self, meta: MetaMethod, mut function: F)
|
|
|
|
where
|
|
|
|
A: FromLuaMulti<'lua>,
|
|
|
|
R: ToLuaMulti<'lua>,
|
|
|
|
F: 'static + MaybeSend + FnMut(&'lua Lua, A) -> Result<R>,
|
|
|
|
{
|
|
|
|
self.meta_methods.push((
|
|
|
|
meta,
|
|
|
|
NonStaticMethod::FunctionMut(Box::new(move |lua, args| {
|
|
|
|
function(lua, A::from_lua_multi(args, lua)?)?.to_lua_multi(lua)
|
|
|
|
})),
|
|
|
|
));
|
|
|
|
}
|
|
|
|
}
|