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