/* -*- 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 are "
				  "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;
		}
	}
}