gtk2/gtk/gtksearchenginesimple.c

379 lines
9.5 KiB
C
Raw Normal View History

/*
* Copyright (C) 2005 Red Hat, Inc
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
2012-02-27 13:01:10 +00:00
* License along with this library. If not, see <http://www.gnu.org/licenses/>.
*
* Author: Alexander Larsson <alexl@redhat.com>
*
* Based on nautilus-search-engine-simple.c
*/
#include "config.h"
#include <gio/gio.h>
#include <gdk/gdk.h>
#include "gtksearchenginesimple.h"
#include "gtkprivate.h"
#include <string.h>
#define BATCH_SIZE 500
2015-02-28 21:13:18 +00:00
typedef struct
{
GtkSearchEngineSimple *engine;
GCancellable *cancellable;
2015-02-28 21:13:18 +00:00
GQueue *directories;
2015-02-28 21:13:18 +00:00
gint n_processed_files;
GList *hits;
2015-02-28 21:13:18 +00:00
GtkQuery *query;
gboolean recursive;
} SearchThreadData;
struct _GtkSearchEngineSimple
{
GtkSearchEngine parent;
GtkQuery *query;
2015-02-28 21:13:18 +00:00
SearchThreadData *active_search;
2015-02-28 21:13:18 +00:00
gboolean query_finished;
GtkSearchEngineSimpleIsIndexed is_indexed_callback;
gpointer is_indexed_data;
GDestroyNotify is_indexed_data_destroy;
};
struct _GtkSearchEngineSimpleClass
{
GtkSearchEngineClass parent_class;
};
G_DEFINE_TYPE (GtkSearchEngineSimple, _gtk_search_engine_simple, GTK_TYPE_SEARCH_ENGINE)
static void
gtk_search_engine_simple_dispose (GObject *object)
{
GtkSearchEngineSimple *simple = GTK_SEARCH_ENGINE_SIMPLE (object);
2015-02-28 21:13:18 +00:00
g_clear_object (&simple->query);
2015-02-28 21:13:18 +00:00
if (simple->active_search)
{
g_cancellable_cancel (simple->active_search->cancellable);
simple->active_search = NULL;
}
_gtk_search_engine_simple_set_indexed_cb (simple, NULL, NULL, NULL);
G_OBJECT_CLASS (_gtk_search_engine_simple_parent_class)->dispose (object);
}
static SearchThreadData *
search_thread_data_new (GtkSearchEngineSimple *engine,
GtkQuery *query)
{
SearchThreadData *data;
const gchar *uri;
GFile *location;
2015-02-28 21:13:18 +00:00
data = g_new0 (SearchThreadData, 1);
2015-02-28 21:13:18 +00:00
data->engine = g_object_ref (engine);
data->directories = g_queue_new ();
data->query = g_object_ref (query);
data->recursive = _gtk_search_engine_get_recursive (GTK_SEARCH_ENGINE (engine));
uri = gtk_query_get_location (query);
2015-02-28 21:13:18 +00:00
if (uri != NULL)
location = g_file_new_for_uri (uri);
else
location = g_file_new_for_path (g_get_home_dir ());
g_queue_push_tail (data->directories, location);
2015-02-28 21:13:18 +00:00
data->cancellable = g_cancellable_new ();
2015-02-28 21:13:18 +00:00
return data;
}
2015-02-28 21:13:18 +00:00
static void
search_thread_data_free (SearchThreadData *data)
{
g_queue_foreach (data->directories, (GFunc)g_object_unref, NULL);
g_queue_free (data->directories);
g_object_unref (data->cancellable);
g_object_unref (data->query);
g_object_unref (data->engine);
g_free (data);
}
static gboolean
search_thread_done_idle (gpointer user_data)
{
SearchThreadData *data;
data = user_data;
2015-02-28 21:13:18 +00:00
if (!g_cancellable_is_cancelled (data->cancellable))
_gtk_search_engine_finished (GTK_SEARCH_ENGINE (data->engine));
2015-02-28 21:13:18 +00:00
data->engine->active_search = NULL;
search_thread_data_free (data);
2015-02-28 21:13:18 +00:00
return FALSE;
}
2015-02-28 21:13:18 +00:00
typedef struct
{
GList *hits;
SearchThreadData *thread_data;
} Batch;
static gboolean
search_thread_add_hits_idle (gpointer user_data)
{
Batch *batch = user_data;
if (!g_cancellable_is_cancelled (batch->thread_data->cancellable))
_gtk_search_engine_hits_added (GTK_SEARCH_ENGINE (batch->thread_data->engine), batch->hits);
g_list_free_full (batch->hits, (GDestroyNotify)_gtk_search_hit_free);
g_free (batch);
2015-02-28 21:13:18 +00:00
return FALSE;
}
static void
send_batch (SearchThreadData *data)
{
Batch *batch;
2015-02-28 21:13:18 +00:00
data->n_processed_files = 0;
2015-02-28 21:13:18 +00:00
if (data->hits)
{
guint id;
batch = g_new (Batch, 1);
batch->hits = data->hits;
batch->thread_data = data;
2015-02-28 21:13:18 +00:00
id = gdk_threads_add_idle (search_thread_add_hits_idle, batch);
g_source_set_name_by_id (id, "[gtk+] search_thread_add_hits_idle");
}
data->hits = NULL;
}
static gboolean
is_indexed (GtkSearchEngineSimple *engine,
GFile *location)
{
if (engine->is_indexed_callback)
{
if (engine->is_indexed_callback (location, engine->is_indexed_data))
{
gchar *uri = g_file_get_uri (location);
g_debug ("Simple search engine: Skipping indexed location: %s\n", uri);
g_free (uri);
return TRUE;
}
}
return FALSE;
}
static void
visit_directory (GFile *dir, SearchThreadData *data)
{
GFileEnumerator *enumerator;
GFileInfo *info;
GFile *child;
const gchar *display_name;
enumerator = g_file_enumerate_children (dir,
G_FILE_ATTRIBUTE_STANDARD_NAME ","
G_FILE_ATTRIBUTE_STANDARD_DISPLAY_NAME ","
G_FILE_ATTRIBUTE_STANDARD_TYPE ","
G_FILE_ATTRIBUTE_STANDARD_IS_HIDDEN ","
G_FILE_ATTRIBUTE_STANDARD_IS_BACKUP ","
G_FILE_ATTRIBUTE_STANDARD_SIZE ","
G_FILE_ATTRIBUTE_STANDARD_CONTENT_TYPE ","
G_FILE_ATTRIBUTE_STANDARD_TARGET_URI ","
G_FILE_ATTRIBUTE_TIME_MODIFIED,
G_FILE_QUERY_INFO_NOFOLLOW_SYMLINKS,
data->cancellable, NULL);
if (enumerator == NULL)
return;
while (g_file_enumerator_iterate (enumerator, &info, &child, data->cancellable, NULL))
{
if (info == NULL)
break;
display_name = g_file_info_get_display_name (info);
if (display_name == NULL)
continue;
if (g_file_info_get_is_hidden (info))
continue;
if (gtk_query_matches_string (data->query, display_name))
{
GtkSearchHit *hit;
hit = g_new (GtkSearchHit, 1);
hit->uri = g_file_get_uri (child);
hit->info = g_object_ref (info);
data->hits = g_list_prepend (data->hits, hit);
}
data->n_processed_files++;
if (data->n_processed_files > BATCH_SIZE)
send_batch (data);
if (data->recursive &&
g_file_info_get_file_type (info) == G_FILE_TYPE_DIRECTORY &&
!is_indexed (data->engine, child))
g_queue_push_tail (data->directories, g_object_ref (child));
}
g_object_unref (enumerator);
}
2015-02-28 21:13:18 +00:00
static gpointer
search_thread_func (gpointer user_data)
{
SearchThreadData *data;
GFile *dir;
guint id;
2015-02-28 21:13:18 +00:00
data = user_data;
2015-02-28 21:13:18 +00:00
while (!g_cancellable_is_cancelled (data->cancellable) &&
(dir = g_queue_pop_head (data->directories)) != NULL)
{
visit_directory (dir, data);
g_object_unref (dir);
}
if (!g_cancellable_is_cancelled (data->cancellable))
send_batch (data);
2015-02-28 21:13:18 +00:00
id = gdk_threads_add_idle (search_thread_done_idle, data);
g_source_set_name_by_id (id, "[gtk+] search_thread_done_idle");
2015-02-28 21:13:18 +00:00
return NULL;
}
static void
gtk_search_engine_simple_start (GtkSearchEngine *engine)
{
GtkSearchEngineSimple *simple;
SearchThreadData *data;
2015-02-28 21:13:18 +00:00
simple = GTK_SEARCH_ENGINE_SIMPLE (engine);
2015-02-28 21:13:18 +00:00
if (simple->active_search != NULL)
return;
2015-02-28 21:13:18 +00:00
if (simple->query == NULL)
return;
2015-02-28 21:13:18 +00:00
data = search_thread_data_new (simple, simple->query);
2015-02-28 21:13:18 +00:00
g_thread_unref (g_thread_new ("file-search", search_thread_func, data));
2015-02-28 21:13:18 +00:00
simple->active_search = data;
}
static void
gtk_search_engine_simple_stop (GtkSearchEngine *engine)
{
GtkSearchEngineSimple *simple;
2015-02-28 21:13:18 +00:00
simple = GTK_SEARCH_ENGINE_SIMPLE (engine);
2015-02-28 21:13:18 +00:00
if (simple->active_search != NULL)
{
g_cancellable_cancel (simple->active_search->cancellable);
simple->active_search = NULL;
}
}
static void
2015-02-28 21:13:18 +00:00
gtk_search_engine_simple_set_query (GtkSearchEngine *engine,
GtkQuery *query)
{
GtkSearchEngineSimple *simple;
2015-02-28 21:13:18 +00:00
simple = GTK_SEARCH_ENGINE_SIMPLE (engine);
2015-02-28 21:13:18 +00:00
if (query)
g_object_ref (query);
if (simple->query)
g_object_unref (simple->query);
simple->query = query;
}
static void
_gtk_search_engine_simple_class_init (GtkSearchEngineSimpleClass *class)
{
GObjectClass *gobject_class;
GtkSearchEngineClass *engine_class;
2015-02-28 21:13:18 +00:00
gobject_class = G_OBJECT_CLASS (class);
gobject_class->dispose = gtk_search_engine_simple_dispose;
2015-02-28 21:13:18 +00:00
engine_class = GTK_SEARCH_ENGINE_CLASS (class);
engine_class->set_query = gtk_search_engine_simple_set_query;
engine_class->start = gtk_search_engine_simple_start;
engine_class->stop = gtk_search_engine_simple_stop;
}
static void
_gtk_search_engine_simple_init (GtkSearchEngineSimple *engine)
{
}
GtkSearchEngine *
_gtk_search_engine_simple_new (void)
{
return g_object_new (GTK_TYPE_SEARCH_ENGINE_SIMPLE, NULL);
}
void
_gtk_search_engine_simple_set_indexed_cb (GtkSearchEngineSimple *engine,
GtkSearchEngineSimpleIsIndexed callback,
gpointer data,
GDestroyNotify destroy)
{
if (engine->is_indexed_data_destroy)
engine->is_indexed_data_destroy (engine->is_indexed_data);
engine->is_indexed_callback = callback;
engine->is_indexed_data = data;
engine->is_indexed_data_destroy = destroy;
}