From 52e62bbf9573d23729e61c307a0fabc2a549d828 Mon Sep 17 00:00:00 2001
From: MihailRis <mexaruc@gmail.com>
Date: Fri, 25 Oct 2024 17:22:54 +0300
Subject: [PATCH] add utf8.encode

---
 src/logic/scripting/lua/libs/api_lua.hpp |  1 +
 src/logic/scripting/lua/libs/libutf8.cpp | 28 ++++++++++++++++++++++++
 src/logic/scripting/lua/lua_engine.cpp   |  1 +
 3 files changed, 30 insertions(+)
 create mode 100644 src/logic/scripting/lua/libs/libutf8.cpp

diff --git a/src/logic/scripting/lua/libs/api_lua.hpp b/src/logic/scripting/lua/libs/api_lua.hpp
index 2ac9c1e8..f9c8b3f8 100644
--- a/src/logic/scripting/lua/libs/api_lua.hpp
+++ b/src/logic/scripting/lua/libs/api_lua.hpp
@@ -35,6 +35,7 @@ extern const luaL_Reg playerlib[];
 extern const luaL_Reg quatlib[];  // quat.cpp
 extern const luaL_Reg timelib[];
 extern const luaL_Reg tomllib[];
+extern const luaL_Reg utf8lib[];
 extern const luaL_Reg vec2lib[];  // vecn.cpp
 extern const luaL_Reg vec3lib[];  // vecn.cpp
 extern const luaL_Reg vec4lib[];  // vecn.cpp
diff --git a/src/logic/scripting/lua/libs/libutf8.cpp b/src/logic/scripting/lua/libs/libutf8.cpp
new file mode 100644
index 00000000..792f2513
--- /dev/null
+++ b/src/logic/scripting/lua/libs/libutf8.cpp
@@ -0,0 +1,28 @@
+#include "api_lua.hpp"
+
+#include <vector>
+
+#include "../lua_custom_types.hpp"
+#include "util/stringutil.hpp"
+
+int l_encode(lua::State* L) {
+    std::string string = lua::require_string(L, 1);
+    if (lua::toboolean(L, 2)) {
+        lua::createtable(L, string.length(), 0);
+        for (size_t i = 0; i < string.length(); i++) {
+            lua::pushinteger(L, string[i]);
+            lua::rawseti(L, i+1);
+        }
+    } else {
+        lua::newuserdata<lua::LuaBytearray>(L, string.length());
+        auto bytearray = lua::touserdata<lua::LuaBytearray>(L, -1);   
+        bytearray->data().reserve(string.length());
+        std::memcpy(bytearray->data().data(), string.data(), string.length());
+    }
+    return 1;
+}
+
+const luaL_Reg utf8lib[] = {
+    {"encode", lua::wrap<l_encode>},
+    {NULL, NULL}
+};
diff --git a/src/logic/scripting/lua/lua_engine.cpp b/src/logic/scripting/lua/lua_engine.cpp
index fd40e612..fe3e75b5 100644
--- a/src/logic/scripting/lua/lua_engine.cpp
+++ b/src/logic/scripting/lua/lua_engine.cpp
@@ -51,6 +51,7 @@ static void create_libs(State* L, StateType stateType) {
     openlib(L, "quat", quatlib);
     openlib(L, "time", timelib);
     openlib(L, "toml", tomllib);
+    openlib(L, "utf8", utf8lib);
     openlib(L, "vec2", vec2lib);
     openlib(L, "vec3", vec3lib);
     openlib(L, "vec4", vec4lib);