cli: gather all found items through a callback function

1 files changed, 117 insertions, 58 deletions

diff --git a/main.c b/main.c
index 78fa0e4..2bdb110 100644
--- a/main.c
+++ b/main.c
@@ -25,10 +25,18 @@
 	#include <json-c/json.h>
 #endif
 
+#include <libubox/list.h>
+
 #include "lexer.h"
 #include "parser.h"
 #include "matcher.h"
 
+
+struct match_item {
+	struct json_object *jsobj;
+	struct list_head list;
+};
+
 static struct json_object *
 parse_json(FILE *fd, const char *source, const char **error)
 {
@@ -91,75 +99,95 @@ print_string(const char *s)
 }
 
 static void
-export_value(struct json_object *jsobj, const char *prefix)
+export_value(struct list_head *matches, const char *prefix)
 {
 	int n, len;
 	bool first = true;
+	struct match_item *item;
+
+	if (list_empty(matches))
+		return;
 
 	if (prefix)
 	{
-		switch (json_object_get_type(jsobj))
+		printf("export %s=", prefix);
+
+		list_for_each_entry(item, matches, list)
 		{
-		case json_type_object:
-			printf("export %s=", prefix);
-			json_object_object_foreach(jsobj, key, val)
+			switch (json_object_get_type(item->jsobj))
 			{
-				if (!val)
-					continue;
+			case json_type_object:
+				; /* a label can only be part of a statement */
+				json_object_object_foreach(item->jsobj, key, val)
+				{
+					if (!val)
+						continue;
 
-				if (!first)
-					printf("\\ ");
+					if (!first)
+						printf("\\ ");
 
-				print_string(key);
-				first = false;
-			}
-			printf("; ");
-			break;
+					print_string(key);
+					first = false;
+				}
+				break;
 
-		case json_type_array:
-			printf("export %s=", prefix);
-			for (n = 0, len = json_object_array_length(jsobj); n < len; n++)
-			{
+			case json_type_array:
+				for (n = 0, len = json_object_array_length(item->jsobj);
+				     n < len; n++)
+				{
+					if (!first)
+						printf("\\ ");
+
+					printf("%d", n);
+					first = false;
+				}
+				break;
+
+			case json_type_boolean:
 				if (!first)
 					printf("\\ ");
+				printf("%d", json_object_get_boolean(item->jsobj));
+				break;
 
-				printf("%d", n);
-				first = false;
-			}
-			printf("; ");
-			break;
-
-		case json_type_boolean:
-			printf("export %s=%d; ", prefix, json_object_get_boolean(jsobj));
-			break;
+			case json_type_int:
+				if (!first)
+					printf("\\ ");
+				printf("%d", json_object_get_int(item->jsobj));
+				break;
 
-		case json_type_int:
-			printf("export %s=%d; ", prefix, json_object_get_int(jsobj));
-			break;
+			case json_type_double:
+				if (!first)
+					printf("\\ ");
+				printf("%f", json_object_get_double(item->jsobj));
+				break;
 
-		case json_type_double:
-			printf("export %s=%f; ", prefix, json_object_get_double(jsobj));
-			break;
+			case json_type_string:
+				if (!first)
+					printf("\\ ");
+				print_string(json_object_get_string(item->jsobj));
+				break;
 
-		case json_type_string:
-			printf("export %s=", prefix);
-			print_string(json_object_get_string(jsobj));
-			printf("; ");
-			break;
+			case json_type_null:
+				break;
+			}
 
-		case json_type_null:
-			break;
+			first = false;
 		}
+
+		printf("; ");
 	}
 	else
 	{
-		printf("%s\n", json_object_to_json_string(jsobj));
+		list_for_each_entry(item, matches, list)
+			printf("%s\n", json_object_to_json_string(item->jsobj));
 	}
 }
 
 static void
-export_type(struct json_object *jsobj, const char *prefix)
+export_type(struct list_head *matches, const char *prefix)
 {
+	bool first = true;
+	struct match_item *item;
 	const char *types[] = {
 		"null",
 		"boolean",
@@ -170,18 +198,48 @@ export_type(struct json_object *jsobj, const char *prefix)
 		"string"
 	};
 
+	if (list_empty(matches))
+		return;
+
 	if (prefix)
-		printf("export %s=%s; ", prefix, types[json_object_get_type(jsobj)]);
+		printf("export %s=", prefix);
+
+	list_for_each_entry(item, matches, list)
+	{
+		if (!first)
+			printf("\\ ");
+
+		printf("%s", types[json_object_get_type(item->jsobj)]);
+		first = false;
+	}
+
+	if (prefix)
+		printf("; ");
 	else
-		printf("%s\n", types[json_object_get_type(jsobj)]);
+		printf("\n");
+}
+
+static void
+match_cb(struct json_object *res, void *priv)
+{
+	struct list_head *h = priv;
+	struct match_item *i = calloc(1, sizeof(*i));
+
+	if (i)
+	{
+		i->jsobj = res;
+		list_add_tail(&i->list, h);
+	}
 }
 
 static bool
 filter_json(int opt, struct json_object *jsobj, char *expr)
 {
 	struct jp_state *state;
-	struct json_object *res = NULL;
 	const char *prefix = NULL;
+	struct list_head matches;
+	struct match_item *item, *tmp;
+	struct json_object *res = NULL;
 
 	state = jp_parse(expr);
 
@@ -193,24 +251,25 @@ filter_json(int opt, struct json_object *jsobj, char *expr)
 		goto out;
 	}
 
-	res = jp_match(state->path, jsobj);
+	INIT_LIST_HEAD(&matches);
 
-	if (res)
-	{
-		prefix = (state->path->type == T_LABEL) ? state->path->str : NULL;
+	res = jp_match(state->path, jsobj, match_cb, &matches);
+	prefix = (state->path->type == T_LABEL) ? state->path->str : NULL;
 
-		switch (opt)
-		{
-		case 't':
-			export_type(res, prefix);
-			break;
+	switch (opt)
+	{
+	case 't':
+		export_type(&matches, prefix);
+		break;
 
-		default:
-			export_value(res, prefix);
-			break;
-		}
+	default:
+		export_value(&matches, prefix);
+		break;
 	}
 
+	list_for_each_entry_safe(item, tmp, &matches, list)
+		free(item);
+
 out:
 	if (state)
 		jp_free(state);