swaybar: add StatusNotifierHost to tray

3 files changed, 191 insertions, 0 deletions

diff --git a/swaybar/meson.build b/swaybar/meson.build
index b8ed2c23..85783a0f 100644
--- a/swaybar/meson.build
+++ b/swaybar/meson.build
@@ -1,4 +1,5 @@
 tray_files = get_option('enable-tray') ? [
+	'tray/host.c',
 	'tray/icon.c',
 	'tray/tray.c',
 	'tray/watcher.c'
diff --git a/swaybar/tray/host.c b/swaybar/tray/host.c
new file mode 100644
index 00000000..3cc90254
--- /dev/null
+++ b/swaybar/tray/host.c
@@ -0,0 +1,177 @@
+#define _POSIX_C_SOURCE 200809L
+#include <stdbool.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include "swaybar/tray/host.h"
+#include "swaybar/tray/tray.h"
+#include "list.h"
+#include "log.h"
+
+static const char *watcher_path = "/StatusNotifierWatcher";
+
+static int cmp_sni_id(const void *item, const void *cmp_to) {
+	const char *sni = item;
+	return strcmp(sni, cmp_to);
+}
+
+static void add_sni(struct swaybar_tray *tray, char *id) {
+	int idx = list_seq_find(tray->items, cmp_sni_id, id);
+	if (idx == -1) {
+		wlr_log(WLR_DEBUG, "Registering Status Notifier Item '%s'", id);
+		char *sni = strdup(id);
+		if (sni) {
+			list_add(tray->items, sni);
+		}
+	}
+}
+
+static int handle_sni_registered(sd_bus_message *msg, void *data,
+		sd_bus_error *error) {
+	char *id;
+	int ret = sd_bus_message_read(msg, "s", &id);
+	if (ret < 0) {
+		wlr_log(WLR_ERROR, "Failed to parse register SNI message: %s", strerror(-ret));
+	}
+
+	struct swaybar_tray *tray = data;
+	add_sni(tray, id);
+
+	return ret;
+}
+
+static int handle_sni_unregistered(sd_bus_message *msg, void *data,
+		sd_bus_error *error) {
+	char *id;
+	int ret = sd_bus_message_read(msg, "s", &id);
+	if (ret < 0) {
+		wlr_log(WLR_ERROR, "Failed to parse unregister SNI message: %s", strerror(-ret));
+	}
+
+	struct swaybar_tray *tray = data;
+	int idx = list_seq_find(tray->items, cmp_sni_id, id);
+	if (idx != -1) {
+		wlr_log(WLR_DEBUG, "Unregistering Status Notifier Item '%s'", id);
+		free(tray->items->items[idx]);
+		list_del(tray->items, idx);
+	}
+	return ret;
+}
+
+static int get_registered_snis_callback(sd_bus_message *msg, void *data,
+		sd_bus_error *error) {
+	if (sd_bus_message_is_method_error(msg, NULL)) {
+		sd_bus_error err = *sd_bus_message_get_error(msg);
+		wlr_log(WLR_ERROR, "Failed to get registered SNIs: %s", err.message);
+		return -sd_bus_error_get_errno(&err);
+	}
+
+	int ret = sd_bus_message_enter_container(msg, 'v', NULL);
+	if (ret < 0) {
+		wlr_log(WLR_ERROR, "Failed to read registered SNIs: %s", strerror(-ret));
+		return ret;
+	}
+
+	char **ids;
+	ret = sd_bus_message_read_strv(msg, &ids);
+	if (ret < 0) {
+		wlr_log(WLR_ERROR, "Failed to read registered SNIs: %s", strerror(-ret));
+		return ret;
+	}
+
+	if (ids) {
+		struct swaybar_tray *tray = data;
+		for (char **id = ids; *id; ++id) {
+			add_sni(tray, *id);
+		}
+	}
+
+	return ret;
+}
+
+static bool register_to_watcher(struct swaybar_host *host) {
+	// this is called asynchronously in case the watcher is owned by this process
+	int ret = sd_bus_call_method_async(host->tray->bus, NULL,
+			host->watcher_interface, watcher_path, host->watcher_interface,
+			"RegisterStatusNotifierHost", NULL, NULL, "s", host->service);
+	if (ret < 0) {
+		wlr_log(WLR_ERROR, "Failed to send register call: %s", strerror(-ret));
+		return false;
+	}
+
+	ret = sd_bus_call_method_async(host->tray->bus, NULL,
+			host->watcher_interface, watcher_path,
+			"org.freedesktop.DBus.Properties", "Get",
+			get_registered_snis_callback, host->tray, "ss",
+			host->watcher_interface, "RegisteredStatusNotifierItems");
+	if (ret < 0) {
+		wlr_log(WLR_ERROR, "Failed to get registered SNIs: %s", strerror(-ret));
+	}
+
+	return ret >= 0;
+}
+
+bool init_host(struct swaybar_host *host, char *protocol,
+		struct swaybar_tray *tray) {
+	size_t len = snprintf(NULL, 0, "org.%s.StatusNotifierWatcher", protocol) + 1;
+	host->watcher_interface = malloc(len);
+	if (!host->watcher_interface) {
+		return false;
+	}
+	snprintf(host->watcher_interface, len, "org.%s.StatusNotifierWatcher", protocol);
+
+	sd_bus_slot *reg_slot = NULL, *unreg_slot = NULL;
+	int ret = sd_bus_match_signal(tray->bus, &reg_slot, host->watcher_interface,
+			watcher_path, host->watcher_interface,
+			"StatusNotifierItemRegistered", handle_sni_registered, tray);
+	if (ret < 0) {
+		wlr_log(WLR_ERROR, "Failed to subscribe to registering events: %s",
+				strerror(-ret));
+		goto error;
+	}
+	ret = sd_bus_match_signal(tray->bus, &unreg_slot, host->watcher_interface,
+			watcher_path, host->watcher_interface,
+			"StatusNotifierItemUnregistered", handle_sni_unregistered, tray);
+	if (ret < 0) {
+		wlr_log(WLR_ERROR, "Failed to subscribe to unregistering events: %s",
+				strerror(-ret));
+		goto error;
+	}
+
+	pid_t pid = getpid();
+	size_t service_len = snprintf(NULL, 0, "org.%s.StatusNotifierHost-%d",
+			protocol, pid) + 1;
+	host->service = malloc(service_len);
+	if (!host->service) {
+		goto error;
+	}
+	snprintf(host->service, service_len, "org.%s.StatusNotifierHost-%d", protocol, pid);
+	ret = sd_bus_request_name(tray->bus, host->service, 0);
+	if (ret < 0) {
+		wlr_log(WLR_DEBUG, "Failed to acquire service name: %s", strerror(-ret));
+		goto error;
+	}
+
+	host->tray = tray;
+	if (!register_to_watcher(host)) {
+		goto error;
+	}
+
+	sd_bus_slot_set_floating(reg_slot, 1);
+	sd_bus_slot_set_floating(unreg_slot, 1);
+
+	wlr_log(WLR_DEBUG, "Registered %s", host->service);
+	return true;
+error:
+	sd_bus_slot_unref(reg_slot);
+	sd_bus_slot_unref(unreg_slot);
+	finish_host(host);
+	return false;
+}
+
+void finish_host(struct swaybar_host *host) {
+	sd_bus_release_name(host->tray->bus, host->service);
+	free(host->service);
+	free(host->watcher_interface);
+}
diff --git a/swaybar/tray/tray.c b/swaybar/tray/tray.c
index c1d3b50b..e760812c 100644
--- a/swaybar/tray/tray.c
+++ b/swaybar/tray/tray.c
@@ -4,8 +4,10 @@
 #include <string.h>
 #include "swaybar/bar.h"
 #include "swaybar/tray/icon.h"
+#include "swaybar/tray/host.h"
 #include "swaybar/tray/tray.h"
 #include "swaybar/tray/watcher.h"
+#include "list.h"
 #include "log.h"
 
 struct swaybar_tray *create_tray(struct swaybar *bar) {
@@ -29,6 +31,11 @@ struct swaybar_tray *create_tray(struct swaybar *bar) {
 	tray->watcher_xdg = create_watcher("freedesktop", tray->bus);
 	tray->watcher_kde = create_watcher("kde", tray->bus);
 
+	tray->items = create_list();
+
+	init_host(&tray->host_xdg, "freedesktop", tray);
+	init_host(&tray->host_kde, "kde", tray);
+
 	init_themes(&tray->themes, &tray->basedirs);
 
 	return tray;
@@ -38,6 +45,12 @@ void destroy_tray(struct swaybar_tray *tray) {
 	if (!tray) {
 		return;
 	}
+	finish_host(&tray->host_xdg);
+	finish_host(&tray->host_kde);
+	for (int i = 0; i < tray->items->length; ++i) {
+		free(tray->items->items[0]);
+	}
+	list_free(tray->items);
 	destroy_watcher(tray->watcher_xdg);
 	destroy_watcher(tray->watcher_kde);
 	sd_bus_flush_close_unref(tray->bus);