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

use mlua::{FromLua, IntoLua, Lua, UserDataFields, UserDataMethods};
use ratatui::text::Text;

use crate::app::settings::{
    color_settings::ColorSettings, key_settings::KeySettings, settings_value::SettingsValue,
    Settings,
};

pub fn register_vec_u8(lua: &Lua) -> mlua::Result<()> {
    lua.register_userdata_type(|data: &mut mlua::UserDataRegistry<Vec<u8>>| {
        data.add_field_method_get("len", |_lua, this| Ok(this.len()));
        data.add_method("get", |_lua, this, index: usize| {
            if let Some(byte) = this.get(index) {
                Ok(*byte)
            } else {
                Err(mlua::Error::RuntimeError("Index out of bounds".to_string()))
            }
        });
        data.add_method_mut("set", |_lua, this, (index, value): (usize, u8)| {
            if let Some(byte) = this.get_mut(index) {
                let old_value = *byte;
                *byte = value;
                Ok(old_value)
            } else {
                Err(mlua::Error::RuntimeError("Index out of bounds".to_string()))
            }
        });
    })?;
    Ok(())
}

pub fn register_text(lua: &Lua) -> mlua::Result<()> {
    // TODO: maybe write a wrapper for Text that allows for easier manipulation
    lua.register_userdata_type(|data: &mut mlua::UserDataRegistry<Text>| {
        data.add_method_mut("push_line", |_lua, this, value: String| {
            this.push_line(value);
            Ok(())
        });
        data.add_method_mut("push_span", |_lua, this, value: String| {
            this.push_span(value);
            Ok(())
        });
    })?;
    Ok(())
}

pub fn register_string(lua: &Lua) -> mlua::Result<()> {
    lua.register_userdata_type(|data: &mut mlua::UserDataRegistry<String>| {
        data.add_method("get", |_lua, this, ()| Ok(this.clone()));
        data.add_method_mut("set", |_lua, this, value: String| {
            let old_value = std::mem::replace(this, value);
            Ok(old_value)
        });
    })?;
    Ok(())
}

pub fn register_usize(lua: &Lua) -> mlua::Result<()> {
    lua.register_userdata_type(|data: &mut mlua::UserDataRegistry<usize>| {
        data.add_method("get", |_lua, this, ()| Ok(*this));
        data.add_method_mut("set", |_lua, this, value: usize| {
            let old_value = std::mem::replace(this, value);
            Ok(old_value)
        });
    })?;
    Ok(())
}

pub fn register_settings(lua: &Lua) -> mlua::Result<()> {
    lua.register_userdata_type(|data: &mut mlua::UserDataRegistry<Settings>| {
        ColorSettings::register_userdata(data);
        KeySettings::register_userdata(data);
        data.add_method("get_custom", |_lua, this, key: String| {
            Ok(this.custom.get(&key).cloned())
        });
        data.add_method_mut(
            "set_custom",
            |lua, this, (key, value): (String, mlua::Value)| {
                if let Some(old_value) = this.custom.get_mut(&key) {
                    let old_value_copy = old_value.clone();
                    if value.is_nil() {
                        this.custom.remove(&key);
                    } else {
                        *old_value = SettingsValue::from_lua(value, lua)?;
                    }
                    old_value_copy.into_lua(lua)
                } else {
                    if !value.is_nil() {
                        this.custom
                            .insert(key, SettingsValue::from_lua(value, lua)?);
                    }
                    Ok(mlua::Value::Nil)
                }
            },
        );
    })?;
    Ok(())
}