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 }