Dreaded_X
eb0c80c4ce
ActionCallback now only has one generics argument that has to implement IntoLuaMulti, this makes ActionCallback much more flexible as it no longer always requires two arguments.
71 lines
1.9 KiB
Rust
71 lines
1.9 KiB
Rust
use std::marker::PhantomData;
|
|
|
|
use futures::future::try_join_all;
|
|
use mlua::{FromLua, IntoLuaMulti};
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub struct ActionCallback<P> {
|
|
callbacks: Vec<mlua::Function>,
|
|
_parameters: PhantomData<P>,
|
|
}
|
|
|
|
// NOTE: For some reason the derive macro combined with PhantomData leads to issues where it
|
|
// requires all types part of P to implement default, even if they never actually get constructed.
|
|
// By manually implemented Default it works fine.
|
|
impl<P> Default for ActionCallback<P> {
|
|
fn default() -> Self {
|
|
Self {
|
|
callbacks: Default::default(),
|
|
_parameters: Default::default(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl<P> FromLua for ActionCallback<P> {
|
|
fn from_lua(value: mlua::Value, _lua: &mlua::Lua) -> mlua::Result<Self> {
|
|
let callbacks = match value {
|
|
mlua::Value::Function(f) => vec![f],
|
|
mlua::Value::Table(table) => table
|
|
.pairs::<mlua::Value, mlua::Function>()
|
|
.map(|pair| {
|
|
let (_, f) = pair?;
|
|
|
|
Ok::<_, mlua::Error>(f)
|
|
})
|
|
.try_collect()?,
|
|
_ => {
|
|
return Err(mlua::Error::FromLuaConversionError {
|
|
from: value.type_name(),
|
|
to: "ActionCallback".into(),
|
|
message: Some("expected function or table of functions".into()),
|
|
});
|
|
}
|
|
};
|
|
|
|
Ok(ActionCallback {
|
|
callbacks,
|
|
_parameters: PhantomData::<P>,
|
|
})
|
|
}
|
|
}
|
|
|
|
// TODO: Return proper error here
|
|
impl<P> ActionCallback<P>
|
|
where
|
|
P: IntoLuaMulti + Sync + Clone,
|
|
{
|
|
pub async fn call(&self, parameters: P) {
|
|
try_join_all(
|
|
self.callbacks
|
|
.iter()
|
|
.map(async |f| f.call_async::<()>(parameters.clone()).await),
|
|
)
|
|
.await
|
|
.unwrap();
|
|
}
|
|
|
|
pub fn is_empty(&self) -> bool {
|
|
self.callbacks.is_empty()
|
|
}
|
|
}
|