1 files changed, 110 insertions, 0 deletions

diff --git a/source/libs/asura-lib-utils/io/binding/_data_buffer.cpp b/source/libs/asura-lib-utils/io/binding/_data_buffer.cpp
new file mode 100644
index 0000000..d9eca36
--- /dev/null
+++ b/source/libs/asura-lib-utils/io/binding/_data_buffer.cpp
@@ -0,0 +1,110 @@
+#include "../data_buffer.h"
+
+using namespace Luax;
+
+namespace AsuraEngine
+{
+	namespace IO
+	{
+
+		LUAX_REGISTRY(DataBuffer)
+		{
+			LUAX_REGISTER_METHODS(state, 
+				{ "New",       _New     },
+				{ "GetBuffer", _GetData },
+				{ "GetSize",   _GetSize },
+				{ "Load",      _Load    },
+				{ "Clear",     _Clear   }
+			);
+		}
+
+		LUAX_POSTPROCESS(DataBuffer)
+		{
+		}
+
+		// databuffer = DataBuffer.New(lstring)
+		// databuffer = DataBuffer.New(size)
+		LUAX_IMPL_METHOD(DataBuffer, _New)
+		{
+			LUAX_STATE(L);
+
+			if (state.IsType(1, LUA_TSTRING))
+			{
+				byte* bytes;
+				size_t size;
+				lua_tolstring(L, 1, &size);
+				DataBuffer* buffer = new DataBuffer(bytes, size);
+				buffer->PushLuaxUserdata(state);
+				return 1;
+			}
+			else if (state.IsType(1, LUA_TNUMBER))
+			{
+				size_t size = lua_tonumber(L, 1);
+				DataBuffer* buffer = new DataBuffer(size);
+				buffer->PushLuaxUserdata(state);
+			}
+			else
+			{
+				return state.ErrorType(1, "number or string");
+			}
+		}
+
+		// lsting, len = databuffer:GetBuffer()
+		LUAX_IMPL_METHOD(DataBuffer, _GetData)
+		{
+			LUAX_SETUP(L, "U");
+
+			DataBuffer* self = state.GetUserdata<DataBuffer>(1);
+			lua_pushlstring(L, self->GetData(), self->GetSize());
+			return 2;
+		}
+
+		// length = databuffer:GetSize()
+		LUAX_IMPL_METHOD(DataBuffer, _GetSize)
+		{
+			LUAX_SETUP(L, "U");
+
+			DataBuffer* self = state.GetUserdata<DataBuffer>(1);
+			lua_pushinteger(L, self->GetSize());
+			return 1;
+		}
+
+		// databuffer:Load(lstring)
+		// databuffer:Load(src)
+		LUAX_IMPL_METHOD(DataBuffer, _Load)
+		{
+			LUAX_STATE(L);
+
+			DataBuffer* buffer = state.GetUserdata<DataBuffer>(1); 
+			const byte* data;
+			size_t size;
+			if (state.IsType(2, LUA_TSTRING))
+			{
+				data = lua_tolstring(L, 2, &size);
+				buffer->Load(data, size);
+				return 0;
+			}
+			else if(state.IsType(2, LUA_TUSERDATA))
+			{
+				DataBuffer* src = state.CheckUserdata<DataBuffer>(2);
+				buffer->Load(*src);
+				return 0;
+			}
+			else
+			{
+				return state.ErrorType(1, "lstring or DataBuffer");
+			}
+		}
+
+		// databuffer:Clear()
+		LUAX_IMPL_METHOD(DataBuffer, _Clear)
+		{
+			LUAX_SETUP(L, "U");
+
+			DataBuffer* self = state.GetUserdata<DataBuffer>(1);
+			self->Clear(); 
+			return 0;
+		}
+
+	}
+}
\ No newline at end of file