From 36621e6be9604517c900adc0d97665e975c2b325 Mon Sep 17 00:00:00 2001
From: chai <chaifix@163.com>
Date: Wed, 22 Aug 2018 13:58:13 +0800
Subject: *update

---
 src/lua/modules/net/net.cpp | 85 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 src/lua/modules/net/net.cpp

(limited to 'src/lua/modules/net/net.cpp')

diff --git a/src/lua/modules/net/net.cpp b/src/lua/modules/net/net.cpp
new file mode 100644
index 0000000..a984920
--- /dev/null
+++ b/src/lua/modules/net/net.cpp
@@ -0,0 +1,85 @@
+#include "lua/modules/luax.h"
+#include "lua/modules/types.h"
+#include "libjin/jin.h"
+#include "lua/common/common.h"
+#include "Buffer.h"
+
+namespace jin
+{
+namespace lua
+{
+    using namespace jin::lua::net;
+	using namespace jin::net;
+
+    static int l_initNetwork(lua_State* L)
+    {
+        jin::net::Net::get()->init();
+        return 1;
+    }
+    
+    static int l_Socket(lua_State* L)
+    {
+        SocketInformation info = { 0 };
+        {
+            const char* socketType = luax_checkstring(L, 1);
+            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);
+            }
+        }
+        Socket* socket = new Socket(info);
+        Proxy* proxy = (Proxy*)luax_newinstance(L, JIN_NETWORK_SOCKET, sizeof(Proxy));
+        proxy->bind(new Ref<Socket>(socket, JIN_NETWORK_SOCKET));
+        return 1;
+    }
+
+    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(new Ref<Buffer>(buffer, JIN_NETWORK_BUFFER));
+        return 1;
+    }
+
+    static const luaL_Reg f[] = {
+        { "init",      l_initNetwork },
+        { "newSocket", l_Socket      },
+        { "newBuffer", 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
\ No newline at end of file
-- 
cgit v1.1-26-g67d0