lua-protobuf-rs 0.3.0

lua protobuf
Documentation 
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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

use crate::descriptor::enum_descriptor::LuaEnumDescriptor;
use crate::message_dyn::LuaMessageDyn;
use crate::runtime_type::LuaRuntimeType;
use derive_more::{Deref, From, Into};
use mlua::prelude::LuaUserData;
use mlua::{Integer, UserDataMethods};
use protobuf::reflect::ReflectValueBox;

#[derive(Debug, Clone, Deref, From, Into)]
pub struct LuaReflectedValueBox(pub ReflectValueBox);

impl LuaUserData for LuaReflectedValueBox {
    fn add_methods<M: UserDataMethods<Self>>(methods: &mut M) {
        methods.add_method("get_type", |_, this, ()| {
            Ok::<LuaRuntimeType, _>(this.get_type().into())
        });

        methods.add_method("get_u32", |_, this, ()| {
            if let ReflectValueBox::U32(v) = this.0 {
                Ok(Some(v))
            } else {
                Ok(None)
            }
        });

        methods.add_method("get_u64", |_, this, ()| {
            if let ReflectValueBox::U64(v) = this.0 {
                Ok(Some(v))
            } else {
                Ok(None)
            }
        });

        methods.add_method("get_i32", |_, this, ()| {
            if let ReflectValueBox::I32(v) = this.0 {
                Ok(Some(v))
            } else {
                Ok(None)
            }
        });

        methods.add_method("get_i64", |_, this, ()| {
            if let ReflectValueBox::I64(v) = this.0 {
                Ok(Some(v))
            } else {
                Ok(None)
            }
        });

        methods.add_method("get_f32", |_, this, ()| {
            if let ReflectValueBox::F32(v) = this.0 {
                Ok(Some(v))
            } else {
                Ok(None)
            }
        });

        methods.add_method("get_f64", |_, this, ()| {
            if let ReflectValueBox::F64(v) = this.0 {
                Ok(Some(v))
            } else {
                Ok(None)
            }
        });

        methods.add_method("get_bool", |_, this, ()| {
            if let ReflectValueBox::Bool(v) = this.0 {
                Ok(Some(v))
            } else {
                Ok(None)
            }
        });

        methods.add_method("get_string", |_, this, ()| {
            if let ReflectValueBox::String(v) = &this.0 {
                Ok(Some(v.clone()))
            } else {
                Ok(None)
            }
        });

        methods.add_method("get_bytes", |_, this, ()| {
            if let ReflectValueBox::Bytes(v) = &this.0 {
                Ok(Some(v.clone()))
            } else {
                Ok(None)
            }
        });

        methods.add_method("get_enum", |lua, this, ()| {
            if let ReflectValueBox::Enum(e, i) = &this.0 {
                let mut mv = mlua::MultiValue::new();
                mv.push_back(mlua::Value::UserData(
                    lua.create_any_userdata(LuaEnumDescriptor(e.clone()))?,
                ));
                mv.push_back(mlua::Value::Integer(Integer::from(*i)));
                Ok(mv)
            } else {
                Ok(mlua::MultiValue::new()) // 空多返回值，Lua 接收 nil
            }
        });

        methods.add_method("get_message", |_, this, ()| {
            if let ReflectValueBox::Message(m) = &this.0 {
                Ok(Some(LuaMessageDyn(m.clone_box())))
            } else {
                Ok(None)
            }
        });
    }
}