/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/*
 * Copyright (C) 2003-2007 Imendio AB
 * Copyright (C) 2007-2008 Collabora Ltd.
 *
 * 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.
 *
 * Authors: Xavier Claessens <xclaesse@gmail.com>
 */

#include <config.h>

#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <glib/gstdio.h>

#include "empathy-log-manager.h"
#include "empathy-log-source-empathy.h"
#include "empathy-contact.h"
#include "empathy-time.h"
#include "empathy-utils.h"

#define DEBUG_FLAG EMPATHY_DEBUG_OTHER
#include "empathy-debug.h"

#define LOG_DIR_CREATE_MODE       (S_IRUSR | S_IWUSR | S_IXUSR)
#define LOG_FILE_CREATE_MODE      (S_IRUSR | S_IWUSR)
#define LOG_DIR_CHATROOMS         "chatrooms"
#define LOG_FILENAME_SUFFIX       ".log"
#define LOG_TIME_FORMAT_FULL      "%Y%m%dT%H:%M:%S"
#define LOG_TIME_FORMAT           "%Y%m%d"
#define LOG_HEADER \
    "<?xml version='1.0' encoding='utf-8'?>\n" \
    "<?xml-stylesheet type=\"text/xsl\" href=\"empathy-log.xsl\"?>\n" \
    "<log>\n"

#define LOG_FOOTER \
    "</log>\n"

static gchar *
log_source_empathy_get_basedir (void)
{
  return g_build_path (G_DIR_SEPARATOR_S, g_get_home_dir (),
      ".gnome2", PACKAGE_NAME, "logs", NULL);
}

static gchar *
log_source_empathy_get_dir (EmpathyLogManager *manager,
                            McAccount *account,
                            const gchar *chat_id,
                            gboolean chatroom)
{
  const gchar *account_id;
  gchar *basedir, *log_dir;

  account_id = mc_account_get_unique_name (account);

  log_dir = log_source_empathy_get_basedir ();

  if (chatroom)
    basedir = g_build_path (G_DIR_SEPARATOR_S, log_dir, account_id,
        LOG_DIR_CHATROOMS, chat_id, NULL);
  else
    basedir = g_build_path (G_DIR_SEPARATOR_S, log_dir,
        account_id, chat_id, NULL);

  g_free (log_dir);

  return basedir;
}

static gchar *
log_source_empathy_get_timestamp_filename (void)
{
  time_t t;
  gchar *time_str;
  gchar *filename;

  t = empathy_time_get_current ();
  time_str = empathy_time_to_string_local (t, LOG_TIME_FORMAT);
  filename = g_strconcat (time_str, LOG_FILENAME_SUFFIX, NULL);

  g_free (time_str);

  return filename;
}

static gchar *
log_source_empathy_get_timestamp_from_message (EmpathyMessage *message)
{
  time_t t;

  t = empathy_message_get_timestamp (message);

  /* We keep the timestamps in the messages as UTC. */
  return empathy_time_to_string_utc (t, LOG_TIME_FORMAT_FULL);
}

static gchar *
log_source_empathy_get_filename (EmpathyLogManager *manager,
                                 McAccount *account,
                                 const gchar *chat_id,
                                 gboolean chatroom)
{
  gchar *basedir;
  gchar *timestamp;
  gchar *filename;

  basedir = log_source_empathy_get_dir (manager, account, chat_id, chatroom);
  timestamp = log_source_empathy_get_timestamp_filename ();
  filename = g_build_filename (basedir, timestamp, NULL);

  g_free (basedir);
  g_free (timestamp);

  return filename;
}

static void
log_source_empathy_add_message (EmpathyLogManager *manager,
                                const gchar *chat_id,
                                gboolean chatroom,
                                EmpathyMessage *message)
{
  FILE *file;
  McAccount *account;
  EmpathyContact *sender;
  const gchar *body_str;
  const gchar *str;
  EmpathyAvatar *avatar;
  gchar *avatar_token = NULL;
  gchar *filename;
  gchar *basedir;
  gchar *body;
  gchar *timestamp;
  gchar *contact_name;
  gchar *contact_id;
  TpChannelTextMessageType msg_type;

  g_return_if_fail (EMPATHY_IS_LOG_MANAGER (manager));
  g_return_if_fail (chat_id != NULL);
  g_return_if_fail (EMPATHY_IS_MESSAGE (message));

  sender = empathy_message_get_sender (message);
  account = empathy_contact_get_account (sender);
  body_str = empathy_message_get_body (message);
  msg_type = empathy_message_get_tptype (message);

  if (G_STR_EMPTY (body_str))
    return;

  filename = log_source_empathy_get_filename (manager, account, chat_id, chatroom);
  basedir = g_path_get_dirname (filename);
  if (!g_file_test (basedir, G_FILE_TEST_EXISTS | G_FILE_TEST_IS_DIR))
    {
      DEBUG ("Creating directory:'%s'", basedir);
      g_mkdir_with_parents (basedir, LOG_DIR_CREATE_MODE);
    }
  g_free (basedir);

  DEBUG ("Adding message: '%s' to file: '%s'", body_str, filename);

  if (!g_file_test (filename, G_FILE_TEST_EXISTS))
    {
      file = g_fopen (filename, "w+");
      if (file)
        g_fprintf (file, LOG_HEADER);

      g_chmod (filename, LOG_FILE_CREATE_MODE);
    }
  else
    {
      file = g_fopen (filename, "r+");
      if (file)
        fseek (file, - strlen (LOG_FOOTER), SEEK_END);
    }

  body = g_markup_escape_text (body_str, -1);
  timestamp = log_source_empathy_get_timestamp_from_message (message);

  str = empathy_contact_get_name (sender);
  contact_name = g_markup_escape_text (str, -1);

  str = empathy_contact_get_id (sender);
  contact_id = g_markup_escape_text (str, -1);

  avatar = empathy_contact_get_avatar (sender);
  if (avatar)
    avatar_token = g_markup_escape_text (avatar->token, -1);

  g_fprintf (file,
       "<message time='%s' id='%s' name='%s' token='%s' isuser='%s' type='%s'>"
       "%s</message>\n" LOG_FOOTER, timestamp, contact_id, contact_name,
       avatar_token ? avatar_token : "",
       empathy_contact_is_user (sender) ? "true" : "false",
       empathy_message_type_to_str (msg_type), body);

  fclose (file);
  g_free (filename);
  g_free (contact_id);
  g_free (contact_name);
  g_free (timestamp);
  g_free (body);
  g_free (avatar_token);
}

static gboolean
log_source_empathy_exists (EmpathyLogManager *manager,
                           McAccount *account,
                           const gchar *chat_id,
                           gboolean chatroom)
{
  gchar *dir;
  gboolean exists;

  dir = log_source_empathy_get_dir (manager, account, chat_id, chatroom);
  exists = g_file_test (dir, G_FILE_TEST_EXISTS | G_FILE_TEST_IS_DIR);
  g_free (dir);

  return exists;
}

static GList *
log_source_empathy_get_dates (EmpathyLogManager *manager,
                              McAccount *account,
                              const gchar *chat_id,
                              gboolean chatroom)
{
  GList *dates = NULL;
  gchar *date;
  gchar *directory;
  GDir *dir;
  const gchar *filename;
  const gchar *p;

  g_return_val_if_fail (EMPATHY_IS_LOG_MANAGER (manager), NULL);
  g_return_val_if_fail (MC_IS_ACCOUNT (account), NULL);
  g_return_val_if_fail (chat_id != NULL, NULL);

  directory = log_source_empathy_get_dir (manager, account, chat_id, chatroom);
  dir = g_dir_open (directory, 0, NULL);
  if (!dir)
    {
      DEBUG ("Could not open directory:'%s'", directory);
      g_free (directory);
      return NULL;
    }

  DEBUG ("Collating a list of dates in:'%s'", directory);

  while ((filename = g_dir_read_name (dir)) != NULL)
    {
      if (!g_str_has_suffix (filename, LOG_FILENAME_SUFFIX))
        continue;

      p = strstr (filename, LOG_FILENAME_SUFFIX);
      date = g_strndup (filename, p - filename);

      if (!date)
        continue;

      if (!g_regex_match_simple ("\\d{8}", date, 0, 0))
        continue;

      dates = g_list_insert_sorted (dates, date, (GCompareFunc) strcmp);
    }

  g_free (directory);
  g_dir_close (dir);

  DEBUG ("Parsed %d dates", g_list_length (dates));

  return dates;
}

static gchar *
log_source_empathy_get_filename_for_date (EmpathyLogManager *manager,
                                          McAccount *account,
                                          const gchar *chat_id,
                                          gboolean chatroom,
                                          const gchar *date)
{
  gchar *basedir;
  gchar *timestamp;
  gchar *filename;

  basedir = log_source_empathy_get_dir (manager, account, chat_id, chatroom);
  timestamp = g_strconcat (date, LOG_FILENAME_SUFFIX, NULL);
  filename = g_build_filename (basedir, timestamp, NULL);

  g_free (basedir);
  g_free (timestamp);

  return filename;
}

static EmpathyLogSearchHit *
log_source_empathy_search_hit_new (EmpathyLogManager *manager,
                                   const gchar *filename)
{
  EmpathyLogSearchHit *hit;
  const gchar *account_name;
  const gchar *end;
  gchar **strv;
  guint len;

  if (!g_str_has_suffix (filename, LOG_FILENAME_SUFFIX))
    return NULL;

  strv = g_strsplit (filename, G_DIR_SEPARATOR_S, -1);
  len = g_strv_length (strv);

  hit = g_slice_new0 (EmpathyLogSearchHit);

  end = strstr (strv[len-1], LOG_FILENAME_SUFFIX);
  hit->date = g_strndup (strv[len-1], end - strv[len-1]);
  hit->chat_id = g_strdup (strv[len-2]);
  hit->is_chatroom = (strcmp (strv[len-3], LOG_DIR_CHATROOMS) == 0);

  if (hit->is_chatroom)
    account_name = strv[len-4];
  else
    account_name = strv[len-3];

  hit->account = mc_account_lookup (account_name);
  hit->filename = g_strdup (filename);

  g_strfreev (strv);

  return hit;
}

static GList *
log_source_empathy_get_messages_for_file (EmpathyLogManager *manager,
                                          const gchar *filename)
{
  GList *messages = NULL;
  xmlParserCtxtPtr ctxt;
  xmlDocPtr doc;
  xmlNodePtr log_node;
  xmlNodePtr node;
  EmpathyLogSearchHit *hit;
  McAccount *account;

  g_return_val_if_fail (EMPATHY_IS_LOG_MANAGER (manager), NULL);
  g_return_val_if_fail (filename != NULL, NULL);

  DEBUG ("Attempting to parse filename:'%s'...", filename);

  if (!g_file_test (filename, G_FILE_TEST_EXISTS))
    {
      DEBUG ("Filename:'%s' does not exist", filename);
      return NULL;
    }

  /* Get the account from the filename */
  hit = log_source_empathy_search_hit_new (manager, filename);
  account = g_object_ref (hit->account);
  empathy_log_manager_search_hit_free (hit);

  /* Create parser. */
  ctxt = xmlNewParserCtxt ();

  /* Parse and validate the file. */
  doc = xmlCtxtReadFile (ctxt, filename, NULL, 0);
  if (!doc)
    {
      g_warning ("Failed to parse file:'%s'", filename);
      xmlFreeParserCtxt (ctxt);
      return NULL;
    }

  /* The root node, presets. */
  log_node = xmlDocGetRootElement (doc);
  if (!log_node)
    {
      xmlFreeDoc (doc);
      xmlFreeParserCtxt (ctxt);
      return NULL;
    }

  /* Now get the messages. */
  for (node = log_node->children; node; node = node->next)
    {
      EmpathyMessage *message;
      EmpathyContact *sender;
      gchar *time;
      time_t t;
      gchar *sender_id;
      gchar *sender_name;
      gchar *sender_avatar_token;
      gchar *body;
      gchar *is_user_str;
      gboolean is_user = FALSE;
      gchar *msg_type_str;
      TpChannelTextMessageType msg_type = TP_CHANNEL_TEXT_MESSAGE_TYPE_NORMAL;

      if (strcmp (node->name, "message") != 0)
        continue;

      body = xmlNodeGetContent (node);
      time = xmlGetProp (node, "time");
      sender_id = xmlGetProp (node, "id");
      sender_name = xmlGetProp (node, "name");
      sender_avatar_token = xmlGetProp (node, "token");
      is_user_str = xmlGetProp (node, "isuser");
      msg_type_str = xmlGetProp (node, "type");

      if (is_user_str)
        is_user = strcmp (is_user_str, "true") == 0;

      if (msg_type_str)
        msg_type = empathy_message_type_from_str (msg_type_str);

      t = empathy_time_parse (time);

      sender = empathy_contact_new_full (account, sender_id, sender_name);
      empathy_contact_set_is_user (sender, is_user);
      if (!G_STR_EMPTY (sender_avatar_token))
        empathy_contact_load_avatar_cache (sender,
            sender_avatar_token);

      message = empathy_message_new (body);
      empathy_message_set_sender (message, sender);
      empathy_message_set_timestamp (message, t);
      empathy_message_set_tptype (message, msg_type);

      messages = g_list_append (messages, message);

      g_object_unref (sender);
      xmlFree (time);
      xmlFree (sender_id);
      xmlFree (sender_name);
      xmlFree (body);
      xmlFree (is_user_str);
      xmlFree (msg_type_str);
    }

  DEBUG ("Parsed %d messages", g_list_length (messages));

  xmlFreeDoc (doc);
  xmlFreeParserCtxt (ctxt);

  return messages;
}

static GList *
log_source_empathy_get_all_files (EmpathyLogManager *manager,
                                  const gchar *dir)
{
  GDir *gdir;
  GList *files = NULL;
  const gchar *name;
  gchar *basedir;

  basedir = dir ? g_strdup (dir) : log_source_empathy_get_basedir ();

  gdir = g_dir_open (basedir, 0, NULL);
  if (!gdir)
    {
      g_free (basedir);
      return NULL;
    }

  while ((name = g_dir_read_name (gdir)) != NULL)
    {
      gchar *filename;

      filename = g_build_filename (basedir, name, NULL);
      if (g_str_has_suffix (filename, LOG_FILENAME_SUFFIX))
        {
          files = g_list_prepend (files, filename);
          continue;
        }

      if (g_file_test (filename, G_FILE_TEST_IS_DIR))
        {
          /* Recursively get all log files */
          files = g_list_concat (files, log_source_empathy_get_all_files (manager, filename));
        }

      g_free (filename);
    }

  g_dir_close (gdir);
  g_free (basedir);

  return files;
}

static GList *
log_source_empathy_search_new (EmpathyLogManager *manager,
                               const gchar *text)
{
  GList *files, *l;
  GList *hits = NULL;
  gchar *text_casefold;

  g_return_val_if_fail (EMPATHY_IS_LOG_MANAGER (manager), NULL);
  g_return_val_if_fail (!G_STR_EMPTY (text), NULL);

  text_casefold = g_utf8_casefold (text, -1);

  files = log_source_empathy_get_all_files (manager, NULL);
  DEBUG ("Found %d log files in total", g_list_length (files));

  for (l = files; l; l = l->next)
    {
      gchar *filename;
      GMappedFile *file;
      gsize length;
      gchar *contents;
      gchar *contents_casefold;

      filename = l->data;

      file = g_mapped_file_new (filename, FALSE, NULL);
      if (!file)
        continue;

      length = g_mapped_file_get_length (file);
      contents = g_mapped_file_get_contents (file);
      contents_casefold = g_utf8_casefold (contents, length);

      g_mapped_file_free (file);

      if (strstr (contents_casefold, text_casefold))
        {
          EmpathyLogSearchHit *hit;

          hit = log_source_empathy_search_hit_new (manager, filename);

          if (hit)
            {
              hits = g_list_prepend (hits, hit);
              DEBUG ("Found text:'%s' in file:'%s' on date:'%s'",
                  text, hit->filename, hit->date);
            }
        }

      g_free (contents_casefold);
      g_free (filename);
    }

  g_list_free (files);
  g_free (text_casefold);

  return hits;
}

static GList *
log_source_empathy_get_chats_for_dir (EmpathyLogManager *manager,
                                      const gchar *dir,
                                      gboolean is_chatroom)
{
  GDir *gdir;
  GList *hits = NULL;
  const gchar *name;

  gdir = g_dir_open (dir, 0, NULL);
  if (!gdir)
    return NULL;

  while ((name = g_dir_read_name (gdir)) != NULL)
    {
      EmpathyLogSearchHit *hit;
      gchar *filename;

      filename = g_build_filename (dir, name, NULL);
      if (strcmp (name, LOG_DIR_CHATROOMS) == 0)
        {
          hits = g_list_concat (hits, log_source_empathy_get_chats_for_dir (manager, dir, TRUE));
          g_free (filename);
          continue;
        }

      hit = g_slice_new0 (EmpathyLogSearchHit);
      hit->chat_id = g_strdup (name);
      hit->is_chatroom = is_chatroom;

      hits = g_list_prepend (hits, hit);
    }

  g_dir_close (gdir);

  return hits;
}


static GList *
log_source_empathy_get_messages_for_date (EmpathyLogManager *manager,
                                          McAccount *account,
                                          const gchar *chat_id,
                                          gboolean chatroom,
                                          const gchar *date)
{
  gchar *filename;
  GList *messages;

  g_return_val_if_fail (EMPATHY_IS_LOG_MANAGER (manager), NULL);
  g_return_val_if_fail (MC_IS_ACCOUNT (account), NULL);
  g_return_val_if_fail (chat_id != NULL, NULL);

  filename = log_source_empathy_get_filename_for_date (manager, account,
      chat_id, chatroom, date);
  messages = log_source_empathy_get_messages_for_file (manager, filename);
  g_free (filename);

  return messages;
}

static GList *
log_source_empathy_get_chats (EmpathyLogManager *manager,
                              McAccount *account)
{
  gchar *basedir;
  gchar *dir;
  GList *hits;

  basedir = log_source_empathy_get_basedir ();
  dir = g_build_filename (basedir, mc_account_get_unique_name (account),
      NULL);
  g_free (basedir);

  hits = log_source_empathy_get_chats_for_dir (manager, dir, FALSE);

  g_free (dir);

  return hits;
}

EmpathyLogSource *
empathy_log_source_empathy_get_source (void)
{
  EmpathyLogSource *source;
  source = g_slice_new0 (EmpathyLogSource);

  source->exists = log_source_empathy_exists;
  source->add_message = log_source_empathy_add_message;
  source->get_dates = log_source_empathy_get_dates;
  source->get_messages_for_date = log_source_empathy_get_messages_for_date;
  source->get_chats = log_source_empathy_get_chats;
  source->search_new = log_source_empathy_search_new;

  return source;
}