path: root/libempathy-gtk/empathy-avatar-chooser.c

/* -*- Mode: C; tab-width: 8; indent-tabs-mode: t; c-basic-offset: 8 -*- */
/*
 * Copyright (C) 2006-2007 Imendio AB.
 * Copyright (C) 2007-2008 Collabora Ltd.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of version 2 of the GNU General Public
 * License as published by the Free Software Foundation.
 *
 * 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 library; if not, write to the
 * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
 * Boston, MA  02110-1301  USA
 *
 * Authors: Based on Novell's e-image-chooser.
 *          Xavier Claessens <xclaesse@gmail.com>
 */

#include "config.h"

#include <string.h>

#include <glib/gi18n-lib.h>
#include <gtk/gtk.h>
#include <gio/gio.h>

#include <libempathy/empathy-utils.h>
#include <libempathy/empathy-tp-contact-factory.h>

#include "empathy-avatar-chooser.h"
#include "empathy-conf.h"
#include "empathy-ui-utils.h"

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

/**
 * SECTION:empathy-avatar-chooser
 * @title: EmpathyAvatarChooser
 * @short_description: A widget used to change avatar
 * @include: libempathy-gtk/empathy-avatar-chooser.h
 *
 * #EmpathyAvatarChooser is a widget which extends #GtkButton to
 * provide a way of changing avatar.
 */

/**
 * EmpathyAvatarChooser:
 * @parent: parent object
 *
 * Widget which extends #GtkButton to provide a way of changing avatar.
 */

#define AVATAR_SIZE_SAVE 96
#define AVATAR_SIZE_VIEW 64
#define DEFAULT_DIR DATADIR"/pixmaps/faces"

#define GET_PRIV(obj) EMPATHY_GET_PRIV (obj, EmpathyAvatarChooser)
typedef struct {
    EmpathyTpContactFactory *factory;
    TpConnection            *connection;
    GtkFileChooser          *chooser_dialog;

    gulong ready_handler_id;

    EmpathyAvatar *avatar;
} EmpathyAvatarChooserPriv;

static void       avatar_chooser_finalize              (GObject              *object);
static void       avatar_chooser_set_connection        (EmpathyAvatarChooser *self,
                            TpConnection         *connection);
static void       avatar_chooser_set_image             (EmpathyAvatarChooser *chooser,
                            EmpathyAvatar        *avatar,
                            GdkPixbuf            *pixbuf,
                            gboolean              set_locally);
static gboolean   avatar_chooser_drag_motion_cb        (GtkWidget            *widget,
                            GdkDragContext       *context,
                            gint                  x,
                            gint                  y,
                            guint                 time,
                            EmpathyAvatarChooser *chooser);
static void       avatar_chooser_drag_leave_cb         (GtkWidget            *widget,
                            GdkDragContext       *context,
                            guint                 time,
                            EmpathyAvatarChooser *chooser);
static gboolean   avatar_chooser_drag_drop_cb          (GtkWidget            *widget,
                            GdkDragContext       *context,
                            gint                  x,
                            gint                  y,
                            guint                 time,
                            EmpathyAvatarChooser *chooser);
static void       avatar_chooser_drag_data_received_cb (GtkWidget            *widget,
                            GdkDragContext       *context,
                            gint                  x,
                            gint                  y,
                            GtkSelectionData     *selection_data,
                            guint                 info,
                            guint                 time,
                            EmpathyAvatarChooser *chooser);
static void       avatar_chooser_clicked_cb            (GtkWidget            *button,
                            EmpathyAvatarChooser *chooser);

enum {
    CHANGED,
    LAST_SIGNAL
};

enum {
    PROP_0,
    PROP_CONNECTION
};

static guint signals [LAST_SIGNAL];

G_DEFINE_TYPE (EmpathyAvatarChooser, empathy_avatar_chooser, GTK_TYPE_BUTTON);

/*
 * Drag and drop stuff
 */
#define URI_LIST_TYPE "text/uri-list"

enum DndTargetType {
    DND_TARGET_TYPE_URI_LIST
};

static const GtkTargetEntry drop_types[] = {
    { URI_LIST_TYPE, 0, DND_TARGET_TYPE_URI_LIST },
};

static void
avatar_chooser_get_property (GObject    *object,
                 guint       param_id,
                 GValue     *value,
                 GParamSpec *pspec)
{
    EmpathyAvatarChooserPriv *priv = GET_PRIV (object);

    switch (param_id) {
    case PROP_CONNECTION:
        g_value_set_object (value, priv->connection);
        break;
    default:
        G_OBJECT_WARN_INVALID_PROPERTY_ID (object, param_id, pspec);
        break;
    }
}

static void
avatar_chooser_set_property (GObject      *object,
                 guint         param_id,
                 const GValue *value,
                 GParamSpec   *pspec)
{
    EmpathyAvatarChooser *self = EMPATHY_AVATAR_CHOOSER (object);

    switch (param_id) {
    case PROP_CONNECTION:
        avatar_chooser_set_connection (self, g_value_get_object (value));
        break;
    default:
        G_OBJECT_WARN_INVALID_PROPERTY_ID (object, param_id, pspec);
        break;
    }
}

static void
empathy_avatar_chooser_class_init (EmpathyAvatarChooserClass *klass)
{
    GObjectClass *object_class = G_OBJECT_CLASS (klass);
    GParamSpec *param_spec;

    object_class->finalize = avatar_chooser_finalize;
    object_class->get_property = avatar_chooser_get_property;
    object_class->set_property = avatar_chooser_set_property;

    /**
     * EmpathyAvatarChooser::changed:
     * @chooser: an #EmpathyAvatarChooser
     *
     * Emitted when the chosen avatar has changed.
     *
     */
    signals[CHANGED] =
        g_signal_new ("changed",
                  G_TYPE_FROM_CLASS (klass),
                  G_SIGNAL_RUN_LAST,
                  0,
                  NULL, NULL,
                  g_cclosure_marshal_VOID__VOID,
                  G_TYPE_NONE, 0);

    /**
     * EmpathyAvatarChooser:connection:
     *
     * The #TpConnection whose avatar should be shown and modified by
     * the #EmpathyAvatarChooser instance.
     */
    param_spec = g_param_spec_object ("connection",
                      "TpConnection",
                      "TpConnection whose avatar should be "
                      "shown and modified by this widget",
                      TP_TYPE_CONNECTION,
                      G_PARAM_READWRITE |
                      G_PARAM_STATIC_STRINGS);
    g_object_class_install_property (object_class,
                     PROP_CONNECTION,
                     param_spec);

    g_type_class_add_private (object_class, sizeof (EmpathyAvatarChooserPriv));
}

static void
empathy_avatar_chooser_init (EmpathyAvatarChooser *chooser)
{
    EmpathyAvatarChooserPriv *priv = G_TYPE_INSTANCE_GET_PRIVATE (chooser,
        EMPATHY_TYPE_AVATAR_CHOOSER, EmpathyAvatarChooserPriv);

    chooser->priv = priv;
    gtk_drag_dest_set (GTK_WIDGET (chooser),
               GTK_DEST_DEFAULT_ALL,
               drop_types,
               G_N_ELEMENTS (drop_types),
               GDK_ACTION_COPY);

    g_signal_connect (chooser, "drag-motion",
              G_CALLBACK (avatar_chooser_drag_motion_cb),
              chooser);
    g_signal_connect (chooser, "drag-leave",
              G_CALLBACK (avatar_chooser_drag_leave_cb),
              chooser);
    g_signal_connect (chooser, "drag-drop",
              G_CALLBACK (avatar_chooser_drag_drop_cb),
              chooser);
    g_signal_connect (chooser, "drag-data-received",
              G_CALLBACK (avatar_chooser_drag_data_received_cb),
              chooser);
    g_signal_connect (chooser, "clicked",
              G_CALLBACK (avatar_chooser_clicked_cb),
              chooser);

    empathy_avatar_chooser_set (chooser, NULL);
}

static void
avatar_chooser_finalize (GObject *object)
{
    EmpathyAvatarChooserPriv *priv;

    priv = GET_PRIV (object);

    avatar_chooser_set_connection (EMPATHY_AVATAR_CHOOSER (object), NULL);
    g_assert (priv->connection == NULL);
    g_assert (priv->factory == NULL);

    if (priv->avatar != NULL) {
        empathy_avatar_unref (priv->avatar);
    }

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

static void
avatar_chooser_set_connection (EmpathyAvatarChooser *self,
                   TpConnection         *connection)
{
    EmpathyAvatarChooserPriv *priv = GET_PRIV (self);

    if (priv->connection != NULL) {
        g_object_unref (priv->connection);
        priv->connection = NULL;

        g_object_unref (priv->factory);
        priv->factory = NULL;
    }

    if (connection != NULL) {
        priv->connection = g_object_ref (connection);
        priv->factory = empathy_tp_contact_factory_dup_singleton (connection);
    }
}

static void
avatar_chooser_error_show (EmpathyAvatarChooser *chooser,
               const gchar          *primary_text,
               const gchar          *secondary_text)
{
    GtkWidget *parent;
    GtkWidget *dialog;

    parent = gtk_widget_get_toplevel (GTK_WIDGET (chooser));
    if (!GTK_IS_WINDOW (parent)) {
        parent = NULL;
    }

    dialog = gtk_message_dialog_new (parent ? GTK_WINDOW (parent) : NULL,
                     GTK_DIALOG_MODAL,
                     GTK_MESSAGE_WARNING,
                     GTK_BUTTONS_CLOSE,
                     "%s", primary_text);

    if (secondary_text != NULL) {
        gtk_message_dialog_format_secondary_text (GTK_MESSAGE_DIALOG (dialog),
                              "%s", secondary_text);
    }

    g_signal_connect (dialog, "response",
              G_CALLBACK (gtk_widget_destroy), NULL);
    gtk_widget_show (dialog);

}

static gboolean
str_in_strv (const gchar  *str,
         gchar **strv)
{
    if (strv == NULL) {
        return FALSE;
    }

    while (*strv != NULL) {
        if (g_str_equal (str, *strv)) {
            return TRUE;
        }
        strv++;
    }
    return FALSE;
}

/* The caller must free the strings stored in satisfactory_format_name and
 * satisfactory_mime_type.
 */
static gboolean
avatar_chooser_need_mime_type_conversion (const gchar *current_mime_type,
                      gchar      **accepted_mime_types,
                      gchar      **satisfactory_format_name,
                      gchar      **satisfactory_mime_type)
{
    gchar   *good_mime_types[] = {"image/jpeg", "image/png", NULL};
    guint    i;
    GSList  *formats, *l;
    gboolean found = FALSE;

    *satisfactory_format_name = NULL;
    *satisfactory_mime_type = NULL;

    /* If there is no accepted format there is nothing we can do */
    if (accepted_mime_types == NULL || *accepted_mime_types == NULL) {
        return TRUE;
    }

    /* If the current mime type is good and accepted, don't change it!
     * jpeg is compress better pictures, but png is better for logos and
     * could have an alpha layer. */
    if (str_in_strv (current_mime_type, good_mime_types) &&
        str_in_strv (current_mime_type, accepted_mime_types)) {
        *satisfactory_mime_type = g_strdup (current_mime_type);
        *satisfactory_format_name = g_strdup (current_mime_type +
                              strlen ("image/"));
        return FALSE;
    }

    /* The current mime type is either not accepted or not good to use.
     * Check if one of the good format is supported... */
    for (i = 0; good_mime_types[i] != NULL;  i++) {
        if (str_in_strv (good_mime_types[i], accepted_mime_types)) {
            *satisfactory_mime_type = g_strdup (good_mime_types[i]);
            *satisfactory_format_name = g_strdup (good_mime_types[i] +
                                  strlen ("image/"));
            return TRUE;
        }
    }

    /* Pick the first supported format we can write */
    formats = gdk_pixbuf_get_formats ();
    for (l = formats; !found && l != NULL; l = l->next) {
        GdkPixbufFormat *format = l->data;
        gchar **format_mime_types;
        gchar **iter;

        if (!gdk_pixbuf_format_is_writable (format)) {
            continue;
        }

        format_mime_types = gdk_pixbuf_format_get_mime_types (format);
        for (iter = format_mime_types; *iter != NULL; iter++) {
            if (str_in_strv (*iter, accepted_mime_types)) {
                *satisfactory_format_name = gdk_pixbuf_format_get_name (format);
                *satisfactory_mime_type = g_strdup (*iter);
                found = TRUE;
                break;
            }
        }
        g_strfreev (format_mime_types);
    }
    g_slist_free (formats);

    return TRUE;
}

static EmpathyAvatar *
avatar_chooser_maybe_convert_and_scale (EmpathyAvatarChooser *chooser,
                    GdkPixbuf            *pixbuf,
                    EmpathyAvatar        *avatar)
{
    EmpathyAvatarChooserPriv *priv = GET_PRIV (chooser);
    guint                     max_width = 0, max_height = 0, max_size = 0;
    gchar                   **mime_types = NULL;
    gboolean                  needs_conversion = FALSE;
    guint                     width, height;
    gchar                    *new_format_name = NULL;
    gchar                    *new_mime_type = NULL;
    gdouble                   min_factor, max_factor;
    gdouble                   factor;
    gchar                    *converted_image_data = NULL;
    gsize                     converted_image_size = 0;

    g_object_get (priv->factory,
        "avatar-mime-types", &mime_types, /* Needs g_strfreev-ing */
        "avatar-max-width", &max_width,
        "avatar-max-height", &max_height,
        "avatar-max-size", &max_size,
        NULL);

    /* Smaller is the factor, smaller will be the image.
     * 0 is an empty image, 1 is the full size. */
    min_factor = 0;
    max_factor = 1;
    factor = 1;

    /* Check if we need to convert to another image format */
    if (avatar_chooser_need_mime_type_conversion (avatar->format,
                              mime_types,
                              &new_format_name,
                              &new_mime_type)) {
        DEBUG ("Format conversion needed, we'll use mime type '%s' "
               "and format name '%s'. Current mime type is '%s'",
               new_mime_type, new_format_name, avatar->format);
        needs_conversion = TRUE;
    }
    g_strfreev (mime_types);

    /* If there is no format we can use, report error to the user. */
    if (new_mime_type == NULL || new_format_name == NULL) {
        avatar_chooser_error_show (chooser, _("Couldn't convert image"),
                _("None of the accepted image formats is "
                  "supported on your system"));
        return NULL;
    }

    /* If width or height are too big, it needs converting. */
    width = gdk_pixbuf_get_width (pixbuf);
    height = gdk_pixbuf_get_height (pixbuf);
    if ((max_width > 0 && width > max_width) ||
        (max_height > 0 && height > max_height)) {
        gdouble h_factor, v_factor;

        h_factor = (gdouble) max_width / width;
        v_factor = (gdouble) max_height / height;
        factor = max_factor = MIN (h_factor, v_factor);

        DEBUG ("Image dimensions (%dx%d) are too big. Max is %dx%d.",
               width, height, max_width, max_height);

        needs_conversion = TRUE;
    }

    /* If the data len is too big and no other conversion is needed,
     * try with a lower factor. */
    if (max_size > 0 && avatar->len > max_size && !needs_conversion) {
        DEBUG ("Image data (%"G_GSIZE_FORMAT" bytes) is too big "
               "(max is %u bytes), conversion needed.",
               avatar->len, max_size);

        factor = 0.5;
        needs_conversion = TRUE;
    }

    /* If no conversion is needed, return the avatar */
    if (!needs_conversion) {
        g_free (new_format_name);
        g_free (new_mime_type);
        return empathy_avatar_ref (avatar);
    }

    do {
        GdkPixbuf *pixbuf_scaled = NULL;
        gboolean   saved;
        gint       new_width, new_height;
        GError    *error = NULL;

        g_free (converted_image_data);

        if (factor != 1) {
            new_width = width * factor;
            new_height = height * factor;
            pixbuf_scaled = gdk_pixbuf_scale_simple (pixbuf,
                                 new_width,
                                 new_height,
                                 GDK_INTERP_HYPER);
        } else {
            new_width = width;
            new_height = height;
            pixbuf_scaled = g_object_ref (pixbuf);
        }

        DEBUG ("Trying with factor %f (%dx%d) and format %s...", factor,
            new_width, new_height, new_format_name);

        saved = gdk_pixbuf_save_to_buffer (pixbuf_scaled,
                           &converted_image_data,
                           &converted_image_size,
                           new_format_name,
                           &error, NULL);
        g_object_unref (pixbuf_scaled);

        if (!saved) {
            g_free (new_format_name);
            g_free (new_mime_type);
            avatar_chooser_error_show (chooser,
                _("Couldn't convert image"),
                error ? error->message : NULL);
            g_clear_error (&error);
            return NULL;
        }

        DEBUG ("Produced an image data of %"G_GSIZE_FORMAT" bytes.",
            converted_image_size);

        if (max_size == 0)
            break;

        /* Make a binary search for the bigest factor that produce
         * an image data size less than max_size */
        if (converted_image_size > max_size)
            max_factor = factor;
        if (converted_image_size < max_size)
            min_factor = factor;
        factor = (min_factor + max_factor)/2;

        /* We are done if either:
         * - min_factor == max_factor. That happens if we resized to
         *   the max required dimension and the produced data size is
         *   less than max_size.
         * - The data size is close enough to max_size. Here we accept
         *   a difference of 1k.
         */
    } while (min_factor != max_factor &&
             abs (max_size - converted_image_size) > 1024);
    g_free (new_format_name);

    /* Takes ownership of new_mime_type and converted_image_data */
    avatar = empathy_avatar_new ((guchar *) converted_image_data,
        converted_image_size, new_mime_type, NULL, NULL);

    return avatar;
}

static void
avatar_chooser_clear_image (EmpathyAvatarChooser *chooser)
{
    EmpathyAvatarChooserPriv *priv = GET_PRIV (chooser);
    GtkWidget *image;

    if (priv->avatar != NULL) {
        empathy_avatar_unref (priv->avatar);
        priv->avatar = NULL;
    }

    image = gtk_image_new_from_icon_name ("stock_person", GTK_ICON_SIZE_DIALOG);
    gtk_button_set_image (GTK_BUTTON (chooser), image);
    g_signal_emit (chooser, signals[CHANGED], 0);
}

static void
avatar_chooser_set_image_from_data (EmpathyAvatarChooser *chooser,
                    gchar                *data,
                    gsize                 size,
                    gboolean              set_locally)
{
    GdkPixbuf     *pixbuf;
    EmpathyAvatar *avatar = NULL;
    gchar         *mime_type = NULL;

    if (data == NULL) {
        avatar_chooser_clear_image (chooser);
        return;
    }

    pixbuf = empathy_pixbuf_from_data_and_mime (data, size, &mime_type);
    if (pixbuf == NULL) {
        g_free (data);
        data = NULL;
        return;
    }

    /* avatar takes ownership of data and mime_type */
    avatar = empathy_avatar_new ((guchar *) data, size, mime_type, NULL, NULL);

    avatar_chooser_set_image (chooser, avatar, pixbuf, set_locally);
}

static void
avatar_chooser_set_image_from_avatar (EmpathyAvatarChooser *chooser,
                      EmpathyAvatar        *avatar,
                      gboolean              set_locally)
{
    GdkPixbuf *pixbuf;
    gchar     *mime_type = NULL;

    g_assert (avatar != NULL);

    pixbuf = empathy_pixbuf_from_data_and_mime ((gchar *) avatar->data,
                            avatar->len,
                            &mime_type);
    if (pixbuf == NULL) {
        DEBUG ("couldn't make a pixbuf from avatar; giving up");
        return;
    }

    if (avatar->format == NULL) {
        avatar->format = mime_type;
    } else {
        if (strcmp (mime_type, avatar->format)) {
            DEBUG ("avatar->format is %s; gdkpixbuf yields %s!",
                avatar->format, mime_type);
        }
        g_free (mime_type);
    }

    empathy_avatar_ref (avatar);

    avatar_chooser_set_image (chooser, avatar, pixbuf, set_locally);
}

static void
avatar_chooser_set_image (EmpathyAvatarChooser *chooser,
              EmpathyAvatar        *avatar,
              GdkPixbuf            *pixbuf,
              gboolean              set_locally)
{
    EmpathyAvatarChooserPriv *priv = GET_PRIV (chooser);
    GdkPixbuf                *pixbuf_view;
    GtkWidget                *image;

    g_assert (avatar != NULL);
    g_assert (pixbuf != NULL);

    if (set_locally) {
        EmpathyAvatar *conv;

        conv = avatar_chooser_maybe_convert_and_scale (chooser,
            pixbuf, avatar);
        empathy_avatar_unref (avatar);

        if (conv == NULL) {
            /* An error occured; don't change the avatar. */
            return;
        }

        avatar = conv;
    }

    if (priv->avatar != NULL) {
        empathy_avatar_unref (priv->avatar);
    }
    priv->avatar = avatar;

    pixbuf_view = empathy_pixbuf_scale_down_if_necessary (pixbuf, AVATAR_SIZE_VIEW);
    image = gtk_image_new_from_pixbuf (pixbuf_view);

    gtk_button_set_image (GTK_BUTTON (chooser), image);
    g_signal_emit (chooser, signals[CHANGED], 0);

    g_object_unref (pixbuf_view);
    g_object_unref (pixbuf);
}

static void
avatar_chooser_set_image_from_file (EmpathyAvatarChooser *chooser,
                    const gchar          *filename)
{
    gchar  *image_data = NULL;
    gsize   image_size = 0;
    GError *error = NULL;

    if (!g_file_get_contents (filename, &image_data, &image_size, &error)) {
        DEBUG ("Failed to load image from '%s': %s", filename,
            error ? error->message : "No error given");

        g_clear_error (&error);
        return;
    }

    avatar_chooser_set_image_from_data (chooser, image_data, image_size, TRUE);
}

static gboolean
avatar_chooser_drag_motion_cb (GtkWidget          *widget,
                  GdkDragContext     *context,
                  gint                x,
                  gint                y,
                  guint               time_,
                  EmpathyAvatarChooser *chooser)
{
    EmpathyAvatarChooserPriv *priv;
    GList                  *p;

    priv = GET_PRIV (chooser);

    for (p = context->targets; p != NULL; p = p->next) {
        gchar *possible_type;

        possible_type = gdk_atom_name (GDK_POINTER_TO_ATOM (p->data));

        if (!strcmp (possible_type, URI_LIST_TYPE)) {
            g_free (possible_type);
            gdk_drag_status (context, GDK_ACTION_COPY, time_);

            return TRUE;
        }

        g_free (possible_type);
    }

    return FALSE;
}

static void
avatar_chooser_drag_leave_cb (GtkWidget          *widget,
                 GdkDragContext     *context,
                 guint               time_,
                 EmpathyAvatarChooser *chooser)
{
}

static gboolean
avatar_chooser_drag_drop_cb (GtkWidget          *widget,
                GdkDragContext     *context,
                gint                x,
                gint                y,
                guint               time_,
                EmpathyAvatarChooser *chooser)
{
    EmpathyAvatarChooserPriv *priv;
    GList                  *p;

    priv = GET_PRIV (chooser);

    if (context->targets == NULL) {
        return FALSE;
    }

    for (p = context->targets; p != NULL; p = p->next) {
        char *possible_type;

        possible_type = gdk_atom_name (GDK_POINTER_TO_ATOM (p->data));
        if (!strcmp (possible_type, URI_LIST_TYPE)) {
            g_free (possible_type);
            gtk_drag_get_data (widget, context,
                       GDK_POINTER_TO_ATOM (p->data),
                       time_);

            return TRUE;
        }

        g_free (possible_type);
    }

    return FALSE;
}

static void
avatar_chooser_drag_data_received_cb (GtkWidget          *widget,
                     GdkDragContext     *context,
                     gint                x,
                     gint                y,
                     GtkSelectionData   *selection_data,
                     guint               info,
                     guint               time_,
                     EmpathyAvatarChooser *chooser)
{
    gchar    *target_type;
    gboolean  handled = FALSE;

    target_type = gdk_atom_name (gtk_selection_data_get_target (selection_data));
    if (!strcmp (target_type, URI_LIST_TYPE)) {
        GFile            *file;
        gchar            *nl;
        gchar            *data = NULL;
        gsize             bytes_read;

        nl = strstr ((gchar *) gtk_selection_data_get_data (selection_data),
                        "\r\n");
        if (nl) {
            gchar *uri;

            uri = g_strndup ((gchar *) gtk_selection_data_get_data (selection_data),
                     nl - (gchar *) gtk_selection_data_get_data (selection_data));

            file = g_file_new_for_uri (uri);
            g_free (uri);
        } else {
            file = g_file_new_for_uri ((gchar *) gtk_selection_data_get_data (
                        selection_data));
        }

        handled = g_file_load_contents (file, NULL, &data, &bytes_read,
                        NULL, NULL);

        if (handled) {
            /* this in turn calls empathy_avatar_new (), which assumes
             * ownership of data.
             */
            avatar_chooser_set_image_from_data (chooser, data,
                                bytes_read,
                                TRUE);
        }

        g_object_unref (file);
    }

    gtk_drag_finish (context, handled, FALSE, time_);
}

static void
avatar_chooser_update_preview_cb (GtkFileChooser       *file_chooser,
                  EmpathyAvatarChooser *chooser)
{
    gchar *filename;

    filename = gtk_file_chooser_get_preview_filename (file_chooser);

    if (filename) {
        GtkWidget *image;
        GdkPixbuf *pixbuf = NULL;
        GdkPixbuf *scaled_pixbuf;

        pixbuf = gdk_pixbuf_new_from_file (filename, NULL);

        image = gtk_file_chooser_get_preview_widget (file_chooser);

        if (pixbuf) {
            scaled_pixbuf = empathy_pixbuf_scale_down_if_necessary (pixbuf, AVATAR_SIZE_SAVE);
            gtk_image_set_from_pixbuf (GTK_IMAGE (image), scaled_pixbuf);
            g_object_unref (scaled_pixbuf);
            g_object_unref (pixbuf);
        } else {
            gtk_image_set_from_stock (GTK_IMAGE (image),
                          "gtk-dialog-question",
                          GTK_ICON_SIZE_DIALOG);
        }

        g_free (filename);
    }

    gtk_file_chooser_set_preview_widget_active (file_chooser, TRUE);
}

static void
avatar_chooser_response_cb (GtkWidget            *widget,
                gint                  response,
                EmpathyAvatarChooser *chooser)
{
    EmpathyAvatarChooserPriv *priv = GET_PRIV (chooser);

    priv->chooser_dialog = NULL;

    if (response == GTK_RESPONSE_OK) {
        gchar *filename;
        gchar *path;

        filename = gtk_file_chooser_get_filename (GTK_FILE_CHOOSER (widget));
        avatar_chooser_set_image_from_file (chooser, filename);
        g_free (filename);

        path = gtk_file_chooser_get_current_folder (GTK_FILE_CHOOSER (widget));
        if (path) {
            empathy_conf_set_string (empathy_conf_get (),
                         EMPATHY_PREFS_UI_AVATAR_DIRECTORY,
                         path);
            g_free (path);
        }
    }
    else if (response == GTK_RESPONSE_NO) {
        /* This corresponds to "No Image", not to "Cancel" */
        avatar_chooser_clear_image (chooser);
    }

    gtk_widget_destroy (widget);
}

static void
avatar_chooser_clicked_cb (GtkWidget            *button,
               EmpathyAvatarChooser *chooser)
{
    GtkFileChooser *chooser_dialog;
    GtkWidget      *image;
    gchar          *saved_dir = NULL;
    const gchar    *default_dir = DEFAULT_DIR;
    const gchar    *pics_dir;
    GtkFileFilter  *filter;
    EmpathyAvatarChooserPriv *priv = GET_PRIV (chooser);

    if (priv->chooser_dialog) {
        gtk_window_present (GTK_WINDOW (priv->chooser_dialog));
        return;
    }

    priv->chooser_dialog = GTK_FILE_CHOOSER (
        gtk_file_chooser_dialog_new (_("Select Your Avatar Image"),
                         empathy_get_toplevel_window (GTK_WIDGET (chooser)),
                         GTK_FILE_CHOOSER_ACTION_OPEN,
                         _("No Image"),
                         GTK_RESPONSE_NO,
                         GTK_STOCK_CANCEL,
                         GTK_RESPONSE_CANCEL,
                         GTK_STOCK_OPEN,
                         GTK_RESPONSE_OK,
                         NULL));
    chooser_dialog = priv->chooser_dialog;
    gtk_window_set_destroy_with_parent (GTK_WINDOW (chooser_dialog), TRUE);

    /* Get special dirs */
    empathy_conf_get_string (empathy_conf_get (),
                 EMPATHY_PREFS_UI_AVATAR_DIRECTORY,
                 &saved_dir);
    if (saved_dir && !g_file_test (saved_dir, G_FILE_TEST_IS_DIR)) {
        g_free (saved_dir);
        saved_dir = NULL;
    }
    if (!g_file_test (default_dir, G_FILE_TEST_IS_DIR)) {
        default_dir = NULL;
    }
    pics_dir = g_get_user_special_dir (G_USER_DIRECTORY_PICTURES);
    if (pics_dir && !g_file_test (pics_dir, G_FILE_TEST_IS_DIR)) {
        pics_dir = NULL;
    }

    /* Set current dir to the last one or to DEFAULT_DIR or to home */
    if (saved_dir) {
        gtk_file_chooser_set_current_folder (chooser_dialog, saved_dir);
    }
    else if (pics_dir) {
        gtk_file_chooser_set_current_folder (chooser_dialog, pics_dir);
    }
    else if (default_dir) {
        gtk_file_chooser_set_current_folder (chooser_dialog, default_dir);
    } else {
        gtk_file_chooser_set_current_folder (chooser_dialog, g_get_home_dir ());
    }

    /* Add shortcuts to special dirs */
    if (saved_dir) {
        gtk_file_chooser_add_shortcut_folder (chooser_dialog, saved_dir, NULL);
    }
    else if (pics_dir) {
        gtk_file_chooser_add_shortcut_folder (chooser_dialog, pics_dir, NULL);
    }
    if (default_dir) {
        gtk_file_chooser_add_shortcut_folder (chooser_dialog, default_dir, NULL);
    }

    /* Setup preview image */
    image = gtk_image_new ();
    gtk_file_chooser_set_preview_widget (chooser_dialog, image);
    gtk_widget_set_size_request (image, AVATAR_SIZE_SAVE, AVATAR_SIZE_SAVE);
    gtk_widget_show (image);
    gtk_file_chooser_set_use_preview_label (chooser_dialog, FALSE);
    g_signal_connect (chooser_dialog, "update-preview",
              G_CALLBACK (avatar_chooser_update_preview_cb),
              chooser);

    /* Setup filers */
    filter = gtk_file_filter_new ();
    gtk_file_filter_set_name (filter, _("Images"));
    gtk_file_filter_add_pixbuf_formats (filter);
    gtk_file_chooser_add_filter (chooser_dialog, filter);
    filter = gtk_file_filter_new ();
    gtk_file_filter_set_name (filter, _("All Files"));
    gtk_file_filter_add_pattern (filter, "*");
    gtk_file_chooser_add_filter (chooser_dialog, filter);

    /* Setup response */
    gtk_dialog_set_default_response (GTK_DIALOG (chooser_dialog), GTK_RESPONSE_OK);
    g_signal_connect (chooser_dialog, "response",
              G_CALLBACK (avatar_chooser_response_cb),
              chooser);

    gtk_widget_show (GTK_WIDGET (chooser_dialog));
    g_free (saved_dir);
}

/**
 * empathy_avatar_chooser_new:
 *
 * Creates a new #EmpathyAvatarChooser.
 *
 * Return value: a new #EmpathyAvatarChooser
 */
GtkWidget *
empathy_avatar_chooser_new ()
{
    return g_object_new (EMPATHY_TYPE_AVATAR_CHOOSER, NULL);
}

/**
 * empathy_avatar_chooser_set:
 * @chooser: an #EmpathyAvatarChooser
 * @avatar: a new #EmpathyAvatar
 *
 * Sets the @chooser to display the avatar indicated by @avatar.
 */
void
empathy_avatar_chooser_set (EmpathyAvatarChooser *chooser,
                EmpathyAvatar        *avatar)
{
    g_return_if_fail (EMPATHY_IS_AVATAR_CHOOSER (chooser));

    if (avatar != NULL) {
        avatar_chooser_set_image_from_avatar (chooser, avatar, FALSE);
    } else {
        avatar_chooser_clear_image (chooser);
    }
}

/**
 * empathy_avatar_chooser_get_image_data:
 * @chooser: an #EmpathyAvatarChooser
 * @data: avatar bytes
 * @data_size: size of @data
 * @mime_type: avatar mime-type
 *
 * Gets image data about the currently selected avatar.
 */
void
empathy_avatar_chooser_get_image_data (EmpathyAvatarChooser  *chooser,
                       const gchar          **data,
                       gsize                 *data_size,
                       const gchar          **mime_type)
{
    EmpathyAvatarChooserPriv *priv;

    g_return_if_fail (EMPATHY_IS_AVATAR_CHOOSER (chooser));

    priv = GET_PRIV (chooser);

    if (priv->avatar != NULL) {
        if (data != NULL) {
            *data = (gchar *) priv->avatar->data;
        }
        if (data_size != NULL) {
            *data_size = priv->avatar->len;
        }
        if (mime_type != NULL) {
            *mime_type = priv->avatar->format;
        }
    } else {
        if (data != NULL) {
            *data = NULL;
        }
        if (data_size != NULL) {
            *data_size = 0;
        }
        if (mime_type != NULL) {
            *mime_type = NULL;
        }
    }
}