/*
 * 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 "notebook.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
close_clicked_cb(GtkWidget G_GNUC_UNUSED *button, MqTabLabel *tab_label)
{
	gtk_widget_hide(tab_label->popover);
	mq_tab_page_close(tab_label->tab_page);
}

static void
new_tab_clicked_cb(GtkWidget G_GNUC_UNUSED *button, MqTabLabel *tab_label)
{
	mq_notebook_insert_sibling(
		MQ_NOTEBOOK(gtk_widget_get_parent(GTK_WIDGET(tab_label))),
		NULL,
		tab_label->tab_page,
		TRUE);
	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);
}

#define BUTTON_ROWS 2
#define BUTTON_COLS 4
#define NEW_BUTTON(Y, X, ICON, TOOLTIP) \
	G_STMT_START { \
		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); \
	} G_STMT_END
#define CLICKED_CB(Y, X, CB) \
	g_signal_connect(buttons[Y * BUTTON_COLS + X], "clicked", \
		G_CALLBACK(CB), tab_label)

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

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

	CLICKED_CB(0, 0, reload_tab_clicked_cb);
	/* TODO: 0, 1: Duplicate tab */
	gtk_widget_set_sensitive(buttons[0 * BUTTON_COLS + 1], FALSE);
	/* TODO: 0, 2: Move tab to new window */
	gtk_widget_set_sensitive(buttons[0 * BUTTON_COLS + 2], FALSE);
	CLICKED_CB(0, 3, close_clicked_cb);
	CLICKED_CB(1, 0, new_tab_clicked_cb);
	CLICKED_CB(1, 1, new_window_clicked_cb);
	/* TODO: 1, 2: Undo close tab */
	gtk_widget_set_sensitive(buttons[1 * BUTTON_COLS + 2], FALSE);

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

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

#undef BUTTON_ROWS
#undef BUTTON_COLS
#undef NEW_BUTTON
#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 == 2) {
		mq_tab_page_close(tab_label->tab_page);
	} else 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-html", 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-html",
		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");
	g_signal_connect(close_button, "clicked",
		G_CALLBACK(close_clicked_cb), tab_label);

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