/*
 * Tab label
 *
 * Copyright (C) 2017  Patrick McDermott
 *
 * This file is part of Marquee.
 *
 * Marquee 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 3 of the License, or
 * (at your option) any later version.
 *
 * Marquee 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 Marquee.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "tab-label.h"

#include <glib.h>
#include <gtk/gtk.h>
#include <webkit2/webkit2.h>

#include "tab-page.h"
#include "web-view.h"

struct _MqTabLabel {
	GtkEventBox    parent_instance;
	MqTabPage     *tab_page;
	WebKitWebView *web_view;
	GtkWidget     *image;
	GtkWidget     *label;
	guint          position;
	const gchar   *title;
	gboolean       scrolling;
	gchar         *scrolled_title;
	GtkWidget     *popover;
};

enum {
	PROP_TAB_PAGE = 1,
	PROP_WEB_VIEW,
	N_PROPERTIES
};

static GParamSpec *obj_properties[N_PROPERTIES] = {NULL,};

struct _MqTabLabelClass {
	GtkEventBoxClass parent_class;
};

G_DEFINE_TYPE(MqTabLabel, mq_tab_label, GTK_TYPE_EVENT_BOX)

static void
reload_tab_clicked_cb(GtkWidget G_GNUC_UNUSED *button, MqTabLabel *tab_label)
{
	webkit_web_view_reload(tab_label->web_view);
	gtk_widget_hide(tab_label->popover);
}

static void
new_tab_clicked_cb(GtkWidget G_GNUC_UNUSED *button, MqTabLabel *tab_label)
{
	mq_tab_page_new(NULL, tab_label->tab_page);
	gtk_widget_hide(tab_label->popover);
}

static void
new_window_clicked_cb(GtkWidget G_GNUC_UNUSED *button, MqTabLabel *tab_label)
{
	mq_application_add_window(
		mq_tab_page_get_application(tab_label->tab_page), NULL);
	gtk_widget_hide(tab_label->popover);
}

static void
tab_list_button_toggled_cb(GtkToggleButton *toggle_button, GtkWidget *tab_list)
{
	if (gtk_toggle_button_get_active(toggle_button)) {
		gtk_widget_show(tab_list);
	} else {
		gtk_widget_hide(tab_list);
	}
}

static void
create_tree_model_recurse(MqTabPage *node, GtkTreeStore *tree_store,
	GtkTreeIter *parent_tree_iter)
{
	GtkTreeIter tree_iter;

	for (; node; node = mq_tab_page_next(node)) {
		gtk_tree_store_append(tree_store, &tree_iter, parent_tree_iter);
		gtk_tree_store_set(tree_store, &tree_iter, 0,
			mq_tab_page_get_title(node), -1);
		create_tree_model_recurse(mq_tab_page_first_child(node),
			tree_store, &tree_iter);
	}
}

static GtkTreeModel *
create_tree_model(MqTabLabel *tab_label)
{
	GtkTreeStore *tree_store;

	tree_store = gtk_tree_store_new(1, G_TYPE_STRING);

	create_tree_model_recurse(mq_tab_page_root(tab_label->tab_page),
		tree_store, NULL);

	return GTK_TREE_MODEL(tree_store);
}

static void
row_activated_cb(GtkTreeView G_GNUC_UNUSED *tree_view, GtkTreePath *tree_path,
	GtkTreeViewColumn G_GNUC_UNUSED *tree_view_column,
	MqTabLabel *tab_label)
{
	gint *indices;
	gint  depth;

	indices = gtk_tree_path_get_indices_with_depth(tree_path, &depth);
	g_assert(depth == 1);
	mq_window_set_current_tab(mq_tab_page_get_window(tab_label->tab_page),
		indices[0] + 1);
	gtk_widget_hide(tab_label->popover);
}

static GtkWidget *
create_tab_list(MqTabLabel *tab_label)
{
	GtkWidget        *tree_view;
	GtkTreeSelection *tree_selection;
	GtkCellRenderer  *cell_renderer;
	GtkWidget        *scrolled_window;

	tree_view = gtk_tree_view_new_with_model(create_tree_model(tab_label));
	tree_selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(tree_view));
	gtk_tree_selection_set_mode(tree_selection, GTK_SELECTION_BROWSE);
	gtk_tree_selection_select_path(tree_selection,
		gtk_tree_path_new_from_indices(
			mq_window_get_current_tab(mq_tab_page_get_window(
					tab_label->tab_page)) - 1, -1));
	gtk_tree_view_set_headers_visible(GTK_TREE_VIEW(tree_view), FALSE);
	gtk_tree_view_set_activate_on_single_click(GTK_TREE_VIEW(tree_view),
		TRUE);
	gtk_tree_view_expand_all(GTK_TREE_VIEW(tree_view));
	gtk_tree_view_set_reorderable(GTK_TREE_VIEW(tree_view), TRUE);
	gtk_tree_view_set_enable_tree_lines(GTK_TREE_VIEW(tree_view), TRUE);
	g_signal_connect(tree_view, "row-activated",
		G_CALLBACK(row_activated_cb), tab_label);

	cell_renderer = gtk_cell_renderer_text_new();
	gtk_tree_view_insert_column_with_attributes(GTK_TREE_VIEW(tree_view),
		-1, NULL, cell_renderer, "text", 0, NULL);

	scrolled_window = gtk_scrolled_window_new(NULL, NULL);
	gtk_scrolled_window_set_min_content_width(
		GTK_SCROLLED_WINDOW(scrolled_window), 400);
	gtk_scrolled_window_set_min_content_height(
		GTK_SCROLLED_WINDOW(scrolled_window), 200);
	gtk_container_add(GTK_CONTAINER(scrolled_window), tree_view);

	return scrolled_window;
}

#define BUTTON_ROWS 2
#define BUTTON_COLS 4
#define NEW_BUTTON(Y, X, ICON, TOOLTIP) \
	do { \
		buttons[Y * BUTTON_COLS + X] = gtk_button_new_from_icon_name(\
			ICON, GTK_ICON_SIZE_BUTTON); \
		gtk_widget_set_tooltip_text(buttons[Y * BUTTON_COLS + X], \
			TOOLTIP); \
		gtk_widget_set_can_focus(buttons[Y * BUTTON_COLS + X], FALSE); \
		gtk_grid_attach(GTK_GRID(button_grid), \
			buttons[Y * BUTTON_COLS + X], X, Y, 1, 1); \
	} while (0)
#define NEW_TOGGLE(Y, X, ICON, TOOLTIP) \
	do { \
		buttons[Y * BUTTON_COLS + X] = gtk_toggle_button_new(); \
		gtk_button_set_image(GTK_BUTTON(buttons[Y * BUTTON_COLS + X]), \
			gtk_image_new_from_icon_name(ICON, \
				GTK_ICON_SIZE_BUTTON)); \
		gtk_widget_set_tooltip_text(buttons[Y * BUTTON_COLS + X], \
			TOOLTIP); \
		gtk_widget_set_can_focus(buttons[Y * BUTTON_COLS + X], FALSE); \
		gtk_grid_attach(GTK_GRID(button_grid), \
			buttons[Y * BUTTON_COLS + X], X, Y, 1, 1); \
	} while (0)
#define CLICKED_CB(Y, X, CB) \
	g_signal_connect(buttons[Y * BUTTON_COLS + X], "clicked", CB, tab_label)

static void
create_tab_popover(GtkWidget *widget, MqTabLabel *tab_label)
{
	GtkWidget *button_grid;
	GtkWidget *buttons[BUTTON_ROWS * BUTTON_COLS];
	GtkWidget *tab_list;
	GtkWidget *tab_list_scrolled_window;
	GtkWidget *box;

	/* Set up button grid. */
	button_grid = gtk_grid_new();
	gtk_widget_set_halign(button_grid, GTK_ALIGN_CENTER);

	/* Set up buttons. */
	NEW_BUTTON(0, 0, "view-refresh",       "Reload tab");
	NEW_BUTTON(0, 1, "edit-copy",          "Duplicate tab");
	NEW_BUTTON(0, 2, "window-new",         "Move tab to new window");
	NEW_BUTTON(0, 3, "window-close",       "Close tab");
	NEW_BUTTON(1, 0, "tab-new-symbolic",   "New tab");
	NEW_BUTTON(1, 1, "window-new",         "New window");
	NEW_BUTTON(1, 2, "edit-undo",          "Undo close tab");
	NEW_TOGGLE(1, 3, "view-list-symbolic", "Tab list...");

	CLICKED_CB(0, 0, G_CALLBACK(reload_tab_clicked_cb));
	CLICKED_CB(1, 0, G_CALLBACK(new_tab_clicked_cb));
	CLICKED_CB(1, 1, G_CALLBACK(new_window_clicked_cb));

	/* Set up the tab list. */
	tab_list = create_tab_list(tab_label);

	/* Set up the tab list scrolled window.
	 *
	 * The following GtkScrolledWindow widget has a hardcoded minimum size,
	 * because there seems to be (in GTK+ versions before 3.22) no way to
	 * set the natural size of GtkScrolledWindow and its GtkViewport.
	 *
	 * See tab-chrome.c for more information. */
	tab_list_scrolled_window = gtk_scrolled_window_new(NULL, NULL);
	gtk_scrolled_window_set_min_content_width(
		GTK_SCROLLED_WINDOW(tab_list_scrolled_window), 400);
	gtk_scrolled_window_set_min_content_height(
		GTK_SCROLLED_WINDOW(tab_list_scrolled_window), 200);
	gtk_container_add(GTK_CONTAINER(tab_list_scrolled_window), tab_list);

	/* Add tab list toggle button handler. */
	g_signal_connect(buttons[1 * BUTTON_COLS + 3], "toggled",
		G_CALLBACK(tab_list_button_toggled_cb),
		tab_list_scrolled_window);

	/* Set up the button rows box. */
	box = gtk_box_new(GTK_ORIENTATION_VERTICAL, 0);
	gtk_box_pack_start(GTK_BOX(box), button_grid,
		TRUE, FALSE, 0);
	gtk_box_pack_start(GTK_BOX(box), tab_list_scrolled_window,
		TRUE, FALSE, 0);

	/* Set up the popover. */
	tab_label->popover = gtk_popover_new(widget);
	gtk_container_add(GTK_CONTAINER(tab_label->popover), box);

	/* NB: gtk_popover_popup() is new in GTK+ 3.22. */
	gtk_widget_show_all(tab_label->popover);
	gtk_widget_hide(tab_list_scrolled_window);
}

#undef BUTTON_ROWS
#undef BUTTON_COLS
#undef NEW_BUTTON
#undef NEW_TOGGLE
#undef CLICKED_CB

static gboolean
button_press_cb(GtkWidget *widget, GdkEventButton *event,
	MqTabLabel *tab_label)
{
	/* Create a popover menu on right click. */
	if (event->button == 3) {
		create_tab_popover(widget, tab_label);
	}

	return FALSE;
}

static void
update_image(MqTabLabel *tab_label, GdkPixbuf *favicon)
{
	if (favicon) {
		gtk_image_set_from_pixbuf(GTK_IMAGE(tab_label->image), favicon);
	} else {
		gtk_image_set_from_icon_name(GTK_IMAGE(tab_label->image),
			"text-x-generic", GTK_ICON_SIZE_BUTTON);
	}
}

static void
update_label(MqTabLabel *tab_label)
{
	const gchar *title;
	gchar       *label;

	title = tab_label->scrolling ? tab_label->scrolled_title :
		tab_label->title;
	label = g_strdup_printf("%d. %s", tab_label->position, title);
	gtk_label_set_text(GTK_LABEL(tab_label->label), label);
	gtk_widget_set_tooltip_text(GTK_WIDGET(tab_label), label);
	g_free(label);
}

static void
favicon_cb(WebKitWebView G_GNUC_UNUSED *web_view,
	GParamSpec G_GNUC_UNUSED *param_spec, MqTabLabel *tab_label)
{
	cairo_surface_t *surface;
	GdkPixbuf       *pixbuf;
	GdkPixbuf       *scaled_pixbuf;

	surface = webkit_web_view_get_favicon(tab_label->web_view);
	scaled_pixbuf = NULL;
	if (surface) {
		pixbuf = gdk_pixbuf_get_from_surface(surface, 0, 0,
			cairo_image_surface_get_width(surface),
			cairo_image_surface_get_height(surface));
		if (pixbuf) {
			scaled_pixbuf = gdk_pixbuf_scale_simple(pixbuf, 16, 16,
				GDK_INTERP_BILINEAR);
			g_object_unref(pixbuf);
		}
	}

	update_image(tab_label, scaled_pixbuf);
}

static void
title_cb(WebKitWebView G_GNUC_UNUSED *web_view,
	GParamSpec G_GNUC_UNUSED *param_spec, MqTabLabel *tab_label)
{
	tab_label->title = webkit_web_view_get_title(tab_label->web_view);
	if (tab_label->scrolling) {
		tab_label->scrolled_title = g_strdup_printf("%s    ",
			tab_label->title);
	}
	update_label(tab_label);
}

static void
set_web_view(MqTabLabel *tab_label, MqWebView *web_view)
{
	tab_label->web_view = WEBKIT_WEB_VIEW(web_view);

	g_signal_connect(web_view, "notify::favicon",
		G_CALLBACK(favicon_cb), tab_label);
	g_signal_connect(web_view, "notify::title",
		G_CALLBACK(title_cb), tab_label);
}

static void
finalize(GObject *object)
{
	MqTabLabel *tab_label;

	tab_label = MQ_TAB_LABEL(object);

	if (tab_label->scrolled_title) {
		g_free(tab_label->scrolled_title);
	}

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

static void
get_property(GObject *object, guint property_id, GValue *value,
	GParamSpec *param_spec)
{
	MqTabLabel *tab_label;

	tab_label = MQ_TAB_LABEL(object);

	switch (property_id) {
		case PROP_TAB_PAGE:
			g_value_set_object(value, tab_label->tab_page);
			break;
		case PROP_WEB_VIEW:
			g_value_set_object(value, tab_label->web_view);
			break;
		default:
			G_OBJECT_WARN_INVALID_PROPERTY_ID(object, property_id,
				param_spec);
			break;
	}
}

static void
set_property(GObject *object, guint property_id, const GValue *value,
	GParamSpec *param_spec)
{
	MqTabLabel *tab_label;

	tab_label = MQ_TAB_LABEL(object);

	switch (property_id) {
		case PROP_TAB_PAGE:
			tab_label->tab_page = g_value_get_object(value);
			break;
		case PROP_WEB_VIEW:
			set_web_view(tab_label, g_value_get_object(value));
			break;
		default:
			G_OBJECT_WARN_INVALID_PROPERTY_ID(object, property_id,
				param_spec);
			break;
	}
}

static void
mq_tab_label_class_init(MqTabLabelClass *klass)
{
	GObjectClass *object_class = G_OBJECT_CLASS(klass);

	object_class->finalize     = finalize;
	object_class->get_property = get_property;
	object_class->set_property = set_property;

	obj_properties[PROP_TAB_PAGE] = g_param_spec_object(
		"tab-page",
		"MqTabPage",
		"The ancestral MqTabPage instance",
		MQ_TYPE_TAB_PAGE,
		G_PARAM_WRITABLE | G_PARAM_CONSTRUCT_ONLY |
		G_PARAM_STATIC_NICK | G_PARAM_STATIC_BLURB);
	obj_properties[PROP_WEB_VIEW] = g_param_spec_object(
		"web-view",
		"MqWebView",
		"The associated MqWebView instance",
		MQ_TYPE_WEB_VIEW,
		G_PARAM_WRITABLE | G_PARAM_CONSTRUCT_ONLY |
		G_PARAM_STATIC_NICK | G_PARAM_STATIC_BLURB);
	g_object_class_install_properties(object_class, N_PROPERTIES,
		obj_properties);
}

static void
mq_tab_label_init(MqTabLabel *tab_label)
{
	GtkWidget *close_button;
	GtkWidget *box;

	tab_label->title = "New tab";

	/* Set up tab image. */
	tab_label->image = gtk_image_new_from_icon_name("text-x-generic",
		GTK_ICON_SIZE_BUTTON);

	/* Set up tab label. */
	tab_label->label = gtk_label_new(NULL);
	gtk_label_set_ellipsize(GTK_LABEL(tab_label->label),
		PANGO_ELLIPSIZE_END);
	gtk_widget_set_hexpand(tab_label->label, TRUE);
	gtk_widget_set_size_request(tab_label->label, 50, 1);

	/* Set up close button. */
	close_button = gtk_button_new_from_icon_name("window-close",
		GTK_ICON_SIZE_BUTTON);
	gtk_button_set_relief(GTK_BUTTON(close_button), GTK_RELIEF_NONE);
	gtk_widget_set_tooltip_text(close_button, "Close tab");

	/* Pack tab box. */
	box = gtk_box_new(GTK_ORIENTATION_HORIZONTAL, 0);
	gtk_box_pack_start(GTK_BOX(box), tab_label->image, FALSE, FALSE, 0);
	gtk_box_pack_start(GTK_BOX(box), tab_label->label, TRUE, TRUE, 0);
	gtk_box_pack_start(GTK_BOX(box), close_button, FALSE, FALSE, 0);
	gtk_widget_show_all(box);

	/* Set up event box. */
	g_signal_connect(tab_label, "button-press-event",
		G_CALLBACK(button_press_cb), tab_label);
	gtk_event_box_set_visible_window(GTK_EVENT_BOX(tab_label), FALSE);
	gtk_container_add(GTK_CONTAINER(tab_label), box);
}

GtkWidget *
mq_tab_label_new(MqTabPage *tab_page, MqWebView *web_view)
{
	return g_object_new(MQ_TYPE_TAB_LABEL,
		"tab-page", tab_page,
		"web-view", web_view,
		NULL);
}

void
mq_tab_label_set_position(MqTabLabel *tab_label, guint position)
{
	tab_label->position = position;
	update_label(tab_label);
}

void
mq_tab_label_begin_scrolling(MqTabLabel *tab_label)
{
	static PangoFontDescription *font_desc = NULL;

	if (!font_desc) {
		font_desc = pango_font_description_new();
		pango_font_description_set_family_static(font_desc,
			"monospace");
	}

	tab_label->scrolling = TRUE;
	tab_label->scrolled_title = g_strdup_printf("%s    ", tab_label->title);
	gtk_widget_override_font(tab_label->label, font_desc);
}

void
mq_tab_label_end_scrolling(MqTabLabel *tab_label)
{
	tab_label->scrolling = FALSE;

	gtk_widget_override_font(tab_label->label, NULL);

	update_label(tab_label);
}

void
mq_tab_label_scroll(MqTabLabel *tab_label)
{
	gchar c[5];  /* Up to 4 bytes for a UTF-8 character, plus NUL */
	gsize i;
	gsize j;

	/* Save the first (possibly multibyte) character. */
	c[0] = tab_label->scrolled_title[0];
	for (i = 1; tab_label->scrolled_title[i] & 0x80; ++i) {
		c[i] = tab_label->scrolled_title[i];
	}
	c[i] = '\0';

	/* Shift all characters. */
	for (j = 0; tab_label->scrolled_title[i]; ++i, ++j) {
		tab_label->scrolled_title[j] = tab_label->scrolled_title[i];
	}

	/* Set the last (possibly multibyte) character. */
	for (--j, i = 0; c[i]; ++i, ++j) {
		tab_label->scrolled_title[j] = c[i];
	}

	update_label(tab_label);
}