#include <string.h>
#include <gtk/gtk.h>
#include <libindicate/listener.h>

#include "im-menu-item.h"

static IndicateListener * listener;
static GHashTable * imHash;
#if 0
static GHashTable * mailHash;
#endif

typedef struct _imHash_t imHash_t;
struct _imHash_t {
	IndicateListenerServer * server;
	IndicateListenerIndicator * indicator;
};

static gboolean
imHash_equal (gconstpointer a, gconstpointer b)
{
	imHash_t * pa, * pb;

	pa = (imHash_t *)a;
	pb = (imHash_t *)b;

	gchar * pas = (gchar *)pa->server;
	gchar * pbs = (gchar *)pb->server;

	guint pai = GPOINTER_TO_UINT(pa->indicator);
	guint pbi = GPOINTER_TO_UINT(pb->indicator);

	return (!strcmp(pas, pbs)) && (pai == pbi);
}

static void
imHash_destroy (gpointer data)
{
	imHash_t * hasher = (imHash_t *)data;

	g_free(hasher);
}

static void
subtype_cb (IndicateListener * listener, IndicateListenerServer * server, IndicateListenerIndicator * indicator, gchar * property, gchar * propertydata, gpointer data)
{
	GtkMenuShell * menushell = GTK_MENU_SHELL(data);
	if (menushell == NULL) {
		g_error("Data in callback is not a menushell");
		return;
	}

	if (property == NULL || strcmp(property, "subtype")) {
		/* We should only ever get subtypes, but just in case */
		g_warning("Subtype callback got a property '%s'", property);
		return;
	}

	if (propertydata == NULL || propertydata[0] == '\0') {
		/* It's possible that this message didn't have a subtype.  That's
		 * okay, but we don't want to display those */
		g_debug("No subtype");
		return;
	}

	g_debug("Message subtype: %s", propertydata);

	if (!strcmp(propertydata, "im")) {
		imHash_t * hasher = g_new(imHash_t, 1);
		hasher->server = server;
		hasher->indicator = indicator;

		g_debug("Building IM Item");
		ImMenuItem * menuitem = im_menu_item_new(listener, server, indicator);
		g_object_ref(G_OBJECT(menuitem));

		g_debug("Adding to IM Hash");
		g_hash_table_insert(imHash, hasher, menuitem);

		g_debug("Placing in Shell");
		gtk_menu_shell_prepend(menushell, menuitem);
#if 0
	} else if (!strcmp(propertydata, "mail")) {
		gpointer pntr_menu_item;
		pntr_menu_item = g_hash_table_lookup(mailHash, server);
		if (pntr_menu_item == NULL) {
			/* If we don't know about it, we need a new menu item */
			GtkWidget * menuitem = mail_menu_item_new(listener, server);
			g_object_ref(menuitem);

			g_hash_table_insert(mailHash, server, menuitem);

			gtk_menu_shell_append(menushell, menuitem);
		} else {
			/* If we do, we need to increment the count */
			MailMenuItem * menu_item = MAIL_MENU_ITEM(pntr_menu_item);
			mail_menu_item_increment(menu_item);
		}
#endif
	}

	return;
}

static void
indicator_added (IndicateListener * listener, IndicateListenerServer * server, IndicateListenerIndicator * indicator, gchar * type, gpointer data)
{
	if (type == NULL || strcmp(type, "message")) {
		/* We only care about message type indicators
		   all of the others can go to the bit bucket */
		g_debug("Ignoreing indicator of type '%s'", type);
		return;
	}
	g_debug("Got a message");

	indicate_listener_get_property(listener, server, indicator, "subtype", subtype_cb, data);	
	return;
}

static void
indicator_removed (IndicateListener * listener, IndicateListenerServer * server, IndicateListenerIndicator * indicator, gchar * type, gpointer data)
{
	g_debug("Removing %s %d", (gchar*)server, (guint)indicator);
	if (type == NULL || strcmp(type, "message")) {
		/* We only care about message type indicators
		   all of the others can go to the bit bucket */
		g_debug("Ignoreing indicator of type '%s'", type);
		return;
	}

	gboolean removed = FALSE;

	/* Look in the IM Hash Table */
	imHash_t hasher;
	hasher.server = server;
	hasher.indicator = indicator;

	GtkWidget * menuitem = GTK_WIDGET(g_hash_table_lookup(imHash, &hasher));
	if (!removed && menuitem != NULL) {
		g_object_ref(menuitem);
		g_hash_table_remove(imHash, &hasher);

		gtk_widget_hide(menuitem);
		gtk_container_remove(GTK_CONTAINER(data), menuitem);

		g_object_unref(menuitem);
		removed = TRUE;
	}

	/* TODO: Look at mail */

	if (!removed) {
		g_warning("We were asked to remove %s %d but we didn't.", (gchar*)server, (guint)indicator);
	}

	return;
}

GtkWidget *
get_menu_item (void)
{
	listener = indicate_listener_new();
	imHash = g_hash_table_new_full(g_direct_hash, imHash_equal,
	                               imHash_destroy, g_object_unref);
#if 0
	mailHash = g_hash_table_new_full(g_direct_hash, g_direct_equal,
	                               NULL, g_object_unref);
#endif

	GtkWidget * mainmenu = gtk_menu_item_new_with_label("Message");

	GtkWidget * submenu = gtk_menu_new();
	gtk_menu_item_set_submenu(GTK_MENU_ITEM(mainmenu), submenu);
	gtk_widget_show(submenu);
	gtk_widget_show(mainmenu);

	g_signal_connect(listener, "indicator-added", G_CALLBACK(indicator_added), submenu);
	g_signal_connect(listener, "indicator-removed", G_CALLBACK(indicator_removed), submenu);

	return mainmenu;
}