/*
 * Tab chrome
 *
 * 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 <stdlib.h>
#include <string.h>

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

#include "tab-chrome.h"
#include "tab.h"
#include "back-forward-button-box.h"
#include "find-toolbar.h"

static void
stop_reload_clicked_cb(GtkToolButton G_GNUC_UNUSED *toolbutton,
	MqTabChrome *chrome)
{
	if (webkit_web_view_is_loading(WEBKIT_WEB_VIEW(chrome->web_view))) {
		webkit_web_view_stop_loading(WEBKIT_WEB_VIEW(chrome->web_view));
	} else {
		webkit_web_view_reload(WEBKIT_WEB_VIEW(chrome->web_view));
	}
}

static void
uri_activate_cb(GtkEntry *entry, MqTabChrome *chrome)
{
	const gchar *uri;

	uri = gtk_entry_get_text(GTK_ENTRY(entry));

	mq_web_view_load_uri(chrome->web_view, uri);
}

static void
home_clicked_cb(GtkToolButton G_GNUC_UNUSED *toolbutton,
	MqTabChrome *chrome)
{
	const gchar *uri;

	uri = mq_config_get_string(chrome->config, "tabs.home");

	mq_web_view_load_uri(chrome->web_view, uri);
}

static void
zoom_out_clicked_cb(GtkButton G_GNUC_UNUSED *button, MqTabChrome *chrome)
{
	mq_web_view_zoom_out(chrome->web_view);
}

static void
zoom_reset_clicked_cb(GtkButton G_GNUC_UNUSED *button, MqTabChrome *chrome)
{
	mq_web_view_zoom_reset(chrome->web_view);
}

static void
zoom_in_clicked_cb(GtkButton G_GNUC_UNUSED *button, MqTabChrome *chrome)
{
	mq_web_view_zoom_in(chrome->web_view);
}

static void
find_clicked_cb(GtkButton G_GNUC_UNUSED *button, MqTabChrome *chrome)
{
	gtk_widget_hide(chrome->menu_popover);
	mq_find_toolbar_reveal(chrome->find_toolbar);
}

static void
fullscreen_clicked_cb(GtkButton G_GNUC_UNUSED *button, MqTabChrome *chrome)
{
	gtk_widget_hide(chrome->menu_popover);
	mq_window_toggle_fullscreen(mq_tab_get_window(chrome->tab));
}

static void
developer_tools_clicked_cb(GtkButton G_GNUC_UNUSED *button, MqTabChrome *chrome)
{
	gtk_widget_hide(chrome->menu_popover);
	webkit_web_inspector_show(webkit_web_view_get_inspector(
			WEBKIT_WEB_VIEW(chrome->web_view)));
}

static void
preferences_clicked_cb(GtkButton G_GNUC_UNUSED *button, MqTabChrome *chrome)
{
	gtk_widget_hide(chrome->menu_popover);
	mq_tab_new("about:preferences", chrome->tab);
	/* TODO: Hack: */
	gtk_notebook_next_page(GTK_NOTEBOOK(chrome->tab->window->notebook));
}

static void
about_clicked_cb(GtkButton G_GNUC_UNUSED *button, MqTabChrome *chrome)
{
	gtk_widget_hide(chrome->menu_popover);
	mq_tab_new("about:", chrome->tab);
	/* TODO: Hack: */
	gtk_notebook_next_page(GTK_NOTEBOOK(chrome->tab->window->notebook));
}

static void
quit_clicked_cb(GtkButton G_GNUC_UNUSED *button, MqTabChrome *chrome)
{
	/* mq_tab_quit() just calls mq_window_quit(), which just calls
	 * mq_application_quit(), which is asynchronous.  So close the menu. */
	gtk_widget_hide(chrome->menu_popover);
	mq_tab_quit(chrome->tab);
}

#define BUTTON_ROWS 6
#define BUTTON_COLS 3
#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_grid_attach(GTK_GRID(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", \
		G_CALLBACK(CB), chrome)

static void
menu_button_clicked_cb(GtkToolButton *tool_button,
	MqTabChrome G_GNUC_UNUSED *chrome)
{
	GtkWidget *grid;
	GtkWidget *buttons[BUTTON_ROWS * BUTTON_COLS];

	/* Set up the grid. */
	grid = gtk_grid_new();

	NEW_BUTTON(0, 0, "zoom-out",               "Zoom out");
	NEW_BUTTON(0, 1, "zoom-original",          "Reset zoom");
	NEW_BUTTON(0, 2, "zoom-in",                "Zoom in");
	NEW_BUTTON(1, 0, "document-open",          "Open file");
	NEW_BUTTON(1, 1, "document-save-as",       "Save page");
	NEW_BUTTON(1, 2, "mail-message-new",       "E-mail link");
	NEW_BUTTON(2, 0, "edit-find",              "Find");
	NEW_BUTTON(2, 1, "document-print-preview", "Print preview");
	NEW_BUTTON(2, 2, "document-print",         "Print");
	NEW_BUTTON(3, 0, "bookmark-new",           "Bookmarks");
	NEW_BUTTON(3, 1, "document-open-recent",   "History");
	NEW_BUTTON(3, 2, "document-save",          "Downloads");
	NEW_BUTTON(4, 0, "view-fullscreen",        "Full screen");
	NEW_BUTTON(4, 1, "document-properties",    "Developer tools");
	NEW_BUTTON(4, 2, "system-run",             "Preferences");
	NEW_BUTTON(5, 0, "help-about",             "About Marquee");
	NEW_BUTTON(5, 2, "application-exit",       "Quit");

	CLICKED_CB(0, 0, zoom_out_clicked_cb);
	CLICKED_CB(0, 1, zoom_reset_clicked_cb);
	CLICKED_CB(0, 2, zoom_in_clicked_cb);
	/* TODO: 1, 0: Open file */
	/* TODO: 1, 1: Save page */
	/* TODO: 1, 2: E-mail link */
	CLICKED_CB(2, 0, find_clicked_cb);
	/* TODO: 2, 1: Print preview */
	/* TODO: 2, 2: Print */
	/* TODO: 3, 0: Bookmarks */
	/* TODO: 3, 1: History */
	/* TODO: 3, 2: Downloads */
	CLICKED_CB(4, 0, fullscreen_clicked_cb);
	CLICKED_CB(4, 1, developer_tools_clicked_cb);
	CLICKED_CB(4, 2, preferences_clicked_cb);
	CLICKED_CB(5, 0, about_clicked_cb);
	CLICKED_CB(5, 2, quit_clicked_cb);

	/* Set up the popover. */
	chrome->menu_popover = gtk_popover_new(GTK_WIDGET(tool_button));
	gtk_container_add(GTK_CONTAINER(chrome->menu_popover), grid);

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

#undef BUTTON_ROWS
#undef BUTTON_COLS
#undef NEW_BUTTON

static void
load_changed_cb(MqWebView *web_view, WebKitLoadEvent load_event,
	MqTabChrome *chrome)
{
	const gchar *uri;

	switch (load_event) {
		case WEBKIT_LOAD_STARTED:
		case WEBKIT_LOAD_REDIRECTED:
		case WEBKIT_LOAD_COMMITTED:
			uri = mq_web_view_get_uri(web_view);
			gtk_entry_set_text(GTK_ENTRY(chrome->uri_entry), uri);
			gtk_entry_set_attributes(GTK_ENTRY(chrome->uri_entry),
				NULL);
			break;
		case WEBKIT_LOAD_FINISHED:
			gtk_entry_set_progress_fraction(
				GTK_ENTRY(chrome->uri_entry), 0.0);
			break;
	}
}

static gboolean
load_failed_cb(WebKitWebView G_GNUC_UNUSED *web_view,
	WebKitLoadEvent G_GNUC_UNUSED load_event,
	gchar G_GNUC_UNUSED *failing_uri, GError G_GNUC_UNUSED *error,
	MqTabChrome *chrome)
{
	chrome->load_failed = TRUE;
	return FALSE;
}

/* TODO: <Esc> key in URI bar should reset to URI of current hovered link, if
 * any, even if different from chrome->hovered_link_uri. */
static void
mouse_target_changed_cb(MqWebView G_GNUC_UNUSED *web_view,
	WebKitHitTestResult *hit_test_result, guint G_GNUC_UNUSED modifiers,
	MqTabChrome *chrome)
{
	const gchar *uri;

	uri = mq_web_view_get_uri(web_view);
	if (webkit_hit_test_result_context_is_link(hit_test_result)) {
		if (gtk_widget_has_focus(chrome->uri_entry)) {
		} else if (chrome->hovered_link_uri &&
				g_strcmp0(gtk_entry_get_text(
					GTK_ENTRY(chrome->uri_entry)),
				chrome->hovered_link_uri) != 0) {
			/* The user has edited the hovered link URI in the URI
			 * bar. */
		} else if (g_strcmp0(gtk_entry_get_text(
					GTK_ENTRY(chrome->uri_entry)), uri) ==
				0) {
			g_free(chrome->hovered_link_uri);
			chrome->hovered_link_uri = g_strdup(
				webkit_hit_test_result_get_link_uri(
					hit_test_result));
			gtk_entry_set_text(GTK_ENTRY(chrome->uri_entry),
				chrome->hovered_link_uri);
			gtk_entry_set_attributes(GTK_ENTRY(chrome->uri_entry),
				chrome->hovered_link_style);
		} else if (gtk_entry_get_attributes(
					GTK_ENTRY(chrome->uri_entry)) ==
				chrome->hovered_link_style) {
			/* The URI bar's text differs from the Web view's URI
			 * because the mouse was already targeting a different
			 * link. */
			g_free(chrome->hovered_link_uri);
			chrome->hovered_link_uri = g_strdup(
				webkit_hit_test_result_get_link_uri(
					hit_test_result));
			gtk_entry_set_text(GTK_ENTRY(chrome->uri_entry),
				chrome->hovered_link_uri);
		}
	} else if (gtk_entry_get_attributes(GTK_ENTRY(chrome->uri_entry)) ==
				chrome->hovered_link_style) {
		if (gtk_widget_has_focus(chrome->uri_entry)) {
		} else if (g_strcmp0(gtk_entry_get_text(
						GTK_ENTRY(chrome->uri_entry)),
				chrome->hovered_link_uri) == 0) {
			/* The user hasn't edited the hovered link URI in the
			 * URI bar. */
			g_free(chrome->hovered_link_uri);
			chrome->hovered_link_uri = NULL;
			gtk_entry_set_text(GTK_ENTRY(chrome->uri_entry), uri);
			gtk_entry_set_attributes(GTK_ENTRY(chrome->uri_entry),
				NULL);
		}
	}
}

static void
load_progress_cb(WebKitWebView *web_view, GParamSpec G_GNUC_UNUSED *paramspec,
	MqTabChrome *chrome)
{
	/*
	 * If loading fails, the WebKitWebView's "estimated-load-progress" is
	 * set to 1.0 after signals like "load-changed" and "load-failed" are
	 * emitted.  So the only way to avoid leaving behind a full progress bar
	 * after, for example, canceling a page load is to save a flag on a
	 * failed load and only update the progress bar if the flag is unset.
	 */
	if (chrome->load_failed) {
		chrome->load_failed = FALSE;
		return;
	}
	gtk_entry_set_progress_fraction(GTK_ENTRY(chrome->uri_entry),
		webkit_web_view_get_estimated_load_progress(web_view));
}

static void
uri_cb(MqWebView *web_view, GParamSpec G_GNUC_UNUSED *paramspec,
	MqTabChrome *chrome)
{
	const gchar *uri;

	uri = mq_web_view_get_uri(web_view);
	gtk_entry_set_text(GTK_ENTRY(chrome->uri_entry), uri);
	gtk_entry_set_attributes(GTK_ENTRY(chrome->uri_entry), NULL);
}

static void
loading_cb(MqWebView *web_view, GParamSpec G_GNUC_UNUSED *paramspec,
	MqTabChrome *chrome)
{
	if (webkit_web_view_is_loading(WEBKIT_WEB_VIEW(web_view))) {
		gtk_tool_button_set_icon_widget(
			GTK_TOOL_BUTTON(chrome->stop_reload_button),
			chrome->stop_icon);
		gtk_tool_button_set_label(
			GTK_TOOL_BUTTON(chrome->stop_reload_button), "Stop");
		gtk_widget_set_tooltip_text(
			GTK_WIDGET(chrome->stop_reload_button),
			"Stop loading the current page");
	} else {
		gtk_tool_button_set_icon_widget(
			GTK_TOOL_BUTTON(chrome->stop_reload_button),
			chrome->reload_icon);
		gtk_tool_button_set_label(
			GTK_TOOL_BUTTON(chrome->stop_reload_button), "Reload");
		gtk_widget_set_tooltip_text(
			GTK_WIDGET(chrome->stop_reload_button),
			"Reload the current page");
	}
	gtk_widget_show_all(GTK_WIDGET(chrome->stop_reload_button));
}

static GtkWidget *
navigation_toolbar_new(MqTabChrome *chrome, const gchar *uri)
{
	GtkToolbar  *navigation_toolbar;
	GtkToolItem *back_forward_tool_item;
	GtkWidget   *back_forward_event_box;
	GtkToolItem *uri_tool_item;
	GtkToolItem *home_button;

	navigation_toolbar = GTK_TOOLBAR(gtk_toolbar_new());

	/* Back/forward button box */
	back_forward_tool_item = gtk_tool_item_new();
	back_forward_event_box = mq_back_forward_button_box_new(chrome->tab,
		chrome->web_view);
	gtk_container_add(GTK_CONTAINER(back_forward_tool_item),
		back_forward_event_box);
	gtk_toolbar_insert(navigation_toolbar, back_forward_tool_item, -1);

	/* Stop/reload button */
	chrome->stop_icon = gtk_image_new_from_icon_name("process-stop",
		GTK_ICON_SIZE_SMALL_TOOLBAR);
	g_object_ref_sink(chrome->stop_icon);
	chrome->reload_icon = gtk_image_new_from_icon_name("view-refresh",
		GTK_ICON_SIZE_SMALL_TOOLBAR);
	g_object_ref_sink(chrome->reload_icon);
	chrome->stop_reload_button = gtk_tool_button_new(chrome->stop_icon,
		"Stop");
	gtk_widget_set_tooltip_text(GTK_WIDGET(chrome->stop_reload_button),
		"Stop loading the current page");
	g_signal_connect(chrome->stop_reload_button, "clicked",
		G_CALLBACK(stop_reload_clicked_cb), chrome);
	gtk_toolbar_insert(navigation_toolbar, chrome->stop_reload_button, -1);

	/* URI bar */
	uri_tool_item = gtk_tool_item_new();
	chrome->uri_entry = gtk_entry_new();
	if (uri) {
		gtk_entry_set_text(GTK_ENTRY(chrome->uri_entry), uri);
	}
	gtk_entry_set_placeholder_text(GTK_ENTRY(chrome->uri_entry),
		"URI...");
	gtk_entry_set_icon_from_icon_name(GTK_ENTRY(chrome->uri_entry),
		GTK_ENTRY_ICON_PRIMARY, "text-x-generic");
	gtk_entry_set_progress_fraction(GTK_ENTRY(chrome->uri_entry), 0.0);
	g_signal_connect(chrome->uri_entry, "activate",
		G_CALLBACK(uri_activate_cb), chrome);
	gtk_container_add(GTK_CONTAINER(uri_tool_item),
		chrome->uri_entry);
	gtk_tool_item_set_expand(uri_tool_item, TRUE);
	gtk_toolbar_insert(navigation_toolbar, uri_tool_item, -1);

	/* URI bar hovered link style */
	chrome->hovered_link_style = pango_attr_list_new();
	pango_attr_list_insert(chrome->hovered_link_style,
		pango_attr_style_new(PANGO_STYLE_ITALIC));

	/* Home button */
	home_button = gtk_tool_button_new(gtk_image_new_from_icon_name(
			"go-home", GTK_ICON_SIZE_SMALL_TOOLBAR), "Home");
	gtk_widget_set_tooltip_text(GTK_WIDGET(home_button),
		"Load the home page");
	g_signal_connect(home_button, "clicked",
		G_CALLBACK(home_clicked_cb), chrome);
	gtk_toolbar_insert(navigation_toolbar, home_button, -1);

	/* Menu button */
	chrome->menu_button = gtk_tool_button_new(
		gtk_image_new_from_icon_name("open-menu-symbolic",
			GTK_ICON_SIZE_SMALL_TOOLBAR), "Menu");
	gtk_widget_set_tooltip_text(GTK_WIDGET(chrome->menu_button),
		"Open menu");
	g_signal_connect(chrome->menu_button, "clicked",
		G_CALLBACK(menu_button_clicked_cb), chrome);
	gtk_toolbar_insert(navigation_toolbar, chrome->menu_button, -1);

	gtk_widget_set_hexpand(GTK_WIDGET(navigation_toolbar), TRUE);

	chrome->load_failed = FALSE;
	chrome->hovered_link_uri = NULL;

	g_signal_connect(chrome->web_view, "load-changed",
		G_CALLBACK(load_changed_cb), chrome);
	g_signal_connect(chrome->web_view, "load-failed",
		G_CALLBACK(load_failed_cb), chrome);
	g_signal_connect(chrome->web_view, "mouse-target-changed",
		G_CALLBACK(mouse_target_changed_cb), chrome);
	g_signal_connect(chrome->web_view, "notify::estimated-load-progress",
		G_CALLBACK(load_progress_cb), chrome);
	g_signal_connect(chrome->web_view, "notify::rewritten-uri",
		G_CALLBACK(uri_cb), chrome);
	g_signal_connect(chrome->web_view, "notify::is-loading",
		G_CALLBACK(loading_cb), chrome);

	return GTK_WIDGET(navigation_toolbar);
}

MqTabChrome *
mq_tab_chrome_new(MqTab *tab, MqFindToolbar *find_toolbar, MqWebView *web_view,
	const gchar *uri)
{
	MqTabChrome *chrome;

	chrome = malloc(sizeof(*chrome));
	chrome->config = mq_application_get_config(mq_tab_get_application(tab));
	chrome->tab = tab;
	chrome->find_toolbar = find_toolbar;
	chrome->web_view = web_view;

	chrome->container = gtk_box_new(GTK_ORIENTATION_VERTICAL, 0);
	gtk_box_pack_start(GTK_BOX(chrome->container),
		navigation_toolbar_new(chrome, uri), FALSE, FALSE, 0);

	return chrome;
}

GtkWidget *
mq_tab_chrome_get_container(MqTabChrome *chrome)
{
	return chrome->container;
}