clean up ClosureSearcher api with hrtb
see - https://stackoverflow.com/questions/35592750/how-does-for-syntax-differ-from-a-regular-lifetime-bound/35595491#35595491 - https://doc.rust-lang.org/nomicon/hrtb.html
This commit is contained in:
parent
801a04802a
commit
406c84dabf
@ -1,5 +1,5 @@
|
|||||||
use rlua::prelude::LuaError;
|
use rlua::prelude::LuaError;
|
||||||
use rlua::{Context, MetaMethod, RegistryKey, Table, UserData, UserDataMethods, Value};
|
use rlua::{Context, Function, MetaMethod, RegistryKey, Table, UserData, UserDataMethods, Value};
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::fs::File;
|
use std::fs::File;
|
||||||
use std::io::Read;
|
use std::io::Read;
|
||||||
@ -132,20 +132,38 @@ where
|
|||||||
/// up an `rlua::Context` with Rust code.
|
/// up an `rlua::Context` with Rust code.
|
||||||
///
|
///
|
||||||
/// Enables exposing `UserData` types to an `rlua::Context`.
|
/// Enables exposing `UserData` types to an `rlua::Context`.
|
||||||
///
|
|
||||||
/// Closures must return an `rlua::Result`-wrapped `&'static str`. This string is
|
|
||||||
/// subsequently passed into `rlua::Context.load()` and evaluated.
|
|
||||||
pub struct ClosureSearcher {
|
pub struct ClosureSearcher {
|
||||||
/// Closures must accept an `rlua::Context` as their only parameter, and can do with
|
/// Closures must accept three parameters:
|
||||||
/// it what they wish. Closures must return an `rlua::Result`-wrapped `&'static str`.
|
///
|
||||||
modules: HashMap<&'static str, Box<dyn Fn(Context) -> rlua::Result<&'static str> + Send>>,
|
/// 1. An `rlua::Context`, which the closure can do what it wants with.
|
||||||
|
///
|
||||||
|
/// 2. An `rlua::Table` containing globals (i.e. Lua’s `_G`), which can be passed
|
||||||
|
/// to `Chunk.set_environment()`.
|
||||||
|
///
|
||||||
|
/// 3. The name of the module to be loaded (`&str`).
|
||||||
|
///
|
||||||
|
/// Closures must return an `rlua::Result`-wrapped `Function`. This `Function`
|
||||||
|
/// acts as the module loader.
|
||||||
|
modules: HashMap<
|
||||||
|
&'static str,
|
||||||
|
Box<
|
||||||
|
dyn for<'ctx> Fn(Context<'ctx>, Table<'ctx>, &str) -> rlua::Result<Function<'ctx>>
|
||||||
|
+ Send,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
|
|
||||||
globals: RegistryKey,
|
globals: RegistryKey,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl ClosureSearcher {
|
impl ClosureSearcher {
|
||||||
pub fn new(
|
pub fn new(
|
||||||
modules: HashMap<&'static str, Box<dyn Fn(Context) -> rlua::Result<&'static str> + Send>>,
|
modules: HashMap<
|
||||||
|
&'static str,
|
||||||
|
Box<
|
||||||
|
dyn for<'ctx> Fn(Context<'ctx>, Table<'ctx>, &str) -> rlua::Result<Function<'ctx>>
|
||||||
|
+ Send,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
globals: RegistryKey,
|
globals: RegistryKey,
|
||||||
) -> Self {
|
) -> Self {
|
||||||
Self { modules, globals }
|
Self { modules, globals }
|
||||||
@ -157,22 +175,20 @@ impl UserData for ClosureSearcher {
|
|||||||
where
|
where
|
||||||
M: UserDataMethods<'lua, Self>,
|
M: UserDataMethods<'lua, Self>,
|
||||||
{
|
{
|
||||||
methods.add_meta_method(MetaMethod::Call, |lua_ctx, this, name: String| {
|
methods.add_meta_method(
|
||||||
let name = name.as_str();
|
MetaMethod::Call,
|
||||||
match this.modules.get(name) {
|
|lua_ctx: Context<'lua>, this, name: String| {
|
||||||
Some(ref loader) => {
|
let name = name.as_str();
|
||||||
let content = loader(lua_ctx)?;
|
match this.modules.get(name) {
|
||||||
Ok(Value::Function(
|
Some(ref closure) => Ok(Value::Function(closure(
|
||||||
lua_ctx
|
lua_ctx,
|
||||||
.load(content)
|
lua_ctx.registry_value::<Table>(&this.globals)?,
|
||||||
.set_name(name)?
|
name,
|
||||||
.set_environment(lua_ctx.registry_value::<Table>(&this.globals)?)?
|
)?)),
|
||||||
.into_function()?,
|
None => Ok(Value::Nil),
|
||||||
))
|
|
||||||
}
|
}
|
||||||
None => Ok(Value::Nil),
|
},
|
||||||
}
|
);
|
||||||
});
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -207,7 +223,13 @@ pub trait AddSearcher {
|
|||||||
/// setup.
|
/// setup.
|
||||||
fn add_closure_searcher(
|
fn add_closure_searcher(
|
||||||
&self,
|
&self,
|
||||||
modules: HashMap<&'static str, Box<dyn Fn(Context) -> rlua::Result<&'static str> + Send>>,
|
modules: HashMap<
|
||||||
|
&'static str,
|
||||||
|
Box<
|
||||||
|
dyn for<'ctx> Fn(Context<'ctx>, Table<'ctx>, &str) -> rlua::Result<Function<'ctx>>
|
||||||
|
+ Send,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
) -> Result<()>;
|
) -> Result<()>;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -266,7 +288,13 @@ impl<'a> AddSearcher for Context<'a> {
|
|||||||
|
|
||||||
fn add_closure_searcher(
|
fn add_closure_searcher(
|
||||||
&self,
|
&self,
|
||||||
modules: HashMap<&'static str, Box<dyn Fn(Context) -> rlua::Result<&'static str> + Send>>,
|
modules: HashMap<
|
||||||
|
&'static str,
|
||||||
|
Box<
|
||||||
|
dyn for<'ctx> Fn(Context<'ctx>, Table<'ctx>, &str) -> rlua::Result<Function<'ctx>>
|
||||||
|
+ Send,
|
||||||
|
>,
|
||||||
|
>,
|
||||||
) -> Result<()> {
|
) -> Result<()> {
|
||||||
let globals = self.globals();
|
let globals = self.globals();
|
||||||
let searchers: Table = globals.get::<_, Table>("package")?.get("searchers")?;
|
let searchers: Table = globals.get::<_, Table>("package")?.get("searchers")?;
|
||||||
|
@ -1,4 +1,4 @@
|
|||||||
use rlua::{Context, Lua, Table, UserData, UserDataMethods, Value};
|
use rlua::{Context, Function, Lua, Table, UserData, UserDataMethods, Value};
|
||||||
use rlua_searcher::{AddSearcher, Result};
|
use rlua_searcher::{AddSearcher, Result};
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::fs::File;
|
use std::fs::File;
|
||||||
@ -166,10 +166,15 @@ fn add_closure_searcher_works() {
|
|||||||
|
|
||||||
let mut modules: HashMap<
|
let mut modules: HashMap<
|
||||||
&'static str,
|
&'static str,
|
||||||
Box<dyn Fn(Context) -> rlua::Result<&'static str> + Send>,
|
Box<
|
||||||
|
dyn for<'ctx> Fn(Context<'ctx>, Table<'ctx>, &str) -> rlua::Result<Function<'ctx>>
|
||||||
|
+ Send,
|
||||||
|
>,
|
||||||
> = HashMap::new();
|
> = HashMap::new();
|
||||||
|
|
||||||
let instrument_loader = Box::new(|lua_ctx: Context| {
|
let instrument_loader: Box<
|
||||||
|
dyn for<'ctx> Fn(Context<'ctx>, Table<'ctx>, &str) -> rlua::Result<Function<'ctx>> + Send,
|
||||||
|
> = Box::new(|lua_ctx, env, name| {
|
||||||
let globals = lua_ctx.globals();
|
let globals = lua_ctx.globals();
|
||||||
let new = lua_ctx.create_function(|_, (name, sound): (String, String)| {
|
let new = lua_ctx.create_function(|_, (name, sound): (String, String)| {
|
||||||
Ok(Instrument::new(name, sound))
|
Ok(Instrument::new(name, sound))
|
||||||
@ -177,7 +182,11 @@ fn add_closure_searcher_works() {
|
|||||||
let tbl = lua_ctx.create_table()?;
|
let tbl = lua_ctx.create_table()?;
|
||||||
tbl.set("new", new)?;
|
tbl.set("new", new)?;
|
||||||
globals.set("instrument", tbl)?;
|
globals.set("instrument", tbl)?;
|
||||||
Ok("return instrument")
|
Ok(lua_ctx
|
||||||
|
.load("return instrument")
|
||||||
|
.set_name(name)?
|
||||||
|
.set_environment(env)?
|
||||||
|
.into_function()?)
|
||||||
});
|
});
|
||||||
|
|
||||||
modules.insert("instrument", instrument_loader);
|
modules.insert("instrument", instrument_loader);
|
||||||
|
Loading…
Reference in New Issue
Block a user