/* pps-search-context.c
 *  this file is part of papers, a gnome document viewer
 *
 * Copyright (C) 2015 Igalia S.L.
 * Copyright (C) 2024 Markus Göllnitz  <camelcasenick@bewares.it>
 *
 * Papers 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 2 of the License, or
 * (at your option) any later version.
 *
 * Papers 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 this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <string.h>

#include "pps-jobs.h"
#include "pps-search-context.h"
#include <papers-view.h>

enum {
	PROP_0,

	PROP_DOCUMENT_MODEL,
	PROP_SEARCH_TERM,
	PROP_ACTIVE,

	NUM_PROPERTIES
};

enum {
	RESULT_ACTIVATED,

	STARTED,
	FINISHED,
	CLEARED,

	N_SIGNALS
};

typedef struct
{
	PpsDocumentModel *model;
	PpsJobFind *job;
	PpsFindOptions options;
	PpsFindOptions supported_options;

	gchar *search_term;
	guint active_use_count;

	gboolean autoselecting;

	GHashTable *per_page_store;
	GListStore *result_model;
	GtkSingleSelection *selection_model;
} PpsSearchContextPrivate;

G_DEFINE_TYPE_WITH_PRIVATE (PpsSearchContext, pps_search_context, G_TYPE_OBJECT)

#define GET_PRIVATE(o) pps_search_context_get_instance_private (o)

static guint signals[N_SIGNALS];

static GParamSpec *props[NUM_PROPERTIES] = {
	NULL,
};

static void
pps_search_context_clear_job (PpsSearchContext *context)
{
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	if (!priv->job)
		return;

	if (!pps_job_is_finished (PPS_JOB (priv->job)))
		pps_job_cancel (PPS_JOB (priv->job));

	g_signal_handlers_disconnect_matched (priv->job, G_SIGNAL_MATCH_DATA, 0, 0, NULL, NULL, context);
	g_clear_object (&priv->job);
}

static void
pps_search_context_set_supported_options (PpsSearchContext *context,
                                          PpsFindOptions options)
{
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	priv->supported_options = options;
}

static void
pps_search_context_setup_document (PpsSearchContext *context,
                                   PpsDocument *document)
{
	if (!document || !PPS_IS_DOCUMENT_FIND (document)) {
		pps_search_context_set_supported_options (context, PPS_FIND_DEFAULT);
		return;
	}

	pps_search_context_set_supported_options (context, pps_document_find_get_supported_options (PPS_DOCUMENT_FIND (document)));
}

static void
document_changed_cb (PpsDocumentModel *model,
                     GParamSpec *pspec,
                     PpsSearchContext *context)
{
	pps_search_context_setup_document (context, pps_document_model_get_document (model));
}

static gchar *
sanitized_substring (const gchar *text,
                     gint start,
                     gint end)
{
	const gchar *p;
	const gchar *start_ptr;
	const gchar *end_ptr;
	guint len = 0;
	g_autofree gchar *retval = NULL;

	if (end - start <= 0)
		return NULL;

	start_ptr = g_utf8_offset_to_pointer (text, start);
	end_ptr = g_utf8_offset_to_pointer (start_ptr, end - start);

	retval = g_malloc (end_ptr - start_ptr + 1);
	p = start_ptr;

	while (p != end_ptr) {
		const gchar *next;

		next = g_utf8_next_char (p);

		if (next != end_ptr) {
			GUnicodeBreakType break_type;

			break_type = g_unichar_break_type (g_utf8_get_char (p));
			if (break_type == G_UNICODE_BREAK_HYPHEN && *next == '\n') {
				p = g_utf8_next_char (next);
				continue;
			}
		}

		if (*p != '\n') {
			strncpy (retval + len, p, next - p);
			len += next - p;
		} else {
			*(retval + len) = ' ';
			len++;
		}

		p = next;
	}

	if (len == 0)
		return NULL;

	retval[len] = 0;

	return g_steal_pointer (&retval);
}

static gchar *
get_surrounding_text_markup (const gchar *text,
                             const gchar *find_text,
                             gboolean case_sensitive,
                             PangoLogAttr *log_attrs,
                             gint log_attrs_length,
                             gint offset,
                             gboolean has_nextline,
                             gboolean hyphen_was_ignored)
{
	gint iter;
	g_autofree gchar *prec = NULL;
	g_autofree gchar *succ = NULL;
	g_autofree gchar *match = NULL;
	gchar *markup;
	gint max_chars;

	iter = MAX (0, offset - 1);
	while (!log_attrs[iter].is_word_start && iter > 0)
		iter--;

	prec = sanitized_substring (text, iter, offset);

	iter = offset;
	offset += g_utf8_strlen (find_text, -1);

	if (has_nextline || g_utf8_offset_to_pointer (text, offset - 1)[0] == '\n') {
		if (has_nextline) {
			offset += 1; /* for newline */
			if (hyphen_was_ignored)
				offset += 1; /* for hyphen */
		}
		match = sanitized_substring (text, iter, offset);
	} else if (!case_sensitive)
		match = g_utf8_substring (text, iter, offset);

	iter = MIN (log_attrs_length, offset + 1);
	max_chars = MIN (log_attrs_length - 1, iter + 100);
	while (TRUE) {
		gint word = iter;

		while (!log_attrs[word].is_word_end && word < max_chars)
			word++;

		if (word > max_chars)
			break;

		iter = word + 1;
	}

	succ = sanitized_substring (text, offset, iter);

	markup = g_markup_printf_escaped ("%s<span weight=\"bold\">%s</span>%s",
	                                  prec ? prec : "", match ? match : find_text, succ ? succ : "");

	return markup;
}

static gchar *
get_page_text (PpsDocument *document,
               PpsPage *page,
               PpsRectangle **areas,
               guint *n_areas)
{
	g_autofree gchar *text = NULL;
	gboolean success;

	text = pps_document_text_get_text (PPS_DOCUMENT_TEXT (document), page);
	success = pps_document_text_get_text_layout (PPS_DOCUMENT_TEXT (document), page, areas, n_areas);

	if (!success)
		return NULL;

	return g_steal_pointer (&text);
}

static gint
get_match_offset (PpsRectangle *areas,
                  guint n_areas,
                  PpsFindRectangle *match,
                  gint offset)
{
	gdouble x, y;
	gint i;

	x = match->x1;
	y = (match->y1 + match->y2) / 2;

	i = offset;

	do {
		PpsRectangle *area = areas + i;
		gdouble area_y = (area->y1 + area->y2) / 2;
		gdouble area_x = (area->x1 + area->x2) / 2;

		if (x >= area->x1 && x < area->x2 &&
		    y >= area->y1 && y <= area->y2 &&
		    area_x >= match->x1 && area_x <= match->x2 &&
		    area_y >= match->y1 && area_y <= match->y2) {
			return i;
		}

		i = (i + 1) % n_areas;
	} while (i != offset);

	return -1;
}

static void
process_matches_idle (PpsSearchContext *context)
{
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);
	PpsDocument *document;
	gint first_match_page = -1;
	g_autoptr (GPtrArray) results_array = g_ptr_array_new ();
	PpsSearchResult **results;
	gsize n_results;
	guint global_index = 0;

	g_return_if_fail (PPS_IS_JOB (priv->job));

	if (!pps_job_find_has_results (priv->job)) {
		g_signal_emit (context, signals[FINISHED], 0, first_match_page);
		pps_search_context_clear_job (context);
		return;
	}

	document = pps_job_get_document (PPS_JOB (priv->job));

	for (guint current_page = 0; current_page < pps_document_get_n_pages (document); current_page++) {
		GList *matches, *l;
		PpsPage *page;
		guint index = 0;
		g_autofree gchar *page_label = NULL;
		g_autofree gchar *page_text = NULL;
		g_autofree PpsRectangle *areas = NULL;
		PpsSearchResult *result = NULL;
		guint n_areas;
		g_autofree PangoLogAttr *text_log_attrs = NULL;
		gulong text_log_attrs_length;
		gint offset;
		GPtrArray *per_page_results_array = g_ptr_array_new_null_terminated (0, g_object_unref, TRUE);

		matches = priv->job->pages[current_page];
		if (!matches)
			continue;

		page = pps_document_get_page (document, current_page);
		page_label = pps_document_get_page_label (document, current_page);
		page_text = get_page_text (document, page, &areas, &n_areas);
		g_object_unref (page);
		if (!page_text)
			continue;

		text_log_attrs_length = g_utf8_strlen (page_text, -1);
		text_log_attrs = g_new0 (PangoLogAttr, text_log_attrs_length + 1);
		pango_get_log_attrs (page_text, -1, -1, NULL, text_log_attrs, text_log_attrs_length + 1);

		if (first_match_page == -1 && current_page >= priv->job->start_page)
			first_match_page = current_page;

		offset = 0;

		for (l = matches; l; l = g_list_next (l)) {
			PpsFindRectangle *match = (PpsFindRectangle *) l->data;
			g_autofree gchar *markup = NULL;
			gint new_offset;

			if (l->prev && ((PpsFindRectangle *) l->prev->data)->next_line) {
				/* Multi-line match. We should delay creation
				 * instead of appending rectangles, but that
				 * requires a lot of refactoring in the
				 * logic around this function. */
				g_assert (result != NULL);
				pps_search_result_append_rectangle (result, match);
				continue;
			}

			new_offset = get_match_offset (areas, n_areas, match, offset);
			if (new_offset == -1) {
				/* It may happen that a text match has no corresponding text area available,
				 * (due to limitations/bugs of Poppler's TextPage->getSelectionWords() used by
				 * poppler-glib poppler_page_get_text_layout_for_area() function) so in that
				 * case we just show matched text because we cannot retrieve surrounding text.
				 * Issue #1943 and related #1545 */
				markup = g_strdup_printf ("<b>%s</b>", priv->job->text);
			} else {
				offset = new_offset;
				markup = get_surrounding_text_markup (page_text,
				                                      priv->job->text,
				                                      priv->job->options & PPS_FIND_CASE_SENSITIVE,
				                                      text_log_attrs,
				                                      text_log_attrs_length,
				                                      offset,
				                                      match->next_line,
				                                      match->after_hyphen);
			}

			result = pps_search_result_new (markup,
			                                page_label,
			                                current_page,
			                                index++,
			                                global_index++,
			                                match);
			g_ptr_array_add (per_page_results_array, g_object_ref (result));
			g_ptr_array_add (results_array, result);
		}

		g_hash_table_insert (priv->per_page_store, GINT_TO_POINTER (current_page), per_page_results_array);
	}

	results = (PpsSearchResult **) g_ptr_array_steal (results_array, &n_results);
	if (n_results > 0)
		g_list_store_splice (priv->result_model, 0, 0, (gpointer *) results, (guint) n_results);

	g_signal_emit (context, signals[FINISHED], 0, first_match_page);

	pps_search_context_clear_job (context);
}

static void
find_job_finished_cb (PpsJobFind *job,
                      PpsSearchContext *context)
{
	g_idle_add_once ((GSourceOnceFunc) process_matches_idle, context);
}

static void
selection_changed_cb (PpsSearchContext *context)
{
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);
	PpsSearchResult *result = PPS_SEARCH_RESULT (gtk_single_selection_get_selected_item (priv->selection_model));

	if (result != NULL && !priv->autoselecting)
		g_signal_emit (context, signals[RESULT_ACTIVATED], 0, result);
}

static void
pps_search_context_dispose (GObject *object)
{
	PpsSearchContext *context = PPS_SEARCH_CONTEXT (object);
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	pps_search_context_clear_job (context);
	g_clear_object (&priv->per_page_store);
	g_clear_object (&priv->result_model);
	g_clear_object (&priv->selection_model);

	G_OBJECT_CLASS (pps_search_context_parent_class)->dispose (object);
}

static void
pps_search_context_init (PpsSearchContext *context)
{
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	priv->per_page_store = g_hash_table_new_full (g_direct_hash, g_direct_equal, NULL, (GDestroyNotify) g_ptr_array_unref);
	priv->result_model = g_list_store_new (PPS_TYPE_SEARCH_RESULT);
	priv->selection_model = gtk_single_selection_new (G_LIST_MODEL (priv->result_model));
	gtk_single_selection_set_autoselect (priv->selection_model, FALSE);
}

static void
pps_search_context_set_property (GObject *object,
                                 guint prop_id,
                                 const GValue *value,
                                 GParamSpec *pspec)
{
	PpsSearchContext *context = PPS_SEARCH_CONTEXT (object);
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	switch (prop_id) {
	case PROP_DOCUMENT_MODEL:
		priv->model = g_value_get_object (value);
		break;
	case PROP_SEARCH_TERM:
		pps_search_context_set_search_term (context, g_value_get_string (value));
		break;
	default:
		G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
	}
}

static void
pps_search_context_get_property (GObject *object,
                                 guint prop_id,
                                 GValue *value,
                                 GParamSpec *pspec)
{
	PpsSearchContext *context = PPS_SEARCH_CONTEXT (object);
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	switch (prop_id) {
	case PROP_SEARCH_TERM:
		g_value_set_string (value, priv->search_term);
		break;
	case PROP_ACTIVE:
		g_value_set_boolean (value, pps_search_context_get_active (context));
		break;
	default:
		G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
	}
}

static void
pps_search_context_constructed (GObject *object)
{
	PpsSearchContext *context = PPS_SEARCH_CONTEXT (object);
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	G_OBJECT_CLASS (pps_search_context_parent_class)->constructed (object);

	g_object_add_weak_pointer (G_OBJECT (priv->model),
	                           (gpointer) &priv->model);

	pps_search_context_setup_document (context, pps_document_model_get_document (priv->model));
	g_signal_connect_object (priv->model, "notify::document",
	                         G_CALLBACK (document_changed_cb),
	                         context, G_CONNECT_DEFAULT);

	g_signal_connect_object (priv->selection_model, "selection-changed",
	                         G_CALLBACK (selection_changed_cb), context,
	                         G_CONNECT_SWAPPED);
}

static void
pps_search_context_class_init (PpsSearchContextClass *klass)
{
	GObjectClass *gobject_class = G_OBJECT_CLASS (klass);

	gobject_class->set_property = pps_search_context_set_property;
	gobject_class->get_property = pps_search_context_get_property;
	gobject_class->dispose = pps_search_context_dispose;
	gobject_class->constructed = pps_search_context_constructed;

	props[PROP_DOCUMENT_MODEL] =
	    g_param_spec_object ("document-model",
	                         "DocumentModel",
	                         "The document model",
	                         PPS_TYPE_DOCUMENT_MODEL,
	                         G_PARAM_WRITABLE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS);

	props[PROP_SEARCH_TERM] =
	    g_param_spec_string ("search-term",
	                         NULL,
	                         "search term for the current context",
	                         "",
	                         G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS | G_PARAM_EXPLICIT_NOTIFY);

	props[PROP_ACTIVE] =
	    g_param_spec_boolean ("active",
	                          NULL,
	                          "search is actively ongoing",
	                          FALSE,
	                          G_PARAM_READABLE | G_PARAM_STATIC_STRINGS | G_PARAM_EXPLICIT_NOTIFY);

	g_object_class_install_properties (gobject_class, NUM_PROPERTIES, props);

	/* Signals */
	signals[RESULT_ACTIVATED] =
	    g_signal_new ("result-activated",
	                  G_TYPE_FROM_CLASS (gobject_class),
	                  G_SIGNAL_RUN_LAST,
	                  0, NULL, NULL,
	                  g_cclosure_marshal_generic,
	                  G_TYPE_NONE, 1,
	                  PPS_TYPE_SEARCH_RESULT);
	signals[STARTED] =
	    g_signal_new ("started",
	                  G_OBJECT_CLASS_TYPE (gobject_class),
	                  G_SIGNAL_RUN_LAST,
	                  0, NULL, NULL,
	                  g_cclosure_marshal_VOID__VOID,
	                  G_TYPE_NONE, 0);
	signals[FINISHED] =
	    g_signal_new ("finished",
	                  G_OBJECT_CLASS_TYPE (gobject_class),
	                  G_SIGNAL_RUN_LAST,
	                  0, NULL, NULL,
	                  g_cclosure_marshal_VOID__INT,
	                  G_TYPE_NONE, 1,
	                  G_TYPE_INT);
	signals[CLEARED] =
	    g_signal_new ("cleared",
	                  G_OBJECT_CLASS_TYPE (gobject_class),
	                  G_SIGNAL_RUN_LAST,
	                  0, NULL, NULL,
	                  g_cclosure_marshal_VOID__VOID,
	                  G_TYPE_NONE, 0);
}

PpsSearchContext *
pps_search_context_new (PpsDocumentModel *model)
{
	return PPS_SEARCH_CONTEXT (g_object_new (PPS_TYPE_SEARCH_CONTEXT,
	                                         "search-term", "",
	                                         "document-model", model,
	                                         NULL));
}

void
pps_search_context_set_search_term (PpsSearchContext *context,
                                    const gchar *search_term)
{
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	if (g_strcmp0 (search_term, priv->search_term) == 0)
		return;

	if (priv->search_term)
		g_free (priv->search_term);

	priv->search_term = g_strdup (search_term);
	g_object_notify_by_pspec (G_OBJECT (context), props[PROP_SEARCH_TERM]);
	pps_search_context_restart (context);
}

/**
 * pps_search_context_get_search_term:
 *
 * Returns: (transfer full)
 */
const gchar *
pps_search_context_get_search_term (PpsSearchContext *context)
{
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	return g_strdup (priv->search_term);
}

void
pps_search_context_set_options (PpsSearchContext *context,
                                PpsFindOptions options)
{
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	if (priv->options == options)
		return;

	priv->options = options;
	pps_search_context_restart (context);
}

PpsFindOptions
pps_search_context_get_options (PpsSearchContext *context)
{
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	return priv->options;
}

/**
 * pps_search_context_get_result_model:
 *
 * Returns: (not nullable) (transfer none): the returned #GtkSingleSelection
 */
GtkSingleSelection *
pps_search_context_get_result_model (PpsSearchContext *context)
{
	g_return_val_if_fail (PPS_IS_SEARCH_CONTEXT (context), NULL);

	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	return priv->selection_model;
}

/**
 * pps_search_context_get_results_on_page:
 *
 * Returns: (not nullable) (transfer full) (element-type PpsSearchResult): #GPtrArray of #PpsSearchResult on the current page
 *
 * Since: 48.0
 */
GPtrArray *
pps_search_context_get_results_on_page (PpsSearchContext *context,
                                        guint page)
{
	g_return_val_if_fail (PPS_IS_SEARCH_CONTEXT (context), NULL);

	PpsSearchContextPrivate *priv = GET_PRIVATE (context);
	GPtrArray *result_array = (GPtrArray *) g_hash_table_lookup (priv->per_page_store, GINT_TO_POINTER (page));

	if (result_array == NULL)
		return g_ptr_array_new_null_terminated (0, g_object_unref, TRUE);

	return g_ptr_array_copy (result_array, (GCopyFunc) g_object_ref, NULL);
}

gboolean
pps_search_context_has_results_on_page (PpsSearchContext *context,
                                        guint page)
{
	g_return_val_if_fail (PPS_IS_SEARCH_CONTEXT (context), false);

	PpsSearchContextPrivate *priv = GET_PRIVATE (context);
	GPtrArray *result_array = (GPtrArray *) g_hash_table_lookup (priv->per_page_store, GINT_TO_POINTER (page));

	return result_array != NULL && result_array->len > 0;
}

/**
 * pps_search_context_activate:
 *
 * Since: 48.0
 */
void
pps_search_context_activate (PpsSearchContext *context)
{
	g_return_if_fail (PPS_IS_SEARCH_CONTEXT (context));

	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	priv->active_use_count++;

	if (priv->active_use_count == 1) {
		g_object_notify_by_pspec (G_OBJECT (context), props[PROP_ACTIVE]);

		/*
		 * If the list model was not filled for a search term that was
		 * set before search was activated, it means we have cancelled
		 * the search job and we need to restart it.
		 *
		 * Otherwise, we can pretend to run a search job, mocking it
		 * with emitting the same signals, because the model is still
		 * up-to-date.
		 */
		if (priv->search_term && priv->search_term[0]) {
			if (g_list_model_get_n_items (G_LIST_MODEL (priv->result_model)) == 0) {
				pps_search_context_restart (context);
			} else {
				g_signal_emit (context, signals[STARTED], 0);
				g_signal_emit (context, signals[FINISHED], 0, -1);
			}
		}
	}
}

/**
 * pps_search_context_release:
 *
 * Since: 48.0
 */
void
pps_search_context_release (PpsSearchContext *context)
{
	g_return_if_fail (PPS_IS_SEARCH_CONTEXT (context));

	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	g_return_if_fail (priv->active_use_count > 0);

	priv->active_use_count--;

	if (priv->active_use_count == 0) {
		g_object_notify_by_pspec (G_OBJECT (context), props[PROP_ACTIVE]);

		pps_search_context_clear_job (context);
	}
}

/**
 * pps_search_context_get_active:
 *
 * Returns: (not nullable): whether search is actively ongoing
 *
 * Since: 48.0
 */
gboolean
pps_search_context_get_active (PpsSearchContext *context)
{
	g_return_val_if_fail (PPS_IS_SEARCH_CONTEXT (context), FALSE);

	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	return priv->active_use_count > 0;
}

void
pps_search_context_restart (PpsSearchContext *context)
{
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	pps_search_context_clear_job (context);
	g_hash_table_remove_all (priv->per_page_store);
	g_list_store_remove_all (priv->result_model);

	if (priv->search_term && priv->search_term[0]) {
		PpsDocument *doc = pps_document_model_get_document (priv->model);

		priv->job = PPS_JOB_FIND (pps_job_find_new (doc,
		                                            pps_document_model_get_page (priv->model),
		                                            pps_document_get_n_pages (doc),
		                                            priv->search_term,
		                                            priv->options));
		g_signal_connect (priv->job, "finished",
		                  G_CALLBACK (find_job_finished_cb),
		                  context);

		g_signal_emit (context, signals[STARTED], 0);
		pps_job_scheduler_push_job (PPS_JOB (priv->job), PPS_JOB_PRIORITY_NONE);
	} else {
		g_signal_emit (context, signals[CLEARED], 0);
	}
}

/**
 * pps_search_context_autoselect_result:
 * @result: a #PpsSearchResult to be the auto-selected one
 *
 * Since: 48.0
 */
void
pps_search_context_autoselect_result (PpsSearchContext *context,
                                      PpsSearchResult *result)
{
	PpsSearchContextPrivate *priv = GET_PRIVATE (context);

	priv->autoselecting = TRUE;
	gtk_single_selection_set_selected (priv->selection_model,
	                                   pps_search_result_get_global_index (result));
	priv->autoselecting = FALSE;
}