jshn: support using characters in elements that do not conform to shell variable...

diff --git a/jshn.c b/jshn.c
index 88e31d1cd0a2243ae6c145b556b7b99119b53366..f7a46dee85b0cbc5450fc0d7838c07035a20d41b 100644 (file)
--- a/jshn.c
+++ b/jshn.c
@@ -150,14 +150,19 @@ static char *get_keys(const char *prefix)
        return getenv(keys);
 }
 
-static void get_var(const char *prefix, const char *name, char **var, char **type)
+static void get_var(const char *prefix, const char **name, char **var, char **type)
 {
-       char *tmpname;
+       char *tmpname, *varname;
 
-       tmpname = alloca(strlen(prefix) + 1 + strlen(name) + 1 + sizeof("TYPE_"));
-       sprintf(tmpname, "TYPE_%s_%s", prefix, name);
+       tmpname = alloca(strlen(prefix) + 1 + strlen(*name) + 1 + sizeof("TYPE_"));
+       sprintf(tmpname, "TYPE_%s_%s", prefix, *name);
        *var = getenv(tmpname + 5);
        *type = getenv(tmpname);
+
+       memcpy(tmpname, "NAME", 4);
+       varname = getenv(tmpname);
+       if (varname)
+               *name = varname;
 }
 
 static json_object *jshn_add_objects(json_object *obj, const char *prefix, bool array);
@@ -167,7 +172,7 @@ static void jshn_add_object_var(json_object *obj, bool array, const char *prefix
        json_object *new;
        char *var, *type;
 
-       get_var(prefix, name, &var, &type);
+       get_var(prefix, &name, &var, &type);
        if (!var || !type)
                return;
 

diff --git a/sh/jshn.sh b/sh/jshn.sh
index 7389f8c2a26ab743b7fb1aeb24f2c27be6138a6b..8ecefcf03f932fadb5a699b2bd9f3bd29664ddca 100644 (file)
--- a/sh/jshn.sh
+++ b/sh/jshn.sh
@@ -19,11 +19,15 @@ json_add_generic() {
        local val="$3"
        local cur="${4:-$JSON_CUR}"
 
-       [ "${cur%%[0-9]*}" = "JSON_ARRAY" ] && {
+       if [ "${cur%%[0-9]*}" = "JSON_ARRAY" ]; then
                eval "local aseq=\"\${SEQ_$cur}\""
                var=$(( ${aseq:-0} + 1 ))
                export -- "SEQ_$cur=$var"
-       }
+       else
+               local name="$(echo -n "$var" | tr -C '[a-zA-Z_]' _)"
+               [[ "$name" == "$var" ]] || export -- "NAME_${cur}_${name}=$var"
+               var="$name"
+       fi
 
        export -- "${cur}_$var=$val"
        export -- "TYPE_${cur}_$var=$type"