path: root/libempathy-gtk/empathy-theme-adium.c

/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/*
 * Copyright (C) 2008 Collabora Ltd.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 * Authors: Xavier Claessens <xclaesse@gmail.com>
 */

#include "config.h"
#include <string.h>

#include <libempathy/empathy-time.h>
#include <libempathy/empathy-utils.h>

#include "empathy-theme-adium.h"

#define DEBUG_FLAG EMPATHY_DEBUG_CHAT
#include <libempathy/empathy-debug.h>

#define GET_PRIV(obj) EMPATHY_GET_PRIV (obj, EmpathyThemeAdium)

typedef struct {
    gchar          *in_content_html;
    gsize           in_content_len;
    gchar          *in_nextcontent_html;
    gsize           in_nextcontent_len;
    gchar          *out_content_html;
    gsize           out_content_len;
    gchar          *out_nextcontent_html;
    gsize           out_nextcontent_len;
    EmpathyContact *last_contact;
    gboolean        ready;
    GList          *message_queue;
    gchar          *default_avatar_filename;
} EmpathyThemeAdiumPriv;

static void theme_adium_iface_init (EmpathyChatViewIface *iface);

G_DEFINE_TYPE_WITH_CODE (EmpathyThemeAdium, empathy_theme_adium,
             WEBKIT_TYPE_WEB_VIEW,
             G_IMPLEMENT_INTERFACE (EMPATHY_TYPE_CHAT_VIEW,
                        theme_adium_iface_init));

static void
theme_adium_load (EmpathyThemeAdium *theme)
{
    EmpathyThemeAdiumPriv *priv = GET_PRIV (theme);
    gchar                 *basedir;
    gchar                 *file;
    gchar                 *template_html;
    gsize                  template_len;
    GString               *string;
    gchar                **strv;
    gchar                 *content;
    gchar                 *css_path;

    /* FIXME: Find a better way to get the theme dir */
    basedir = g_build_filename (g_get_home_dir (), "Contents", "Resources", NULL);

    /* Load html files */
    file = g_build_filename (basedir, "Template.html", NULL);
    g_file_get_contents (file, &template_html, &template_len, NULL);
    g_free (file);

    file = g_build_filename (basedir, "Incoming", "Content.html", NULL);
    g_file_get_contents (file, &priv->in_content_html, &priv->in_content_len, NULL);
    g_free (file);

    file = g_build_filename (basedir, "Incoming", "NextContent.html", NULL);
    g_file_get_contents (file, &priv->in_nextcontent_html, &priv->in_nextcontent_len, NULL);
    g_free (file);

    file = g_build_filename (basedir, "Outgoing", "Content.html", NULL);
    g_file_get_contents (file, &priv->out_content_html, &priv->out_content_len, NULL);
    g_free (file);

    file = g_build_filename (basedir, "Outgoing", "NextContent.html", NULL);
    g_file_get_contents (file, &priv->out_nextcontent_html, &priv->out_nextcontent_len, NULL);
    g_free (file);

    css_path = g_build_filename (basedir, "main.css", NULL);

    /* Replace %@ with the needed information in the template html */
    strv = g_strsplit (template_html, "%@", 5);
    string = g_string_sized_new (template_len);
    g_string_append (string, strv[0]);
    g_string_append (string, basedir);
    g_string_append (string, strv[1]);
    g_string_append (string, css_path);
    g_string_append (string, strv[2]);
    g_string_append (string, ""); /* We don't want header */
    g_string_append (string, strv[3]);
    g_string_append (string, ""); /* We have no footer */
    g_string_append (string, strv[4]);
    content = g_string_free (string, FALSE);

    /* Load the template */
    webkit_web_view_load_html_string (WEBKIT_WEB_VIEW (theme),
                      content, basedir);

    g_free (basedir);
    g_free (content);
    g_free (template_html);
    g_free (css_path);
    g_strfreev (strv);
}

static gchar *
theme_adium_escape_script (const gchar *text)
{
    const gchar *cur = text;
    GString     *string;

    string = g_string_sized_new (strlen (text));
    while (!G_STR_EMPTY (cur)) {
        switch (*cur) {
        case '\\':
            g_string_append (string, "\\\\");   
            break;
        case '\"':
            g_string_append (string, "\\\"");
            break;
        case '\r':
            g_string_append (string, "<br/>");
            break;
        case '\n':
            break;
        default:
            g_string_append_c (string, *cur);
        }
        cur++;
    }

    return g_string_free (string, FALSE);
}

static gchar *
theme_adium_escape_body (const gchar *body)
{
    gchar *ret, *iter;

    /* Replace \n by \r so it will be replaced by <br/> */
    ret = g_strdup (body);
    for (iter = ret; *iter != '\0'; iter++) {
        if (*iter == '\n') {
            *iter = '\r';
        }
    }

    return ret;
}

static const gchar *
theme_adium_get_default_avatar_filename (EmpathyThemeAdium *theme)
{
    EmpathyThemeAdiumPriv *priv = GET_PRIV (theme);
    GtkIconTheme          *icon_theme;
    GtkIconInfo           *icon_info;
    gint                   w, h;
    gint                   size = 48;

    /* Lazy initialization */
    if (priv->default_avatar_filename) {
        return priv->default_avatar_filename;
    }

    icon_theme = gtk_icon_theme_get_default ();

    if (gtk_icon_size_lookup (GTK_ICON_SIZE_DIALOG, &w, &h)) {
        size = (w + h) / 2;
    }

    icon_info = gtk_icon_theme_lookup_icon (icon_theme, "stock_person", size, 0);
    priv->default_avatar_filename = g_strdup (gtk_icon_info_get_filename (icon_info));
    gtk_icon_info_free (icon_info);

    return priv->default_avatar_filename;
}

static void
theme_adium_scroll_down (EmpathyChatView *view)
{
    /* Not implemented */
}

#define FOLLOW(cur, str) (!strncmp (cur, str, strlen (str)))
static void
theme_adium_append_message (EmpathyChatView *view,
                EmpathyMessage  *msg)
{
    EmpathyThemeAdium     *theme = EMPATHY_THEME_ADIUM (view);
    EmpathyThemeAdiumPriv *priv = GET_PRIV (theme);
    EmpathyContact        *sender;
    gchar                 *body;
    const gchar           *name;
    EmpathyAvatar         *avatar;
    const gchar           *avatar_filename = NULL;
    time_t                 timestamp;
    gsize                  len;
    GString               *string;
    gchar                 *cur = NULL;
    gchar                 *prev;
    gchar                 *script;
    gchar                 *escape;
    const gchar           *func;

    if (!priv->ready) {
        priv->message_queue = g_list_prepend (priv->message_queue,
                              g_object_ref (msg));
        return;
    }

    /* Get information */
    sender = empathy_message_get_sender (msg);
    timestamp = empathy_message_get_timestamp (msg);
    body = theme_adium_escape_body (empathy_message_get_body (msg));
    name = empathy_contact_get_name (sender);
    avatar = empathy_contact_get_avatar (sender);
    if (avatar) {
        avatar_filename = avatar->filename;
    }
    if (!avatar_filename) {
        avatar_filename = theme_adium_get_default_avatar_filename (theme);
    }

    /* Get the right html/func to add the message */
    if (priv->last_contact &&
        empathy_contact_equal (priv->last_contact, sender)) {
        func = "appendNextMessage";
        if (empathy_contact_is_user (sender)) {
            cur = priv->out_nextcontent_html;
            len = priv->out_nextcontent_len;
        }
        if (!cur) {
            cur = priv->in_nextcontent_html;
            len = priv->in_nextcontent_len;
        }
    }
    if (!cur) {
        func = "appendMessage";
        if (empathy_contact_is_user (sender)) {
            cur = priv->out_content_html;
            len = priv->out_content_len;
        }
        if (!cur) {
            cur = priv->in_content_html;
            len = priv->in_content_len;
        }
    }

    /* Make some search-and-replece in the html code */
    prev = cur;
    string = g_string_sized_new (len + strlen (body));
    while ((cur = strchr (cur, '%'))) {
        const gchar *replace = NULL;
        gchar       *dup_replace = NULL;
        gchar       *fin = NULL;

        if (FOLLOW (cur, "%message%")) {
            replace = body;
        } else if (FOLLOW (cur, "%userIconPath%")) {
            replace = avatar_filename;
        } else if (FOLLOW (cur, "%sender%")) {
            replace = name;
        } else if (FOLLOW (cur, "%time")) {
            gchar *format = NULL;
            gchar *start;
            gchar *end;

            /* Extract the time format it provided. */
            if (*(start = cur + strlen("%time")) == '{') {
                start++;
                end = strstr (start, "}%");
                if (!end) /* Invalid string */
                    continue;
                format = g_strndup (start, end - start);
                fin = end + 1;
            } 

            dup_replace = empathy_time_to_string_local (timestamp,
                format ? format : EMPATHY_TIME_FORMAT_DISPLAY_SHORT);
            replace = dup_replace;
            g_free (format);
        } else {
            cur++;
            continue;
        }

        /* Here we have a replacement to make */
        g_string_append_len (string, prev, cur - prev);
        g_string_append (string, replace);
        g_free (dup_replace);

        /* And update the pointers */
        if (fin) {
            prev = cur = fin + 1;
        } else {
            prev = cur = strchr (cur + 1, '%') + 1;
        }
    }
    g_string_append (string, prev);

    /* Execute a js to add the message */
    cur = g_string_free (string, FALSE);
    escape = theme_adium_escape_script (cur);
    script = g_strdup_printf("%s(\"%s\")", func, escape);
    webkit_web_view_execute_script (WEBKIT_WEB_VIEW (view), script);

    /* Keep the sender of the last displayed message */
    if (priv->last_contact) {
        g_object_unref (priv->last_contact);
    }
    priv->last_contact = g_object_ref (sender);

    g_free (body);
    g_free (cur);
    g_free (script);
}
#undef FOLLOW

static void
theme_adium_append_event (EmpathyChatView *view,
              const gchar     *str)
{
    /* Not implemented */
}

static void
theme_adium_scroll (EmpathyChatView *view,
            gboolean         allow_scrolling)
{
    /* Not implemented */
}

static gboolean
theme_adium_get_has_selection (EmpathyChatView *view)
{
    /* Not implemented */
    return FALSE;
}

static void
theme_adium_clear (EmpathyChatView *view)
{
    /* Not implemented */
}

static gboolean
theme_adium_find_previous (EmpathyChatView *view,
               const gchar     *search_criteria,
               gboolean         new_search)
{
    /* Not implemented */
    return FALSE;
}

static gboolean
theme_adium_find_next (EmpathyChatView *view,
               const gchar     *search_criteria,
               gboolean         new_search)
{
    /* Not implemented */
    return FALSE;
}

static void
theme_adium_find_abilities (EmpathyChatView *view,
                const gchar    *search_criteria,
                gboolean       *can_do_previous,
                gboolean       *can_do_next)
{
    /* Not implemented */
}

static void
theme_adium_highlight (EmpathyChatView *view,
               const gchar     *text)
{
    /* Not implemented */
}

static void
theme_adium_copy_clipboard (EmpathyChatView *view)
{
    /* Not implemented */
}

static void
theme_adium_iface_init (EmpathyChatViewIface *iface)
{
    iface->append_message = theme_adium_append_message;
    iface->append_event = theme_adium_append_event;
    iface->scroll = theme_adium_scroll;
    iface->scroll_down = theme_adium_scroll_down;
    iface->get_has_selection = theme_adium_get_has_selection;
    iface->clear = theme_adium_clear;
    iface->find_previous = theme_adium_find_previous;
    iface->find_next = theme_adium_find_next;
    iface->find_abilities = theme_adium_find_abilities;
    iface->highlight = theme_adium_highlight;
    iface->copy_clipboard = theme_adium_copy_clipboard;
}

static void
theme_adium_load_finished_cb (WebKitWebView  *view,
                  WebKitWebFrame *frame,
                  gpointer        user_data)
{
    EmpathyThemeAdiumPriv *priv = GET_PRIV (view);
    EmpathyChatView       *chat_view = EMPATHY_CHAT_VIEW (view);

    DEBUG ("Page loaded");
    priv->ready = TRUE;

    /* Display queued messages */
    priv->message_queue = g_list_reverse (priv->message_queue);
    while (priv->message_queue) {
        EmpathyMessage *message = priv->message_queue->data;

        theme_adium_append_message (chat_view, message);
        priv->message_queue = g_list_remove (priv->message_queue, message);
        g_object_unref (message);
    }
}

static void
theme_adium_finalize (GObject *object)
{
    EmpathyThemeAdiumPriv *priv = GET_PRIV (object);

    g_free (priv->in_content_html);
    g_free (priv->in_nextcontent_html);
    g_free (priv->out_content_html);
    g_free (priv->out_nextcontent_html);
    g_free (priv->default_avatar_filename);

    if (priv->last_contact) {
        g_object_unref (priv->last_contact);
    }

    G_OBJECT_CLASS (empathy_theme_adium_parent_class)->finalize (object);
}

static void
empathy_theme_adium_class_init (EmpathyThemeAdiumClass *klass)
{
    GObjectClass *object_class = G_OBJECT_CLASS (klass);
    
    object_class->finalize = theme_adium_finalize;

    g_type_class_add_private (object_class, sizeof (EmpathyThemeAdiumPriv));
}

static void
empathy_theme_adium_init (EmpathyThemeAdium *theme)
{
    EmpathyThemeAdiumPriv *priv = G_TYPE_INSTANCE_GET_PRIVATE (theme,
        EMPATHY_TYPE_THEME_ADIUM, EmpathyThemeAdiumPriv);

    theme->priv = priv; 

    theme_adium_load (theme);

    g_signal_connect (theme, "load-finished",
              G_CALLBACK (theme_adium_load_finished_cb),
              NULL);
}

EmpathyThemeAdium *
empathy_theme_adium_new (void)
{
    return g_object_new (EMPATHY_TYPE_THEME_ADIUM, NULL);
}