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