/*

 * roster.c     -- Local roster implementation

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

 */

#define _GNU_SOURCE     /* for strcasestr() */

#include <string.h>

#include "roster.h"

/* This is a private structure type for the roster */

typedef struct {

  const gchar *name;

  const gchar *jid;

  const gchar *status_msg;

  guint type;

  enum imstatus status;

  guint flags;

  // list: user -> points to his group; group -> points to its users list

  GSList *list;

} roster;

/* ### Variables ### */

static int hide_offline_buddies;

static GSList *groups;

static GSList *unread_list;

GList *buddylist;

GList *current_buddy;

GList *alternate_buddy;

/* ### Roster functions ### */

// Comparison function used to search in the roster (compares jids and types)

static gint roster_compare_jid_type(roster *a, roster *b) {

  if (! (a->type & b->type))

    return -1; // arbitrary (but should be != , of course)

  return strcasecmp(a->jid, b->jid);

}

// Comparison function used to sort the roster (by name)

static gint roster_compare_name(roster *a, roster *b) {

  return strcasecmp(a->name, b->name);

}

// Finds a roster element (user, group, agent...), by jid or name

// If roster_type is 0, returns match of any type.

// Returns the roster GSList element, or NULL if jid/name not found

GSList *roster_find(const char *jidname, enum findwhat type, guint roster_type)

{

  GSList *sl_roster_elt = groups;

  GSList *res;

  roster sample;

  GCompareFunc comp;

  if (!jidname) return NULL;

  if (!roster_type)

    roster_type = ROSTER_TYPE_USER|ROSTER_TYPE_AGENT|ROSTER_TYPE_GROUP;

  sample.type = roster_type;

  if (type == jidsearch) {

    sample.jid = jidname;

    comp = (GCompareFunc)&roster_compare_jid_type;

  } else if (type == namesearch) {

    sample.name = jidname;

    comp = (GCompareFunc)&roster_compare_name;

  } else

    return NULL;    // should not happen

  while (sl_roster_elt) {

    roster *roster_elt = (roster*)sl_roster_elt->data;

    if (roster_type & ROSTER_TYPE_GROUP) {

      if ((type == namesearch) && !strcasecmp(jidname, roster_elt->name))

        return sl_roster_elt;

    } else {

      res = g_slist_find_custom(roster_elt->list, &sample, comp);

      if (res)

        return res;

    }

    sl_roster_elt = g_slist_next(sl_roster_elt);

  }

  return NULL;

}

// Returns pointer to new group, or existing group with that name

GSList *roster_add_group(const char *name)

{

  roster *roster_grp;

  // #1 Check name doesn't already exist

  if (!roster_find(name, namesearch, ROSTER_TYPE_GROUP)) {

    // #2 Create the group node

    roster_grp = g_new0(roster, 1);

    roster_grp->name = g_strdup(name);

    roster_grp->type = ROSTER_TYPE_GROUP;

    // #3 Insert (sorted)

    groups = g_slist_insert_sorted(groups, roster_grp,

            (GCompareFunc)&roster_compare_name);

  }

  return roster_find(name, namesearch, ROSTER_TYPE_GROUP);

}

// Returns a pointer to the new user, or existing user with that name

GSList *roster_add_user(const char *jid, const char *name, const char *group,

        guint type)

{

  roster *roster_usr;

  roster *my_group;

  GSList *slist;

  if ((type != ROSTER_TYPE_USER) && (type != ROSTER_TYPE_AGENT)) {

    // XXX Error message?

    return NULL;

  }

  // Let's be arbitrary: default group has an empty name ("").

  if (!group)  group = "";

  // #1 Check this user doesn't already exist

  slist = roster_find(jid, jidsearch, ROSTER_TYPE_USER|ROSTER_TYPE_AGENT);

  if (slist) return slist;

  // #2 add group if necessary

  slist = roster_add_group(group);

  if (!slist) return NULL;

  my_group = (roster*)slist->data;

  // #3 Create user node

  roster_usr = g_new0(roster, 1);

  roster_usr->jid   = g_strdup(jid);

  if (name) {

    roster_usr->name  = g_strdup(name);

  } else {

    gchar *p, *str = g_strdup(jid);

    p = strstr(str, "/");

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

    roster_usr->name = g_strdup(str);

    g_free(str);

  }

  roster_usr->type  = type; //ROSTER_TYPE_USER;

  roster_usr->list  = slist;    // (my_group SList element)

  // #4 Insert node (sorted)

  my_group->list = g_slist_insert_sorted(my_group->list, roster_usr,

          (GCompareFunc)&roster_compare_name);

  return roster_find(jid, jidsearch, type);

}

// Removes user (jid) from roster, frees allocated memory

void roster_del_user(const char *jid)

{

  GSList *sl_user, *sl_group;

  GSList **sl_group_listptr;

  roster *roster_usr;

  GSList *node;

  sl_user = roster_find(jid, jidsearch, ROSTER_TYPE_USER|ROSTER_TYPE_AGENT);

  if (sl_user == NULL)

    return;

  roster_usr = (roster*)sl_user->data;

  // Remove (if present) from unread messages list

  node = g_slist_find(unread_list, roster_usr);

  if (node) unread_list = g_slist_delete_link(unread_list, node);

  // Let's free memory (jid, name, status message)

  if (roster_usr->jid)        g_free((gchar*)roster_usr->jid);

  if (roster_usr->name)       g_free((gchar*)roster_usr->name);

  if (roster_usr->status_msg) g_free((gchar*)roster_usr->status_msg);

  g_free(roster_usr);

  // That's a little complex, we need to dereference twice

  sl_group = ((roster*)sl_user->data)->list;

  sl_group_listptr = &((roster*)(sl_group->data))->list;

  *sl_group_listptr = g_slist_delete_link(*sl_group_listptr, sl_user);

  // We need to rebuild the list

  if (current_buddy)

    buddylist_build();

  // TODO What we could do, too, is to check if the deleted node is

  // current_buddy, in which case we could move current_buddy to the

  // previous (or next) node.

}

// Free all roster data and call buddylist_build() to free the buddylist.

void roster_free(void)

{

  GSList *sl_grp = groups;

  // Free unread_list

  if (unread_list) {

    g_slist_free(unread_list);

    unread_list = NULL;

  }

  // Walk through groups

  while (sl_grp) {

    roster *roster_grp = (roster*)sl_grp->data;

    GSList *sl_usr = roster_grp->list;

    // Walk through this group users

    while (sl_usr) {

      roster *roster_usr = (roster*)sl_usr->data;

      // Free name and jid

      if (roster_usr->jid)        g_free((gchar*)roster_usr->jid);

      if (roster_usr->name)       g_free((gchar*)roster_usr->name);

      if (roster_usr->status_msg) g_free((gchar*)roster_usr->status_msg);

      g_free(roster_usr);

      sl_usr = g_slist_next(sl_usr);

    }

    // Free group's users list

    if (roster_grp->list)

      g_slist_free(roster_grp->list);

    // Free group's name and jid

    if (roster_grp->jid)  g_free((gchar*)roster_grp->jid);

    if (roster_grp->name) g_free((gchar*)roster_grp->name);

    g_free(roster_grp);

    sl_grp = g_slist_next(sl_grp);

  }

  // Free groups list

  if (groups) {

    g_slist_free(groups);

    groups = NULL;

    // Update (i.e. free) buddylist

    if (buddylist)

      buddylist_build();

  }

}

void roster_setstatus(const char *jid, enum imstatus bstat,

        const char *status_msg)

{

  GSList *sl_user;

  roster *roster_usr;

  sl_user = roster_find(jid, jidsearch, ROSTER_TYPE_USER|ROSTER_TYPE_AGENT);

  // If we can't find it, we add it

  if (sl_user == NULL)

    sl_user = roster_add_user(jid, NULL, NULL, ROSTER_TYPE_USER);

  roster_usr = (roster*)sl_user->data;

  roster_usr->status = bstat;

  if (roster_usr->status_msg) {

    g_free((gchar*)roster_usr->status_msg);

    roster_usr->status_msg = NULL;

  }

  if (status_msg)

    roster_usr->status_msg = g_strdup(status_msg);

}

//  roster_setflags()

// Set one or several flags to value (TRUE/FALSE)

void roster_setflags(const char *jid, guint flags, guint value)

{

  GSList *sl_user;

  roster *roster_usr;

  sl_user = roster_find(jid, jidsearch, ROSTER_TYPE_USER|ROSTER_TYPE_AGENT);

  if (sl_user == NULL)

    return;

  roster_usr = (roster*)sl_user->data;

  if (value)

    roster_usr->flags |= flags;

  else

    roster_usr->flags &= ~flags;

}

//  roster_msg_setflag()

// Set the ROSTER_FLAG_MSG to the given value for the given jid.

// It will update the buddy's group message flag.

// Update the unread messages list too.

void roster_msg_setflag(const char *jid, guint value)

{

  GSList *sl_user;

  roster *roster_usr, *roster_grp;

  sl_user = roster_find(jid, jidsearch, ROSTER_TYPE_USER|ROSTER_TYPE_AGENT);

  if (sl_user == NULL)

    return;

  roster_usr = (roster*)sl_user->data;

  roster_grp = (roster*)roster_usr->list->data;

  if (value) {

    // Message flag is TRUE.  This is easy, we just have to set both flags

    // to TRUE...

    roster_usr->flags |= ROSTER_FLAG_MSG;

    roster_grp->flags |= ROSTER_FLAG_MSG; // group

    // Append the roster_usr to unread_list, but avoid duplicates

    if (!g_slist_find(unread_list, roster_usr))

      unread_list = g_slist_append(unread_list, roster_usr);

  } else {

    // Message flag is FALSE.

    guint msg = FALSE;

    roster_usr->flags &= ~ROSTER_FLAG_MSG;

    if (unread_list) {

      GSList *node = g_slist_find(unread_list, roster_usr);

      if (node) unread_list = g_slist_delete_link(unread_list, node);

    }

    // For the group value we need to watch all buddies in this group;

    // if one is flagged, then the group will be flagged.

    // I will re-use sl_user and roster_usr here, as they aren't used

    // anymore.

    sl_user = roster_grp->list;

    while (sl_user) {

      roster_usr = (roster*)sl_user->data;

      if (roster_usr->flags & ROSTER_FLAG_MSG) {

        msg = TRUE;

        break;

      }

      sl_user = g_slist_next(sl_user);

    }

    if (!msg)

      roster_grp->flags &= ~ROSTER_FLAG_MSG;

    else

      roster_grp->flags |= ROSTER_FLAG_MSG;

      // Actually the "else" part is useless, because the group

      // ROSTER_FLAG_MSG should already be set...

  }

}

void roster_settype(const char *jid, guint type)

{

  GSList *sl_user;

  roster *roster_usr;

  if ((sl_user = roster_find(jid, jidsearch, 0)) == NULL)

    return;

  roster_usr = (roster*)sl_user->data;

  roster_usr->type = type;

}

enum imstatus roster_getstatus(const char *jid)

{

  GSList *sl_user;

  roster *roster_usr;

  sl_user = roster_find(jid, jidsearch, ROSTER_TYPE_USER|ROSTER_TYPE_AGENT);

  if (sl_user == NULL)

    return offline; // Not in the roster, anyway...

  roster_usr = (roster*)sl_user->data;

  return roster_usr->status;

}

const char *roster_getstatusmsg(const char *jid)

{

  GSList *sl_user;

  roster *roster_usr;

  sl_user = roster_find(jid, jidsearch, ROSTER_TYPE_USER|ROSTER_TYPE_AGENT);

  if (sl_user == NULL)

    return offline; // Not in the roster, anyway...

  roster_usr = (roster*)sl_user->data;

  return roster_usr->status_msg;

}

guint roster_gettype(const char *jid)

{

  GSList *sl_user;

  roster *roster_usr;

  if ((sl_user = roster_find(jid, jidsearch, 0)) == NULL)

    return 0;

  roster_usr = (roster*)sl_user->data;

  return roster_usr->type;

}

inline guint roster_exists(const char *jidname, enum findwhat type,

        guint roster_type)

{

  if (roster_find(jidname, type, roster_type))

    return TRUE;

  return FALSE;

}