use mlua::{FromLuaMulti, Function, IntoLuaMulti, Lua, Result};

use super::ownedfunction::OwnedFunction;

pub trait Runnable<A, R>
where
    Self: Send,
    R: ?Sized
{
    fn run<'lua>(&self, arg: A, lua: &'lua Lua) -> Result<R>
    where
        A: IntoLuaMulti<'lua>,
        R: FromLuaMulti<'lua>;
}

impl<A, R, T> Runnable<A, R> for T
where
    T: Fn(A) -> R,
    Self: Send,
    R: ?Sized
{
    fn run<'lua>(&self, arg: A, _: &'lua Lua) -> Result<R>
    where
        A: IntoLuaMulti<'lua>,
        R: FromLuaMulti<'lua>,
    {
        Ok(self(arg))
    }
}

impl<A, R> Runnable<A, R> for OwnedFunction
where
    Self: Send,
    R: ?Sized
{
    fn run<'lua>(&self, arg: A, lua: &'lua Lua) -> Result<R>
    where
        A: IntoLuaMulti<'lua>,
        R: FromLuaMulti<'lua>,
    {
        let func: Function = self.get(lua);
        func.call::<A, R>(arg)
    }
}