fbffe85dd7ea9db2b60d699b4b9bb6609c10a060
[project/libubox.git] / sh / jshn.sh
1 # functions for parsing and generating json
2
3 _json_get_var() {
4         # dest=$1
5         # var=$2
6         eval "$1=\"\$${JSON_PREFIX}$2\""
7 }
8
9 _json_set_var() {
10         # var=$1
11         local ___val="$2"
12         eval "${JSON_PREFIX}$1=\"\$___val\""
13 }
14
15 __jshn_raw_append() {
16         # var=$1
17         local value="$2"
18         local sep="${3:- }"
19
20         eval "export -- \"$1=\${$1:+\${$1}\${value:+\$sep}}\$value\""
21 }
22
23 _jshn_append() {
24         # var=$1
25         local _a_value="$2"
26         eval "${JSON_PREFIX}$1=\"\${${JSON_PREFIX}$1} \$_a_value\""
27 }
28
29 _get_var() {
30         # var=$1
31         # value=$2
32         eval "$1=\"\$$2\""
33 }
34
35 _set_var() {
36         # var=$1
37         local __val="$2"
38         eval "$1=\"\$__val\""
39 }
40
41 _json_inc() {
42         # var=$1
43         # dest=$2
44
45         let "${JSON_PREFIX}$1 += 1" "$2 = ${JSON_PREFIX}$1"
46 }
47
48 _json_add_generic() {
49         # type=$1
50         # name=$2
51         # value=$3
52         # cur=$4
53
54         local var
55         if [ "${4%%[0-9]*}" = "JSON_ARRAY" ]; then
56                 _json_inc "SEQ_$4" var
57         else
58                 local name="${2//[^a-zA-Z0-9_]/_}"
59                 [[ "$name" == "$2" ]] || export -- "${JSON_PREFIX}NAME_${4}_${name}=$2"
60                 var="$name"
61         fi
62
63         local cur_var=
64         export -- \
65                 "${JSON_PREFIX}${4}_$var=$3" \
66                 "${JSON_PREFIX}TYPE_${4}_$var=$1"
67         _jshn_append "JSON_UNSET" "${4}_$var"
68         _jshn_append "KEYS_$4" "$var"
69 }
70
71 _json_add_table() {
72         # name=$1
73         # type=$2
74         # itype=$3
75         local cur seq
76
77         _json_get_var cur JSON_CUR
78         _json_inc JSON_SEQ seq
79
80         local table="JSON_$3$seq"
81         _json_set_var "UP_$table" "$cur"
82         export -- "${JSON_PREFIX}KEYS_$table="
83         unset "${JSON_PREFIX}SEQ_$table"
84         _json_set_var JSON_CUR "$table"
85         _jshn_append "JSON_UNSET" "$table"
86
87         _json_add_generic "$2" "$1" "$table" "$cur"
88 }
89
90 _json_close_table() {
91         local _s_cur
92
93         _json_get_var _s_cur JSON_CUR
94         _json_get_var "${JSON_PREFIX}JSON_CUR" "UP_$_s_cur"
95         unset "${JSON_PREFIX}UP_$_s_cur"
96 }
97
98 json_set_namespace() {
99         local _new="$1"
100         local _old="$2"
101
102         [ -n "$_old" ] && _set_var "$_old" "$JSON_PREFIX"
103         JSON_PREFIX="$_new"
104 }
105
106 json_cleanup() {
107         local unset tmp
108
109         _json_get_var unset JSON_UNSET
110         for tmp in $unset JSON_VAR; do
111                 unset \
112                         ${JSON_PREFIX}UP_$tmp \
113                         ${JSON_PREFIX}KEYS_$tmp \
114                         ${JSON_PREFIX}SEQ_$tmp \
115                         ${JSON_PREFIX}TYPE_$tmp \
116                         ${JSON_PREFIX}NAME_$tmp \
117                         ${JSON_PREFIX}$tmp
118         done
119
120         unset \
121                 ${JSON_PREFIX}JSON_SEQ \
122                 ${JSON_PREFIX}JSON_CUR \
123                 ${JSON_PREFIX}JSON_UNSET
124 }
125
126 json_init() {
127         json_cleanup
128         export -n ${JSON_PREFIX}JSON_SEQ=0
129         export -- \
130                 ${JSON_PREFIX}JSON_CUR="JSON_VAR" \
131                 ${JSON_PREFIX}KEYS_JSON_VAR= \
132                 ${JSON_PREFIX}TYPE_JSON_VAR=
133 }
134
135 json_add_object() {
136         _json_add_table "$1" object TABLE
137 }
138
139 json_close_object() {
140         _json_close_table
141 }
142
143 json_add_array() {
144         _json_add_table "$1" array ARRAY 
145 }
146
147 json_close_array() {
148         _json_close_table
149 }
150
151 json_add_string() {
152         local cur
153         _json_get_var cur JSON_CUR
154         _json_add_generic string "$1" "$2" "$cur"
155 }
156
157 json_add_int() {
158         local cur
159         _json_get_var cur JSON_CUR
160         _json_add_generic int "$1" "$2" "$cur"
161 }
162
163 json_add_boolean() {
164         local cur
165         _json_get_var cur JSON_CUR
166         _json_add_generic boolean "$1" "$2" "$cur"
167 }
168
169 json_add_double() {
170         local cur
171         _json_get_var cur JSON_CUR
172         _json_add_generic double "$1" "$2" "$cur"
173 }
174
175 # functions read access to json variables
176
177 json_load() {
178         eval "`jshn -r "$1"`"
179 }
180
181 json_dump() {
182         jshn "$@" ${JSON_PREFIX:+-p "$JSON_PREFIX"} -w 
183 }
184
185 json_get_type() {
186         local __dest="$1"
187         local __cur
188
189         _json_get_var __cur JSON_CUR
190         local __var="${JSON_PREFIX}TYPE_${__cur}_${2//[^a-zA-Z0-9_]/_}"
191         eval "export -- \"$__dest=\${$__var}\"; [ -n \"\${$__var+x}\" ]"
192 }
193
194 json_get_keys() {
195         local __dest="$1"
196         local _tbl_cur
197
198         if [ -n "$2" ]; then
199                 json_get_var _tbl_cur "$2"
200         else
201                 _json_get_var _tbl_cur JSON_CUR
202         fi
203         local __var="${JSON_PREFIX}KEYS_${_tbl_cur}"
204         eval "export -- \"$__dest=\${$__var}\"; [ -n \"\${$__var+x}\" ]"
205 }
206
207 json_get_values() {
208         local _v_dest="$1"
209         local _v_keys _v_val _select=
210         local _json_no_warning=1
211
212         unset "$_v_dest"
213         [ -n "$2" ] && {
214                 json_select "$2" || return 1
215                 _select=1
216         }
217
218         json_get_keys _v_keys
219         set -- $_v_keys
220         while [ "$#" -gt 0 ]; do
221                 json_get_var _v_val "$1"
222                 __jshn_raw_append "$_v_dest" "$_v_val"
223                 shift
224         done
225         [ -n "$_select" ] && json_select ..
226
227         return 0
228 }
229
230 json_get_var() {
231         local __dest="$1"
232         local __cur
233
234         _json_get_var __cur JSON_CUR
235         local __var="${JSON_PREFIX}${__cur}_${2//[^a-zA-Z0-9_]/_}"
236         eval "export -- \"$__dest=\${$__var:-$3}\"; [ -n \"\${$__var+x}\${3+x}\" ]"
237 }
238
239 json_get_vars() {
240         while [ "$#" -gt 0 ]; do
241                 local _var="$1"; shift
242                 if [ "$_var" != "${_var#*:}" ]; then
243                         json_get_var "${_var%%:*}" "${_var%%:*}" "${_var#*:}"
244                 else
245                         json_get_var "$_var" "$_var"
246                 fi
247         done
248 }
249
250 json_select() {
251         local target="$1"
252         local type
253         local cur
254
255         [ -z "$1" ] && {
256                 _json_set_var JSON_CUR "JSON_VAR"
257                 return 0
258         }
259         [[ "$1" == ".." ]] && {
260                 _json_get_var cur JSON_CUR
261                 _json_get_var cur "UP_$cur"
262                 unset "${JSON_PREFIX}UP_$cur"
263                 _json_set_var JSON_CUR "$cur"
264                 return 0
265         }
266         json_get_type type "$target"
267         case "$type" in
268                 object|array)
269                         json_get_var cur "$target"
270                         _json_get_var "${JSON_PREFIX}UP_$cur" JSON_CUR
271                         _json_set_var JSON_CUR "$cur"
272                 ;;
273                 *)
274                         [ -n "$_json_no_warning" ] || \
275                                 echo "WARNING: Variable '$target' does not exist or is not an array/object"
276                         return 1
277                 ;;
278         esac
279 }
280
281 json_is_a() {
282         local type
283
284         json_get_type type "$1"
285         [ "$type" = "$2" ]
286 }