add string case conversion functions: lower and upper

3 files changed, 127 insertions, 0 deletions

diff --git a/core/src/xmake/engine.c b/core/src/xmake/engine.c
index 532be1adc..d81828f1f 100644
--- a/core/src/xmake/engine.c
+++ b/core/src/xmake/engine.c
@@ -286,6 +286,8 @@ tb_int_t xm_string_lastof(lua_State *lua);
 tb_int_t xm_string_convert(lua_State *lua);
 tb_int_t xm_string_endswith(lua_State *lua);
 tb_int_t xm_string_startswith(lua_State *lua);
+tb_int_t xm_string_lower(lua_State *lua);
+tb_int_t xm_string_upper(lua_State *lua);
 
 // the process functions
 tb_int_t xm_process_open(lua_State *lua);
@@ -589,6 +591,8 @@ static luaL_Reg const g_string_functions[] = {
     { "convert", xm_string_convert },
     { "endswith", xm_string_endswith },
     { "startswith", xm_string_startswith },
+    { "lower", xm_string_lower },
+    { "upper", xm_string_upper },
     { tb_null, tb_null },
 };
 
diff --git a/core/src/xmake/string/case.c b/core/src/xmake/string/case.c
new file mode 100644
index 000000000..4c6fb54ca
--- /dev/null
+++ b/core/src/xmake/string/case.c
@@ -0,0 +1,116 @@
+/*!A cross-platform build utility based on Lua
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ * Copyright (C) 2015-present, Xmake Open Source Community.
+ *
+ * @author      ruki
+ * @file        case.c
+ *
+ */
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * trace
+ */
+#define TB_TRACE_MODULE_NAME "case"
+#define TB_TRACE_MODULE_DEBUG (0)
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * includes
+ */
+#include "prefix.h"
+#include "../utils/charset.h"
+#include <wctype.h>
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * helper
+ */
+static tb_int_t xm_string_case(lua_State* lua, tb_bool_t lower) {
+
+    // check
+    tb_assert_and_check_return_val(lua, 0);
+
+    // get the string
+    size_t           size = 0;
+    tb_char_t const* str  = luaL_checklstring(lua, 1, &size);
+    tb_check_return_val(str, 0);
+
+    // empty?
+    if (!size) {
+        lua_pushstring(lua, "");
+        return 1;
+    }
+
+    // find charsets
+    xm_charset_entry_ref_t fcharset = xm_charset_find_by_name("utf8");
+#if defined(TB_CONFIG_OS_WINDOWS)
+    xm_charset_entry_ref_t tcharset = xm_charset_find_by_name("utf16");
+#else
+    xm_charset_entry_ref_t tcharset = xm_charset_find_by_name("utf32");
+#endif
+    tb_check_return_val(fcharset && tcharset, 0);
+
+    // convert string
+    tb_long_t  dst_size = 0;
+    tb_size_t  dst_maxn = (tb_size_t)(size + 1) * 4;
+    tb_byte_t* dst_data = tb_malloc_bytes(dst_maxn);
+    if (dst_data && dst_maxn &&
+        (dst_size = tb_charset_conv_data(fcharset->type, tcharset->type, (tb_byte_t const*)str, (tb_size_t)size, dst_data, dst_maxn)) >= 0 &&
+        dst_size < dst_maxn) {
+
+        // to lower/upper
+#if defined(TB_CONFIG_OS_WINDOWS)
+        tb_uint16_t* p = (tb_uint16_t*)dst_data;
+        tb_size_t    n = (tb_size_t)dst_size / 2;
+#else
+        tb_uint32_t* p = (tb_uint32_t*)dst_data;
+        tb_size_t    n = (tb_size_t)dst_size / 4;
+#endif
+        while (n--) {
+            *p = (tb_wchar_t)(lower? towlower((wint_t)*p) : towupper((wint_t)*p));
+            p++;
+        }
+
+        // convert string back
+        tb_long_t  res_size = 0;
+        tb_size_t  res_maxn = dst_maxn * 2;
+        tb_byte_t* res_data = tb_malloc_bytes(res_maxn);
+        if (res_data && res_maxn &&
+            (res_size = tb_charset_conv_data(tcharset->type, fcharset->type, dst_data, (tb_size_t)dst_size, res_data, res_maxn)) >= 0 &&
+            res_size < res_maxn) {
+            lua_pushlstring(lua, (tb_char_t const*)res_data, res_size);
+        } else {
+            lua_pushnil(lua);
+        }
+        if (res_data) tb_free(res_data);
+    } else {
+        lua_pushnil(lua);
+    }
+    
+    // free data
+    if (dst_data) tb_free(dst_data);
+
+    // ok
+    return 1;
+}
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * implementation
+ */
+tb_int_t xm_string_lower(lua_State* lua) {
+    return xm_string_case(lua, tb_true);
+}
+
+tb_int_t xm_string_upper(lua_State* lua) {
+    return xm_string_case(lua, tb_false);
+}
diff --git a/tests/modules/string/test.lua b/tests/modules/string/test.lua
index 3f52dc593..6bb3b6f78 100644
--- a/tests/modules/string/test.lua
+++ b/tests/modules/string/test.lua
@@ -106,3 +106,10 @@ function test_replace(t)
     t:are_equal(("123$xyz456xyz789xyz"):replace("123$", "000", {plain = true}), "000xyz456xyz789xyz")
     t:are_equal(("123xyz$456xyz$789xyz$"):replace("xyz$", "000", {plain = true}), "123000456000789000")
 end
+
+function test_case(t)
+    t:are_equal(("Hello"):lower(), "hello")
+    t:are_equal(("Hello"):upper(), "HELLO")
+    t:are_equal(("Звезда Хэнсин"):lower(), "звезда хэнсин")
+    t:are_equal(("Звезда Хэнсин"):upper(), "ЗВЕЗДА ХЭНСИН")
+end