summaryrefslogtreecommitdiff
path: root/lualibs-util-jsn.lua
blob: bbe25d89df89c2df30685b93d86e0968e3272ed4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
if not modules then modules = { } end modules ['util-jsn'] = {
    version   = 1.001,
    comment   = "companion to m-json.mkiv",
    author    = "Hans Hagen, PRAGMA-ADE, Hasselt NL",
    copyright = "PRAGMA ADE / ConTeXt Development Team",
    license   = "see context related readme files"
}

-- Of course we could make a nice complete parser with proper error messages but
-- as json is generated programmatically errors are systematic and we can assume
-- a correct stream. If not, we have some fatal error anyway. So, we can just rely
-- on strings being strings (apart from the unicode escape which is not in 5.1) and
-- as we first catch known types we just assume that anything else is a number.
--
-- Reminder for me: check usage in framework and extend when needed. Also document
-- it in the cld lib documentation.

local P, V, R, S, C, Cc, Cs, Ct, Cf, Cg = lpeg.P, lpeg.V, lpeg.R, lpeg.S, lpeg.C, lpeg.Cc, lpeg.Cs, lpeg.Ct, lpeg.Cf, lpeg.Cg
local lpegmatch = lpeg.match
local format = string.format
local utfchar = utf.char
local concat = table.concat

local tonumber, tostring, rawset, type = tonumber, tostring, rawset, type

local json      = utilities.json or { }
utilities.json  = json

-- moduledata      = moduledata or { }
-- moduledata.json = json

-- \\ \/ \b \f \n \r \t \uHHHH

local lbrace     = P("{")
local rbrace     = P("}")
local lparent    = P("[")
local rparent    = P("]")
local comma      = P(",")
local colon      = P(":")
local dquote     = P('"')

local whitespace = lpeg.patterns.whitespace
local optionalws = whitespace^0

local escapes    = {
 -- ["\\"] = "\\",  -- lua will escape these
 -- ["/"]  = "/",   -- no need to escape this one
    ["b"]  = "\010",
    ["f"]  = "\014",
    ["n"]  = "\n",
    ["r"]  = "\r",
    ["t"]  = "\t",
}

local escape_un  = C(P("\\u") / "0x" * S("09","AF","af")) / function(s) return utfchar(tonumber(s)) end
local escape_bs  = P([[\]]) / "" * (P(1) / escapes) -- if not found then P(1) is returned i.e. the to be escaped char

local jstring    = dquote * Cs((escape_un + escape_bs + (1-dquote))^0) * dquote
local jtrue      = P("true")  * Cc(true)
local jfalse     = P("false") * Cc(false)
local jnull      = P("null")  * Cc(nil)
local jnumber    = (1-whitespace-rparent-rbrace-comma)^1 / tonumber

local key        = jstring

local jsonconverter = { "value",
    object   = lbrace * Cf(Ct("") * V("pair") * (comma * V("pair"))^0,rawset) * rbrace,
    pair     = Cg(optionalws * key * optionalws * colon * V("value")),
    array    = Ct(lparent * V("value") * (comma * V("value"))^0 * rparent),
    value    = optionalws * (jstring + V("object") + V("array") + jtrue + jfalse + jnull + jnumber + #rparent) * optionalws,
}

-- local jsonconverter = { "value",
--     object   = lbrace * Cf(Ct("") * V("pair") * (comma * V("pair"))^0,rawset) * rbrace,
--     pair     = Cg(optionalws * V("string") * optionalws * colon * V("value")),
--     array    = Ct(lparent * V("value") * (comma * V("value"))^0 * rparent),
--     string   = jstring,
--     value    = optionalws * (V("string") + V("object") + V("array") + jtrue + jfalse + jnull + jnumber) * optionalws,
-- }

-- lpeg.print(jsonconverter) -- size 181

function json.tolua(str)
    return lpegmatch(jsonconverter,str)
end

local function tojson(value,t) -- we could optimize #t
    local kind = type(value)
    if kind == "table" then
        local done = false
        local size = #value
        if size == 0 then
            for k, v in next, value do
                if done then
                    t[#t+1] = ","
                else
                    t[#t+1] = "{"
                    done = true
                end
                t[#t+1] = format("%q:",k)
                tojson(v,t)
            end
            if done then
                t[#t+1] = "}"
            else
                t[#t+1] = "{}"
            end
        elseif size == 1 then
            -- we can optimize for non tables
            t[#t+1] = "["
            tojson(value[1],t)
            t[#t+1] = "]"
        else
            for i=1,size do
                if done then
                    t[#t+1] = ","
                else
                    t[#t+1] = "["
                    done = true
                end
                tojson(value[i],t)
            end
            t[#t+1] = "]"
        end
    elseif kind == "string"  then
        t[#t+1] = format("%q",value)
    elseif kind == "number" then
        t[#t+1] = value
    elseif kind == "boolean" then
        t[#t+1] = tostring(value)
    end
    return t
end

function json.tostring(value)
    -- todo optimize for non table
    local kind = type(value)
    if kind == "table" then
        return concat(tojson(value,{}),"")
    elseif kind == "string" or kind == "number" then
        return value
    else
        return tostring(value)
    end
end

-- local tmp = [[ { "a" : true, "b" : [ 123 , 456E-10, { "a" : true, "b" : [ 123 , 456 ] } ] } ]]

-- tmp = json.tolua(tmp)
-- inspect(tmp)
-- tmp = json.tostring(tmp)
-- inspect(tmp)
-- tmp = json.tolua(tmp)
-- inspect(tmp)
-- tmp = json.tostring(tmp)
-- inspect(tmp)

-- inspect(json.tostring(true))