fix and improve os.args

4 files changed, 130 insertions, 1 deletions

diff --git a/core/src/xmake/machine.c b/core/src/xmake/machine.c
index 625f60f63..70ab72d61 100644
--- a/core/src/xmake/machine.c
+++ b/core/src/xmake/machine.c
@@ -58,6 +58,7 @@ typedef struct __xm_machine_t
 
 // the os functions
 tb_int_t xm_os_argv(lua_State* lua);
+tb_int_t xm_os_args(lua_State* lua);
 tb_int_t xm_os_find(lua_State* lua);
 tb_int_t xm_os_link(lua_State* lua);
 tb_int_t xm_os_isdir(lua_State* lua);
@@ -209,6 +210,7 @@ tb_int_t xm_curses_register(lua_State* lua);
 static luaL_Reg const g_os_functions[] = 
 {
     { "argv",           xm_os_argv      }
+,   { "args",           xm_os_args      }
 ,   { "find",           xm_os_find      }
 ,   { "link",           xm_os_link      }
 ,   { "isdir",          xm_os_isdir     }
diff --git a/core/src/xmake/makefile b/core/src/xmake/makefile
index 6def8df5b..f5fa74257 100644
--- a/core/src/xmake/makefile
+++ b/core/src/xmake/makefile
@@ -15,6 +15,7 @@ xmake_C_FILES 			+= \
 						xmake \
 						machine \
 						os/argv \
+						os/args \
 						os/find \
 						os/link \
 						os/isdir \
diff --git a/core/src/xmake/os/args.c b/core/src/xmake/os/args.c
new file mode 100644
index 000000000..419e6db54
--- /dev/null
+++ b/core/src/xmake/os/args.c
@@ -0,0 +1,126 @@
+/*!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-2020, TBOOX Open Source Group.
+ *
+ * @author      ruki
+ * @file        args.c
+ *
+ */
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * trace
+ */
+#define TB_TRACE_MODULE_NAME                "os.args"
+#define TB_TRACE_MODULE_DEBUG               (0)
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * includes
+ */
+#include "prefix.h"
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * private implementation
+ */
+static tb_void_t tb_os_args_append(tb_string_ref_t result, tb_char_t const* cstr, tb_size_t size)
+{
+    // check
+    tb_assert_and_check_return(size < TB_PATH_MAXN);
+
+    // wrap and escape characters
+    tb_char_t ch;
+    tb_size_t n = 0;
+    tb_char_t const* p = cstr;
+    tb_bool_t wrap_quote = tb_false;
+    tb_char_t buff[TB_PATH_MAXN];
+    tb_size_t m = tb_arrayn(buff);
+    while ((ch = *p) && n < m)
+    {
+        if (ch == '\"') 
+        {
+            if (n < m) buff[n++] = '\\';
+            wrap_quote = tb_true;
+        }
+        else if (ch == ' ' || ch == '(' || ch == ')') wrap_quote = tb_true;
+        if (n < m) buff[n++] = ch;
+        p++;
+    }
+    tb_assert_and_check_return(n < m);
+    buff[n] = '\0';
+
+    // wrap "" and escape '\\' if exists escape characters and spaces?
+    if (wrap_quote) 
+    {
+        tb_string_chrcat(result, '\"');
+        tb_size_t i = 0;
+        tb_char_t ch;
+        for (i = 0; i < n; i++)
+        {
+            ch = buff[i];
+            if (ch == '\\') // escape the '\\' characters in ""
+                tb_string_chrcat(result, '\\');
+            tb_string_chrcat(result, ch);
+        }
+        tb_string_chrcat(result, '\"');
+    }
+    else if (n) tb_string_cstrncat(result, buff, n);
+}
+
+/* //////////////////////////////////////////////////////////////////////////////////////
+ * implementation
+ */
+tb_int_t xm_os_args(lua_State* lua)
+{
+    // check
+    tb_assert_and_check_return_val(lua, 0);
+
+    // init result
+    tb_string_t result;
+    tb_string_init(&result);
+
+    // make string from arguments list
+    if (lua_istable(lua, 1))
+    {
+        tb_size_t i = 0;
+        tb_size_t n = lua_objlen(lua, 1);
+        for (i = 1; i <= n; i++)
+        {
+            // add space 
+            if (i != 1) tb_string_chrcat(&result, ' ');
+
+            // add argument
+            lua_pushnumber(lua, (tb_int_t)i);
+            lua_rawget(lua, 1);
+            size_t size = 0;
+            tb_char_t const* cstr = luaL_checklstring(lua, -1, &size);
+            if (cstr && size)
+                tb_os_args_append(&result, cstr, size);
+            lua_pop(lua, 1);
+        }
+    }
+    else
+    {
+        size_t size = 0;
+        tb_char_t const* cstr = luaL_checklstring(lua, 1, &size);
+        if (cstr && size)
+            tb_os_args_append(&result, cstr, size);
+    }
+
+    // return result
+    tb_size_t size = tb_string_size(&result);
+    if (size) lua_pushlstring(lua, tb_string_cstr(&result), size);
+    else lua_pushnil(lua);
+    tb_string_exit(&result);
+    return 1;
+}
diff --git a/core/src/xmake/os/argv.c b/core/src/xmake/os/argv.c
index d1c74d4d9..db7b6ced2 100644
--- a/core/src/xmake/os/argv.c
+++ b/core/src/xmake/os/argv.c
@@ -22,7 +22,7 @@
 /* //////////////////////////////////////////////////////////////////////////////////////
  * trace
  */
-#define TB_TRACE_MODULE_NAME                "argv"
+#define TB_TRACE_MODULE_NAME                "os.argv"
 #define TB_TRACE_MODULE_DEBUG               (0)
 
 /* //////////////////////////////////////////////////////////////////////////////////////