#[macro_use]
mod util;

#[tokio::test]
async fn ops_assign() {
    // assign number
    test!(i32 {
        local a = 3;
        return  a;
    }, 3);
    test!(i32 {
        local a = -4;
        return  a;
    }, -4);

    // assign combination
    test!(usize {
        local a = 1;
        local b = 2;
        return a + b;
    }, 3);
    test!(usize {
        local a = 1;
        local b = a*2;
        return b;
    }, 2);
    test!(usize {
        local a = 4; local b = a*2;
        return b;
    }, 8);
    test!(usize {
        local a = 3 local b = a*2;
        return b;
    }, 6);

    // assign string
    test!(String {
        local a = "ok";
        return  a;
    }, "ok");
    test!(String {
        local a = "abc"; local b = "def";
        return a .. b;
    }, "abcdef");

    // assign function
    test!(String {
        local function foo()
            return "OOOKKK"
        end
        local ok = foo();
        return ok
    }, "OOOKKK");

    // multiple-assign
    test!(usize {
        local x = 33
        local a, b = 10, 2*x
        return a + b
    }, 76);
    #[cfg(unstable)]
    test!(Option<usize> {
        local a, b, c = 1, 2
        return c
    }, None);
    #[cfg(unstable)]
    test!(Vec<usize> {
        local a, b = 1, 2, 3;
        return {a, b}
    }, vec![1, 2]);
    test!(Vec<usize> {
        local function foo()
            return 1, 2
        end
        local x, y = foo()
        return {x, y}
    }, vec![1, 2]);
    test!(Vec<usize> {
        local s, e = string.find("hello Lua users", "Lua");
        return {s, e}
    }, vec![7, 9]);

    // swap
    test!(Vec<usize> {
        local x = 3;
        local y = 9;
        x, y = y, x;
        return {x, y}
    }, vec![9, 3]);
    test!(Vec<usize> {
        local p = {3, 9};
        local i = 1;
        local j = 2;
        p[i], p[j] = p[j], p[i];
        return p
    }, vec![9, 3]);
}

#[tokio::test]
async fn ops_arith_add() {
    test!(i32 { return  3 + 4 }, 7);
    test!(i32 { return  -3 + 4 }, 1);
    test!(i32 { return  3 + -4 }, -1);
    test!(i32 { return  -3 + -4 }, -7);
}

#[tokio::test]
async fn ops_arith_sub() {
    test!(i32 { return  3 - 4; }, -1);
    test!(i32 { return  -3 - 4; }, -7);
    test!(i32 { return  3 - -4; }, 7);
    test!(i32 { return  -3 - -4; }, 1);
}

#[tokio::test]
async fn ops_arith_mul() {
    test!(i32 { return  3 * 4; }, 12);
    test!(i32 { return  -3 * 4; }, -12);
    test!(i32 { return  3 * -4; }, -12);
    test!(i32 { return  -3 * -4; }, 12);
}

#[tokio::test]
async fn ops_arith_div() {
    test!(i32 { return  4 / 4; }, 1);
    test!(i32 { return  -8 / 4; }, -2);
    test!(i32 { return  8 / -4; }, -2);
    test!(i32 { return  -8 / -4; }, 2);
}

#[tokio::test]
async fn ops_rel() {
    test!(bool { return  4 == 4; }, true);
    test!(bool { return  4 ~= 4; }, false);
    test!(bool { return  4 > 4; }, false);
    test!(bool { return  -8 < 4; }, true);
    test!(bool { return  8 >= -4; }, true);
    test!(bool { return  -8 <= -4; }, true);
}

#[tokio::test]
async fn ops_logical() {
    test!(u32 { return 4 and 5; }, 5);
    test!(Option<u32> { return nil and 5; }, None);
    test!(usize { return 4 or 5; }, 4);
    test!(usize { return false or 5; }, 5);
    test!(bool { return not false; }, true);
}

#[tokio::test]
async fn ops_trinary() {
    test!(u32 { return true and 3 or 2; }, 3);
    test!(u32 {
        local x = 10;
        local y = 11;
        return (x > y) and 1 or 2;
    }, 2);
}

#[tokio::test]
async fn ops_concat() {
    test!(String { return "hello" .. "world"; }, "helloworld");
    test!(String { local a = "hello" return a .. "world"; }, "helloworld");
}

#[tokio::test]
async fn ops_precedence() {
    test!(bool {
        local a = 8;
        local b = 10;
        local i = 4;
        local p = a+i < b/2+1;
        local q = (a+i) < ((b/2)+1);
        return p == q
    }, true);

    test!(bool {
        local x = 111;
        local p = 5+x^2*8;
        local q = 5+((x^2)*8);
        return p == q
    }, true);

    test!(bool {
        local a = 10;
        local y = 3;
        local z = 5;
        local p = a < y and y <= z;
        local q = (a < y) and (y <= z);
        return p == q
    }, true);

    test!(i32 { local x = 2; return -x^2; }, -4);
    test!(i32 { local x = 2; return (-x)^2; }, 4);

    test!(u64 {
        local x = 2;
        local y = 3;
        local z = 2;
        return x^y^z;
    }, 512);

    test!(u64 {
        local x = 2;
        local y = 3;
        local z = 2;
        return (x^y)^z;
    }, 64);
}

#[tokio::test]
async fn table_ctor() {
    test!(Vec<String> {
        return {"A", "B", "C"}
    }, vec!["A".to_owned(), "B".into(), "C".into()]);
    test!(Vec<String> {
        return {[1]="A", [2]="B", [3]="C"}
    }, vec!["A".to_owned(), "B".into(), "C".into()]);
    test!(Vec<usize> {
        return {math.abs(-1), math.abs(-2), math.abs(-3)}
    }, vec![1usize, 2, 3]);

    // Redis discards the index 0.
    // 127.0.0.1:6379> eval 'return {[0]="A", "B", "C"}' 0
    // 1) "B"
    // 2) "C"
    test!(Vec<String> {
        return {[0]="A", "B", "C"}
    }, vec!["B".to_owned(), "C".into()]);

    // Having only non-number keys result in an empty list or set in redis.
    // 127.0.0.1:6379> eval "return {x=10, y=10}" 0
    // (empty list or set)
    test!(Vec<String> { return {x=10, y=100} }, Vec::<String>::new());
    test!(Vec<String> { return {["x"]=10, ["y"]=100} }, Vec::<String>::new());

    test!(bool {
        local a = {x=10, y=100}
        return a.x == 10 and a.y == 100
    }, true);
    test!(bool {
        local a = {["x"]=1000, ["y"]=100}
        return a.x == 1000 and a.y == 100
    }, true);
    test!(bool {
        local p = {color="blue", thickness=2, npoints=4,
                   {x=0,   y=0},
                   {x=-10, y=0},
                   {x=-10, y=1},
                   {x=0,   y=1}};
        return p.color == "blue" and p.thickness == 2 and p.npoints == 4
            and p[1].x == 0 and p[1].y == 0
            and p[2].x == -10 and p[2].y == 0
            and p[3].x == -10 and p[3].y == 1
            and p[4].x == 0 and p[4].y == 1
    }, true);

    test!(bool {
        local opnames = {["+"] = "add", ["-"] = "sub",
                   ["*"] = "mul", ["/"] = "div"}
        local i = 20; local s = "-";
        local a = {[i+0] = s, [i+1] = s..s, [i+2] = s..s..s}
        return opnames["-"] == "sub" and a[22] == "---"
    }, true);

    test!(Vec<String> { return {x=10, y=45; "one", "two", "three"} },
          vec!["one".to_owned(), "two".into(), "three".into()]);
}

#[tokio::test]
async fn ctrl() {
    test!(usize {
        local a = -10;
        if a < 0 then a = 0 end;
        return a
    }, 0);

    test!(isize {
        local a = -10;
        local b = 20;
        if a < b then return a else return b end;
    }, -10);

    test!(isize {
        local op = "*";
        local a = 2;
        local b = 4;
        local r = 0;
        if op == "+" then
            r = a + b
        elseif op == "-" then
            r = a - b
        elseif op == "*" then
            r = a*b
        elseif op == "/" then
            r = a/b
        else
            return r;
        end
        return r;
    }, 8);

    test!(isize {
        local s = 0;
        local a = {1, 2, 4, 8};
        local i = 1;
        while a[i] do
            s = s + a[i];
            i = i + 1
        end;
        return s
    }, 15);

    test!(usize {
        local a = {1,2,3,4,5};
        local i = 1;
        local s = 0;
        repeat
            s = s + a[i];
            i = i + 1;
        until i == 6
        return s
    }, 15);

    test!(usize {
        local s = 0
        for v = 0,10,2 do
            s = s + v
        end
        return s
    }, 2 + 4 + 6 + 8 + 10);

    test!(usize {
        local s = 0;
        for v = 0,10 do
            s = s + v;
        end;
        return s
    }, 55);

    #[cfg(unstable)]
    test!(usize {
        local a = {1,3,3,4,2,5,6,1,3,4};
        local value = 5;
        local found = nil
        for i=1,10 do
            if a[i] == value then
                found = i      -- save value of 'i'
                break
            end
        end
        return found
    }, 6);

    test!(usize {
        local a = {3, 3, 3};
        local s = 0;
        for i,v in ipairs(a) do
            s = s + i + v;
        end
        return s
    }, 15);

    test!(usize {
        local a = {3, 3, 3};
        local s = 0;
        for k in pairs(a) do
            s = s + k;
        end
        return s
    }, 6);

    test!(usize {
        do return 1 end
        return 2
    }, 1);
}

#[tokio::test]
async fn types_bool() {
    test!(bool { return true }, true);
    test!(bool { return false }, false);
    test!(bool { return 1 }, true);
    test!(bool { return 0 }, false);
    test!(bool { return nil }, false);
}

#[tokio::test]
async fn types_num() {
    test!(usize { return 4 }, 4);
    // Indeed, Redis protocol doesn't support floating points.
    test!(f64 { return 0.4 }, 0.0);
    test!(bool { return 0.4 == 0.4 }, true);
    // FIXME: full-moon cannot parse at the moment.
    // test!(bool { return 0.457e-3 == 0.457e-3 }, true);
    // test!(bool { return 0.3e12 == 0.3e12 }, true);
    // test!(bool { return 5e+20 == 5e+20 }, true);
}

#[tokio::test]
async fn types_string() {
    test!(String { return "one string" }, "one string");
    // Indeed, a single quotation is not supported by Rust.
    // test!(String { return 'one string' }, "one string");
    test!(
        String {
            local s1 = "one string";
            local s2 = string.gsub(s1, "one", "another");
            return s2
        },
        "another string"
    );

    test!(String { return "one line\nnext\"in quotes\", 'in quotes'" },
          "one line\nnext\"in quotes\", 'in quotes'");

    // FIXME: This is not supported by full-moon.
    // let s = String {
    //     page = [[
    // <HTML>
    // <HEAD>
    // <TITLE>An HTML Page</TITLE>
    // </HEAD>
    // <BODY>
    //  <A HREF="http://www.lua.org">Lua</A>
    //  [[a text between double brackets]]
    // </BODY>
    // </HTML>
    //     ]]
    //         return page
    // });
    // println!("{}", s);

    test!(usize { return "10" + 1 }, 11);
    test!(String { return "10 + 1" }, "10 + 1");
    // FIXME: full-moon cannot parse at the moment.
    // test!(bool { return "-5.3e-10"*"2" == -1.06e-09 }, true);
    test!(String { return 10 .. 20 }, "1020");
    test!(String { return tostring(10) }, "10");
}

#[tokio::test]
async fn table() {
    test!(Vec<String> { return {} }, Vec::<String>::new());
    test!(usize { local a = {} local k = "x" a[k] = 10 return a[k] }, 10);
    test!(usize { local a = {} local k = "x" a[k] = 10 return a["x"] + a["x"] }, 20);
    test!(usize { local a = {} a["x"] = 10 local b = a a["x"] = 20 return b["x"] }, 20);
    test!(Vec<String> { local a = {} a["x"] = 10 local b = a a = nil return b },
          Vec::<String>::new());
    let v: Vec<_> = (1usize..=1000).map(|i| i * 2).collect();
    test!(Vec<usize> {
        local a = {}
        for i=1,1000 do a[i] = i * 2 end
        return a
    }, v);
}

#[tokio::test]
async fn func() {
    test!(Vec<String> {
        local logtable = {}

        local function logit(msg)
            logtable[#logtable+1] = msg
            end

            logit("foo")
            logit("bar")

            return logtable
    }, vec!["foo".to_owned(), "bar".into()]);

    test!(Vec<String> {
        local network = {
            {name = "grauna",  IP = "210.26.30.34"},
            {name = "arraial", IP = "210.26.30.23"},
            {name = "lua",     IP = "210.26.23.12"},
            {name = "derain",  IP = "210.26.23.20"},
        };
        table.sort(network, function (a,b)
            return (a.name > b.name)
        end);
        local res = {}
        for i,v in ipairs(network) do
            res[i] = v.IP
        end
        return res
    }, vec!["210.26.23.12".to_owned(), "210.26.30.34".into(), "210.26.23.20".into(), "210.26.30.23".into()]);

    // variadic args
    //
    // FIXME: selene complains that arg not found.
    // test!(String {
    //     function print (...)
    //         local printResult = ""
    //         for i,v in ipairs(arg) do
    //             printResult = printResult .. tostring(v) .. "\t"
    //         end
    //         printResult = printResult .. "\n"
    //     end
    //     return print("sa", "wa", "ki")
    // }, "sawaki");

    // named args
    test!(String {
        local function rename (arg)
            return "old=" .. arg.old .. ",new=" .. arg.new
        end
        return rename{old="a", new="b"}
    }, "old=a,new=b");

    // TODO: Test closure
    // TODO: Test methods
}