mlua-socket 0.2.7

A Rust-native implementation of LuaSocket for mlua.
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

mod bind;
mod connect;
mod gettime;
mod sleep;
mod tcp;
mod tcp4;
mod tcp6;
mod tcp_bind;
mod tcp_client;
mod tcp_close;
mod tcp_connect;
mod tcp_getfd;
mod tcp_getsockname;
mod tcp_listen;
mod tcp_receive;
mod tcp_send;
mod tcp_server;
mod tcp_setoption;
mod tcp_settimeout;
mod tcp_shutdown;
mod udp;
mod udp4;
mod udp6;
mod udp_close;
mod udp_getfd;
mod udp_receive;
mod udp_receivefrom;
mod udp_sendto;
mod udp_setoption;
mod udp_setpeername;
mod udp_setsockname;
mod udp_settimeout;

use mlua::{Error, Lua, Table, Value};

pub fn preload(lua: &Lua) -> Result<(), Error> {
    // Configure module table submodules
    let table = lua.create_table()?;
    table.set("dns", lua.load("return require('socket.dns')").eval::<Table>()?)?;
    table.set("headers", lua.load("return require('socket.headers')").eval::<Table>()?)?;
    table.set("url", lua.load("return require('socket.url')").eval::<Table>()?)?;

    // Configure module table direct functions
    table.set("bind", lua.create_function(bind::handle)?)?;
    table.set("connect", lua.create_function(connect::handle)?)?;
    table.set("gettime", lua.create_function(gettime::handle)?)?;
    table.set("sleep", lua.create_function(sleep::handle)?)?;
    table.set("tcp", lua.create_function(tcp4::handle)?)?;
    table.set("tcp4", lua.create_function(tcp4::handle)?)?;
    table.set("tcp6", lua.create_function(tcp6::handle)?)?;
    table.set("udp", lua.create_function(udp4::handle)?)?;
    table.set("udp4", lua.create_function(udp4::handle)?)?;
    table.set("udp6", lua.create_function(udp6::handle)?)?;

    // Configure module table direct values
    table.set("_SOCKETINVALID", Value::Number(-1.))?;

    // Configure module metatable
    let metatable = lua.create_table()?;
    table.set_metatable(Some(metatable))?;

    // Preload module
    let globals = lua.globals();
    let package: Table = globals.get("package")?;
    let loaded: Table = package.get("loaded")?;
    loaded.set("socket.core", table)?;

    Ok(())
}

#[cfg(test)]
mod tests {
    use mlua::{Lua, Table};
    use std::error::Error;

    #[test]
    fn load() -> Result<(), Box<dyn Error>> {
        let lua = Lua::new();
        crate::preload(&lua)?;
        let module: Table = lua.load("return require('socket.core')").eval()?;
        assert!(module.contains_key("bind")?);
        Ok(())
    }
}