/*

 * jabglue.c    -- Jabber protocol handling

 *

 * Copyright (C) 2005 Mikael Berthe <bmikael@lists.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

 */

#define _GNU_SOURCE  /* We need glibc for strptime */

#include "../libjabber/jabber.h"

#include "jabglue.h"

#include "roster.h"

#include "screen.h"

#include "hooks.h"

#include "utils.h"

#include "settings.h"

#include "hbuf.h"

#include "histolog.h"

#define JABBERPORT      5222

#define JABBERSSLPORT   5223

#define JABBER_AGENT_GROUP "Jabber Agents"

jconn jc;

static time_t LastPingTime;

static unsigned int KeepaliveDelay;

static int s_id;

static int regmode, regdone;

static enum imstatus mystatus = offline;

static gchar *mystatusmsg;

static unsigned char online;

char imstatus2char[imstatus_size+1] = {

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

};

static enum {

  STATE_CONNECTING,

  STATE_GETAUTH,

  STATE_SENDAUTH,

  STATE_LOGGED

} jstate;

struct T_presence {

  enum imstatus st;

  const char *msg;

};

static void statehandler(jconn, int);

static void packethandler(jconn, jpacket);

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.

static char *jidtodisp(const char *jid)

{

  char *ptr;

  char *alias;

  alias = g_strdup(jid);

  if ((ptr = strchr(alias, '/')) != NULL) {

    *ptr = 0;

  }

  return alias;

}

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

                  const char *resource)

{

  char *jid = g_new(char, 3 +

                    strlen(username) + strlen(servername) + strlen(resource));

  strcpy(jid, username);

  if (!strchr(jid, '@')) {

    strcat(jid, "@");

    strcat(jid, servername);

  }

  strcat(jid, "/");

  strcat(jid, resource);

  return jid;

}

inline unsigned char jb_getonline(void)

{

  return online;

}

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

                 int ssl, const char *pass)

{

  char *utf8_jid;

  if (!port) {

    if (ssl)

      port = JABBERSSLPORT;

    else

      port = JABBERPORT;

  }

  jb_disconnect();

  utf8_jid = to_utf8(jid);

  if (!utf8_jid) return jc;

  s_id = 1;

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

  g_free(utf8_jid);

  /* 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;

  // announce it to  everyone else

  jb_setstatus(offline, NULL, "");

  // announce it to the user

  statehandler(jc, JCONN_STATE_OFF);

  jab_delete(jc);

  //free(jc); XXX

  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;

}

void jb_main()

{

  xmlnode x, z;

  char *cid;

  if (!online) {

    safe_usleep(10000);

    return;

  }

  if (jc && jc->state == JCONN_STATE_CONNECTING) {

    safe_usleep(75000);

    jab_start(jc);

    return;

  }

  jab_poll(jc, 50);

  if (jstate == STATE_CONNECTING) {

    if (jc) {

      x = jutil_iqnew(JPACKET__GET, NS_AUTH);

      cid = jab_getid(jc);

      xmlnode_put_attrib(x, "id", cid);

      // id = atoi(cid);

      z = xmlnode_insert_tag(xmlnode_get_tag(x, "query"), "username");

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

      jab_send(jc, x);

      xmlnode_free(x);

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

  }

  // Keepalive

  if (KeepaliveDelay) {

    time_t now;

    time(&now);

    if (now > LastPingTime + KeepaliveDelay)

      jb_keepalive();

  }

}

inline enum imstatus jb_getstatus()

{

  return mystatus;

}

inline const char *jb_getstatusmsg()

{

  return mystatusmsg;

}

static void roompresence(gpointer room, void *presencedata)

{

  const char *jid;

  const char *nickname;

  char *to;

  struct T_presence *pres = presencedata;

  if (!buddy_isresource(room))

    return;

  jid = buddy_getjid(room);

  if (!jid) return;

  nickname = buddy_getnickname(room);

  if (!nickname) return;

  to = g_strdup_printf("%s/%s", jid, nickname);

  jb_setstatus(pres->st, to, pres->msg);

  g_free(to);

}

//  presnew(status, recipient, message)

// Create an xmlnode with default presence attributes

// Note: the caller must free the node after use

static xmlnode presnew(enum imstatus st, const char *recipient,

                        const char *msg)

{

  unsigned int prio;

  xmlnode x;

  x = jutil_presnew(JPACKET__UNKNOWN, 0, 0);

  if (recipient)

    xmlnode_put_attrib(x, "to", recipient);

  switch(st) {

    case away:

        xmlnode_insert_cdata(xmlnode_insert_tag(x, "show"), "away",

                             (unsigned) -1);

        break;

    case dontdisturb:

        xmlnode_insert_cdata(xmlnode_insert_tag(x, "show"), "dnd",

                             (unsigned) -1);

        break;

    case freeforchat:

        xmlnode_insert_cdata(xmlnode_insert_tag(x, "show"), "chat",

                             (unsigned) -1);

        break;

    case notavail:

        xmlnode_insert_cdata(xmlnode_insert_tag(x, "show"), "xa",

                             (unsigned) -1);

        break;

    case invisible:

        xmlnode_put_attrib(x, "type", "invisible");

        break;

    case offline:

        xmlnode_put_attrib(x, "type", "unavailable");

        break;

    default:

        break;

  }

  prio = settings_opt_get_int("priority");

  if (prio) {

    char strprio[8];

    snprintf(strprio, 8, "%u", prio);

    xmlnode_insert_cdata(xmlnode_insert_tag(x, "priority"),

                         strprio, (unsigned) -1);

  }

  if (msg) {

    gchar *utf8_msg = to_utf8(msg);

    xmlnode_insert_cdata(xmlnode_insert_tag(x, "status"), utf8_msg,

                         (unsigned) -1);

    g_free(utf8_msg);

  }

  return x;

}

void jb_setstatus(enum imstatus st, const char *recipient, const char *msg)

{

  xmlnode x;

  struct T_presence room_presence;

  if (!online) return;

  if (msg) {

    // The status message has been specified.  We'll use it, unless it is

    // "-" which is a special case (option meaning "no status message").

    if (!strcmp(msg, "-"))

      msg = "";

  } else {

    // No status message specified; we'll use:

    // a) the default status message (if provided by the user);

    // b) the current status message;

    // c) no status message (i.e. an empty one).

    msg = settings_get_status_msg(st);

    if (!msg) {

      if (mystatusmsg)

        msg = mystatusmsg;

      else

        msg = "";

    }

  }

  x = presnew(st, recipient, msg);

  jab_send(jc, x);

  xmlnode_free(x);

  // If we didn't change our _global_ status, we are done

  if (recipient) return;

  // Send presence to chatrooms

  if (st != invisible) {

    room_presence.st = st;

    room_presence.msg = msg;

    foreach_buddy(ROSTER_TYPE_ROOM, &roompresence, &room_presence);

  }

  // We'll need to update the roster if we switch to/from offline because

  // we don't know the presences of buddies when offline...

  if (mystatus == offline || st == offline)

    update_roster = TRUE;

  hk_mystatuschange(0, mystatus, st, msg);

  mystatus = st;

  if (msg != mystatusmsg) {

    if (mystatusmsg)

      g_free(mystatusmsg);

    if (*msg)

      mystatusmsg = g_strdup(msg);

    else

      mystatusmsg = NULL;

  }

}

void jb_send_msg(const char *jid, const char *text, int type,

                 const char *subject)

{

  xmlnode x;

  gchar *strtype;

  gchar *buffer = to_utf8(text);

  if (!online) return;

  if (type == ROSTER_TYPE_ROOM)

    strtype = TMSG_GROUPCHAT;

  else

    strtype = TMSG_CHAT;

  x = jutil_msgnew(strtype, (char*)jid, NULL, (char*)buffer);