1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

mod pairs;

use crate::{
    api::{LuaApi, PushApi},
    state::LuaState,
};

fn lua_type(ls: &mut dyn LuaApi) -> usize {
    let tp = ls.ty_id(1);
    let ty_name = ls.tp_name(tp);
    ls.push_string(&ty_name);
    1
}

fn pcall(ls: &mut dyn LuaApi) -> usize {
    let n_args = ls.get_top() - 1;
    let status = ls.pcall(n_args, -1, 0);
    ls.push_boolean(status == 0);
    ls.insert(1);
    ls.get_top() as usize
}

fn error(ls: &mut dyn LuaApi) -> usize {
    ls.error() as usize
}

fn get_meta_table(ls: &mut dyn LuaApi) -> usize {
    if !ls.get_meta_table(1) {
        ls.push_nil();
    }
    1
}

fn set_meta_table(ls: &mut dyn LuaApi) -> usize {
    ls.set_meta_table(1);
    1
}

fn lua_print(ls: &mut dyn LuaApi) -> usize {
    let n_args = ls.get_top();
    for i in 1..=n_args {
        if ls.is_boolean(i) {
            print!("{}", ls.to_boolean(i))
        } else if ls.is_string(i) {
            print!("{}", ls.to_string(i))
        } else {
            print!("{}", ls.tp_name(ls.ty_id(i)));
        }
        if i < n_args {
            print!("\t")
        }
    }
    println!();
    0
}

impl LuaState {
    pub fn aux_lib(&mut self) {
        self.register("type", lua_type);
        self.register("print", lua_print);
        self.register("getmetatable", get_meta_table);
        self.register("setmetatable", set_meta_table);
        self.register("next", pairs::next);
        self.register("pairs", pairs::pairs);
        self.register("ipairs", pairs::ipairs);
        self.register("error", error);
        self.register("pcall", pcall);
        self.register("print", lua_print);
    }
}