aboutsummaryrefslogtreecommitdiff
path: root/src/lua/net/luaopen_net.cpp
blob: 89a79f3c1a69369e665e301e71c51c0e16562900 (plain)
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
#include "lua/luax.h"
#include "libjin/jin.h"
#include "../luaopen_types.h"
#include "lua_net_Buffer.h"

namespace jin
{
namespace lua
{

    using namespace jin::net;

    static int l_initNetwork(lua_State* L)
    {
        jin::net::Net::get()->init();
        return 1;
    }
    
    // jin.net.toSocket(lightuserdata)
    static int l_toSocket(lua_State*L)
    {
        Proxy* proxy = (Proxy*)luax_checktype(L, 1, JIN_NETWORK_SOCKET);
        //Proxy * socketProxy = (Proxy*)luax_newinstance(L, JIN_NETWORK_SOCKET, sizeof(Proxy));

        return 1;
    }

    // jin.net.Socket()
    static int l_Socket(lua_State* L)
    {
        const char* socketType = luax_checkstring(L, 1);
        SocketInformation info = { 0 };
        if (strcmp(socketType, "TCP") == 0)
            info.type = SocketType::TCP;
        else if (strcmp(socketType, "UDP") == 0)
            info.type = SocketType::UDP;
        else
        {
            luax_error(L, "jin.net.Socket() first paramter wrong, must be TCP or UDP");
            return 0;
        }
        // type, port
        if (luax_gettop(L) == 2)
        {
            info.port = luax_checkinteger(L, 2);
        }
        // type, address, port
        else if (luax_gettop(L) == 3)
        {
            if (luax_isstringstrict(L, 2))
                info.address = tk_strtohl(luax_checkstring(L, 2));
            else if(luax_isintegerstrict(L, 2))
                info.address = luax_checkinteger(L, 2);
            info.port = luax_checkinteger(L, 3);
        }
        Proxy* proxy = (Proxy*)luax_newinstance(L, JIN_NETWORK_SOCKET, sizeof(Proxy));
        Socket* socket = new Socket(info);
        proxy->bind(socket, JIN_NETWORK_SOCKET);
        return 1;
    }

    // jin.net.Buffer()
    static int l_Buffer(lua_State* L)
    {
        int size = luax_checkinteger(L, 1);
        Proxy* proxy = (Proxy*)luax_newinstance(L, JIN_NETWORK_BUFFER, sizeof(Proxy));
        net::Buffer* buffer = new net::Buffer(size);
        proxy->bind(buffer, JIN_NETWORK_BUFFER);
        return 1;
    }

    static const luaL_Reg f[] = {
        { "init", l_initNetwork },
        { "toSocket", l_toSocket }, 
        { "Socket", l_Socket },
        { "Buffer", l_Buffer },
        { 0, 0 }
    };

    extern int luaopen_Socket(lua_State* L);
    extern int luaopen_Buffer(lua_State* L);

    int luaopen_net(lua_State* L)
    {
        luaopen_Socket(L);
        luaopen_Buffer(L);

        luax_newlib(L, f);

        return 1;
    }

} // lua
} // jin