use std::ops::Deref;
use mlua::{FromLua, IntoLua, Lua, ObjectLike, Result, Table, Value};
use crate::{Converters, Fetches, Http, HttpMessage, LogLevel};
#[derive(Clone)]
pub struct Txn {
class: Table,
pub c: Converters,
pub f: Fetches,
pub(crate) r#priv: Value,
}
impl Txn {
#[inline]
pub fn http(&self) -> Result<Http> {
self.class.get("http")
}
pub fn http_req(&self) -> Result<HttpMessage> {
self.class.get("http_req")
}
pub fn http_res(&self) -> Result<HttpMessage> {
self.class.get("http_res")
}
#[inline]
pub fn log(&self, level: LogLevel, msg: impl AsRef<str>) -> Result<()> {
let msg = msg.as_ref();
self.class.call_method("log", (level, msg))
}
#[inline]
pub fn deflog(&self, msg: impl AsRef<str>) -> Result<()> {
self.class.call_method("deflog", msg.as_ref())
}
#[inline]
pub fn get_priv<R: FromLua>(&self) -> Result<R> {
self.class.call_method("get_priv", ())
}
#[inline]
pub fn set_priv(&self, val: impl IntoLua) -> Result<()> {
self.class.call_method("set_priv", val)
}
#[inline]
pub fn get_var<R: FromLua>(&self, name: &str) -> Result<R> {
self.class.call_method("get_var", name)
}
#[inline]
pub fn set_var(&self, name: &str, val: impl IntoLua) -> Result<()> {
self.class.call_method("set_var", (name, val))
}
#[inline]
pub fn set_var_if_exists(&self, name: &str, val: impl IntoLua) -> Result<()> {
self.class.call_method("set_var", (name, val, true))
}
#[inline]
pub fn unset_var(&self, name: &str) -> Result<()> {
self.class.call_method("unset_var", name)
}
#[inline]
pub fn set_loglevel(&self, level: LogLevel) -> Result<()> {
self.class.call_method("set_loglevel", level)
}
}
impl FromLua for Txn {
#[inline]
fn from_lua(value: Value, lua: &Lua) -> Result<Self> {
let class = Table::from_lua(value, lua)?;
Ok(Txn {
c: class.get("c")?,
f: class.get("f")?,
class,
r#priv: Value::Nil,
})
}
}
impl Deref for Txn {
type Target = Table;
#[inline]
fn deref(&self) -> &Self::Target {
&self.class
}
}