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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93

use neige_infra::LUA_RIDX_GLOBALS;

use crate::{api::CallApi, state::LuaState, value::value::LuaValue};

pub trait SetApi {
    fn set_table(&mut self, idx: isize);
    fn set_field(&mut self, idx: isize, key: &str);
    fn set_i(&mut self, idx: isize, i: i64);
    fn set_meta_table(&mut self, idx: isize);
    fn raw_set(&mut self, idx: isize);
    fn raw_set_i(&mut self, idx: isize, i: i64);
    fn set_global(&mut self, name: &str);
}

impl SetApi for LuaState {
    fn set_table(&mut self, idx: isize) {
        let t = self.stack_get(idx);
        let v = self.stack_pop();
        let k = self.stack_pop();
        self.inline_set_table(t, k, v, false)
    }

    fn set_field(&mut self, idx: isize, key: &str) {
        let t = self.stack_get(idx);
        let v = self.stack_pop();
        self.inline_set_table(t, LuaValue::Str(key.into()), v, false)
    }

    fn set_i(&mut self, idx: isize, i: i64) {
        let t = self.stack_get(idx);
        let v = self.stack_pop();
        self.inline_set_table(t, LuaValue::Integer(i), v, false)
    }

    fn set_meta_table(&mut self, idx: isize) {
        let val = self.stack_get(idx);
        let mt_tb = self.stack_pop();
        match mt_tb {
            LuaValue::Nil => self.inline_set_meta_table(&val, None),
            LuaValue::Table(tb) => self.inline_set_meta_table(&val, Some(tb)),
            _ => panic!("table expected!"),
        }
    }

    fn raw_set(&mut self, idx: isize) {
        let t = self.stack_get(idx);
        let v = self.stack_pop();
        let k = self.stack_pop();
        self.inline_set_table(t, k, v, true)
    }

    fn raw_set_i(&mut self, idx: isize, i: i64) {
        let t = self.stack_get(idx);
        let v = self.stack_pop();
        self.inline_set_table(t, LuaValue::Integer(i), v, true);
    }

    fn set_global(&mut self, name: &str) {
        let t = self.registry_get(&LuaValue::Integer(LUA_RIDX_GLOBALS));
        let v = self.stack_pop();
        self.inline_set_table(t, LuaValue::Str(name.into()), v, false)
    }
}

impl LuaState {
    fn inline_set_table(&mut self, t: LuaValue, k: LuaValue, v: LuaValue, raw: bool) {
        if let LuaValue::Table(tbl) = &t {
            if raw || !tbl.get(&k).is_nil() || !tbl.has_meta_field("__newindex") {
                tbl.put(k, v);
                return;
            }
        }
        if !raw {
            let mf = self.inline_get_meta_field(&t, "__newindex");
            match &mf {
                LuaValue::Table(_) => {
                    self.inline_set_table(mf, k, v, false);
                    return;
                }
                LuaValue::Function(_) => {
                    self.stack_push(mf);
                    self.stack_push(t);
                    self.stack_push(k);
                    self.stack_push(v);
                    self.call(3, 0);
                    return;
                }
                _ => {}
            }
        }
        panic!("not a table!")
    }
}