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

use std::marker::PhantomData;

use mlua::{AnyUserData, Error, IntoLua, Lua, UserData};

use crate::{EnumGenerator, RecordGenerator, ToTypename, Type, TypeBody, TypeName};

/// A userdata which can be used as a static proxy
pub trait StaticUserdata: UserData + 'static {}
impl<T: UserData + 'static> StaticUserdata for T {}

/// A newtype storing proxy userdata created via [`mlua::Lua::create_proxy`].
///
/// the `TypeName` for this struct is implemented as the `TypeName` for `T` concatenated with "Class".
/// For example, if your type is called "MyType", the proxy would have "MyTypeClass" for a `TypeName`.
///
/// the documentation for this proxy receives only `static` methods, i.e. those created via:
/// - `TealDataMethods::add_function`
/// - `TealDataMethods::add_meta_function`
/// - `TealDataMethods::add_meta_function_mut`
/// - `TealDataMethods::add_function_mut`
/// - `TealDataMethods::add_async_function`
/// - `TealDataFields::add_field_function_get`
/// - `TealDataFields::add_field_function_set`
/// - `TealDataFields::add_meta_field_with`
///
/// The type documentation is overriden as well.
pub struct UserDataProxy<'lua, T: StaticUserdata> {
    user_data: AnyUserData<'lua>,
    ph_: PhantomData<T>,
}

impl<'lua, T: StaticUserdata> UserDataProxy<'lua, T> {
    /// Creates a new UserDataProxy
    pub fn new(lua: &'lua Lua) -> Result<Self, Error> {
        Ok(Self {
            user_data: lua.create_proxy::<T>()?,
            ph_: Default::default(),
        })
    }
}

impl<T: StaticUserdata + ToTypename> ToTypename for UserDataProxy<'_, T> {
    fn to_typename() -> crate::Type {
        let mut x = T::to_typename();
        if let Type::Single(x) = &mut x {
            x.name = format!("Class{}", x.name).into();
        }
        x
    }
}

impl<T: StaticUserdata + TypeBody + ToTypename> TypeBody for UserDataProxy<'_, T> {
    fn get_type_body() -> crate::TypeGenerator {
        let generator = T::get_type_body();
        // extract only "functions"
        let type_name = Self::get_type_parts();
        let type_name_string = type_name[..type_name.len() - 1]
            .iter()
            .map(|v| v.to_string())
            .collect::<Vec<_>>()
            .join("");
        match generator {
            crate::TypeGenerator::Record(record_generator) => {
                crate::TypeGenerator::Record(Box::new(RecordGenerator {
                    // we overwrite anything which is not static
                    type_name,
                    type_doc: format!("Collection of static methods for [`{}`].", type_name_string),
                    fields: Default::default(),
                    methods: Default::default(),
                    mut_methods: Default::default(),
                    meta_method: Default::default(),
                    meta_method_mut: Default::default(),
                    ..record_generator.as_ref().clone()
                }))
            }
            crate::TypeGenerator::Enum(enum_generator) => {
                crate::TypeGenerator::Enum(EnumGenerator {
                    name: Self::get_type_parts(),
                    ..enum_generator
                })
            }
        }
    }
}

impl<'lua, T: StaticUserdata> IntoLua<'lua> for UserDataProxy<'lua, T> {
    fn into_lua(self, lua: &'lua Lua) -> mlua::Result<mlua::Value<'lua>> {
        self.user_data.into_lua(lua)
    }
}