1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
|
#define _XOPEN_SOURCE 700
#include <string.h>
#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>
#include <unistd.h>
#include "log.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);
struct json_tokener *tok = json_tokener_new_ex(256);
if (!tok) {
sway_abort("Unable to get json tokener.");
}
tree = json_tokener_parse_ex(tok, res, len);
if (!tree || tok->err != json_tokener_success) {
sway_abort("Unable to parse IPC response as JSON: %s", json_tokener_error_desc(tok->err));
}
json_object *success;
json_object_object_get_ex(tree, "success", &success);
if (success && !json_object_get_boolean(success)) {
json_object *error;
json_object_object_get_ex(tree, "error", &error);
sway_abort("IPC request failed: %s", json_object_get_string(error));
}
json_object_put(success);
json_tokener_free(tok);
}
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);
if (!outputs) {
sway_abort("Unabled to get focused output. No nodes in tree.");
}
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;
}
|
