diff options
Diffstat (limited to 'libcaja-private/caja-search-engine-simple.c')
| -rw-r--r-- | libcaja-private/caja-search-engine-simple.c | 464 |
1 files changed, 464 insertions, 0 deletions
diff --git a/libcaja-private/caja-search-engine-simple.c b/libcaja-private/caja-search-engine-simple.c new file mode 100644 index 00000000..62bffd5e --- /dev/null +++ b/libcaja-private/caja-search-engine-simple.c @@ -0,0 +1,464 @@ +/* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 8; tab-width: 8 -*- */ +/* + * Copyright (C) 2005 Red Hat, Inc + * + * Caja 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. + * + * Caja 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; see the file COPYING. If not, + * write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, + * Boston, MA 02111-1307, USA. + * + * Author: Alexander Larsson <[email protected]> + * + */ + +#include <config.h> +#include "caja-search-engine-simple.h" + +#include <string.h> +#include <glib.h> + +#include <eel/eel-gtk-macros.h> +#include <eel/eel-glib-extensions.h> +#include <gio/gio.h> + +#define BATCH_SIZE 500 + +typedef struct +{ + CajaSearchEngineSimple *engine; + GCancellable *cancellable; + + GList *mime_types; + char **words; + GList *found_list; + + GQueue *directories; /* GFiles */ + + GHashTable *visited; + + gint n_processed_files; + GList *uri_hits; +} SearchThreadData; + + +struct CajaSearchEngineSimpleDetails +{ + CajaQuery *query; + + SearchThreadData *active_search; + + gboolean query_finished; +}; + + +static void caja_search_engine_simple_class_init (CajaSearchEngineSimpleClass *class); +static void caja_search_engine_simple_init (CajaSearchEngineSimple *engine); + +G_DEFINE_TYPE (CajaSearchEngineSimple, + caja_search_engine_simple, + CAJA_TYPE_SEARCH_ENGINE); + +static CajaSearchEngineClass *parent_class = NULL; + +static void +finalize (GObject *object) +{ + CajaSearchEngineSimple *simple; + + simple = CAJA_SEARCH_ENGINE_SIMPLE (object); + + if (simple->details->query) + { + g_object_unref (simple->details->query); + simple->details->query = NULL; + } + + g_free (simple->details); + + EEL_CALL_PARENT (G_OBJECT_CLASS, finalize, (object)); +} + +static SearchThreadData * +search_thread_data_new (CajaSearchEngineSimple *engine, + CajaQuery *query) +{ + SearchThreadData *data; + char *text, *lower, *normalized, *uri; + GFile *location; + + data = g_new0 (SearchThreadData, 1); + + data->engine = engine; + data->directories = g_queue_new (); + data->visited = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, NULL); + uri = caja_query_get_location (query); + location = NULL; + if (uri != NULL) + { + location = g_file_new_for_uri (uri); + g_free (uri); + } + if (location == NULL) + { + location = g_file_new_for_path ("/"); + } + g_queue_push_tail (data->directories, location); + + text = caja_query_get_text (query); + normalized = g_utf8_normalize (text, -1, G_NORMALIZE_NFD); + lower = g_utf8_strdown (normalized, -1); + data->words = g_strsplit (lower, " ", -1); + g_free (text); + g_free (lower); + g_free (normalized); + + data->mime_types = caja_query_get_mime_types (query); + + data->cancellable = g_cancellable_new (); + + return data; +} + +static void +search_thread_data_free (SearchThreadData *data) +{ + g_queue_foreach (data->directories, + (GFunc)g_object_unref, NULL); + g_queue_free (data->directories); + g_hash_table_destroy (data->visited); + g_object_unref (data->cancellable); + g_strfreev (data->words); + eel_g_list_free_deep (data->mime_types); + eel_g_list_free_deep (data->uri_hits); + g_free (data); +} + +static gboolean +search_thread_done_idle (gpointer user_data) +{ + SearchThreadData *data; + + data = user_data; + + if (!g_cancellable_is_cancelled (data->cancellable)) + { + caja_search_engine_finished (CAJA_SEARCH_ENGINE (data->engine)); + data->engine->details->active_search = NULL; + } + + search_thread_data_free (data); + + return FALSE; +} + +typedef struct +{ + GList *uris; + SearchThreadData *thread_data; +} SearchHits; + + +static gboolean +search_thread_add_hits_idle (gpointer user_data) +{ + SearchHits *hits; + + hits = user_data; + + if (!g_cancellable_is_cancelled (hits->thread_data->cancellable)) + { + caja_search_engine_hits_added (CAJA_SEARCH_ENGINE (hits->thread_data->engine), + hits->uris); + } + + eel_g_list_free_deep (hits->uris); + g_free (hits); + + return FALSE; +} + +static void +send_batch (SearchThreadData *data) +{ + SearchHits *hits; + + data->n_processed_files = 0; + + if (data->uri_hits) + { + hits = g_new (SearchHits, 1); + hits->uris = data->uri_hits; + hits->thread_data = data; + g_idle_add (search_thread_add_hits_idle, hits); + } + data->uri_hits = NULL; +} + +#define STD_ATTRIBUTES \ + G_FILE_ATTRIBUTE_STANDARD_NAME "," \ + G_FILE_ATTRIBUTE_STANDARD_DISPLAY_NAME "," \ + G_FILE_ATTRIBUTE_STANDARD_IS_HIDDEN "," \ + G_FILE_ATTRIBUTE_STANDARD_TYPE "," \ + G_FILE_ATTRIBUTE_ID_FILE + +static void +visit_directory (GFile *dir, SearchThreadData *data) +{ + GFileEnumerator *enumerator; + GFileInfo *info; + GFile *child; + const char *mime_type, *display_name; + char *lower_name, *normalized; + gboolean hit; + int i; + GList *l; + const char *id; + gboolean visited; + + enumerator = g_file_enumerate_children (dir, + data->mime_types != NULL ? + STD_ATTRIBUTES "," + G_FILE_ATTRIBUTE_STANDARD_CONTENT_TYPE + : + STD_ATTRIBUTES + , + 0, data->cancellable, NULL); + + if (enumerator == NULL) + { + return; + } + + while ((info = g_file_enumerator_next_file (enumerator, data->cancellable, NULL)) != NULL) + { + if (g_file_info_get_is_hidden (info)) + { + goto next; + } + + display_name = g_file_info_get_display_name (info); + if (display_name == NULL) + { + goto next; + } + + normalized = g_utf8_normalize (display_name, -1, G_NORMALIZE_NFD); + lower_name = g_utf8_strdown (normalized, -1); + g_free (normalized); + + hit = TRUE; + for (i = 0; data->words[i] != NULL; i++) + { + if (strstr (lower_name, data->words[i]) == NULL) + { + hit = FALSE; + break; + } + } + g_free (lower_name); + + if (hit && data->mime_types) + { + mime_type = g_file_info_get_content_type (info); + hit = FALSE; + + for (l = data->mime_types; mime_type != NULL && l != NULL; l = l->next) + { + if (g_content_type_equals (mime_type, l->data)) + { + hit = TRUE; + break; + } + } + } + + child = g_file_get_child (dir, g_file_info_get_name (info)); + + if (hit) + { + data->uri_hits = g_list_prepend (data->uri_hits, g_file_get_uri (child)); + } + + data->n_processed_files++; + if (data->n_processed_files > BATCH_SIZE) + { + send_batch (data); + } + + if (g_file_info_get_file_type (info) == G_FILE_TYPE_DIRECTORY) + { + id = g_file_info_get_attribute_string (info, G_FILE_ATTRIBUTE_ID_FILE); + visited = FALSE; + if (id) + { + if (g_hash_table_lookup_extended (data->visited, + id, NULL, NULL)) + { + visited = TRUE; + } + else + { + g_hash_table_insert (data->visited, g_strdup (id), NULL); + } + } + + if (!visited) + { + g_queue_push_tail (data->directories, g_object_ref (child)); + } + } + + g_object_unref (child); +next: + g_object_unref (info); + } + + g_object_unref (enumerator); +} + + +static gpointer +search_thread_func (gpointer user_data) +{ + SearchThreadData *data; + GFile *dir; + GFileInfo *info; + const char *id; + + data = user_data; + + /* Insert id for toplevel directory into visited */ + dir = g_queue_peek_head (data->directories); + info = g_file_query_info (dir, G_FILE_ATTRIBUTE_ID_FILE, 0, data->cancellable, NULL); + if (info) + { + id = g_file_info_get_attribute_string (info, G_FILE_ATTRIBUTE_ID_FILE); + if (id) + { + g_hash_table_insert (data->visited, g_strdup (id), NULL); + } + g_object_unref (info); + } + + while (!g_cancellable_is_cancelled (data->cancellable) && + (dir = g_queue_pop_head (data->directories)) != NULL) + { + visit_directory (dir, data); + g_object_unref (dir); + } + send_batch (data); + + g_idle_add (search_thread_done_idle, data); + + return NULL; +} + +static void +caja_search_engine_simple_start (CajaSearchEngine *engine) +{ + CajaSearchEngineSimple *simple; + SearchThreadData *data; + + simple = CAJA_SEARCH_ENGINE_SIMPLE (engine); + + if (simple->details->active_search != NULL) + { + return; + } + + if (simple->details->query == NULL) + { + return; + } + + data = search_thread_data_new (simple, simple->details->query); + + g_thread_create (search_thread_func, data, FALSE, NULL); + + simple->details->active_search = data; +} + +static void +caja_search_engine_simple_stop (CajaSearchEngine *engine) +{ + CajaSearchEngineSimple *simple; + + simple = CAJA_SEARCH_ENGINE_SIMPLE (engine); + + if (simple->details->active_search != NULL) + { + g_cancellable_cancel (simple->details->active_search->cancellable); + simple->details->active_search = NULL; + } +} + +static gboolean +caja_search_engine_simple_is_indexed (CajaSearchEngine *engine) +{ + return FALSE; +} + +static void +caja_search_engine_simple_set_query (CajaSearchEngine *engine, CajaQuery *query) +{ + CajaSearchEngineSimple *simple; + + simple = CAJA_SEARCH_ENGINE_SIMPLE (engine); + + if (query) + { + g_object_ref (query); + } + + if (simple->details->query) + { + g_object_unref (simple->details->query); + } + + simple->details->query = query; +} + +static void +caja_search_engine_simple_class_init (CajaSearchEngineSimpleClass *class) +{ + GObjectClass *gobject_class; + CajaSearchEngineClass *engine_class; + + parent_class = g_type_class_peek_parent (class); + + gobject_class = G_OBJECT_CLASS (class); + gobject_class->finalize = finalize; + + engine_class = CAJA_SEARCH_ENGINE_CLASS (class); + engine_class->set_query = caja_search_engine_simple_set_query; + engine_class->start = caja_search_engine_simple_start; + engine_class->stop = caja_search_engine_simple_stop; + engine_class->is_indexed = caja_search_engine_simple_is_indexed; +} + +static void +caja_search_engine_simple_init (CajaSearchEngineSimple *engine) +{ + engine->details = g_new0 (CajaSearchEngineSimpleDetails, 1); +} + + +CajaSearchEngine * +caja_search_engine_simple_new (void) +{ + CajaSearchEngine *engine; + + engine = g_object_new (CAJA_TYPE_SEARCH_ENGINE_SIMPLE, NULL); + + return engine; +} |