sh/jshn.sh

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