forked from lda/telodendria
Allow encoding user-provided strings.
This commit is contained in:
parent
50e599f1cd
commit
efdf168085
2 changed files with 159 additions and 102 deletions
2
TODO.txt
2
TODO.txt
|
@ -30,7 +30,7 @@ Milestone: v0.3.0
|
||||||
[~] Simple command line tool for working with JSON
|
[~] Simple command line tool for working with JSON
|
||||||
[x] Pretty-print Json
|
[x] Pretty-print Json
|
||||||
[x] Query fields for use in shell scripts.
|
[x] Query fields for use in shell scripts.
|
||||||
[ ] Encode user-provided JSON strings
|
[x] Encode user-provided JSON strings
|
||||||
[ ] json man page
|
[ ] json man page
|
||||||
|
|
||||||
[ ] Move configuration to database
|
[ ] Move configuration to database
|
||||||
|
|
259
tools/src/json.c
259
tools/src/json.c
|
@ -30,94 +30,91 @@
|
||||||
#include <Memory.h>
|
#include <Memory.h>
|
||||||
#include <Json.h>
|
#include <Json.h>
|
||||||
|
|
||||||
int
|
#define FLAG_ENCODE (1 << 0)
|
||||||
main(int argc, char **argv)
|
#define FLAG_SELECT (1 << 1)
|
||||||
|
|
||||||
|
static void
|
||||||
|
usage(char *prog)
|
||||||
{
|
{
|
||||||
HashMap *json;
|
fprintf(stderr, "Usage: %s [-s query|-e str]\n", prog);
|
||||||
|
}
|
||||||
|
|
||||||
json = JsonDecode(stdin);
|
static void
|
||||||
|
query(char *select, HashMap * json)
|
||||||
|
{
|
||||||
|
char *key;
|
||||||
|
JsonValue *val = JsonValueObject(json);
|
||||||
|
|
||||||
if (!json)
|
key = strtok(select, "->");
|
||||||
|
|
||||||
|
while (key)
|
||||||
{
|
{
|
||||||
fprintf(stderr, "Malformed JSON.\n");
|
char keyName[128];
|
||||||
return 1;
|
size_t arrInd;
|
||||||
}
|
int expectArr = 0;
|
||||||
|
int func = 0;
|
||||||
|
|
||||||
if (argc > 1)
|
expectArr = (sscanf(key, "%127[^[][%lu]", keyName, &arrInd) == 2);
|
||||||
{
|
|
||||||
char *select = argv[1];
|
|
||||||
char *key;
|
|
||||||
JsonValue *val = JsonValueObject(json);
|
|
||||||
|
|
||||||
key = strtok(select, "->");
|
if (keyName[0] == '@')
|
||||||
|
|
||||||
while (key)
|
|
||||||
{
|
{
|
||||||
char keyName[128];
|
if (strcmp(keyName + 1, "length") == 0)
|
||||||
size_t arrInd;
|
|
||||||
int expectArr = 0;
|
|
||||||
int func = 0;
|
|
||||||
|
|
||||||
expectArr = (sscanf(key, "%127[^[][%lu]", keyName, &arrInd) == 2);
|
|
||||||
|
|
||||||
if (keyName[0] == '@')
|
|
||||||
{
|
{
|
||||||
if (strcmp(keyName + 1, "length") == 0)
|
switch (JsonValueType(val))
|
||||||
{
|
{
|
||||||
switch (JsonValueType(val))
|
case JSON_ARRAY:
|
||||||
{
|
val = JsonValueInteger(ArraySize(JsonValueAsArray(val)));
|
||||||
case JSON_ARRAY:
|
break;
|
||||||
val = JsonValueInteger(ArraySize(JsonValueAsArray(val)));
|
case JSON_STRING:
|
||||||
break;
|
val = JsonValueInteger(strlen(JsonValueAsString(val)));
|
||||||
case JSON_STRING:
|
break;
|
||||||
val = JsonValueInteger(strlen(JsonValueAsString(val)));
|
default:
|
||||||
break;
|
val = NULL;
|
||||||
default:
|
break;
|
||||||
val = NULL;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
else if (JsonValueType(val) == JSON_OBJECT && strcmp(keyName + 1, "keys") == 0)
|
|
||||||
{
|
|
||||||
HashMap *obj = JsonValueAsObject(val);
|
|
||||||
Array *arr = ArrayCreate();
|
|
||||||
char *k;
|
|
||||||
void *v;
|
|
||||||
|
|
||||||
while (HashMapIterate(obj, &k, &v))
|
|
||||||
{
|
|
||||||
ArrayAdd(arr, JsonValueString(k));
|
|
||||||
}
|
|
||||||
|
|
||||||
val = JsonValueArray(arr);
|
|
||||||
}
|
|
||||||
else if (JsonValueType(val) == JSON_STRING && strcmp(keyName + 1, "decode") == 0)
|
|
||||||
{
|
|
||||||
printf("%s", JsonValueAsString(val));
|
|
||||||
val = NULL;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
val = NULL;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
func = 1;
|
|
||||||
}
|
}
|
||||||
else if (keyName[0] == '!')
|
else if (JsonValueType(val) == JSON_OBJECT && strcmp(keyName + 1, "keys") == 0)
|
||||||
{
|
{
|
||||||
if (JsonValueType(val) == JSON_OBJECT)
|
HashMap *obj = JsonValueAsObject(val);
|
||||||
{
|
Array *arr = ArrayCreate();
|
||||||
HashMap *obj = JsonValueAsObject(val);
|
char *k;
|
||||||
|
void *v;
|
||||||
|
|
||||||
JsonValueFree(HashMapDelete(obj, keyName + 1));
|
while (HashMapIterate(obj, &k, &v))
|
||||||
|
{
|
||||||
|
ArrayAdd(arr, JsonValueString(k));
|
||||||
}
|
}
|
||||||
else if (JsonValueType(val) == JSON_ARRAY)
|
|
||||||
{
|
|
||||||
size_t i;
|
|
||||||
|
|
||||||
sscanf(keyName + 1, "%lu", &i);
|
val = JsonValueArray(arr);
|
||||||
|
}
|
||||||
|
else if (JsonValueType(val) == JSON_STRING && strcmp(keyName + 1, "decode") == 0)
|
||||||
|
{
|
||||||
|
printf("%s\n", JsonValueAsString(val));
|
||||||
|
val = NULL;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
val = NULL;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
func = 1;
|
||||||
|
}
|
||||||
|
else if (keyName[0] == '!')
|
||||||
|
{
|
||||||
|
if (JsonValueType(val) == JSON_OBJECT)
|
||||||
|
{
|
||||||
|
HashMap *obj = JsonValueAsObject(val);
|
||||||
|
|
||||||
|
JsonValueFree(HashMapDelete(obj, keyName + 1));
|
||||||
|
}
|
||||||
|
else if (JsonValueType(val) == JSON_ARRAY)
|
||||||
|
{
|
||||||
|
size_t i;
|
||||||
|
|
||||||
|
if (sscanf(keyName + 1, "%lu", &i) == 1)
|
||||||
|
{
|
||||||
JsonValueFree(ArrayDelete(JsonValueAsArray(val), i));
|
JsonValueFree(ArrayDelete(JsonValueAsArray(val), i));
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
|
@ -125,47 +122,107 @@ main(int argc, char **argv)
|
||||||
val = NULL;
|
val = NULL;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
func = 1;
|
|
||||||
}
|
}
|
||||||
|
else
|
||||||
if (!func)
|
|
||||||
{
|
|
||||||
if (JsonValueType(val) == JSON_OBJECT)
|
|
||||||
{
|
|
||||||
val = HashMapGet(JsonValueAsObject(val), keyName);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
val = NULL;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (expectArr && JsonValueType(val) == JSON_ARRAY)
|
|
||||||
{
|
|
||||||
val = ArrayGet(JsonValueAsArray(val), arrInd);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!val)
|
|
||||||
{
|
{
|
||||||
|
val = NULL;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
key = strtok(NULL, "->");
|
func = 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (val)
|
if (!func)
|
||||||
{
|
{
|
||||||
JsonEncodeValue(val, stdout, JSON_PRETTY);
|
if (JsonValueType(val) == JSON_OBJECT)
|
||||||
|
{
|
||||||
|
val = HashMapGet(JsonValueAsObject(val), keyName);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
val = NULL;
|
||||||
|
break;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (expectArr && JsonValueType(val) == JSON_ARRAY)
|
||||||
|
{
|
||||||
|
val = ArrayGet(JsonValueAsArray(val), arrInd);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!val)
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
key = strtok(NULL, "->");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (val)
|
||||||
|
{
|
||||||
|
JsonEncodeValue(val, stdout, JSON_PRETTY);
|
||||||
printf("\n");
|
printf("\n");
|
||||||
}
|
}
|
||||||
else
|
}
|
||||||
|
|
||||||
|
static void
|
||||||
|
encode(char *str)
|
||||||
|
{
|
||||||
|
JsonValue *val = JsonValueString(str);
|
||||||
|
|
||||||
|
JsonEncodeValue(val, stdout, JSON_DEFAULT);
|
||||||
|
JsonValueFree(val);
|
||||||
|
printf("\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
int
|
||||||
|
main(int argc, char **argv)
|
||||||
|
{
|
||||||
|
HashMap *json;
|
||||||
|
int flag = 0;
|
||||||
|
int ch;
|
||||||
|
char *input;
|
||||||
|
|
||||||
|
while ((ch = getopt(argc, argv, "s:e:")) != -1)
|
||||||
{
|
{
|
||||||
JsonEncode(json, stdout, JSON_PRETTY);
|
switch (ch)
|
||||||
JsonFree(json);
|
{
|
||||||
|
case 's':
|
||||||
|
flag = FLAG_SELECT;
|
||||||
|
input = optarg;
|
||||||
|
break;
|
||||||
|
case 'e':
|
||||||
|
flag = FLAG_ENCODE;
|
||||||
|
input = optarg;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
usage(argv[0]);
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (flag != FLAG_ENCODE)
|
||||||
|
{
|
||||||
|
json = JsonDecode(stdin);
|
||||||
|
|
||||||
|
if (!json)
|
||||||
|
{
|
||||||
|
fprintf(stderr, "Malformed JSON.\n");
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
switch (flag)
|
||||||
|
{
|
||||||
|
case FLAG_SELECT:
|
||||||
|
query(input, json);
|
||||||
|
break;
|
||||||
|
case FLAG_ENCODE:
|
||||||
|
encode(input);
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
JsonEncode(json, stdout, JSON_PRETTY);
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
MemoryFreeAll();
|
MemoryFreeAll();
|
||||||
|
|
Loading…
Reference in a new issue