extern crate libc;
extern crate aw;

use aw::Lua;
use aw::LuaTable;

#[test]
fn basic() {
    let mut lua = Lua::new();
    let val: Option<i32> = lua.exec_string("return 5;");
    assert_eq!(val.unwrap(), 5);
}

#[test]
fn syntax_error() {
    let mut lua = Lua::new();
    let val: Option<()> = lua.exec_string("aw");
    assert!(val.is_none());
}

#[test]
fn execution_error() {
    let mut lua = Lua::new();

    let val: Option<()> = lua.exec_string("return a:hello()");
    assert!(val.is_none());
}

#[test]
fn call_and_read_table() {
    let mut lua = Lua::new();

    let mut val: LuaTable = lua.exec_string("return {1, 2, 3};").unwrap();
    assert_eq!(val.query::<u8, _>(1).unwrap(), 1);
    assert_eq!(val.query::<u8, _>(2).unwrap(), 2);
    assert_eq!(val.query::<u8, _>(3).unwrap(), 3);
}

#[test]
fn simple_function() {
    let mut lua = Lua::new();

    fn ret5() -> i32 {
        5
    };
    lua.set("ret5", aw::function0(ret5));

    let val: i32 = lua.exec_string("return ret5()").unwrap();
    assert_eq!(val, 5);
}

#[test]
fn one_argument() {
    let mut lua = Lua::new();

    fn plus_one(val: i32) -> i32 {
        val + 1
    };
    lua.set("plus_one", aw::function1(plus_one));

    let val: i32 = lua.exec_string("return plus_one(3)").unwrap();
    assert_eq!(val, 4);
}

#[test]
fn two_arguments() {
    let mut lua = Lua::new();

    fn add(val1: i32, val2: i32) -> i32 {
        val1 + val2
    };
    lua.set("add", aw::function2(add));

    let val: i32 = lua.exec_string("return add(3, 7)").unwrap();
    assert_eq!(val, 10);
}

#[test]
fn wrong_arguments_types() {
    let mut lua = Lua::new();

    fn add(val1: i32, val2: i32) -> i32 {
        val1 + val2
    };
    lua.set("add", aw::function2(add));
    let val: Option<i32> = lua.exec_string("return add(3, \"hello\")");
    match val {
        None => (),
        _ => panic!(),
    }
}

#[test]
fn closures() {
    let mut lua = Lua::new();

    lua.set("add", aw::function2(|a: i32, b: i32| a + b));
    lua.set("sub", aw::function2(|a: i32, b: i32| a - b));

    let val1: i32 = lua.exec_string("return add(3, 7)").unwrap();
    assert_eq!(val1, 10);

    let val2: i32 = lua.exec_string("return sub(5, 2)").unwrap();
    assert_eq!(val2, 3);
}

#[test]
fn closures_lifetime() {
    fn t<F>(f: F)
    where
        F: Fn(i32, i32) -> i32,
    {
        let mut lua = Lua::new();

        lua.set("add", aw::function2(f));

        let val1: i32 = lua.exec_string("return add(3, 7)").unwrap();
        assert_eq!(val1, 10);
    }

    t(|a, b| a + b);
}

#[test]
fn closures_extern_access() {
    let mut a = 5;

    {
        let mut lua = Lua::new();

        lua.set("inc", aw::function0(|| a += 1));
        for _ in 0..15 {
            let _: () = lua.exec_string("inc()").unwrap();
        }
    }

    assert_eq!(a, 20)
}

#[test]
fn test_exec_func() {
    let mut lua = Lua::new();
    {
        let mut index = 5;
        lua.set("add", aw::function1(|a: i32| index += a));
        let success: i32 = lua.exec_func1("add", 3);
        assert!(success == 0);
        assert_eq!(index, 8);
    }
    {
        let mut index = 5;
        lua.set(
            "sub",
            aw::function3(|a: i32, b: u32, _c: String| index -= a + b as i32),
        );
        let success: i32 = lua.exec_func3("sub", 3, 1, "".to_string());
        assert!(success == 0);
        assert_eq!(index, 1);
    }
}

#[test]
fn test_exec_func_by_param() {
    let mut lua = Lua::new();
    lua.openlibs();
    let func = r"
        function test() 
            return testRust(1, 2, 3);
        end

        function test2(...) 
            local t = {...}
            local sum = 0
            for _,v in ipairs(t) do
                sum = sum + v
            end
            return sum
        end
    ";

    extern "C" fn test_rust(lua: *mut aw::lua_State) -> libc::c_int {
        let mut lua_ob = Lua::from_existing_state(lua, false);
        let _: Option<()> = lua_ob.exec_func("test2");
        1
    }
    lua.register("testRust", test_rust);
    let _: Option<()> = lua.exec_string(func);
    let ret: Option<i32> = lua.exec_string("return test();");
    assert_eq!(ret.unwrap(), 6);
}