lib.rs

 1use std::collections::{HashMap, HashSet};
 2
 3use mlua::{Error, FromLua, Function, Lua, LuaSerdeExt, ToLua, UserData, Value};
 4
 5pub mod runtime;
 6pub use runtime::*;
 7use serde::{de::DeserializeOwned, Deserialize, Serialize};
 8
 9impl Runtime for Lua {
10    type Module = String;
11
12    fn init(module: Self::Module) -> Option<Self> {
13        let lua = Lua::new();
14        lua.load(&module).exec().ok()?;
15        return Some(lua);
16    }
17
18    fn handles(&self) -> Handles {
19        let mut globals = HashSet::new();
20        for pair in self.globals().pairs::<String, Value>() {
21            if let Ok((k, _)) = pair {
22                globals.insert(k);
23            }
24        }
25
26        globals
27    }
28
29    fn val<T: DeserializeOwned>(&self, name: String) -> Option<T> {
30        let val: Value = self.globals().get(name).ok()?;
31        Some(self.from_value(val).ok()?)
32    }
33
34    fn call<T: Serialize + DeserializeOwned>(&self, name: String, arg: T) -> Option<T> {
35        let fun: Function = self.globals().get(name).ok()?;
36        let arg: Value = self.to_value(&arg).ok()?;
37        let result = fun.call(arg).ok()?;
38        Some(self.from_value(result).ok()?)
39    }
40}