add jshn (JSON SHell Notation), a small utility and shell library for parsing and generating json data
SVN-Revision: 25547
This commit is contained in:
Felix Fietkau
parent
f1ac8cf285
commit
9d4d2587d7
4 changed files with 405 additions and 0 deletions
32
package/jshn/Makefile
Normal file
32
package/jshn/Makefile
Normal file
|
|
@ -0,0 +1,32 @@
|
|||
include $(TOPDIR)/rules.mk
|
||||
|
||||
PKG_NAME:=jshn
|
||||
PKG_VERSION:=0.1
|
||||
PKG_RELEASE=1
|
||||
|
||||
include $(INCLUDE_DIR)/package.mk
|
||||
|
||||
define Package/jshn
|
||||
SECTION:=utils
|
||||
CATEGORY:=Utilities
|
||||
DEPENDS:=+libjson
|
||||
TITLE:=JSON SHell Notation
|
||||
endef
|
||||
|
||||
define Package/jshn/description
|
||||
Library for parsing and generating JSON from shell scripts
|
||||
endef
|
||||
|
||||
TARGET_CFLAGS += -I$(STAGING_DIR)/usr/include --std=gnu99
|
||||
|
||||
define Build/Compile
|
||||
$(TARGET_CC) $(TARGET_CFLAGS) -o $(PKG_BUILD_DIR)/jshn src/jshn.c $(TARGET_LDFLAGS) -ljson
|
||||
endef
|
||||
|
||||
define Package/jshn/install
|
||||
$(INSTALL_DIR) $(1)/bin $(1)/lib/functions
|
||||
$(INSTALL_BIN) $(PKG_BUILD_DIR)/jshn $(1)/bin
|
||||
$(INSTALL_DATA) ./files/jshn.sh $(1)/lib/functions
|
||||
endef
|
||||
|
||||
$(eval $(call BuildPackage,jshn))
|
||||
19
package/jshn/example.txt
Normal file
19
package/jshn/example.txt
Normal file
|
|
@ -0,0 +1,19 @@
|
|||
. /lib/functions/jshn.sh
|
||||
|
||||
# generating json data
|
||||
json_init
|
||||
json_add_string "msg" "Hello, world!"
|
||||
json_add_object "test"
|
||||
json_add_int "testdata" "1"
|
||||
json_close_object
|
||||
MSG=`json_dump`
|
||||
# MSG now contains: { "msg": "Hello, world!", "test": { "testdata": 1 } }
|
||||
|
||||
|
||||
# parsing json data
|
||||
json_load "$MSG"
|
||||
json_select test
|
||||
json_get_var var1 testdata
|
||||
json_select ..
|
||||
json_get_var var2 msg
|
||||
echo "msg: $var2 - testdata: $var1"
|
||||
109
package/jshn/files/jshn.sh
Normal file
109
package/jshn/files/jshn.sh
Normal file
|
|
@ -0,0 +1,109 @@
|
|||
[ -z "$N" ] && . /etc/functions.sh
|
||||
|
||||
# functions for parsing and generating json
|
||||
|
||||
json_init() {
|
||||
[ -n "$JSON_UNSET" ] && eval "unset $JSON_UNSET"
|
||||
export -- JSON_SEQ=0 JSON_STACK= JSON_CUR="JSON_VAR" JSON_UNSET=
|
||||
}
|
||||
|
||||
json_add_generic() {
|
||||
local type="$1"
|
||||
local var="$2"
|
||||
local val="$3"
|
||||
local cur="${4:-$JSON_CUR}"
|
||||
|
||||
export ${NO_EXPORT:+-n} -- "${cur}_$var=$val"
|
||||
export ${NO_EXPORT:+-n} -- "TYPE_${cur}_$var=$type"
|
||||
append JSON_UNSET "${cur}_$var TYPE_${cur}_$var"
|
||||
append "KEYS_${cur}" "$var"
|
||||
}
|
||||
|
||||
json_add_table() {
|
||||
JSON_SEQ=$(($JSON_SEQ + 1))
|
||||
append JSON_STACK "$JSON_CUR"
|
||||
local table="JSON_TABLE$JSON_SEQ"
|
||||
export ${NO_EXPORT:+-n} -- "UP_$table=$JSON_CUR"
|
||||
JSON_CUR="$table"
|
||||
}
|
||||
|
||||
json_add_object() {
|
||||
local cur="$JSON_CUR"
|
||||
json_add_table
|
||||
json_add_generic object "$1" "$JSON_CUR" "$cur"
|
||||
}
|
||||
|
||||
json_close_object() {
|
||||
local oldstack="$JSON_STACK"
|
||||
export "KEYS_${JSON_CUR}"
|
||||
JSON_CUR="${JSON_STACK##* }"
|
||||
JSON_STACK="${JSON_STACK% *}"
|
||||
[[ "$oldstack" == "$JSON_STACK" ]] && JSON_STACK=
|
||||
}
|
||||
|
||||
json_add_array() {
|
||||
local cur="$JSON_CUR"
|
||||
json_add_table
|
||||
json_add_generic array "$1" "$JSON_CUR" "$cur"
|
||||
}
|
||||
|
||||
json_close_array() {
|
||||
json_close_object
|
||||
}
|
||||
|
||||
json_add_string() {
|
||||
json_add_generic string "$1" "$2"
|
||||
}
|
||||
|
||||
json_add_int() {
|
||||
json_add_generic int "$1" "$2"
|
||||
}
|
||||
|
||||
json_add_boolean() {
|
||||
json_add_generic boolean "$1" "$2"
|
||||
}
|
||||
|
||||
# functions read access to json variables
|
||||
|
||||
json_load() {
|
||||
eval `jshn -r "$1"`
|
||||
}
|
||||
|
||||
json_dump() {
|
||||
jshn -w
|
||||
}
|
||||
|
||||
json_get_type() {
|
||||
local dest="$1"
|
||||
local var="$2"
|
||||
eval "export ${NO_EXPORT:+-n} -- \"$dest=\${TYPE_${JSON_CUR}_$var}\""
|
||||
}
|
||||
|
||||
json_get_var() {
|
||||
local dest="$1"
|
||||
local var="$2"
|
||||
eval "export ${NO_EXPORT:+-n} -- \"$dest=\${${JSON_CUR}_$var}\""
|
||||
}
|
||||
|
||||
json_select() {
|
||||
local target="$1"
|
||||
local type
|
||||
|
||||
[ -z "$1" ] && {
|
||||
JSON_CUR="JSON_VAR"
|
||||
return
|
||||
}
|
||||
[[ "$1" == ".." ]] && {
|
||||
eval "JSON_CUR=\"\${UP_$JSON_CUR}\""
|
||||
return;
|
||||
}
|
||||
json_get_type type "$target"
|
||||
case "$type" in
|
||||
object|array)
|
||||
json_get_var JSON_CUR "$target"
|
||||
;;
|
||||
*)
|
||||
echo "WARNING: Variable '$target' does not exist or is not an array/object"
|
||||
;;
|
||||
esac
|
||||
}
|
||||
245
package/jshn/src/jshn.c
Normal file
245
package/jshn/src/jshn.c
Normal file
|
|
@ -0,0 +1,245 @@
|
|||
#include <json/json.h>
|
||||
#include <libubox/list.h>
|
||||
#include <string.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdio.h>
|
||||
#include <stdbool.h>
|
||||
#include <ctype.h>
|
||||
#include <getopt.h>
|
||||
|
||||
#define MAX_VARLEN 256
|
||||
|
||||
static int add_json_element(const char *key, json_object *obj);
|
||||
|
||||
static int add_json_object(json_object *obj)
|
||||
{
|
||||
int ret = 0;
|
||||
|
||||
json_object_object_foreach(obj, key, val) {
|
||||
ret = add_json_element(key, val);
|
||||
if (ret)
|
||||
break;
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
static int add_json_array(struct array_list *a)
|
||||
{
|
||||
char seq[12];
|
||||
int i, len;
|
||||
int ret;
|
||||
|
||||
for (i = 0, len = array_list_length(a); i < len; i++) {
|
||||
sprintf(seq, "%d", i);
|
||||
ret = add_json_element(seq, array_list_get_idx(a, i));
|
||||
if (ret)
|
||||
return ret;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void add_json_string(const char *str)
|
||||
{
|
||||
char *ptr = (char *) str;
|
||||
int len;
|
||||
char *c;
|
||||
|
||||
while ((c = strchr(ptr, '\'')) != NULL) {
|
||||
len = c - ptr;
|
||||
if (len > 0)
|
||||
fwrite(ptr, len, 1, stdout);
|
||||
ptr = c + 1;
|
||||
c = "'\\''";
|
||||
fwrite(c, strlen(c), 1, stdout);
|
||||
}
|
||||
len = strlen(ptr);
|
||||
if (len > 0)
|
||||
fwrite(ptr, len, 1, stdout);
|
||||
}
|
||||
|
||||
static void write_key_string(const char *key)
|
||||
{
|
||||
while (*key) {
|
||||
putc(isalnum(*key) ? *key : '_', stdout);
|
||||
key++;
|
||||
}
|
||||
}
|
||||
|
||||
static int add_json_element(const char *key, json_object *obj)
|
||||
{
|
||||
char *type;
|
||||
|
||||
if (!obj)
|
||||
return -1;
|
||||
|
||||
switch (json_object_get_type(obj)) {
|
||||
case json_type_object:
|
||||
type = "object";
|
||||
break;
|
||||
case json_type_array:
|
||||
type = "array";
|
||||
break;
|
||||
case json_type_string:
|
||||
type = "string";
|
||||
break;
|
||||
case json_type_boolean:
|
||||
type = "boolean";
|
||||
break;
|
||||
case json_type_int:
|
||||
type = "int";
|
||||
break;
|
||||
default:
|
||||
return -1;
|
||||
}
|
||||
|
||||
fprintf(stdout, "json_add_%s '", type);
|
||||
write_key_string(key);
|
||||
|
||||
switch (json_object_get_type(obj)) {
|
||||
case json_type_object:
|
||||
fprintf(stdout, "';\n");
|
||||
add_json_object(obj);
|
||||
fprintf(stdout, "json_close_object;\n");
|
||||
break;
|
||||
case json_type_array:
|
||||
fprintf(stdout, "';\n");
|
||||
add_json_array(json_object_get_array(obj));
|
||||
fprintf(stdout, "json_close_array;\n");
|
||||
break;
|
||||
case json_type_string:
|
||||
fprintf(stdout, "' '");
|
||||
add_json_string(json_object_get_string(obj));
|
||||
fprintf(stdout, "';\n");
|
||||
break;
|
||||
case json_type_boolean:
|
||||
fprintf(stdout, "' %d;\n", json_object_get_boolean(obj));
|
||||
break;
|
||||
case json_type_int:
|
||||
fprintf(stdout, "' %d;\n", json_object_get_int(obj));
|
||||
break;
|
||||
default:
|
||||
return -1;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int jshn_parse(const char *str)
|
||||
{
|
||||
json_object *obj;
|
||||
|
||||
obj = json_tokener_parse(str);
|
||||
if (is_error(obj) || json_object_get_type(obj) != json_type_object) {
|
||||
fprintf(stderr, "Failed to parse message data\n");
|
||||
return 1;
|
||||
}
|
||||
fprintf(stdout, "json_init;\n");
|
||||
add_json_object(obj);
|
||||
fflush(stdout);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static char *get_keys(const char *prefix)
|
||||
{
|
||||
char *keys;
|
||||
|
||||
keys = alloca(strlen(prefix) + sizeof("KEYS_") + 1);
|
||||
sprintf(keys, "KEYS_%s", prefix);
|
||||
return getenv(keys);
|
||||
}
|
||||
|
||||
static void get_var(const char *prefix, const char *name, char **var, char **type)
|
||||
{
|
||||
char *tmpname;
|
||||
|
||||
tmpname = alloca(strlen(prefix) + 1 + strlen(name) + 1 + sizeof("TYPE_"));
|
||||
sprintf(tmpname, "TYPE_%s_%s", prefix, name);
|
||||
*var = getenv(tmpname + 5);
|
||||
*type = getenv(tmpname);
|
||||
}
|
||||
|
||||
static json_object *jshn_add_objects(json_object *obj, const char *prefix, bool array);
|
||||
|
||||
static void jshn_add_object_var(json_object *obj, bool array, const char *prefix, const char *name)
|
||||
{
|
||||
json_object *new;
|
||||
char *var, *type;
|
||||
|
||||
get_var(prefix, name, &var, &type);
|
||||
if (!var || !type)
|
||||
return;
|
||||
|
||||
if (!strcmp(type, "array")) {
|
||||
new = json_object_new_array();
|
||||
jshn_add_objects(new, var, true);
|
||||
} else if (!strcmp(type, "object")) {
|
||||
new = json_object_new_object();
|
||||
jshn_add_objects(new, var, false);
|
||||
} else if (!strcmp(type, "string")) {
|
||||
new = json_object_new_string(var);
|
||||
} else if (!strcmp(type, "int")) {
|
||||
new = json_object_new_int(atoi(var));
|
||||
} else if (!strcmp(type, "boolean")) {
|
||||
new = json_object_new_boolean(!!atoi(var));
|
||||
} else {
|
||||
return;
|
||||
}
|
||||
|
||||
if (array)
|
||||
json_object_array_add(obj, new);
|
||||
else
|
||||
json_object_object_add(obj, name, new);
|
||||
}
|
||||
|
||||
static json_object *jshn_add_objects(json_object *obj, const char *prefix, bool array)
|
||||
{
|
||||
char *keys, *key, *brk;
|
||||
|
||||
keys = get_keys(prefix);
|
||||
if (!keys || !obj)
|
||||
goto out;
|
||||
|
||||
for (key = strtok_r(keys, " ", &brk); key;
|
||||
key = strtok_r(NULL, " ", &brk)) {
|
||||
jshn_add_object_var(obj, array, prefix, key);
|
||||
}
|
||||
|
||||
out:
|
||||
return obj;
|
||||
}
|
||||
|
||||
static int jshn_format(void)
|
||||
{
|
||||
json_object *obj;
|
||||
|
||||
obj = json_object_new_object();
|
||||
jshn_add_objects(obj, "JSON_VAR", false);
|
||||
fprintf(stdout, "%s\n", json_object_to_json_string(obj));
|
||||
json_object_put(obj);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int usage(const char *progname)
|
||||
{
|
||||
fprintf(stderr, "Usage: %s -r <message>|-w\n", progname);
|
||||
return 2;
|
||||
}
|
||||
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
int ch;
|
||||
|
||||
while ((ch = getopt(argc, argv, "r:w")) != -1) {
|
||||
switch(ch) {
|
||||
case 'r':
|
||||
return jshn_parse(optarg);
|
||||
case 'w':
|
||||
return jshn_format();
|
||||
default:
|
||||
return usage(argv[0]);
|
||||
}
|
||||
}
|
||||
return usage(argv[0]);
|
||||
}
|
||||
Loading…
Reference in a new issue