/*
 * e-attachment-tree-view.c
 *
 * This program 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 of the License, or (at your option) version 3.
 *
 * 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with the program; if not, see <http://www.gnu.org/licenses/>  
 *
 *
 * Copyright (C) 1999-2008 Novell, Inc. (www.novell.com)
 *
 */

#include "e-attachment-tree-view.h"

#include <glib/gi18n.h>
#include <gdk/gdkkeysyms.h>

#include "e-attachment.h"
#include "e-attachment-store.h"
#include "e-attachment-view.h"

#define E_ATTACHMENT_TREE_VIEW_GET_PRIVATE(obj) \
	(G_TYPE_INSTANCE_GET_PRIVATE \
	((obj), E_TYPE_ATTACHMENT_TREE_VIEW, EAttachmentTreeViewPrivate))

struct _EAttachmentTreeViewPrivate {
	EAttachmentViewPrivate view_priv;
};

enum {
	PROP_0,
	PROP_EDITABLE
};

static gpointer parent_class;

static void
attachment_tree_view_set_property (GObject *object,
                                   guint property_id,
                                   const GValue *value,
                                   GParamSpec *pspec)
{
	switch (property_id) {
		case PROP_EDITABLE:
			e_attachment_view_set_editable (
				E_ATTACHMENT_VIEW (object),
				g_value_get_boolean (value));
			return;
	}

	G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec);
}

static void
attachment_tree_view_get_property (GObject *object,
                                   guint property_id,
                                   GValue *value,
                                   GParamSpec *pspec)
{
	switch (property_id) {
		case PROP_EDITABLE:
			g_value_set_boolean (
				value, e_attachment_view_get_editable (
				E_ATTACHMENT_VIEW (object)));
			return;
	}

	G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec);
}

static void
attachment_tree_view_dispose (GObject *object)
{
	e_attachment_view_dispose (E_ATTACHMENT_VIEW (object));

	/* Chain up to parent's dispose() method. */
	G_OBJECT_CLASS (parent_class)->dispose (object);
}

static void
attachment_tree_view_finalize (GObject *object)
{
	e_attachment_view_finalize (E_ATTACHMENT_VIEW (object));

	/* Chain up to parent's finalize() method. */
	G_OBJECT_CLASS (parent_class)->finalize (object);
}

static void
attachment_tree_view_render_size (GtkTreeViewColumn *column,
                                  GtkCellRenderer *renderer,
                                  GtkTreeModel *model,
                                  GtkTreeIter *iter)
{
	gchar *display_size;
	gint column_id;
	guint64 size;

	column_id = E_ATTACHMENT_STORE_COLUMN_SIZE;
	gtk_tree_model_get (model, iter, column_id, &size, -1);

	display_size = g_format_size_for_display ((goffset) size);
	g_object_set (renderer, "text", display_size, NULL);
	g_free (display_size);
}

static gboolean
attachment_tree_view_button_press_event (GtkWidget *widget,
                                         GdkEventButton *event)
{
	EAttachmentView *view = E_ATTACHMENT_VIEW (widget);

	if (event->button == 3 && event->type == GDK_BUTTON_PRESS) {
		e_attachment_view_show_popup_menu (view, event);
		return TRUE;
	}

	/* Chain up to parent's button_press_event() method. */
	return GTK_WIDGET_CLASS (parent_class)->
		button_press_event (widget, event);
}

static gboolean
attachment_tree_view_key_press_event (GtkWidget *widget,
                                      GdkEventKey *event)
{
	EAttachmentView *view = E_ATTACHMENT_VIEW (widget);

	if (event->keyval == GDK_Delete) {
		e_attachment_view_remove_selected (view, TRUE);
		return TRUE;
	}

	/* Chain up to parent's key_press_event() method. */
	return GTK_WIDGET_CLASS (parent_class)->
		key_press_event (widget, event);
}

static gboolean
attachment_tree_view_drag_motion (GtkWidget *widget,
                                  GdkDragContext *context,
                                  gint x,
                                  gint y,
                                  guint time)
{
	EAttachmentView *view = E_ATTACHMENT_VIEW (widget);

	return e_attachment_view_drag_motion (view, context, x, y, time);
}

static void
attachment_tree_view_drag_data_received (GtkWidget *widget,
                                         GdkDragContext *context,
                                         gint x,
                                         gint y,
                                         GtkSelectionData *selection,
                                         guint info,
                                         guint time)
{
	EAttachmentView *view = E_ATTACHMENT_VIEW (widget);

	e_attachment_view_drag_data_received (
		view, context, x, y, selection, info, time);
}

static gboolean
attachment_tree_view_popup_menu (GtkWidget *widget)
{
	EAttachmentView *view = E_ATTACHMENT_VIEW (widget);

	e_attachment_view_show_popup_menu (view, NULL);

	return TRUE;
}

static EAttachmentViewPrivate *
attachment_tree_view_get_private (EAttachmentView *view)
{
	EAttachmentTreeViewPrivate *priv;

	priv = E_ATTACHMENT_TREE_VIEW_GET_PRIVATE (view);

	return &priv->view_priv;
}

static EAttachmentStore *
attachment_tree_view_get_store (EAttachmentView *view)
{
	GtkTreeView *tree_view;
	GtkTreeModel *model;

	tree_view = GTK_TREE_VIEW (view);
	model = gtk_tree_view_get_model (tree_view);

	return E_ATTACHMENT_STORE (model);
}

static GtkTreePath *
attachment_tree_view_get_path_at_pos (EAttachmentView *view,
                                      gint x,
                                      gint y)
{
	GtkTreeView *tree_view;
	GtkTreePath *path;
	gboolean row_exists;

	tree_view = GTK_TREE_VIEW (view);

	row_exists = gtk_tree_view_get_path_at_pos (
		tree_view, x, y, &path, NULL, NULL, NULL);

	return row_exists ? path : NULL;
}

static GList *
attachment_tree_view_get_selected_paths (EAttachmentView *view)
{
	GtkTreeView *tree_view;
	GtkTreeSelection *selection;

	tree_view = GTK_TREE_VIEW (view);
	selection = gtk_tree_view_get_selection (tree_view);

	return gtk_tree_selection_get_selected_rows (selection, NULL);
}

static gboolean
attachment_tree_view_path_is_selected (EAttachmentView *view,
                                       GtkTreePath *path)
{
	GtkTreeView *tree_view;
	GtkTreeSelection *selection;

	tree_view = GTK_TREE_VIEW (view);
	selection = gtk_tree_view_get_selection (tree_view);

	return gtk_tree_selection_path_is_selected (selection, path);
}

static void
attachment_tree_view_select_path (EAttachmentView *view,
                                  GtkTreePath *path)
{
	GtkTreeView *tree_view;
	GtkTreeSelection *selection;

	tree_view = GTK_TREE_VIEW (view);
	selection = gtk_tree_view_get_selection (tree_view);

	gtk_tree_selection_select_path (selection, path);
}

static void
attachment_tree_view_unselect_path (EAttachmentView *view,
                                    GtkTreePath *path)
{
	GtkTreeView *tree_view;
	GtkTreeSelection *selection;

	tree_view = GTK_TREE_VIEW (view);
	selection = gtk_tree_view_get_selection (tree_view);

	gtk_tree_selection_unselect_path (selection, path);
}

static void
attachment_tree_view_select_all (EAttachmentView *view)
{
	GtkTreeView *tree_view;
	GtkTreeSelection *selection;

	tree_view = GTK_TREE_VIEW (view);
	selection = gtk_tree_view_get_selection (tree_view);

	gtk_tree_selection_select_all (selection);
}

static void
attachment_tree_view_unselect_all (EAttachmentView *view)
{
	GtkTreeView *tree_view;
	GtkTreeSelection *selection;

	tree_view = GTK_TREE_VIEW (view);
	selection = gtk_tree_view_get_selection (tree_view);

	gtk_tree_selection_unselect_all (selection);
}

static void
attachment_tree_view_class_init (EAttachmentTreeViewClass *class)
{
	GObjectClass *object_class;
	GtkWidgetClass *widget_class;

	parent_class = g_type_class_peek_parent (class);
	g_type_class_add_private (class, sizeof (EAttachmentViewPrivate));

	object_class = G_OBJECT_CLASS (class);
	object_class->set_property = attachment_tree_view_set_property;
	object_class->get_property = attachment_tree_view_get_property;
	object_class->dispose = attachment_tree_view_dispose;
	object_class->finalize = attachment_tree_view_finalize;

	widget_class = GTK_WIDGET_CLASS (class);
	widget_class->button_press_event = attachment_tree_view_button_press_event;
	widget_class->key_press_event = attachment_tree_view_key_press_event;
	widget_class->drag_motion = attachment_tree_view_drag_motion;
	widget_class->drag_data_received = attachment_tree_view_drag_data_received;
	widget_class->popup_menu = attachment_tree_view_popup_menu;

	g_object_class_override_property (
		object_class, PROP_EDITABLE, "editable");
}

static void
attachment_tree_view_iface_init (EAttachmentViewIface *iface)
{
	iface->get_private = attachment_tree_view_get_private;
	iface->get_store = attachment_tree_view_get_store;

	iface->get_path_at_pos = attachment_tree_view_get_path_at_pos;
	iface->get_selected_paths = attachment_tree_view_get_selected_paths;
	iface->path_is_selected = attachment_tree_view_path_is_selected;
	iface->select_path = attachment_tree_view_select_path;
	iface->unselect_path = attachment_tree_view_unselect_path;
	iface->select_all = attachment_tree_view_select_all;
	iface->unselect_all = attachment_tree_view_unselect_all;
}

static void
attachment_tree_view_init (EAttachmentTreeView *tree_view)
{
	GtkTreeSelection *selection;
	GtkTreeViewColumn *column;
	GtkCellRenderer *renderer;

	tree_view->priv = E_ATTACHMENT_TREE_VIEW_GET_PRIVATE (tree_view);

	e_attachment_view_init (E_ATTACHMENT_VIEW (tree_view));

	gtk_tree_view_set_rules_hint (GTK_TREE_VIEW (tree_view), TRUE);

	selection = gtk_tree_view_get_selection (GTK_TREE_VIEW (tree_view));
	gtk_tree_selection_set_mode (selection, GTK_SELECTION_MULTIPLE);

	column = gtk_tree_view_column_new ();
	gtk_tree_view_column_set_expand (column, TRUE);
	gtk_tree_view_column_set_spacing (column, 3);
	gtk_tree_view_column_set_title (column, _("Name"));
	gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), column);

	renderer = gtk_cell_renderer_pixbuf_new ();
	gtk_tree_view_column_pack_start (column, renderer, FALSE);

	g_object_set (renderer, "stock-size", GTK_ICON_SIZE_MENU, NULL);

	gtk_tree_view_column_add_attribute (
		column, renderer, "gicon",
		E_ATTACHMENT_STORE_COLUMN_ICON);

	renderer = gtk_cell_renderer_text_new ();
	g_object_set (renderer, "ellipsize", PANGO_ELLIPSIZE_END, NULL);
	gtk_tree_view_column_pack_start (column, renderer, TRUE);

	gtk_tree_view_column_add_attribute (
		column, renderer, "text",
		E_ATTACHMENT_STORE_COLUMN_DISPLAY_NAME);

	column = gtk_tree_view_column_new ();
	gtk_tree_view_column_set_title (column, _("Size"));
	gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), column);

	renderer = gtk_cell_renderer_text_new ();
	gtk_tree_view_column_pack_start (column, renderer, TRUE);

	gtk_tree_view_column_set_cell_data_func (
		column, renderer, (GtkTreeCellDataFunc)
		attachment_tree_view_render_size, NULL, NULL);

	column = gtk_tree_view_column_new ();
	gtk_tree_view_column_set_title (column, _("Type"));
	gtk_tree_view_append_column (GTK_TREE_VIEW (tree_view), column);

	renderer = gtk_cell_renderer_text_new ();
	gtk_tree_view_column_pack_start (column, renderer, TRUE);

	gtk_tree_view_column_add_attribute (
		column, renderer, "text",
		E_ATTACHMENT_STORE_COLUMN_CONTENT_TYPE);
}

GType
e_attachment_tree_view_get_type (void)
{
	static GType type = 0;

	if (G_UNLIKELY (type == 0)) {
		static const GTypeInfo type_info = {
			sizeof (EAttachmentTreeViewClass),
			(GBaseInitFunc) NULL,
			(GBaseFinalizeFunc) NULL,
			(GClassInitFunc) attachment_tree_view_class_init,
			(GClassFinalizeFunc) NULL,
			NULL,  /* class_data */
			sizeof (EAttachmentTreeView),
			0,     /* n_preallocs */
			(GInstanceInitFunc) attachment_tree_view_init,
			NULL   /* value_table */
		};

		static const GInterfaceInfo iface_info = {
			(GInterfaceInitFunc) attachment_tree_view_iface_init,
			(GInterfaceFinalizeFunc) NULL,
			NULL   /* interface_data */
		};

		type = g_type_register_static (
			GTK_TYPE_TREE_VIEW, "EAttachmentTreeView",
			&type_info, 0);

		g_type_add_interface_static (
			type, E_TYPE_ATTACHMENT_VIEW, &iface_info);
	}

	return type;
}

GtkWidget *
e_attachment_tree_view_new (void)
{
	return g_object_new (E_TYPE_ATTACHMENT_TREE_VIEW, NULL);
}