/*

 * jabglue.c    -- Jabber protocol handling

 *

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

 * Parts come from the centericq project:

 * Copyright (C) 2002-2005 by Konstantin Klyagin <konst@konst.org.ua>

 *

 * 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 "../libjabber/jabber.h"

#include "jabglue.h"

#include "jab_priv.h"

#include "roster.h"

#include "screen.h"

#include "hooks.h"

#include "utils.h"

#include "settings.h"

#include "hbuf.h"

#include "histolog.h"

#include "commands.h"

#include "pgp.h"

#include "otr.h"

#include "fifo.h"

#define JABBERPORT      5222

#define JABBERSSLPORT   5223

#define RECONNECTION_TIMEOUT    60L

jconn jc;

guint AutoConnection;

enum enum_jstate jstate;

char imstatus2char[imstatus_size+1] = {

    '_', 'o', 'f', 'd', 'n', 'a', 'i', '\0'

};

static char *imstatus_showmap[] = {

  "",

  "",

  "chat",

  "dnd",

  "xa",

  "away",

  ""

};

static time_t LastPingTime;

static unsigned int KeepaliveDelay;

static enum imstatus mystatus = offline;

static enum imstatus mywantedstatus = available;

static gchar *mystatusmsg;

static unsigned char online;

static void statehandler(jconn, int);

static void packethandler(jconn, jpacket);

static void handle_state_events(char* from, xmlnode xmldata);

static int evscallback_invitation(eviqs *evp, guint evcontext);

static void logger(jconn j, int io, const char *buf)

{

  scr_LogPrint(LPRINT_DEBUG, "%03s: %s", ((io == 0) ? "OUT" : "IN"), buf);

}

//  jidtodisp(jid)

// Strips the resource part from the jid

// The caller should g_free the result after use.

char *jidtodisp(const char *fjid)

{

  char *ptr;

  char *alias;

  alias = g_strdup(fjid);

  if ((ptr = strchr(alias, JID_RESOURCE_SEPARATOR)) != NULL) {

    *ptr = 0;

  }

  return alias;

}

char *compose_jid(const char *username, const char *servername,

                  const char *resource)

{

  char *fjid;

  if (!strchr(username, JID_DOMAIN_SEPARATOR)) {

    fjid = g_strdup_printf("%s%c%s%c%s", username,

                           JID_DOMAIN_SEPARATOR, servername,

                           JID_RESOURCE_SEPARATOR, resource);

  } else {

    fjid = g_strdup_printf("%s%c%s", username,

                           JID_RESOURCE_SEPARATOR, resource);

  }

  return fjid;

}

unsigned char jb_getonline(void)

{

  return online;

}

jconn jb_connect(const char *fjid, const char *server, unsigned int port,

                 int ssl, const char *pass)

{

  if (!port) {

    if (ssl)

      port = JABBERSSLPORT;

    else

      port = JABBERPORT;

  }

  jb_disconnect();

  if (!fjid) return jc;

  jc = jab_new((char*)fjid, (char*)pass, (char*)server, port, ssl);

  /* These 3 functions can deal with a NULL jc, no worry... */

  jab_logger(jc, logger);

  jab_packet_handler(jc, &packethandler);

  jab_state_handler(jc, &statehandler);

  if (jc && jc->user) {

    online = TRUE;

    jstate = STATE_CONNECTING;

    statehandler(0, -1);

    jab_start(jc);

  }

  return jc;

}

void jb_disconnect(void)

{

  if (!jc) return;

  if (online) {

    // Launch pre-disconnect internal hook

    hook_execute_internal("hook-pre-disconnect");

    // Announce it to  everyone else

    jb_setstatus(offline, NULL, "", FALSE);

    // End the XML flow

    jb_send_raw("</stream:stream>");

    /*

    // Free status message

    g_free(mystatusmsg);

    mystatusmsg = NULL;

    */

  }

  // Announce it to the user

  statehandler(jc, JCONN_STATE_OFF);

  jab_delete(jc);

  jc = NULL;

}

inline void jb_reset_keepalive()

{

  time(&LastPingTime);

}

void jb_send_raw(const char *str)

{

  if (jc && online && str)

    jab_send_raw(jc, str);

}

void jb_keepalive()

{

  if (jc && online)

    jab_send_raw(jc, "  \t  ");

  jb_reset_keepalive();

}

void jb_set_keepalive_delay(unsigned int delay)

{

  KeepaliveDelay = delay;

}

//  check_connection()

// Check if we've been disconnected for a while (predefined timeout),

// and if so try to reconnect.

static void check_connection(void)

{

  static time_t disconnection_timestamp = 0L;

  time_t now;

  // Maybe we're voluntarily offline...

  if (!AutoConnection)

    return;

  // Are we totally disconnected?

  if (jc && jc->state != JCONN_STATE_OFF) {

    disconnection_timestamp = 0L;

    return;

  }

  time(&now);

  if (!disconnection_timestamp) {

    disconnection_timestamp = now;

    return;

  }

  // If the reconnection_timeout is reached, try to reconnect.

  if (now > disconnection_timestamp + RECONNECTION_TIMEOUT) {

    mcabber_connect();

    disconnection_timestamp = 0L;

  }

}

void jb_main()

{

  time_t now;

  fd_set fds;

  long tmout;

  struct timeval tv;

  static time_t last_eviqs_check = 0L;

  int maxfd = 0;

  int fifofd;

  FD_ZERO(&fds);

  FD_SET(0, &fds);

  if (jc && jc->fd > 0) {

    FD_SET(jc->fd, &fds);

    maxfd = jc->fd;

  }

  fifofd = fifo_get_fd();

  if (fifofd > 0) {

    FD_SET(fifofd, &fds);

    maxfd = MAX(maxfd, fifofd);

  }

  tv.tv_sec = 60;

  tv.tv_usec = 0;

  if (!online || (jc && jc->state == JCONN_STATE_CONNECTING)) {

    if (online) {

      // We're connecting and we need to reduce the timeout.

      tv.tv_sec = 0;

      tv.tv_usec = 250000;

    } else {

      tv.tv_sec = 30;

      // Let's first update the screen, we could sleep for a long time...

      scr_DoUpdate();

    }

    // If we're not connected, sleep for a while...

    select(maxfd + 1, &fds, NULL, NULL, &tv);

    if (!online)

      check_connection();

    else

      jab_start(jc);

    return;

  }

  time(&now);

  if (KeepaliveDelay) {

    if (now >= LastPingTime + (time_t)KeepaliveDelay)

      tv.tv_sec = 0;

    else

      tv.tv_sec = LastPingTime + (time_t)KeepaliveDelay - now;

  }

  // Check auto-away timeout

  tmout = scr_GetAutoAwayTimeout(now);

  if (tv.tv_sec > tmout)

    tv.tv_sec = tmout;

#if defined JEP0022 || defined JEP0085

  // Check composing timeout

  tmout = scr_GetChatStatesTimeout(now);

  if (tv.tv_sec > tmout)

    tv.tv_sec = tmout;

#endif

  if (!tv.tv_sec)

    tv.tv_usec = 350000;

  scr_DoUpdate();

  if (select(maxfd + 1, &fds, NULL, NULL, &tv) > 0) {

    if (FD_ISSET(jc->fd, &fds))

      jab_poll(jc, 0);

  }

  if (jstate == STATE_CONNECTING) {

    if (jc) {

      eviqs *iqn;

      xmlnode z;

      iqn = iqs_new(JPACKET__GET, NS_AUTH, "auth", IQS_DEFAULT_TIMEOUT);

      iqn->callback = &iqscallback_auth;

      z = xmlnode_insert_tag(xmlnode_get_tag(iqn->xmldata, "query"),

                             "username");

      xmlnode_insert_cdata(z, jc->user->user, (unsigned) -1);

      jab_send(jc, iqn->xmldata);

      jstate = STATE_GETAUTH;

    }

    if (!jc || jc->state == JCONN_STATE_OFF) {

      scr_LogPrint(LPRINT_LOGNORM, "Unable to connect to the server");

      online = FALSE;

    }

  }

  if (!jc) {

    statehandler(jc, JCONN_STATE_OFF);

  } else if (jc->state == JCONN_STATE_OFF || jc->fd == -1) {

    statehandler(jc, JCONN_STATE_OFF);

  }

  time(&now);

  // Check for EV & IQ requests timeouts

  if (now > last_eviqs_check + 20) {

    iqs_check_timeout(now);

    evs_check_timeout(now);

    last_eviqs_check = now;

  }

  // Keepalive

  if (KeepaliveDelay) {

    if (now > LastPingTime + (time_t)KeepaliveDelay)

      jb_keepalive();

  }

}

inline enum imstatus jb_getstatus()

{

  return mystatus;

}

inline const char *jb_getstatusmsg()

{

  return mystatusmsg;

}

inline void update_last_use(void)

{

  iqlast = time(NULL);

}

//  insert_entity_capabilities(presence_stanza)

// Entity Capabilities (XEP-0115)

static void insert_entity_capabilities(xmlnode x)

{

  xmlnode y;

  const char *ver = entity_version();

  char *exts, *exts2;

  exts = NULL;

  y = xmlnode_insert_tag(x, "c");

  xmlnode_put_attrib(y, "xmlns", NS_CAPS);

  xmlnode_put_attrib(y, "node", MCABBER_CAPS_NODE);

  xmlnode_put_attrib(y, "ver", ver);

#ifdef JEP0085

  if (!chatstates_disabled) {

    exts2 = g_strjoin(" ", "csn", exts, NULL);

    g_free(exts);

    exts = exts2;

  }

#endif

  if (!settings_opt_get_int("iq_last_disable")) {

    exts2 = g_strjoin(" ", "iql", exts, NULL);

    g_free(exts);

    exts = exts2;

  }

  if (exts) {

    xmlnode_put_attrib(y, "ext", exts);

    g_free(exts);

  }

}

static void roompresence(gpointer room, void *presencedata)

{

  const char *bjid;

  const char *nickname;

  char *to;

  struct T_presence *pres = presencedata;

  if (!buddy_getinsideroom(room))