/*
* Copyright © 2013, 2014 Red Hat, Inc.
*
* This program 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.
*
* This program 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, see <http://www.gnu.org/licenses/>.
*/
#include "config.h"
#include <string.h>
#include "terminal-app.h"
#include "terminal-debug.h"
#include "terminal-libgsystem.h"
#include "terminal-screen-container.h"
#include "terminal-search-provider.h"
#include "terminal-search-provider-gdbus-generated.h"
#include "terminal-window.h"
struct _TerminalSearchProvider
{
GObject parent;
TerminalSearchProvider2 *skeleton;
};
struct _TerminalSearchProviderClass
{
GObjectClass parent_class;
};
G_DEFINE_TYPE (TerminalSearchProvider, terminal_search_provider, G_TYPE_OBJECT)
static char *
normalize_casefold_and_unaccent (const char *str)
{
gs_free char *casefolded = NULL, *normalized = NULL;
char *retval = NULL;
if (str == NULL)
goto out;
normalized = g_utf8_normalize (str, -1, G_NORMALIZE_ALL_COMPOSE);
casefolded = g_utf8_casefold (normalized, -1);
retval = g_str_to_ascii (casefolded, NULL);
out:
return retval;
}
static char **
normalize_casefold_and_unaccent_terms (const char* const *terms)
{
char **casefolded_terms;
guint i, n;
n = g_strv_length ((char **) terms);
casefolded_terms = g_new (char *, n + 1);
for (i = 0; i < n; i++)
casefolded_terms[i] = normalize_casefold_and_unaccent (terms[i]);
casefolded_terms[n] = NULL;
return casefolded_terms;
}
static gboolean
match_terms (const char *str,
const char* const *terms)
{
gs_free char *casefolded_str = NULL;
gboolean matches = TRUE;
guint i;
if (str == NULL)
{
matches = FALSE;
goto out;
}
casefolded_str = normalize_casefold_and_unaccent (str);
for (i = 0; terms[i] != NULL; i++)
{
if (strstr (casefolded_str, terms[i]) == NULL)
{
matches = FALSE;
break;
}
}
out:
return matches;
}
static gboolean
handle_get_initial_result_set_cb (TerminalSearchProvider2 *skeleton,
GDBusMethodInvocation *invocation,
const char *const *terms,
gpointer user_data)
{
GList *l, *screens = NULL, *windows;
gs_unref_ptrarray GPtrArray *results;
TerminalApp *app;
gs_strfreev char **casefolded_terms = NULL;
_terminal_debug_print (TERMINAL_DEBUG_SEARCH, "GetInitialResultSet started\n");
app = terminal_app_get ();
windows = gtk_application_get_windows (GTK_APPLICATION (app));
for (l = windows; l != NULL; l = l->next)
{
TerminalWindow *window = (TerminalWindow*)(l->data);
GList *c, *containers;
if (!TERMINAL_IS_WINDOW (l->data))
continue;
containers = terminal_window_list_screen_containers (window);
for (c = containers; c != NULL; c = c->next)
{
TerminalScreenContainer *container = TERMINAL_SCREEN_CONTAINER (c->data);
TerminalScreen *screen;
screen = terminal_screen_container_get_screen (container);
screens = g_list_prepend (screens, screen);
}
}
casefolded_terms = normalize_casefold_and_unaccent_terms (terms);
results = g_ptr_array_new_with_free_func (g_free);
for (l = screens; l != NULL; l = l->next)
{
TerminalScreen *screen = TERMINAL_SCREEN (l->data);
gs_free char *cmdline = NULL, *process = NULL;
const char *cwd, *title;
cwd = vte_terminal_get_current_directory_uri (VTE_TERMINAL (screen));
title = terminal_screen_get_title (screen);
terminal_screen_has_foreground_process (screen, &process, &cmdline);
if (match_terms (cwd, (const char *const *) casefolded_terms) ||
match_terms (title, (const char *const *) casefolded_terms) ||
match_terms (process, (const char *const *) casefolded_terms) ||
match_terms (cmdline, (const char *const *) casefolded_terms))
{
const char *uuid;
uuid = terminal_screen_get_uuid (screen);
g_ptr_array_add (results, g_strdup (uuid));
_terminal_debug_print (TERMINAL_DEBUG_SEARCH, "Search hit: %s\n", uuid);
}
}
g_ptr_array_add (results, NULL);
terminal_search_provider2_complete_get_initial_result_set (skeleton,
invocation,
(const char *const *) results->pdata);
_terminal_debug_print (TERMINAL_DEBUG_SEARCH, "GetInitialResultSet completed\n");
return TRUE;
}
static gboolean
handle_get_subsearch_result_set_cb (TerminalSearchProvider2 *skeleton,
GDBusMethodInvocation *invocation,
const char *const *previous_results,
const char *const *terms,
gpointer user_data)
{
gs_unref_ptrarray GPtrArray *results;
TerminalApp *app;
gs_strfreev char **casefolded_terms = NULL;
guint i;
_terminal_debug_print (TERMINAL_DEBUG_SEARCH, "GetSubsearchResultSet started\n");
app = terminal_app_get ();
casefolded_terms = normalize_casefold_and_unaccent_terms (terms);
results = g_ptr_array_new_with_free_func (g_free);
for (i = 0; previous_results[i] != NULL; i++)
{
TerminalScreen *screen;
gs_free char *cmdline = NULL, *process = NULL;
const char *cwd, *title;
screen = terminal_app_get_screen_by_uuid (app, previous_results[i]);
if (screen == NULL)
{
_terminal_debug_print (TERMINAL_DEBUG_SEARCH, "Not a screen: %s\n", previous_results[i]);
continue;
}
cwd = vte_terminal_get_current_directory_uri (VTE_TERMINAL (screen));
title = terminal_screen_get_title (screen);
terminal_screen_has_foreground_process (screen, &process, &cmdline);
if (match_terms (cwd, (const char *const *) casefolded_terms) ||
match_terms (title, (const char *const *) casefolded_terms) ||
match_terms (process, (const char *const *) casefolded_terms) ||
match_terms (cmdline, (const char *const *) casefolded_terms))
{
g_ptr_array_add (results, g_strdup (previous_results[i]));
_terminal_debug_print (TERMINAL_DEBUG_SEARCH, "Search hit: %s\n", previous_results[i]);
}
}
g_ptr_array_add (results, NULL);
terminal_search_provider2_complete_get_subsearch_result_set (skeleton,
invocation,
(const char *const *) results->pdata);
_terminal_debug_print (TERMINAL_DEBUG_SEARCH, "GetSubsearchResultSet completed\n");
return TRUE;
}
static gboolean
handle_get_result_metas_cb (TerminalSearchProvider2 *skeleton,
GDBusMethodInvocation *invocation,
const char *const *results,
gpointer user_data)
{
GVariantBuilder builder;
TerminalApp *app;
guint i;
_terminal_debug_print (TERMINAL_DEBUG_SEARCH, "GetResultMetas started\n");
app = terminal_app_get ();
g_variant_builder_init (&builder, G_VARIANT_TYPE ("aa{sv}"));
for (i = 0; results[i] != NULL; i++)
{
TerminalScreen *screen;
const char *title;
gs_free char *escaped_text = NULL;
gs_free char *text = NULL;
screen = terminal_app_get_screen_by_uuid (app, results[i]);
if (screen == NULL)
{
_terminal_debug_print (TERMINAL_DEBUG_SEARCH, "Not a screen: %s\n", results[i]);
continue;
}
title = terminal_screen_get_title (screen);
if (terminal_screen_has_foreground_process (screen, NULL, NULL)) {
VteTerminal *terminal = VTE_TERMINAL (screen);
long cursor_row;
vte_terminal_get_cursor_position (terminal, NULL, &cursor_row);
text = vte_terminal_get_text_range (terminal,
MAX(0, cursor_row - 1),
0,
cursor_row + 1,
vte_terminal_get_column_count (terminal) - 1,
NULL, NULL, NULL);
}
g_variant_builder_open (&builder, G_VARIANT_TYPE ("a{sv}"));
g_variant_builder_add (&builder, "{sv}", "id", g_variant_new_string (results[i]));
g_variant_builder_add (&builder, "{sv}", "name", g_variant_new_string (title));
if (text != NULL)
{
escaped_text = g_markup_escape_text (text, -1);
g_variant_builder_add (&builder, "{sv}", "description", g_variant_new_string (escaped_text));
}
g_variant_builder_close (&builder);
_terminal_debug_print (TERMINAL_DEBUG_SEARCH, "Meta for %s: %s\n", results[i], title);
}
terminal_search_provider2_complete_get_result_metas (skeleton, invocation, g_variant_builder_end (&builder));
_terminal_debug_print (TERMINAL_DEBUG_SEARCH, "GetResultMetas completed\n");
return TRUE;
}
static gboolean
handle_activate_result_cb (TerminalSearchProvider2 *skeleton,
GDBusMethodInvocation *invocation,
const char *identifier,
const char* const *terms,
guint timestamp,
gpointer user_data)
{
GtkWidget *toplevel;
TerminalApp *app;
TerminalScreen *screen;
app = terminal_app_get ();
screen = terminal_app_get_screen_by_uuid (app, identifier);
if (screen == NULL)
goto out;
toplevel = gtk_widget_get_toplevel (GTK_WIDGET (screen));
if (!gtk_widget_is_toplevel (toplevel))
goto out;
terminal_window_switch_screen (TERMINAL_WINDOW (toplevel), screen);
gtk_window_present_with_time (GTK_WINDOW (toplevel), timestamp);
_terminal_debug_print (TERMINAL_DEBUG_SEARCH, "ActivateResult: %s\n", identifier);
out:
terminal_search_provider2_complete_activate_result (skeleton, invocation);
return TRUE;
}
static void
terminal_search_provider_init (TerminalSearchProvider *provider)
{
provider->skeleton = terminal_search_provider2_skeleton_new ();
g_signal_connect (provider->skeleton, "handle-get-initial-result-set",
G_CALLBACK (handle_get_initial_result_set_cb), provider);
g_signal_connect (provider->skeleton, "handle-get-subsearch-result-set",
G_CALLBACK (handle_get_subsearch_result_set_cb), provider);
g_signal_connect (provider->skeleton, "handle-get-result-metas",
G_CALLBACK (handle_get_result_metas_cb), provider);
g_signal_connect (provider->skeleton, "handle-activate-result",
G_CALLBACK (handle_activate_result_cb), provider);
}
static void
terminal_search_provider_dispose (GObject *object)
{
TerminalSearchProvider *provider = TERMINAL_SEARCH_PROVIDER (object);
g_clear_object (&provider->skeleton);
G_OBJECT_CLASS (terminal_search_provider_parent_class)->dispose (object);
}
static void
terminal_search_provider_class_init (TerminalSearchProviderClass *klass)
{
GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
gobject_class->dispose = terminal_search_provider_dispose;
}
TerminalSearchProvider *
terminal_search_provider_new (void)
{
return g_object_new (TERMINAL_TYPE_SEARCH_PROVIDER, NULL);
}
gboolean
terminal_search_provider_dbus_register (TerminalSearchProvider *provider,
GDBusConnection *connection,
const char *object_path,
GError **error)
{
return g_dbus_interface_skeleton_export (G_DBUS_INTERFACE_SKELETON (provider->skeleton),
connection,
object_path,
error);
}
void
terminal_search_provider_dbus_unregister (TerminalSearchProvider *provider,
GDBusConnection *connection,
const char *object_path)
{
if (g_dbus_interface_skeleton_has_connection (G_DBUS_INTERFACE_SKELETON (provider->skeleton), connection))
g_dbus_interface_skeleton_unexport_from_connection (G_DBUS_INTERFACE_SKELETON (provider->skeleton),
connection);
}