improve json

1 files changed, 126 insertions, 4 deletions

diff --git a/xmake/core/base/json.lua b/xmake/core/base/json.lua
index 26dbce173..ef17bf822 100644
--- a/xmake/core/base/json.lua
+++ b/xmake/core/base/json.lua
@@ -19,11 +19,133 @@
 --
 
 -- define module: json
-local json = json or {}
+local json  = json or {}
+local cjson = cjson or {}
 
--- hide some cjson members
-json._VERSION = nil
-json._NAME    = nil
+-- laod modules
+local io    = require("base/io")
+local utils = require("base/utils")
+
+-- export null
+json.null = cjson.null
+
+-- decode json string to the lua table
+--
+-- @param jsonstr       the json string
+-- @param opt           the options
+--                      - max_depth
+--                      - invalid_numbers
+--
+-- @return              the lua table
+--
+function json.decode(jsonstr, opt)
+    local cjsonobj = cjson
+    if opt then
+        if opt.max_depth then
+            if cjsonobj == cjson then
+                cjsonobj = cjson.new()
+            end
+            cjsonobj.decode_max_depth(opt.max_depth)
+        end
+        if opt.invalid_numbers then
+            if cjsonobj == cjson then
+                cjsonobj = cjson.new()
+            end
+            cjsonobj.decode_invalid_numbers(opt.invalid_numbers)
+        end
+    end
+    local ok, luatable_or_errors = utils.trycall(cjsonobj.decode, nil, jsonstr)
+    if not ok then
+        return nil, string.format("decode json failed, %s", luatable_or_errors)
+    end
+    return luatable_or_errors
+end
+
+-- encode lua table to the json string
+--
+-- @param luatable      the lua table
+-- @param opt           the options
+--                      - max_depth
+--                      - invalid_numbers
+--                      - keep_buffer
+--                      - number_precision
+--
+-- @return              the json string
+--
+function json.encode(luatable, opt)
+    local cjsonobj = cjson
+    if opt then
+        if opt.max_depth then
+            if cjsonobj == cjson then
+                cjsonobj = cjson.new()
+            end
+            cjsonobj.encode_max_depth(opt.max_depth)
+        end
+        if opt.invalid_numbers then
+            if cjsonobj == cjson then
+                cjsonobj = cjson.new()
+            end
+            cjsonobj.encode_invalid_numbers(opt.invalid_numbers)
+        end
+        if opt.keep_buffer then
+            if cjsonobj == cjson then
+                cjsonobj = cjson.new()
+            end
+            cjsonobj.encode_keep_buffer(opt.keep_buffer)
+        end
+        if opt.number_precision then
+            if cjsonobj == cjson then
+                cjsonobj = cjson.new()
+            end
+            cjsonobj.encode_number_precision(opt.number_precision)
+        end
+    end
+    local ok, jsonstr_or_errors = utils.trycall(cjsonobj.encode, nil, luatable)
+    if not ok then
+        return nil, string.format("encode json failed, %s", jsonstr_or_errors)
+    end
+    return jsonstr_or_errors
+end
+
+-- load json file to the lua table
+--
+-- @param filepath      the json file path
+-- @param opt           the options
+--                      - encoding for io/file, e.g. utf8, utf16, utf16le, utf16be ..
+--                      - continuation for io/read (concat string with the given continuation characters)
+--                      - max_depth
+--                      - invalid_numbers
+--
+-- @return              the lua table
+--
+function json.loadfile(filepath, opt)
+    local filedata, errors = io.readfile(filepath, opt)
+    if not filedata then
+        return nil, errors
+    end
+    return json.decode(filedata, opt)
+end
+
+-- save lua table to the json file 
+--
+-- @param filepath      the json file path
+-- @param luatable      the lua table
+-- @param opt           the options
+--                      - encoding for io/file, e.g. utf8, utf16, utf16le, utf16be ..
+--                      - max_depth
+--                      - invalid_numbers
+--                      - keep_buffer
+--                      - number_precision
+--
+-- @return              the json string
+--
+function json.savefile(filepath, luatable, opt)
+    local jsonstr, errors = json.encode(luatable, opt)
+    if not jsonstr then
+        return false, errors
+    end
+    return io.writefile(filepath, jsonstr, opt)
+end
 
 -- return module: json
 return json