Merge pull request #909 from zandrmartin/grab-focused

implement "focused container" option for swaygrab

1 files changed, 123 insertions, 0 deletions

diff --git a/swaygrab/json.c b/swaygrab/json.c
new file mode 100644
index 00000000..7cd73cbe
--- /dev/null
+++ b/swaygrab/json.c
@@ -0,0 +1,123 @@
+#include <stdio.h>
+#include <stdbool.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include "ipc-client.h"
+#include "swaygrab/json.h"
+
+static json_object *tree;
+
+void init_json_tree(int socketfd) {
+	uint32_t len = 0;
+	char *res = ipc_single_command(socketfd, IPC_GET_TREE, NULL, &len);
+	tree = json_tokener_parse(res);
+}
+
+void free_json_tree() {
+	json_object_put(tree);
+}
+
+static bool is_focused(json_object *c) {
+	json_object *focused;
+	json_object_object_get_ex(c, "focused", &focused);
+	return json_object_get_boolean(focused);
+}
+
+static json_object *get_focused_container_r(json_object *c) {
+	json_object *name;
+	json_object_object_get_ex(c, "name", &name);
+	if (is_focused(c)) {
+		return c;
+	} else {
+		json_object *nodes, *node, *child;
+		json_object_object_get_ex(c, "nodes", &nodes);
+		int i;
+		for (i = 0; i < json_object_array_length(nodes); i++) {
+			node = json_object_array_get_idx(nodes, i);
+
+			if ((child = get_focused_container_r(node))) {
+				return child;
+			}
+		}
+
+		json_object_object_get_ex(c, "floating_nodes", &nodes);
+		for (i = 0; i < json_object_array_length(nodes); i++) {
+			node = json_object_array_get_idx(nodes, i);
+
+			if ((child = get_focused_container_r(node))) {
+				return child;
+			}
+		}
+
+	}
+
+	return NULL;
+}
+
+json_object *get_focused_container() {
+	return get_focused_container_r(tree);
+}
+
+char *get_focused_output() {
+	json_object *outputs, *output, *name;
+	json_object_object_get_ex(tree, "nodes", &outputs);
+
+	for (int i = 0; i < json_object_array_length(outputs); i++) {
+		output = json_object_array_get_idx(outputs, i);
+
+		if (get_focused_container_r(output)) {
+			json_object_object_get_ex(output, "name", &name);
+			return strdup(json_object_get_string(name));
+		}
+	}
+
+	return NULL;
+}
+
+char *create_payload(const char *output, struct wlc_geometry *g) {
+	char *payload_str = malloc(256);
+	json_object *payload = json_object_new_object();
+
+	json_object_object_add(payload, "output", json_object_new_string(output));
+	json_object_object_add(payload, "x", json_object_new_int(g->origin.x));
+	json_object_object_add(payload, "y", json_object_new_int(g->origin.y));
+	json_object_object_add(payload, "w", json_object_new_int(g->size.w));
+	json_object_object_add(payload, "h", json_object_new_int(g->size.h));
+
+	snprintf(payload_str, 256, "%s", json_object_to_json_string(payload));
+	return strdup(payload_str);
+}
+
+struct wlc_geometry *get_container_geometry(json_object *container) {
+	struct wlc_geometry *geo = malloc(sizeof(struct wlc_geometry));
+	json_object *rect, *x, *y, *w, *h;
+
+	json_object_object_get_ex(container, "rect", &rect);
+	json_object_object_get_ex(rect, "x", &x);
+	json_object_object_get_ex(rect, "y", &y);
+	json_object_object_get_ex(rect, "width", &w);
+	json_object_object_get_ex(rect, "height", &h);
+
+	geo->origin.x = json_object_get_int(x);
+	geo->origin.y = json_object_get_int(y);
+	geo->size.w = json_object_get_int(w);
+	geo->size.h = json_object_get_int(h);
+
+	return geo;
+}
+
+json_object *get_output_container(const char *output) {
+	json_object *outputs, *json_output, *name;
+	json_object_object_get_ex(tree, "nodes", &outputs);
+
+	for (int i = 0; i < json_object_array_length(outputs); i++) {
+		json_output = json_object_array_get_idx(outputs, i);
+		json_object_object_get_ex(json_output, "name", &name);
+
+		if (strcmp(json_object_get_string(name), output) == 0) {
+			return json_output;
+		}
+	}
+
+	return NULL;
+}