diff options
Diffstat (limited to 'gtk/gtksearchenginequartz.c')
-rw-r--r-- | gtk/gtksearchenginequartz.c | 274 |
1 files changed, 274 insertions, 0 deletions
diff --git a/gtk/gtksearchenginequartz.c b/gtk/gtksearchenginequartz.c new file mode 100644 index 0000000000..5d3d5004e5 --- /dev/null +++ b/gtk/gtksearchenginequartz.c @@ -0,0 +1,274 @@ +/* + * Copyright (C) 2007 Kristian Rietveld <kris@gtk.org> + * + * 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 + * License along with this library; if not, write to the + * Free Software Foundation, Inc., 59 Temple Place - Suite 330, + */ + +#include <Cocoa/Cocoa.h> +#include <quartz/gdkquartz.h> + +#include "gtksearchenginequartz.h" + +/* This file is a mixture of an objective-C object and a GObject, + * so be careful to not confuse yourself. + */ + +#define QUARTZ_POOL_ALLOC NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init] +#define QUARTZ_POOL_RELEASE [pool release] + + +/* Definition of objective-c object */ +@interface ResultReceiver : NSObject +{ + int submitted_hits; + GtkSearchEngine *engine; +} + +- (void) setEngine:(GtkSearchEngine *)quartz_engine; + +- (void) queryUpdate:(id)sender; +- (void) queryProgress:(id)sender; +- (void) queryFinished:(id)sender; + +@end + + +/* Definition of GObject */ +struct _GtkSearchEngineQuartzPrivate +{ + GtkQuery *query; + + ResultReceiver *receiver; + NSMetadataQuery *ns_query; + + gboolean query_finished; +}; + +G_DEFINE_TYPE (GtkSearchEngineQuartz, _gtk_search_engine_quartz, GTK_TYPE_SEARCH_ENGINE); + + +/* Implementation of the objective-C object */ +@implementation ResultReceiver + +- (void) setEngine:(GtkSearchEngine *)quartz_engine +{ + g_return_if_fail (GTK_IS_SEARCH_ENGINE (quartz_engine)); + + engine = quartz_engine; + submitted_hits = 0; +} + +- (void) submitHits:(NSMetadataQuery *)ns_query +{ + int i; + GList *hits = NULL; + + /* Here we submit hits "submitted_hits" to "resultCount" */ + for (i = submitted_hits; i < [ns_query resultCount]; i++) + { + id result = [ns_query resultAtIndex:i]; + char *result_path; + + result_path = g_strdup_printf ("file://%s", [[result valueForAttribute:@"kMDItemPath"] cString]); + hits = g_list_prepend (hits, result_path); + } + + _gtk_search_engine_hits_added (engine, hits); + g_list_free (hits); + + submitted_hits = [ns_query resultCount]; + + /* The beagle backend stops at 1000 hits, so guess we do so too here. + * It works pretty snappy on my MacBook, if we get rid of this limit + * we are almost definitely going to need some code to submit hits + * in batches. + */ + if (submitted_hits > 1000) + [ns_query stopQuery]; +} + +- (void) queryUpdate:(id)sender +{ + NSMetadataQuery *ns_query = [sender object]; + + [self submitHits:ns_query]; +} + +- (void) queryProgress:(id)sender +{ + NSMetadataQuery *ns_query = [sender object]; + + [self submitHits:ns_query]; +} + +- (void) queryFinished:(id)sender +{ + NSMetadataQuery *ns_query = [sender object]; + + [self submitHits:ns_query]; + + _gtk_search_engine_finished (engine); + submitted_hits = 0; +} + +@end + +/* Implementation of the GObject */ + +static void +gtk_search_engine_quartz_finalize (GObject *object) +{ + GtkSearchEngineQuartz *quartz; + + QUARTZ_POOL_ALLOC; + + quartz = GTK_SEARCH_ENGINE_QUARTZ (object); + + [[NSNotificationCenter defaultCenter] removeObserver:quartz->priv->receiver]; + + [quartz->priv->ns_query release]; + [quartz->priv->receiver release]; + + QUARTZ_POOL_RELEASE; + + if (quartz->priv->query) + { + g_object_unref (quartz->priv->query); + quartz->priv->query = NULL; + } + + G_OBJECT_CLASS (_gtk_search_engine_quartz_parent_class)->finalize (object); +} + +static void +gtk_search_engine_quartz_start (GtkSearchEngine *engine) +{ + GtkSearchEngineQuartz *quartz; + + QUARTZ_POOL_ALLOC; + + quartz = GTK_SEARCH_ENGINE_QUARTZ (engine); + + [quartz->priv->ns_query startQuery]; + + QUARTZ_POOL_RELEASE; +} + +static void +gtk_search_engine_quartz_stop (GtkSearchEngine *engine) +{ + GtkSearchEngineQuartz *quartz; + + QUARTZ_POOL_ALLOC; + + quartz = GTK_SEARCH_ENGINE_QUARTZ (engine); + + [quartz->priv->ns_query stopQuery]; + + QUARTZ_POOL_RELEASE; +} + +static gboolean +gtk_search_engine_quartz_is_indexed (GtkSearchEngine *engine) +{ + return TRUE; +} + +static void +gtk_search_engine_quartz_set_query (GtkSearchEngine *engine, + GtkQuery *query) +{ + GtkSearchEngineQuartz *quartz; + + QUARTZ_POOL_ALLOC; + + quartz = GTK_SEARCH_ENGINE_QUARTZ (engine); + + if (query) + g_object_ref (query); + + if (quartz->priv->query) + g_object_unref (quartz->priv->query); + + quartz->priv->query = query; + + /* We create a query to look for ".*text.*" in the text contents of + * all indexed files. (Should we also search for text in file and folder + * names?). + */ + [quartz->priv->ns_query setPredicate: + [NSPredicate predicateWithFormat: + [NSString stringWithFormat:@"(kMDItemTextContent LIKE[cd] \"*%s*\")", + _gtk_query_get_text (query)]]]; + + QUARTZ_POOL_RELEASE; +} + +static void +_gtk_search_engine_quartz_class_init (GtkSearchEngineQuartzClass *class) +{ + GObjectClass *gobject_class; + GtkSearchEngineClass *engine_class; + + gobject_class = G_OBJECT_CLASS (class); + gobject_class->finalize = gtk_search_engine_quartz_finalize; + + engine_class = GTK_SEARCH_ENGINE_CLASS (class); + engine_class->set_query = gtk_search_engine_quartz_set_query; + engine_class->start = gtk_search_engine_quartz_start; + engine_class->stop = gtk_search_engine_quartz_stop; + engine_class->is_indexed = gtk_search_engine_quartz_is_indexed; + + g_type_class_add_private (gobject_class, sizeof (GtkSearchEngineQuartzPrivate)); +} + +static void +_gtk_search_engine_quartz_init (GtkSearchEngineQuartz *engine) +{ + QUARTZ_POOL_ALLOC; + + engine->priv = G_TYPE_INSTANCE_GET_PRIVATE (engine, GTK_TYPE_SEARCH_ENGINE_QUARTZ, GtkSearchEngineQuartzPrivate); + + engine->priv->ns_query = [[NSMetadataQuery alloc] init]; + engine->priv->receiver = [[ResultReceiver alloc] init]; + + [engine->priv->receiver setEngine:GTK_SEARCH_ENGINE (engine)]; + + [[NSNotificationCenter defaultCenter] addObserver:engine->priv->receiver + selector:@selector(queryUpdate:) + name:@"NSMetadataQueryDidUpdateNotification" + object:engine->priv->ns_query]; + [[NSNotificationCenter defaultCenter] addObserver:engine->priv->receiver + selector:@selector(queryFinished:) + name:@"NSMetadataQueryDidFinishGatheringNotification" + object:engine->priv->ns_query]; + + [[NSNotificationCenter defaultCenter] addObserver:engine->priv->receiver + selector:@selector(queryProgress:) + name:@"NSMetadataQueryGatheringProgressNotification" + object:engine->priv->ns_query]; + + QUARTZ_POOL_RELEASE; +} + +GtkSearchEngine * +_gtk_search_engine_quartz_new (void) +{ +#ifdef GDK_WINDOWING_QUARTZ + return g_object_new (GTK_TYPE_SEARCH_ENGINE_QUARTZ, NULL); +#else + return NULL; +#endif +} |