1 files changed, 403 insertions, 0 deletions
diff --git a/systray/item.c b/systray/item.c
new file mode 100644
index 0000000..8a13181
--- /dev/null
+++ b/systray/item.c
@@ -0,0 +1,403 @@
+#include "item.h"
+
+#include "helpers.h"
+#include "icon.h"
+#include "watcher.h"
+
+#include <dbus/dbus.h>
+
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+// IWYU pragma: no_include "dbus/dbus-protocol.h"
+// IWYU pragma: no_include "dbus/dbus-shared.h"
+
+#define RULEBSIZE 256
+#define MIN(A, B) ((A) < (B) ? (A) : (B))
+
+static const char *match_string =
+	"type='signal',"
+	"sender='%s',"
+	"interface='" SNI_NAME
+	"',"
+	"member='NewIcon'";
+
+static Watcher *
+item_get_watcher(const Item *item)
+{
+	if (!item)
+		return NULL;
+
+	return item->watcher;
+}
+
+static DBusConnection *
+item_get_connection(const Item *item)
+{
+	if (!item || !item->watcher)
+		return NULL;
+
+	return item->watcher->conn;
+}
+
+static const uint8_t *
+extract_image(DBusMessageIter *iter, dbus_int32_t *width, dbus_int32_t *height,
+              int *size)
+{
+	DBusMessageIter vals, bytes;
+	const uint8_t *buf;
+
+	dbus_message_iter_recurse(iter, &vals);
+	if (dbus_message_iter_get_arg_type(&vals) != DBUS_TYPE_INT32)
+		goto fail;
+	dbus_message_iter_get_basic(&vals, width);
+
+	dbus_message_iter_next(&vals);
+	if (dbus_message_iter_get_arg_type(&vals) != DBUS_TYPE_INT32)
+		goto fail;
+	dbus_message_iter_get_basic(&vals, height);
+
+	dbus_message_iter_next(&vals);
+	if (dbus_message_iter_get_arg_type(&vals) != DBUS_TYPE_ARRAY)
+		goto fail;
+	dbus_message_iter_recurse(&vals, &bytes);
+	if (dbus_message_iter_get_arg_type(&bytes) != DBUS_TYPE_BYTE)
+		goto fail;
+	dbus_message_iter_get_fixed_array(&bytes, &buf, size);
+	if (size == 0)
+		goto fail;
+
+	return buf;
+
+fail:
+	return NULL;
+}
+
+static int
+select_image(DBusMessageIter *iter, int target_width)
+{
+	DBusMessageIter vals;
+	dbus_int32_t cur_width;
+	int i = 0;
+
+	do {
+		dbus_message_iter_recurse(iter, &vals);
+		if (dbus_message_iter_get_arg_type(&vals) != DBUS_TYPE_INT32)
+			return -1;
+		dbus_message_iter_get_basic(&vals, &cur_width);
+		if (cur_width >= target_width)
+			return i;
+
+		i++;
+	} while (dbus_message_iter_next(iter));
+
+	/* return last index if desired not found */
+	return i--;
+}
+
+static void
+menupath_ready_handler(DBusPendingCall *pending, void *data)
+{
+	Item *item = data;
+
+	DBusError err = DBUS_ERROR_INIT;
+	DBusMessage *reply = NULL;
+	DBusMessageIter iter, opath;
+	char *path_dup = NULL;
+	const char *path;
+
+	reply = dbus_pending_call_steal_reply(pending);
+	if (!reply)
+		goto fail;
+
+	if (dbus_set_error_from_message(&err, reply)) {
+		fprintf(stderr, "DBus Error: %s - %s: Couldn't get menupath\n",
+		        err.name, err.message);
+		goto fail;
+	}
+
+	dbus_message_iter_init(reply, &iter);
+	if (dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_VARIANT)
+		goto fail;
+	dbus_message_iter_recurse(&iter, &opath);
+	if (dbus_message_iter_get_arg_type(&opath) != DBUS_TYPE_OBJECT_PATH)
+		goto fail;
+	dbus_message_iter_get_basic(&opath, &path);
+
+	path_dup = strdup(path);
+	if (!path_dup)
+		goto fail;
+
+	item->menu_busobj = path_dup;
+
+	dbus_message_unref(reply);
+	dbus_pending_call_unref(pending);
+	return;
+
+fail:
+	free(path_dup);
+	dbus_error_free(&err);
+	if (reply)
+		dbus_message_unref(reply);
+	if (pending)
+		dbus_pending_call_unref(pending);
+}
+
+/*
+ * Gets the Id dbus property, which is the name of the application,
+ * most of the time...
+ * The initial letter will be used as a fallback icon
+ */
+static void
+id_ready_handler(DBusPendingCall *pending, void *data)
+{
+	Item *item = data;
+
+	DBusError err = DBUS_ERROR_INIT;
+	DBusMessage *reply = NULL;
+	DBusMessageIter iter, string;
+	Watcher *watcher;
+	char *id_dup = NULL;
+	const char *id;
+
+	watcher = item_get_watcher(item);
+
+	reply = dbus_pending_call_steal_reply(pending);
+	if (!reply)
+		goto fail;
+
+	if (dbus_set_error_from_message(&err, reply)) {
+		fprintf(stderr, "DBus Error: %s - %s: Couldn't get appid\n",
+		        err.name, err.message);
+		goto fail;
+	}
+
+	dbus_message_iter_init(reply, &iter);
+	if (dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_VARIANT)
+		goto fail;
+	dbus_message_iter_recurse(&iter, &string);
+	if (dbus_message_iter_get_arg_type(&string) != DBUS_TYPE_STRING)
+		goto fail;
+	dbus_message_iter_get_basic(&string, &id);
+
+	id_dup = strdup(id);
+	if (!id_dup)
+		goto fail;
+	item->appid = id_dup;
+
+	/* Don't trigger update if this item already has a real icon */
+	if (!item->icon)
+		watcher_update_trays(watcher);
+
+	dbus_message_unref(reply);
+	dbus_pending_call_unref(pending);
+	return;
+
+fail:
+	dbus_error_free(&err);
+	if (id_dup)
+		free(id_dup);
+	if (reply)
+		dbus_message_unref(reply);
+	if (pending)
+		dbus_pending_call_unref(pending);
+}
+
+static void
+pixmap_ready_handler(DBusPendingCall *pending, void *data)
+{
+	Item *item = data;
+
+	DBusMessage *reply = NULL;
+	DBusMessageIter iter, array, select, strct;
+	Icon *icon = NULL;
+	Watcher *watcher;
+	dbus_int32_t width, height;
+	int selected_index, size;
+	const uint8_t *buf;
+
+	watcher = item_get_watcher(item);
+
+	reply = dbus_pending_call_steal_reply(pending);
+	if (!reply || dbus_message_get_type(reply) == DBUS_MESSAGE_TYPE_ERROR)
+		goto fail;
+	dbus_message_iter_init(reply, &iter);
+	if (dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_VARIANT)
+		goto fail;
+	dbus_message_iter_recurse(&iter, &array);
+	if (dbus_message_iter_get_arg_type(&array) != DBUS_TYPE_ARRAY)
+		goto fail;
+	dbus_message_iter_recurse(&array, &select);
+	if (dbus_message_iter_get_arg_type(&select) != DBUS_TYPE_STRUCT)
+		goto fail;
+	selected_index = select_image(&select, 22); // Get the 22*22 image
+	if (selected_index < 0)
+		goto fail;
+
+	dbus_message_iter_recurse(&array, &strct);
+	if (dbus_message_iter_get_arg_type(&strct) != DBUS_TYPE_STRUCT)
+		goto fail;
+	for (int i = 0; i < selected_index; i++)
+		dbus_message_iter_next(&strct);
+	buf = extract_image(&strct, &width, &height, &size);
+	if (!buf)
+		goto fail;
+
+	if (!item->icon) {
+		/* First icon */
+		icon = createicon(buf, width, height, size);
+		if (!icon)
+			goto fail;
+		item->icon = icon;
+		watcher_update_trays(watcher);
+
+	} else if (memcmp(item->icon->buf_orig, buf,
+	                  MIN(item->icon->size_orig, (size_t)size)) != 0) {
+		/* New icon */
+		destroyicon(item->icon);
+		item->icon = NULL;
+		icon = createicon(buf, width, height, size);
+		if (!icon)
+			goto fail;
+		item->icon = icon;
+		watcher_update_trays(watcher);
+
+	} else {
+		/* Icon didn't change */
+	}
+
+	dbus_message_unref(reply);
+	dbus_pending_call_unref(pending);
+	return;
+
+fail:
+	if (icon)
+		destroyicon(icon);
+	if (reply)
+		dbus_message_unref(reply);
+	if (pending)
+		dbus_pending_call_unref(pending);
+}
+
+static DBusHandlerResult
+handle_newicon(Item *item, DBusConnection *conn, DBusMessage *msg)
+{
+	const char *sender = dbus_message_get_sender(msg);
+
+	if (sender && strcmp(sender, item->busname) == 0) {
+		request_property(conn, item->busname, item->busobj,
+		                 "IconPixmap", SNI_IFACE, pixmap_ready_handler,
+		                 item);
+
+		return DBUS_HANDLER_RESULT_HANDLED;
+
+	} else {
+		return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
+	}
+}
+
+static DBusHandlerResult
+filter_bus(DBusConnection *conn, DBusMessage *msg, void *data)
+{
+	Item *item = data;
+
+	if (dbus_message_is_signal(msg, SNI_IFACE, "NewIcon"))
+		return handle_newicon(item, conn, msg);
+	else
+		return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
+}
+
+Item *
+createitem(const char *busname, const char *busobj, Watcher *watcher)
+{
+	DBusConnection *conn;
+	Item *item;
+	char *busname_dup = NULL;
+	char *busobj_dup = NULL;
+	char match_rule[RULEBSIZE];
+
+	item = calloc(1, sizeof(Item));
+	busname_dup = strdup(busname);
+	busobj_dup = strdup(busobj);
+	if (!item || !busname_dup || !busobj_dup)
+		goto fail;
+
+	conn = watcher->conn;
+	item->busname = busname_dup;
+	item->busobj = busobj_dup;
+	item->watcher = watcher;
+
+	request_property(conn, busname, busobj, "IconPixmap", SNI_IFACE,
+	                 pixmap_ready_handler, item);
+
+	request_property(conn, busname, busobj, "Id", SNI_IFACE,
+	                 id_ready_handler, item);
+
+	request_property(conn, busname, busobj, "Menu", SNI_IFACE,
+	                 menupath_ready_handler, item);
+
+	if (snprintf(match_rule, sizeof(match_rule), match_string, busname) >=
+	    RULEBSIZE) {
+		goto fail;
+	}
+
+	if (!dbus_connection_add_filter(conn, filter_bus, item, NULL))
+		goto fail;
+	dbus_bus_add_match(conn, match_rule, NULL);
+
+	return item;
+
+fail:
+	free(busname_dup);
+	free(busobj_dup);
+	return NULL;
+}
+
+void
+destroyitem(Item *item)
+{
+	DBusConnection *conn;
+	char match_rule[RULEBSIZE];
+
+	conn = item_get_connection(item);
+
+	if (snprintf(match_rule, sizeof(match_rule), match_string,
+	             item->busname) < RULEBSIZE) {
+		dbus_bus_remove_match(conn, match_rule, NULL);
+		dbus_connection_remove_filter(conn, filter_bus, item);
+	}
+	if (item->icon)
+		destroyicon(item->icon);
+	free(item->menu_busobj);
+	free(item->busname);
+	free(item->busobj);
+	free(item->appid);
+	free(item);
+}
+
+void
+item_activate(Item *item)
+{
+	DBusConnection *conn;
+	DBusMessage *msg = NULL;
+	dbus_int32_t x = 0, y = 0;
+
+	conn = item_get_connection(item);
+
+	if (!(msg = dbus_message_new_method_call(item->busname, item->busobj,
+	                                         SNI_IFACE, "Activate")) ||
+	    !dbus_message_append_args(msg, DBUS_TYPE_INT32, &x, DBUS_TYPE_INT32,
+	                              &y, DBUS_TYPE_INVALID) ||
+	    !dbus_connection_send_with_reply(conn, msg, NULL, -1)) {
+		goto fail;
+	}
+
+	dbus_message_unref(msg);
+	return;
+
+fail:
+	if (msg)
+		dbus_message_unref(msg);
+}