Display workspaces on swaybar

1 files changed, 118 insertions, 1 deletions

diff --git a/swaybar/main.c b/swaybar/main.c
index e252a2db..c7c4ed47 100644
--- a/swaybar/main.c
+++ b/swaybar/main.c
@@ -1,10 +1,15 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <stdint.h>
+#include <stdbool.h>
+#include <json-c/json.h>
+#include "ipc-client.h"
 #include "readline.h"
 #include "client/registry.h"
 #include "client/window.h"
 #include "client/pango.h"
+#include "stringop.h"
 #include "log.h"
 
 #define MARGIN 5
@@ -27,10 +32,21 @@ struct colors {
 	struct box_colors binding_mode;
 };
 
+struct workspace {
+	int num;
+	char *name;
+	bool focused;
+	bool visible;
+	bool urgent;
+};
+
+list_t *workspaces = NULL;
+int socketfd;
 FILE *command;
-char *line;
+char *line, *output;
 struct registry *registry;
 struct window *window;
+
 struct colors colors = {
 	.background = 0x000000FF,
 	.statusline = 0xFFFFFFFF,
@@ -89,20 +105,114 @@ void update() {
 }
 
 void render() {
+	// Clear
 	cairo_save(window->cairo);
 	cairo_set_operator(window->cairo, CAIRO_OPERATOR_CLEAR);
 	cairo_paint(window->cairo);
 	cairo_restore(window->cairo);
 
+	// Background
 	cairo_set_source_u32(window->cairo, colors.background);
 	cairo_paint(window->cairo);
 
+	// Command output
 	cairo_set_source_u32(window->cairo, colors.statusline);
 	int width, height;
 	get_text_size(window, &width, &height, "%s", line);
 
 	cairo_move_to(window->cairo, window->width - MARGIN - width, MARGIN);
 	pango_printf(window, "%s", line);
+
+	// Workspaces
+	int x = 0;
+	int i;
+	for (i = 0; i < workspaces->length; ++i) {
+		struct workspace *ws = workspaces->items[i];
+		get_text_size(window, &width, &height, "%s", ws->name);
+		struct box_colors box_colors;
+		if (ws->urgent) {
+			box_colors = colors.urgent_workspace;
+		} else if (ws->focused) {
+			box_colors = colors.focused_workspace;
+		} else if (ws->visible) {
+			box_colors = colors.active_workspace;
+		} else {
+			box_colors = colors.inactive_workspace;
+		}
+		cairo_set_source_u32(window->cairo, box_colors.background);
+		cairo_rectangle(window->cairo, x, 0, width + MARGIN * 2, window->height);
+		cairo_fill(window->cairo);
+
+		cairo_set_source_u32(window->cairo, box_colors.border);
+		cairo_rectangle(window->cairo, x, 0, width + MARGIN * 2, window->height);
+		cairo_stroke(window->cairo);
+
+		cairo_set_source_u32(window->cairo, box_colors.text);
+		cairo_move_to(window->cairo, x + MARGIN, MARGIN);
+		pango_printf(window, "%s", ws->name);
+
+		x += width + MARGIN * 2 + MARGIN;
+	}
+}
+
+void ipc_update_workspaces() {
+	if (workspaces) {
+		free_flat_list(workspaces);
+	}
+	workspaces = create_list();
+
+	uint32_t len = 0;
+	char *res = ipc_single_command(socketfd, IPC_GET_WORKSPACES, NULL, &len);
+	json_object *results = json_tokener_parse(res);
+
+	int i;
+	for (i = 0; i < json_object_array_length(results); ++i) {
+		json_object *ws_json = json_object_array_get_idx(results, i);
+		json_object *num, *name, *visible, *focused, *out, *urgent;
+		json_object_object_get_ex(ws_json, "num", &num);
+		json_object_object_get_ex(ws_json, "name", &name);
+		json_object_object_get_ex(ws_json, "visible", &visible);
+		json_object_object_get_ex(ws_json, "focused", &focused);
+		json_object_object_get_ex(ws_json, "output", &out);
+		json_object_object_get_ex(ws_json, "urgent", &urgent);
+
+		if (strcmp(json_object_get_string(out), output) == 0) {
+			struct workspace *ws = malloc(sizeof(struct workspace));
+			ws->num = json_object_get_int(num);
+			ws->name = strdup(json_object_get_string(name));
+			ws->visible = json_object_get_boolean(visible);
+			ws->focused = json_object_get_boolean(focused);
+			ws->urgent = json_object_get_boolean(urgent);
+			list_add(workspaces, ws);
+			sway_log(L_INFO, "Handling workspace %s", ws->name);
+		}
+
+		json_object_put(num);
+		json_object_put(name);
+		json_object_put(visible);
+		json_object_put(focused);
+		json_object_put(out);
+		json_object_put(urgent);
+		json_object_put(ws_json);
+	}
+
+	json_object_put(results);
+	free(res);
+}
+
+void bar_ipc_init(int outputi) {
+	uint32_t len = 0;
+	char *res = ipc_single_command(socketfd, IPC_GET_OUTPUTS, NULL, &len);
+	json_object *outputs = json_tokener_parse(res);
+	json_object *info = json_object_array_get_idx(outputs, outputi);
+	json_object *name;
+	json_object_object_get_ex(info, "name", &name);
+	output = strdup(json_object_get_string(name));
+	free(res);
+	json_object_put(outputs);
+	sway_log(L_INFO, "Running on output %s", output);
+
+	ipc_update_workspaces();
 }
 
 int main(int argc, char **argv) {
@@ -132,6 +242,13 @@ int main(int argc, char **argv) {
 	line = malloc(1024);
 	line[0] = '\0';
 
+	char *socket_path = get_socketpath();
+	if (!socket_path) {
+		sway_abort("Unable to retrieve socket path");
+	}
+	socketfd = ipc_open_socket(socket_path);
+	bar_ipc_init(desired_output);
+
 	do {
 		if (window_prerender(window) && window->cairo) {
 			update();