/*

 * hooks.c      -- Hooks layer

 *

 * Copyright (C) 2005-2010 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, see <http://www.gnu.org/licenses/>.

 */

#include <loudmouth/loudmouth.h>

#include <stdlib.h>

#include <string.h>

#include <sys/types.h>

#include <unistd.h>

#include "hooks.h"

#include "screen.h"

#include "roster.h"

#include "histolog.h"

#include "hbuf.h"

#include "settings.h"

#include "utils.h"

#include "utf8.h"

#include "commands.h"

#include "main.h"

#ifdef MODULES_ENABLE

#include <glib.h>

typedef struct {

  hk_handler_t handler;

  gint      priority;

  gpointer  userdata;

  guint     hid;

} hook_list_data_t;

static GHashTable *hk_handler_hash = NULL;

//  _new_hook_id()

// Return a unique Hook Id

static guint _new_hook_id(void)

{

  static guint hidcounter;

  return ++hidcounter;

}

//  _new_hook_queue(hookname)

// Create a new hash table entry with a GSList pointer for the specified hook

static GSList **_new_hook_queue(const gchar *hookname)

{

  GSList **p;

  // Create the hash table if needed.

  if (!hk_handler_hash) {

    hk_handler_hash = g_hash_table_new_full(&g_str_hash, &g_str_equal,

                                            &g_free, &g_free);

    if (!hk_handler_hash) {

      scr_log_print(LPRINT_LOGNORM, "Couldn't create hook hash table!");

      return NULL;

    }

  }

  // Add a queue for the requested hook

  p = g_new(GSList*, 1);

  *p = NULL;

  g_hash_table_insert(hk_handler_hash, g_strdup(hookname), p);

  return p;

}

static gint _hk_compare_prio(hook_list_data_t *a, hook_list_data_t *b)

{

  if (a->priority > b->priority)

    return 1;

  return 0;

}

//  hk_add_handler(handler, hookname, priority, userdata)

// Create a hook handler and a hook hash entry if needed.

// Return the handler id.

guint hk_add_handler(hk_handler_t handler, const gchar *hookname,

                     gint priority, gpointer userdata)

{

  GSList **hqueue = NULL;

  hook_list_data_t *h = g_new(hook_list_data_t, 1);

  h->handler  = handler;

  h->priority = priority;

  h->userdata = userdata;

  h->hid      = _new_hook_id();

  if (hk_handler_hash)

    hqueue = g_hash_table_lookup(hk_handler_hash, hookname);

  if (!hqueue)

    hqueue = _new_hook_queue(hookname);

  if (!hqueue)

    return 0;

  *hqueue = g_slist_insert_sorted(*hqueue, h, (GCompareFunc)_hk_compare_prio);

  return h->hid;

}

static gint _hk_queue_search_cb(hook_list_data_t *a, guint *hid)

{

  if (a->hid == *hid)

    return 0;

  return 1;

}

//  hk_del_handler(hookname, hook_id)

// Remove the handler with specified hook id from the hookname queue.

// The hash entry is removed if the queue is empty.

void hk_del_handler(const gchar *hookname, guint hid)

{

  GSList **hqueue;

  GSList *el;

  if (!hid)

    return;

  hqueue = g_hash_table_lookup(hk_handler_hash, hookname);

  if (!hqueue) {

    scr_log_print(LPRINT_LOGNORM, "*ERROR*: Couldn't remove hook handler!");

    return;

  }

  el = g_slist_find_custom(*hqueue, &hid,

                           (GCompareFunc)_hk_queue_search_cb);

  if (el) {

    g_free(el->data);

    *hqueue = g_slist_delete_link(*hqueue, el);

    // Remove hook hash table entry if the hook queue is empty

    if (!*hqueue)

      g_hash_table_remove(hk_handler_hash, hookname);

  }

}

//  hk_run_handlers(hookname, args)

// Process all hooks for the "hookname" event.

// Note that the processing is interrupted as soon as one of the handlers

// do not return HOOK_HANDLER_RESULT_ALLOW_MORE_HANDLERS (i.e. 0).

guint hk_run_handlers(const gchar *hookname, hk_arg_t *args)

{

  GSList **hqueue;

  GSList *h;

  guint ret = 0;

  if (!hk_handler_hash)

    return 0;

  hqueue = g_hash_table_lookup(hk_handler_hash, hookname);

  if (!hqueue)

    return 0; // Should we use a special code?

  for (h = *hqueue; h; h = g_slist_next(h)) {

    hook_list_data_t *data = h->data;

    ret = (data->handler)(hookname, args, data->userdata);

    if (ret) break;

  }

  return ret;

}

#endif

static char *extcmd;

static const char *COMMAND_ME = "/me ";

void hk_message_in(const char *bjid, const char *resname,

                   time_t timestamp, const char *msg, LmMessageSubType type,

                   guint encrypted, gboolean carbon)

{

  int new_guy = FALSE;

  int is_groupchat = FALSE; // groupchat message

  int is_room = FALSE;      // window is a room window

  int log_muc_conf = FALSE;

  int active_window = FALSE;

  int message_flags = 0;

  guint rtype = ROSTER_TYPE_USER;

  char *wmsg = NULL, *bmsg = NULL, *mmsg = NULL;

  GSList *roster_usr;

  unsigned mucnicklen = 0;

  const char *ename = NULL;

  gboolean attention = FALSE, mucprivmsg = FALSE;

  gboolean error_msg_subtype = (type == LM_MESSAGE_SUB_TYPE_ERROR);

#ifdef MODULES_ENABLE

  gchar strdelay[32];

  if (timestamp)

    to_iso8601(strdelay, timestamp);

  else

    strdelay[0] = '\0';

#endif

  if (encrypted == ENCRYPTED_PGP)

    message_flags |= HBB_PREFIX_PGPCRYPT;

  else if (encrypted == ENCRYPTED_OTR)

    message_flags |= HBB_PREFIX_OTRCRYPT;

  if (carbon)

    message_flags |= HBB_PREFIX_CARBON;

  if (type == LM_MESSAGE_SUB_TYPE_GROUPCHAT) {

    rtype = ROSTER_TYPE_ROOM;

    is_groupchat = TRUE;

    log_muc_conf = settings_opt_get_int("log_muc_conf");

    if (!resname) {

      message_flags = HBB_PREFIX_INFO | HBB_PREFIX_NOFLAG;

      resname = "";

      wmsg = bmsg = g_strdup_printf("~ %s", msg);

    } else {

      wmsg = bmsg = g_strdup_printf("<%s> %s", resname, msg);

      mucnicklen = strlen(resname) + 2;

      if (!strncmp(msg, COMMAND_ME, strlen(COMMAND_ME)))

        wmsg = mmsg = g_strdup_printf("*%s %s", resname, msg+4);

    }

  } else {

    bmsg = g_strdup(msg);

    if (!strncmp(msg, COMMAND_ME, strlen(COMMAND_ME))) {

      gchar *shortid = g_strdup(bjid);

      if (settings_opt_get_int("buddy_me_fulljid") == FALSE) {

        gchar *p = strchr(shortid, '@'); // Truncate the jid

        if (p)

          *p = '\0';

      }

      wmsg = mmsg = g_strdup_printf("*%s %s", shortid, msg+4);

      g_free(shortid);

    } else

      wmsg = (char*) msg;

  }

#ifdef MODULES_ENABLE

  {

    guint h_result;

    hk_arg_t args[] = {

      { "jid", bjid },

      { "resource", resname },

      { "message", msg },

      { "groupchat", is_groupchat ? "true" : "false" },

      { "delayed", strdelay },

      { "error", error_msg_subtype ? "true" : "false" },

      { "carbon", carbon ? "true" : "false" },

      { NULL, NULL },

    };

    h_result = hk_run_handlers(HOOK_PRE_MESSAGE_IN, args);

    if (h_result == HOOK_HANDLER_RESULT_NO_MORE_HANDLER_DROP_DATA) {

      scr_LogPrint(LPRINT_DEBUG, "Message dropped (hook result).");

      g_free(bmsg);

      g_free(mmsg);

      return;

    }

  }

#endif

  // If this user isn't in the roster, we add it

  roster_usr = roster_find(bjid, jidsearch, 0);

  if (!roster_usr) {

    new_guy = TRUE;

    roster_usr = roster_add_user(bjid, NULL, NULL, rtype, sub_none, -1);

    if (!roster_usr) { // Shouldn't happen...

      scr_LogPrint(LPRINT_LOGNORM, "ERROR: unable to add buddy!");

      g_free(bmsg);

      g_free(mmsg);

      return;

    }

  } else if (is_groupchat) {

    // Make sure the type is ROOM

    buddy_settype(roster_usr->data, ROSTER_TYPE_ROOM);

  }

  is_room = !!(buddy_gettype(roster_usr->data) & ROSTER_TYPE_ROOM);

  if (is_room) {

    if (!is_groupchat) {

      // This is a private message from a room participant

      g_free(bmsg);

      if (!resname) {

        resname = "";

        wmsg = bmsg = g_strdup(msg);

      } else {

        wmsg = bmsg = g_strdup_printf("PRIV#<%s> %s", resname, msg);

        if (!strncmp(msg, COMMAND_ME, strlen(COMMAND_ME))) {

          g_free(mmsg);

          wmsg = mmsg = g_strdup_printf("PRIV#*%s %s", resname, msg+4);

        }

        mucprivmsg = TRUE;

      }

      message_flags |= HBB_PREFIX_HLIGHT;

    } else {

      // This is a regular chatroom message.

      const char *nick = buddy_getnickname(roster_usr->data);

      if (nick) {

        // Let's see if we are the message sender, in which case we'll

        // highlight it.

        if (resname && !strcmp(resname, nick)) {

          message_flags |= HBB_PREFIX_HLIGHT_OUT;

        } else {

          // We're not the sender.  Can we see our nick?

          const char *msgptr = msg;

          while ((msgptr = strcasestr(msgptr, nick)) != NULL) {

            const char *leftb, *rightb;

            // The message contains our nick.  Let's check it's not

            // in the middle of another word (i.e. preceded/followed

            // immediately by an alphanumeric character or an underscore.

            rightb = msgptr+strlen(nick);

            if (msgptr == msg)

              leftb = NULL;

            else

              leftb = prev_char((char*)msgptr, msg);

            msgptr = next_char((char*)msgptr);

            // Check left boundary

            if (leftb && (iswalnum(get_char(leftb)) || get_char(leftb) == '_'))

              continue;

            // Check right boundary

            if (!iswalnum(get_char(rightb)) && get_char(rightb) != '_')

              attention = TRUE;

            if (attention && !settings_opt_get_int("muc_disable_nick_hl"))

              message_flags |= HBB_PREFIX_HLIGHT;

          }

        }

      }

    }

  } else if (settings_opt_get_int("roster_autolock_resource")) {

    buddy_setactiveresource(roster_usr->data, resname);

    scr_update_chat_status(FALSE);

  }

  if (error_msg_subtype) {

    message_flags = HBB_PREFIX_ERR | HBB_PREFIX_IN;

    scr_LogPrint(LPRINT_LOGNORM, "Error message received from <%s>", bjid);

  }

  // Note: the hlog_write should not be called first, because in some

  // cases scr_write_incoming_message() will load the history and we'd

  // have the message twice...

  scr_write_incoming_message(bjid, wmsg, timestamp, message_flags, mucnicklen);

  // Set urgent (a.k.a. "attention") flag

  {

    guint uip;

    if (is_groupchat) {

      if (attention)      uip = ROSTER_UI_PRIO_MUC_HL_MESSAGE;

      else                uip = ROSTER_UI_PRIO_MUC_MESSAGE;

    } else {

      if (mucprivmsg)     uip = ROSTER_UI_PRIO_MUC_PRIV_MESSAGE;

      else if (attention) uip = ROSTER_UI_PRIO_ATTENTION_MESSAGE;

      else                uip = ROSTER_UI_PRIO_PRIVATE_MESSAGE;

    }

    scr_setattentionflag_if_needed(bjid, FALSE, uip, prio_max);

  }

  // We don't log the modified message, but the original one

  if (wmsg == mmsg)

    wmsg = bmsg;

  // - We don't log the message if it is an error message

  // - We don't log the message if it is a private conf. message

  // - We don't log the message if it is groupchat message and the log_muc_conf

  //   option is off (and it is not a history line)

  if (!(message_flags & HBB_PREFIX_ERR) &&

      (!is_room || (is_groupchat && log_muc_conf && !timestamp)))

    hlog_write_message(bjid, timestamp, 0, wmsg);

  if (settings_opt_get_int("events_ignore_active_window") &&

      current_buddy && scr_get_chatmode()) {

    gpointer bud = BUDDATA(current_buddy);

    if (bud) {

      const char *cjid = buddy_getjid(bud);

      if (cjid && !strcasecmp(cjid, bjid))

        active_window = TRUE;

    }

  }