use mlua::{self, Lua, LuaSerdeExt, Table};
pub fn add_parse_util(lua: &Lua) -> Result<(), mlua::Error> {
let parse_table = lua.create_table()?;
let json_fn = lua.create_function(|lua, json_str: String| {
let value: serde_json::Value = serde_json::from_str(&json_str)
.map_err(|e| mlua::Error::RuntimeError(e.to_string()))?;
lua.to_value(&value)
})?;
parse_table.set("json", json_fn)?;
let yaml_fn = lua.create_function(|lua, yaml_str: String| {
let value: serde_yaml::Value = serde_yaml::from_str(&yaml_str)
.map_err(|e| mlua::Error::RuntimeError(e.to_string()))?;
lua.to_value(&value)
})?;
parse_table.set("yaml", yaml_fn)?;
let toml_fn = lua.create_function(|lua, toml_str: String| {
let value: toml::Value = toml::from_str(&toml_str)
.map_err(|e| mlua::Error::RuntimeError(e.to_string()))?;
lua.to_value(&value)
})?;
parse_table.set("toml", toml_fn)?;
let checksum_fn =
lua.create_function(|_, (content, file_name): (String, String)| {
for line in content.lines() {
let parts: Vec<&str> = line.split_whitespace().collect();
if parts.len() == 2 && parts.get(1) == Some(&file_name.as_str())
{
return Ok(parts.first().map(ToString::to_string));
}
}
Ok(None)
})?;
parse_table.set("checksumFile", checksum_fn)?;
let utils_table: Table = lua.globals().get("UTILS")?;
utils_table.set("PARSE", parse_table)?;
Ok(())
}