/* Copyright 2009,2010 Myhailo Danylenko

 *

 * Service discovery requests

 *

 * This file is part of mcabber-disco

 *

 * mcabber-disco is free software: you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation, either version 2 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program.  If not, see <http://www.gnu.org/licenses/>. */

#include <glib.h>

#include <loudmouth/loudmouth.h>

#include <string.h>

#include "disco.h"

#include "config.h"

//

// module description

//

void disco_init   (void);

void disco_uninit (void);

#define DESCRIPTION ( "Service discovery requests\nProvides command /disco" )

module_info_t info_disco = {

	.branch      = MCABBER_BRANCH,

	.api         = MCABBER_API_VERSION,

	.version     = PROJECT_VERSION,

	.description = DESCRIPTION,

	.requires    = NULL,

	.init        = disco_init,

	.uninit      = disco_uninit,

	.next        = NULL,

};

//

// private types

//

#define DISCO_INFO_REQUEST  ( 0 )

#define DISCO_ITEMS_REQUEST ( 1 )

// lm message handler userdata

typedef struct {

	LmMessageHandler *   reply_handler;

	gpointer             data;

	GDestroyNotify       notify;

	guint                type;

	union {

		disco_info_handler_t  info;

		disco_items_handler_t items;

	} handler;

} disco_request_t;

// user request disco handler userdata (common for info and items)

typedef struct {

	gchar *jid;

	gchar *node;

} disco_handler_t;

//

// globals

//

static gpointer disco_cmid     = NULL;

#endif

static guint    disco_cid      = 0;

static guint    disco_hid      = 0;

static GSList * disco_requests = NULL;

//

// destroyers

//

static void disco_request_free ( disco_request_t * cb )

{

	disco_requests = g_slist_remove ( disco_requests, cb );

	if ( cb -> reply_handler ) {

		lm_message_handler_invalidate ( cb -> reply_handler );

#ifdef HAVE_LM_CONNECTION_UNREGISTER_REPLY_HANDLER

		if ( lconnection )

			lm_connection_unregister_reply_handler ( lconnection, cb -> reply_handler );

#endif

	}

	if ( cb -> notify )

		cb -> notify ( cb -> data );

	g_slice_free ( disco_request_t, cb );

	return;

}

static void disco_handler_destroy_notify ( gpointer data )

{

	disco_handler_t * cb = data;

	g_free ( cb -> jid );

	g_free ( cb -> node );

	g_slice_free ( disco_handler_t, cb );

	return;

}

//

// lm reply handlers

//

static LmHandlerResult disco_reply_handler (LmMessageHandler *handler, LmConnection *connection, LmMessage *message, gpointer udata)

{

	disco_request_t * cb     = udata;

	LmHandlerResult   result = LM_HANDLER_RESULT_REMOVE_MESSAGE;

	switch (lm_message_get_sub_type (message)) {

	case LM_MESSAGE_SUB_TYPE_RESULT:

		{

			LmMessageNode *node = lm_message_get_node (message);

			node = lm_message_node_get_child (node, "query");

			if ( cb -> type == DISCO_INFO_REQUEST ) {

				GSList * identities = NULL;

				GSList * features   = NULL;

				// check xmlns

				if (!node || g_strcmp0 (lm_message_node_get_attribute (node, "xmlns"), NS_DISCO_INFO))

					break;

				// parse request results

				if (node->children)

					for (node = node->children; node; node = node->next)

						if (!strcasecmp (node->name, "identity")) {

							disco_identity_t *identity = g_slice_new (disco_identity_t);

							identity -> category = lm_message_node_get_attribute (node, "category");

							identity -> type     = lm_message_node_get_attribute (node, "type");

							identity -> name     = lm_message_node_get_attribute (node, "name");

							identity -> reserved = NULL;

							identities = g_slist_append (identities, identity);

						} else if (!strcasecmp (node->name, "feature"))

							features = g_slist_insert_sorted (features, (gpointer) lm_message_node_get_attribute (node, "var"), (GCompareFunc) g_strcmp0);

				// call handler

				cb -> handler.info (identities, features, cb -> data);

				{ // free resources

					GSList *iel;

					for (iel = identities; iel; iel = iel -> next)

						g_slice_free (disco_identity_t, iel -> data);

					g_slist_free (identities);

					g_slist_free (features);

				}

			} else { // items request

				GSList * items = NULL;

				// check xmlns

				if (!node || g_strcmp0 (lm_message_node_get_attribute (node, "xmlns"), NS_DISCO_ITEMS))

					break;

				// parse request results

				if (node->children)

					for (node = node->children; node; node = node->next)

						if (!strcasecmp (node->name, "item")) {

							disco_item_t *item = g_slice_new (disco_item_t);

							item -> name = lm_message_node_get_attribute (node, "name");

							item -> jid  = lm_message_node_get_attribute (node, "jid");

							item -> node = lm_message_node_get_attribute (node, "node");

							items = g_slist_append (items, item);

						}

				// call handler

				cb -> handler.items (items, cb -> data);

				{ // free resources

					GSList *iel;

					for (iel = items; iel; iel = iel -> next)

						g_slice_free (disco_item_t, iel -> data);

					g_slist_free (items);

				}

			}

		}

		break;

	case LM_MESSAGE_SUB_TYPE_ERROR:

		{

			LmMessageNode *node   = lm_message_get_node (message);

			const gchar   *from   = lm_message_node_get_attribute (node, "from");

			const gchar   *type;

			const gchar   *reason;

			node = lm_message_node_get_child (node, "error");

			type = lm_message_node_get_attribute (node, "type");

			if (node->children)

				reason = node->children->name;

			else

				reason = "undefined";

			// XXX: we need to inform user, but do we really need to print this on every possible error?

			if ( cb -> type == DISCO_INFO_REQUEST ) {

				scr_log_print (LPRINT_LOGNORM, "disco: Service info discovery for %s failed: %s - %s", from, type, reason);

				cb -> handler.info (NULL, NULL, cb -> data);

			} else {

				scr_log_print (LPRINT_LOGNORM, "disco: Service items discovery for %s failed: %s - %s", from, type, reason);

				cb -> handler.items (NULL, cb -> data);

			}

		}

		break;

	default:

		result = LM_HANDLER_RESULT_ALLOW_MORE_HANDLERS;

		break;

	}

	disco_request_free ( cb );

	return result;

}

//

// disco requests sending

//

#if 0

static void disco_features_cb (gconstpointer data, gpointer userdata)

{

	const gchar  *feature  = data;

	GSList      **features = userdata;

	*features = g_slist_insert_sorted (*features, (gpointer) feature, (GCompareFunc) g_strcmp0);

	return;

}

#endif

gpointer disco_info_request (const gchar *jid, const gchar *dnode, disco_info_handler_t handler, gpointer userdata, GDestroyNotify notify)

{

	if (!handler || !xmpp_is_online ()) {

		if ( handler )

			handler ( NULL, NULL, userdata ); // XXX

		if ( notify )

			notify ( userdata );

		return NULL;

	}

#if 0

	if (0 && !dnode) { // FIXME: no way to get identity(ies) from caps

		gchar       *bjid     = jidtodisp (jid);

		GSList      *buddy    = roster_find (bjid, jidsearch, ROSTER_TYPE_USER | ROSTER_TYPE_ROOM | ROSTER_TYPE_AGENT);

		const gchar *resource = strchr (jid, JID_RESOURCE_SEPARATOR);

		g_free (bjid);

		if (buddy) {

			const gchar *hash = buddy_resource_getcaps (BUDDATA(buddy), resource); // ?? will it all work?

			if (hash) { // cached result

				GSList *identities = NULL;

				GSList *features   = NULL;

				caps_foreach_feature (hash, disco_features_cb, &features);

				handler (identities, features, userdata);

				if (notify)

					notify (userdata);

				return NULL;

			}

		}

	}

#endif

	{ // send request

		LmMessage *        request;

		LmMessageNode *    node;

		LmMessageHandler * lhandler;

		GError *           error    = NULL;

		disco_request_t *  cb       = g_slice_new ( disco_request_t );

		request = lm_message_new_with_sub_type (jid, LM_MESSAGE_TYPE_IQ, LM_MESSAGE_SUB_TYPE_GET);

		node    = lm_message_get_node (request);

		node    = lm_message_node_add_child (node, "query", NULL);

		lm_message_node_set_attribute (node, "xmlns", NS_DISCO_INFO);

		if (dnode)

			lm_message_node_set_attribute (node, "node", dnode);

		lhandler = lm_message_handler_new (disco_reply_handler, cb, NULL);

		cb -> reply_handler = lhandler;

		cb -> type          = DISCO_INFO_REQUEST;

		cb -> handler.info  = handler;

		cb -> data          = userdata;

		cb -> notify        = notify;

		disco_requests = g_slist_append ( disco_requests, cb );

		lm_connection_send_with_reply (lconnection, request, lhandler, &error);

		if (error) {

			scr_log_print (LPRINT_DEBUG, "disco: Error sending disco request: %s.", error -> message);

			g_error_free (error);

			handler ( NULL, NULL, userdata ); // XXX

			disco_request_free ( cb );

			cb = NULL;

		}

		lm_message_handler_unref (lhandler);

		lm_message_unref (request);

		return cb;

	}

}

gpointer disco_items_request (const gchar *jid, const gchar *dnode, disco_items_handler_t handler, gpointer userdata, GDestroyNotify notify)

{

	if (!handler || !xmpp_is_online ()) {

		if ( handler )

			handler ( NULL, userdata ); // XXX

		if ( notify )

			notify ( userdata );

		return NULL;

	}

	{ // send request

		LmMessage *        request;

		LmMessageNode *    node;

		LmMessageHandler * lhandler;

		GError *           error    = NULL;

		disco_request_t *  cb       = g_slice_new (disco_request_t);

		request = lm_message_new_with_sub_type (jid, LM_MESSAGE_TYPE_IQ, LM_MESSAGE_SUB_TYPE_GET);

		node    = lm_message_get_node (request);

		node    = lm_message_node_add_child (node, "query", NULL);

		lm_message_node_set_attribute (node, "xmlns", NS_DISCO_ITEMS);

		if (dnode)

			lm_message_node_set_attribute (node, "node", dnode);

		lhandler = lm_message_handler_new (disco_reply_handler, cb, NULL);

		cb -> reply_handler = lhandler;

		cb -> type          = DISCO_ITEMS_REQUEST;

		cb -> handler.items = handler;

		cb -> data          = userdata;

		cb -> notify        = notify;

		disco_requests = g_slist_append ( disco_requests, cb );

		lm_connection_send_with_reply (lconnection, request, lhandler, &error);

		if (error) {

			scr_log_print (LPRINT_DEBUG, "disco: Error sending disco request: %s.", error -> message);

			g_error_free (error);

			handler ( NULL, userdata );

			disco_request_free ( cb );

			cb = NULL;

		}

		lm_message_handler_unref (lhandler);

		lm_message_unref (request);

		return cb;

	}

}