/*
 * 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 <gtk/gtk.h>
#include <webkit2/webkit2.h>

#include "tab-chrome.h"

static void
back_clicked_cb(GtkToolButton __attribute__((unused)) *toolbutton,
	MqTabChrome *chrome)
{
	webkit_web_view_go_back(chrome->web_view);
}

static void
forward_clicked_cb(GtkToolButton __attribute__((unused)) *toolbutton,
	MqTabChrome *chrome)
{
	webkit_web_view_go_forward(chrome->web_view);
}

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

static void
uri_activate_cb(GtkEntry *entry, MqTabChrome *chrome)
{
	webkit_web_view_load_uri(chrome->web_view,
		gtk_entry_get_text(GTK_ENTRY(entry)));
}

static GtkWidget *
navigation_toolbar_new(MqTabChrome *chrome, gchar *uri)
{
	GtkToolbar  *navigation_toolbar;
	GtkToolItem *uri_toolitem;

	navigation_toolbar = GTK_TOOLBAR(gtk_toolbar_new());

	/* Back button */
	chrome->back_button = gtk_tool_button_new(
		gtk_image_new_from_icon_name("go-previous",
			GTK_ICON_SIZE_SMALL_TOOLBAR), "Back");
	gtk_widget_set_tooltip_text(GTK_WIDGET(chrome->back_button),
		"Go back one page");
	g_signal_connect(chrome->back_button, "clicked",
		G_CALLBACK(back_clicked_cb), chrome);
	gtk_toolbar_insert(navigation_toolbar, chrome->back_button, -1);

	/* Forward button */
	chrome->forward_button = gtk_tool_button_new(
		gtk_image_new_from_icon_name("go-next",
			GTK_ICON_SIZE_SMALL_TOOLBAR), "Forward");
	gtk_widget_set_tooltip_text(GTK_WIDGET(chrome->forward_button),
		"Go forward one page");
	g_signal_connect(chrome->forward_button, "clicked",
		G_CALLBACK(forward_clicked_cb), chrome);
	gtk_toolbar_insert(navigation_toolbar, chrome->forward_button, -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_toolitem = gtk_tool_item_new();
	chrome->uri_entry = gtk_entry_new();
	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_toolitem),
		chrome->uri_entry);
	gtk_tool_item_set_expand(uri_toolitem, TRUE);
	gtk_toolbar_insert(navigation_toolbar, uri_toolitem, -1);

	gtk_widget_set_hexpand(GTK_WIDGET(navigation_toolbar), TRUE);

	chrome->load_failed = FALSE;

	return GTK_WIDGET(navigation_toolbar);
}

MqTabChrome *
mq_tab_chrome_new(gchar *uri, GtkWidget *tab_image, GtkWidget *tab_label)
{
	MqTabChrome *chrome;

	chrome = malloc(sizeof(*chrome));

	chrome->tab_image = tab_image;
	chrome->tab_label = tab_label;

	chrome->container = gtk_grid_new();
	gtk_grid_attach(GTK_GRID(chrome->container),
		navigation_toolbar_new(chrome, uri), 0, 0, 1, 1);

	return chrome;
}

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

static void
update_tab_image(MqTabChrome *chrome)
{
	gtk_image_set_from_surface(GTK_IMAGE(chrome->tab_image),
		webkit_web_view_get_favicon(chrome->web_view));
}

static void
update_tab_label(MqTabChrome *chrome)
{
	gtk_label_set_text(GTK_LABEL(chrome->tab_label),
		g_strdup_printf("%d. %s", chrome->tab_position + 1,
			webkit_web_view_get_title(chrome->web_view)));
}

void
mq_tab_chrome_update_tab_position(MqTabChrome *chrome, guint position)
{
	chrome->tab_position = position;
	update_tab_label(chrome);
}

static void
load_changed_cb(WebKitWebView *web_view, WebKitLoadEvent load_event,
	MqTabChrome *chrome)
{
	switch (load_event) {
		case WEBKIT_LOAD_STARTED:
		case WEBKIT_LOAD_REDIRECTED:
		case WEBKIT_LOAD_COMMITTED:
			gtk_entry_set_text(GTK_ENTRY(chrome->uri_entry),
				webkit_web_view_get_uri(web_view));
			break;
		case WEBKIT_LOAD_FINISHED:
			gtk_entry_set_progress_fraction(
				GTK_ENTRY(chrome->uri_entry), 0.0);
			break;
	}

	gtk_widget_set_sensitive(GTK_WIDGET(chrome->back_button),
		webkit_web_view_can_go_back(web_view));
	gtk_widget_set_sensitive(GTK_WIDGET(chrome->forward_button),
		webkit_web_view_can_go_forward(web_view));
}

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

static void
load_progress_cb(WebKitWebView *web_view,
	GParamSpec __attribute__((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(WebKitWebView *web_view, GParamSpec __attribute__((unused)) *paramspec,
	MqTabChrome *chrome)
{
	gtk_entry_set_text(GTK_ENTRY(chrome->uri_entry),
		webkit_web_view_get_uri(web_view));
}

static void
favicon_cb(WebKitWebView __attribute__((unused)) *web_view,
	GParamSpec __attribute__((unused)) *paramspec, MqTabChrome *chrome)
{
	update_tab_image(chrome);
}

static void
title_cb(WebKitWebView __attribute__((unused)) *web_view,
	GParamSpec __attribute__((unused)) *paramspec, MqTabChrome *chrome)
{
	update_tab_label(chrome);
}

static void
loading_cb(WebKitWebView *web_view,
	GParamSpec __attribute__((unused)) *paramspec, MqTabChrome *chrome)
{
	if (webkit_web_view_is_loading(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 void
connect_web_view(MqTabChrome *chrome)
{
	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, "notify::estimated-load-progress",
		G_CALLBACK(load_progress_cb), chrome);
	g_signal_connect(chrome->web_view, "notify::uri",
		G_CALLBACK(uri_cb), chrome);
	/* FIXME: Doesn't work? */
	g_signal_connect(chrome->web_view, "notify::favicon",
		G_CALLBACK(favicon_cb), chrome);
	g_signal_connect(chrome->web_view, "notify::title",
		G_CALLBACK(title_cb), chrome);
	g_signal_connect(chrome->web_view, "notify::is-loading",
		G_CALLBACK(loading_cb), chrome);
}

void
mq_tab_chrome_set_web_view(MqTabChrome *chrome, WebKitWebView *web_view)
{
	chrome->web_view = web_view;
	connect_web_view(chrome);
}