/*

 * xmpp_iqrequest.c -- Jabber IQ request handling

 *

 * Copyright (C) 2008-2009 Frank Zschockelt <mcabber@freakysoft.de>

 * Copyright (C) 2005-2009 Mikael Berthe <mikael@lilotux.net>

 *

 * This program 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, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307

 * USA

 */

#include <string.h>

#include <stdlib.h>

#include "xmpp_helper.h"

#include "xmpp_iq.h"

#include "screen.h"

#include "utils.h"

#include "settings.h"

#include "hooks.h"

#include "hbuf.h"

extern LmMessageNode *bookmarks;

extern LmMessageNode *rosternotes;

static LmHandlerResult cb_roster(LmMessageHandler *h, LmConnection *c,

                                 LmMessage *m, gpointer user_data);

static LmHandlerResult cb_version(LmMessageHandler *h, LmConnection *c,

                                  LmMessage *m, gpointer user_data);

static LmHandlerResult cb_time(LmMessageHandler *h, LmConnection *c,

                               LmMessage *m, gpointer user_data);

static LmHandlerResult cb_last(LmMessageHandler *h, LmConnection *c,

                               LmMessage *m, gpointer user_data);

static LmHandlerResult cb_vcard(LmMessageHandler *h, LmConnection *c,

                               LmMessage *m, gpointer user_data);

static struct IqRequestHandlers

{

  const gchar *xmlns;

  const gchar *querytag;

  LmHandleMessageFunction handler;

} iq_request_handlers[] = {

  {NS_ROSTER, "query", &cb_roster},

  {NS_VERSION,"query", &cb_version},

  {NS_TIME,   "query", &cb_time},

  {NS_LAST,   "query", &cb_last},

  {NS_VCARD,  "vCard", &cb_vcard},

  {NULL, NULL, NULL}

};

// Enum for vCard attributes

enum vcard_attr {

  vcard_home    = 1<<0,

  vcard_work    = 1<<1,

  vcard_postal  = 1<<2,

  vcard_voice   = 1<<3,

  vcard_fax     = 1<<4,

  vcard_cell    = 1<<5,

  vcard_inet    = 1<<6,

  vcard_pref    = 1<<7,

};

void xmpp_iq_request(const char *fulljid, const char *xmlns)

{

  LmMessage *iq;

  LmMessageNode *query;

  LmMessageHandler *handler;

  int i;

  iq = lm_message_new_with_sub_type(fulljid, LM_MESSAGE_TYPE_IQ,

                                    LM_MESSAGE_SUB_TYPE_GET);

  for (i = 0; strcmp(iq_request_handlers[i].xmlns, xmlns) != 0 ; ++i)

       ;

  query = lm_message_node_add_child(iq->node,

                                    iq_request_handlers[i].querytag,

                                    NULL);

  lm_message_node_set_attribute(query, "xmlns", xmlns);

  handler = lm_message_handler_new(iq_request_handlers[i].handler,

  lm_message_handler_unref(handler);

  lm_message_unref(iq);

}

//  This callback is reached when mcabber receives the first roster update

// after the connection.

static LmHandlerResult cb_roster(LmMessageHandler *h, LmConnection *c,

                                 LmMessage *m, gpointer user_data)

{

  LmMessageNode *x;

  const char *ns;

  // Only execute the hook if the roster has been successfully retrieved

  if (lm_message_get_sub_type(m) != LM_MESSAGE_SUB_TYPE_RESULT)

    return LM_HANDLER_RESULT_ALLOW_MORE_HANDLERS;

  x = lm_message_node_find_child(m->node, "query");

  if (!x)

    return LM_HANDLER_RESULT_ALLOW_MORE_HANDLERS;

  ns = lm_message_node_get_attribute(x, "xmlns");

  if (ns && !strcmp(ns, NS_ROSTER))

    handle_iq_roster(NULL, c, m, user_data);

  // Post-login stuff

  hk_postconnect();

  return LM_HANDLER_RESULT_REMOVE_MESSAGE;

}

static LmHandlerResult cb_version(LmMessageHandler *h, LmConnection *c,

                                  LmMessage *m, gpointer user_data)

{

  LmMessageNode *ansqry;

  const char *p, *bjid;

  char *buf, *tmp;

  // Check IQ result sender

  bjid = lm_message_get_from(m);

  if (!bjid)

    bjid = lm_connection_get_jid(lconnection); // No from means our JID...

  if (!bjid) {

    scr_LogPrint(LPRINT_LOGNORM, "Invalid IQ:version result (no sender name).");

    return LM_HANDLER_RESULT_REMOVE_MESSAGE;

  }

  // Check for error message

  if (lm_message_get_sub_type(m) == LM_MESSAGE_SUB_TYPE_ERROR) {

    scr_LogPrint(LPRINT_LOGNORM, "Received error IQ message (%s)", bjid);

    display_server_error(lm_message_node_get_child(m->node, "error"));

    return LM_HANDLER_RESULT_REMOVE_MESSAGE;

  }

  // Check message contents

  ansqry = lm_message_node_get_child(m->node, "query");

  if (!ansqry) {

    scr_LogPrint(LPRINT_LOGNORM, "Invalid IQ:version result from <%s>!", bjid);

    return LM_HANDLER_RESULT_REMOVE_MESSAGE;

  }

  buf = g_strdup_printf("Received IQ:version result from <%s>", bjid);

  scr_LogPrint(LPRINT_LOGNORM, "%s", buf);

  // bjid should now really be the "bare JID", let's strip the resource

  tmp = strchr(bjid, JID_RESOURCE_SEPARATOR);

  if (tmp) *tmp = '\0';

  scr_WriteIncomingMessage(bjid, buf, 0, HBB_PREFIX_INFO, 0);

  g_free(buf);

  // Get result data...

  p = lm_message_node_get_child_value(ansqry, "name");

  if (p) {

    buf = g_strdup_printf("Name:    %s", p);

    scr_WriteIncomingMessage(bjid, buf,

                             0, HBB_PREFIX_INFO | HBB_PREFIX_CONT, 0);

    g_free(buf);

  }

  p = lm_message_node_get_child_value(ansqry, "version");

  if (p) {

    buf = g_strdup_printf("Version: %s", p);

    scr_WriteIncomingMessage(bjid, buf,

                             0, HBB_PREFIX_INFO | HBB_PREFIX_CONT, 0);

    g_free(buf);

  }

  p = lm_message_node_get_child_value(ansqry, "os");

  if (p) {

    buf = g_strdup_printf("OS:      %s", p);

    scr_WriteIncomingMessage(bjid, buf,

                             0, HBB_PREFIX_INFO | HBB_PREFIX_CONT, 0);

    g_free(buf);

  }

  return LM_HANDLER_RESULT_REMOVE_MESSAGE;

}

static LmHandlerResult cb_time(LmMessageHandler *h, LmConnection *c,

                               LmMessage *m, gpointer user_data)

{

  LmMessageNode *ansqry;

  const char *p, *bjid;

  char *buf, *tmp;

  // Check IQ result sender

  bjid = lm_message_get_from(m);

  if (!bjid)

    bjid = lm_connection_get_jid(lconnection); // No from means our JID...

  if (!bjid) {

    scr_LogPrint(LPRINT_LOGNORM, "Invalid IQ:time result (no sender name).");

    return LM_HANDLER_RESULT_REMOVE_MESSAGE;

  }

  // Check for error message

  if (lm_message_get_sub_type(m) == LM_MESSAGE_SUB_TYPE_ERROR) {

    scr_LogPrint(LPRINT_LOGNORM, "Received error IQ message (%s)", bjid);

    display_server_error(lm_message_node_get_child(m->node, "error"));

    return LM_HANDLER_RESULT_REMOVE_MESSAGE;

  }

  // Check message contents

  ansqry = lm_message_node_get_child(m->node, "query");

  if (!ansqry) {

    scr_LogPrint(LPRINT_LOGNORM, "Invalid IQ:time result from <%s>!", bjid);

    return LM_HANDLER_RESULT_REMOVE_MESSAGE;

  }

  buf = g_strdup_printf("Received IQ:time result from <%s>", bjid);

  scr_LogPrint(LPRINT_LOGNORM, "%s", buf);

  // bjid should now really be the "bare JID", let's strip the resource

  tmp = strchr(bjid, JID_RESOURCE_SEPARATOR);

  if (tmp) *tmp = '\0';

  scr_WriteIncomingMessage(bjid, buf, 0, HBB_PREFIX_INFO, 0);

  g_free(buf);

  // Get result data...

  p = lm_message_node_get_child_value(ansqry, "utc");

  if (p) {

    buf = g_strdup_printf("UTC:  %s", p);

    scr_WriteIncomingMessage(bjid, buf,

                             0, HBB_PREFIX_INFO | HBB_PREFIX_CONT, 0);

    g_free(buf);

  }

  p = lm_message_node_get_child_value(ansqry, "tz");

  if (p) {

    buf = g_strdup_printf("TZ:   %s", p);

    scr_WriteIncomingMessage(bjid, buf,

                             0, HBB_PREFIX_INFO | HBB_PREFIX_CONT, 0);

    g_free(buf);

  }

  p = lm_message_node_get_child_value(ansqry, "display");

  if (p) {

    buf = g_strdup_printf("Time: %s", p);

    scr_WriteIncomingMessage(bjid, buf,

                             0, HBB_PREFIX_INFO | HBB_PREFIX_CONT, 0);

    g_free(buf);

  }

  return LM_HANDLER_RESULT_REMOVE_MESSAGE;

}

static LmHandlerResult cb_last(LmMessageHandler *h, LmConnection *c,

                               LmMessage *m, gpointer user_data)

{

  LmMessageNode *ansqry;

  const char *p, *bjid;

  char *buf, *tmp;

  // Check IQ result sender

  bjid = lm_message_get_from(m);

  if (!bjid)

    bjid = lm_connection_get_jid(lconnection); // No from means our JID...

  if (!bjid) {

    scr_LogPrint(LPRINT_LOGNORM, "Invalid IQ:last result (no sender name).");

    return LM_HANDLER_RESULT_REMOVE_MESSAGE;

  }

  // Check for error message

  if (lm_message_get_sub_type(m) == LM_MESSAGE_SUB_TYPE_ERROR) {

    scr_LogPrint(LPRINT_LOGNORM, "Received error IQ message (%s)", bjid);

    display_server_error(lm_message_node_get_child(m->node, "error"));

    return LM_HANDLER_RESULT_REMOVE_MESSAGE;

  }

  // Check message contents

  ansqry = lm_message_node_get_child(m->node, "query");

  if (!ansqry) {

    scr_LogPrint(LPRINT_LOGNORM, "Invalid IQ:version result from <%s>!", bjid);

    return LM_HANDLER_RESULT_REMOVE_MESSAGE;

  }

  buf = g_strdup_printf("Received IQ:last result from <%s>", bjid);

  scr_LogPrint(LPRINT_LOGNORM, "%s", buf);

  // bjid should now really be the "bare JID", let's strip the resource

  tmp = strchr(bjid, JID_RESOURCE_SEPARATOR);

  if (tmp) *tmp = '\0';