#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);
}