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

#include "find-toolbar.h"
#include "web-view.h"

struct _MqFindToolbar {
	GtkRevealer           parent_instance;
	MqWebView            *web_view;
	GtkWidget            *search_entry;
	GtkWidget            *matches_label;
	gboolean              match_case;
	gboolean              searching;
	WebKitFindController *find_controller;
};

enum {
	PROP_WEB_VIEW = 1,
	N_PROPERTIES
};

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

struct _MqFindToolbarClass {
	GtkRevealerClass parent_class;
};

G_DEFINE_TYPE(MqFindToolbar, mq_find_toolbar, GTK_TYPE_REVEALER)

static void
search(MqFindToolbar *find_toolbar, gboolean forward)
{
	guint32 find_options;

	find_options = WEBKIT_FIND_OPTIONS_WRAP_AROUND;
	if (!find_toolbar->match_case) {
		find_options |= WEBKIT_FIND_OPTIONS_CASE_INSENSITIVE;
	}
	if (!forward) {
		find_options |= WEBKIT_FIND_OPTIONS_BACKWARDS;
	}
	webkit_find_controller_search(find_toolbar->find_controller,
		gtk_entry_get_text(GTK_ENTRY(find_toolbar->search_entry)),
		find_options, G_MAXUINT);
	find_toolbar->searching = TRUE;
}

static void
search_finished(MqFindToolbar *find_toolbar)
{
	find_toolbar->searching = FALSE;
	webkit_find_controller_search_finish(find_toolbar->find_controller);
}

static void
hide(MqFindToolbar *find_toolbar)
{
	gtk_revealer_set_reveal_child(GTK_REVEALER(find_toolbar),
		FALSE);
	gtk_label_set_text(GTK_LABEL(find_toolbar->matches_label), NULL);
	search_finished(find_toolbar);
}

static void
search_changed_cb(GtkSearchEntry G_GNUC_UNUSED *entry,
	MqFindToolbar *find_toolbar)
{
	search(find_toolbar, TRUE);
}

static gboolean
search_key_press_event_cb(GtkSearchEntry G_GNUC_UNUSED *entry,
	GdkEventKey *event, MqFindToolbar *find_toolbar)
{
	switch (event->keyval) {
		case GDK_KEY_Escape:
			hide(find_toolbar);
			return TRUE;
		case GDK_KEY_Return:
		case GDK_KEY_KP_Enter:
		case GDK_KEY_ISO_Enter:
			search(find_toolbar,
				!(event->state & GDK_SHIFT_MASK));
			return TRUE;
		default:
			return FALSE;
	}
}

static void
prev_clicked_cb(GtkButton G_GNUC_UNUSED *button, MqFindToolbar *find_toolbar)
{
	/* Calling this method before webkit_find_controller_search() or
	 * webkit_find_controller_count_matches() is a programming error. */
	if (find_toolbar->searching) {
		webkit_find_controller_search_previous(
			find_toolbar->find_controller);
	} else {
		search(find_toolbar, FALSE);
	}
}

static void
next_clicked_cb(GtkButton G_GNUC_UNUSED *button, MqFindToolbar *find_toolbar)
{
	/* Calling this method before webkit_find_controller_search() or
	 * webkit_find_controller_count_matches() is a programming error. */
	if (find_toolbar->searching) {
		webkit_find_controller_search_next(
			find_toolbar->find_controller);
	} else {
		search(find_toolbar, TRUE);
	}
}

static void
match_case_toggled_cb(GtkToggleButton *toggle_button,
	MqFindToolbar *find_toolbar)
{
	find_toolbar->match_case = gtk_toggle_button_get_active(toggle_button);
	search(find_toolbar, TRUE);
}

static void
close_clicked_cb(GtkButton G_GNUC_UNUSED *button, MqFindToolbar *find_toolbar)
{
	hide(find_toolbar);
}

static void
found_text_cb(WebKitFindController G_GNUC_UNUSED *find_controller,
	guint match_count, MqFindToolbar *find_toolbar)
{
	gchar *text;

	if (match_count == 1) {
		text = g_strdup("1 match");
	} else {
		text = g_strdup_printf("%u matches", match_count);
	}
	gtk_label_set_text(GTK_LABEL(find_toolbar->matches_label), text);
	g_free(text);

}

static void
failed_to_find_text_cb(WebKitFindController G_GNUC_UNUSED *find_controller,
	MqFindToolbar *find_toolbar)
{
	gtk_label_set_text(GTK_LABEL(find_toolbar->matches_label),
		"No matches");
}

static void
constructed(GObject *object)
{
	MqFindToolbar *find_toolbar;

	if (G_OBJECT_CLASS(mq_find_toolbar_parent_class)->constructed) {
		G_OBJECT_CLASS(mq_find_toolbar_parent_class)->constructed(
			object);
	}

	find_toolbar = MQ_FIND_TOOLBAR(object);

	find_toolbar->find_controller = webkit_web_view_get_find_controller(
		WEBKIT_WEB_VIEW(find_toolbar->web_view));
	g_signal_connect(find_toolbar->find_controller, "found-text",
		G_CALLBACK(found_text_cb), find_toolbar);
	g_signal_connect(find_toolbar->find_controller, "failed-to-find-text",
		G_CALLBACK(failed_to_find_text_cb), find_toolbar);
}

static void
get_property(GObject *object, guint property_id, GValue *value,
	GParamSpec *pspec)
{
	MqFindToolbar *find_toolbar;

	find_toolbar = MQ_FIND_TOOLBAR(object);

	switch (property_id) {
		case PROP_WEB_VIEW:
			g_value_set_object(value, find_toolbar->web_view);
			break;
		default:
			G_OBJECT_WARN_INVALID_PROPERTY_ID(object, property_id,
				pspec);
			break;
	}
}

static void
set_property(GObject *object, guint property_id, const GValue *value,
	GParamSpec *pspec)
{
	MqFindToolbar *find_toolbar;

	find_toolbar = MQ_FIND_TOOLBAR(object);

	switch (property_id) {
		case PROP_WEB_VIEW:
			find_toolbar->web_view = g_value_get_object(value);
			break;
		default:
			G_OBJECT_WARN_INVALID_PROPERTY_ID(object, property_id,
				pspec);
			break;
	}
}

static void
mq_find_toolbar_class_init(MqFindToolbarClass *klass)
{
	GObjectClass *object_class = G_OBJECT_CLASS(klass);

	object_class->constructed  = constructed;
	object_class->get_property = get_property;
	object_class->set_property = set_property;

	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_find_toolbar_init(MqFindToolbar *find_toolbar)
{
	GtkWidget *close_button;
	GtkWidget *prev_button;
	GtkWidget *next_button;
	GtkWidget *match_case_button;
	GtkWidget *box;

	/* Search entry */
	find_toolbar->search_entry = gtk_search_entry_new();
	g_signal_connect(find_toolbar->search_entry, "search-changed",
		G_CALLBACK(search_changed_cb), find_toolbar);
	g_signal_connect(find_toolbar->search_entry, "key-press-event",
		G_CALLBACK(search_key_press_event_cb), find_toolbar);

	/* Previous button */
	prev_button = gtk_button_new_from_icon_name("go-up",
		GTK_ICON_SIZE_BUTTON);
	gtk_widget_set_tooltip_text(prev_button, "Find previous occurrence");
	gtk_widget_set_can_focus(prev_button, FALSE);
	g_signal_connect(prev_button, "clicked",
		G_CALLBACK(prev_clicked_cb), find_toolbar);

	/* Next button */
	next_button = gtk_button_new_from_icon_name("go-down",
		GTK_ICON_SIZE_BUTTON);
	gtk_widget_set_tooltip_text(next_button, "Find next occurrence");
	gtk_widget_set_can_focus(next_button, FALSE);
	g_signal_connect(next_button, "clicked",
		G_CALLBACK(next_clicked_cb), find_toolbar);

	/* Case sensitivity button */
	match_case_button = gtk_toggle_button_new_with_label("Match case");
	gtk_widget_set_tooltip_text(match_case_button,
		"Search with case sensitivity");
	gtk_widget_set_can_focus(match_case_button, FALSE);
	g_signal_connect(match_case_button, "toggled",
		G_CALLBACK(match_case_toggled_cb), find_toolbar);

	/* Matches label */
	find_toolbar->matches_label = gtk_label_new(NULL);

	/* 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 find bar");
	gtk_widget_set_can_focus(close_button, FALSE);
	g_signal_connect(close_button, "clicked",
		G_CALLBACK(close_clicked_cb), find_toolbar);

	/* Box */
	box = gtk_box_new(GTK_ORIENTATION_HORIZONTAL, 0);
	gtk_box_pack_start(GTK_BOX(box), find_toolbar->search_entry,
		FALSE, FALSE, 0);
	gtk_box_pack_start(GTK_BOX(box), prev_button, FALSE, FALSE, 0);
	gtk_box_pack_start(GTK_BOX(box), next_button, FALSE, FALSE, 0);
	gtk_box_pack_start(GTK_BOX(box), match_case_button, FALSE, FALSE, 0);
	gtk_box_pack_start(GTK_BOX(box), find_toolbar->matches_label,
		FALSE, FALSE, 0);
	gtk_box_pack_end(GTK_BOX(box), close_button, FALSE, FALSE, 0);

	/* Revealer */
	gtk_revealer_set_transition_type(GTK_REVEALER(find_toolbar),
		GTK_REVEALER_TRANSITION_TYPE_SLIDE_DOWN);
	gtk_container_add(GTK_CONTAINER(find_toolbar), box);

	find_toolbar->match_case = FALSE;
	find_toolbar->searching = FALSE;
}

GtkWidget *
mq_find_toolbar_new(MqWebView *web_view)
{
	return GTK_WIDGET(g_object_new(MQ_TYPE_FIND_TOOLBAR,
		"web-view", web_view,
		NULL));
}

void
mq_find_toolbar_reveal(MqFindToolbar *find_toolbar)
{
	gtk_revealer_set_reveal_child(GTK_REVEALER(find_toolbar), TRUE);
	gtk_widget_grab_focus(find_toolbar->search_entry);
}