summaryrefslogtreecommitdiff
path: root/libcaja-private/caja-search-engine-tracker.c
diff options
context:
space:
mode:
Diffstat (limited to 'libcaja-private/caja-search-engine-tracker.c')
-rw-r--r--libcaja-private/caja-search-engine-tracker.c580
1 files changed, 580 insertions, 0 deletions
diff --git a/libcaja-private/caja-search-engine-tracker.c b/libcaja-private/caja-search-engine-tracker.c
new file mode 100644
index 00000000..90be40b3
--- /dev/null
+++ b/libcaja-private/caja-search-engine-tracker.c
@@ -0,0 +1,580 @@
+/* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 8; tab-width: 8 -*- */
+/*
+ * Copyright (C) 2005 Mr Jamie McCracken
+ *
+ * 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: Jamie McCracken <[email protected]>
+ *
+ */
+
+#include <config.h>
+#include "caja-search-engine-tracker.h"
+#include <eel/eel-gtk-macros.h>
+#include <eel/eel-glib-extensions.h>
+#include <gmodule.h>
+#include <string.h>
+
+
+typedef struct _TrackerClient TrackerClient;
+
+typedef enum
+{
+ TRACKER_0_6 = 1 << 0,
+ TRACKER_0_7 = 1 << 1,
+ TRACKER_0_8 = 1 << 2
+} TrackerVersion;
+
+
+/* tracker 0.6 API */
+typedef void (*TrackerArrayReply) (char **result, GError *error, gpointer user_data);
+
+static TrackerClient * (*tracker_connect) (gboolean enable_warnings,
+ gint timeout) = NULL;
+static void (*tracker_disconnect) (TrackerClient *client) = NULL;
+static void (*tracker_cancel_last_call) (TrackerClient *client) = NULL;
+static int (*tracker_get_version) (TrackerClient *client, GError **error) = NULL;
+
+
+static void (*tracker_search_metadata_by_text_async) (TrackerClient *client,
+ const char *query,
+ TrackerArrayReply callback,
+ gpointer user_data) = NULL;
+static void (*tracker_search_metadata_by_text_and_mime_async) (TrackerClient *client,
+ const char *query,
+ const char **mimes,
+ TrackerArrayReply callback,
+ gpointer user_data) = NULL;
+static void (*tracker_search_metadata_by_text_and_location_async) (TrackerClient *client,
+ const char *query,
+ const char *location,
+ TrackerArrayReply callback,
+ gpointer user_data) = NULL;
+static void (*tracker_search_metadata_by_text_and_mime_and_location_async) (TrackerClient *client,
+ const char *query,
+ const char **mimes,
+ const char *location,
+ TrackerArrayReply callback,
+ gpointer user_data) = NULL;
+
+
+/* tracker 0.8 API */
+typedef enum
+{
+ TRACKER_CLIENT_ENABLE_WARNINGS = 1 << 0
+} TrackerClientFlags;
+
+typedef void (*TrackerReplyGPtrArray) (GPtrArray *result,
+ GError *error,
+ gpointer user_data);
+
+static TrackerClient * (*tracker_client_new) (TrackerClientFlags flags,
+ gint timeout) = NULL;
+static gchar * (*tracker_sparql_escape) (const gchar *str) = NULL;
+static guint (*tracker_resources_sparql_query_async) (TrackerClient *client,
+ const gchar *query,
+ TrackerReplyGPtrArray callback,
+ gpointer user_data) = NULL;
+
+
+static struct TrackerDlMapping
+{
+ const char *fn_name;
+ gpointer *fn_ptr_ref;
+ TrackerVersion versions;
+} tracker_dl_mapping[] =
+{
+#define MAP(a,v) { #a, (gpointer *)&a, v }
+ MAP (tracker_connect, TRACKER_0_6 | TRACKER_0_7),
+ MAP (tracker_disconnect, TRACKER_0_6 | TRACKER_0_7),
+ MAP (tracker_get_version, TRACKER_0_6),
+ MAP (tracker_cancel_last_call, TRACKER_0_6 | TRACKER_0_7 | TRACKER_0_8),
+ MAP (tracker_search_metadata_by_text_async, TRACKER_0_6 | TRACKER_0_7),
+ MAP (tracker_search_metadata_by_text_and_location_async, TRACKER_0_6 | TRACKER_0_7),
+ MAP (tracker_client_new, TRACKER_0_8),
+ MAP (tracker_sparql_escape, TRACKER_0_8),
+ MAP (tracker_resources_sparql_query_async, TRACKER_0_8)
+#undef MAP
+};
+
+
+static TrackerVersion
+open_libtracker (void)
+{
+ static gboolean done = FALSE;
+ static TrackerVersion version = 0;
+ gpointer x;
+
+ if (!done)
+ {
+ int i;
+ GModule *tracker;
+ GModuleFlags flags;
+
+ done = TRUE;
+ flags = G_MODULE_BIND_LAZY | G_MODULE_BIND_LOCAL;
+
+ tracker = g_module_open ("libtracker-client-0.8.so.0", flags);
+ version = TRACKER_0_8;
+
+ if (!tracker)
+ {
+ tracker = g_module_open ("libtracker-client-0.7.so.0", flags);
+
+ if (tracker && !g_module_symbol (tracker, "tracker_resources_sparql_query_async", &x))
+ {
+ version = TRACKER_0_7;
+ }
+ }
+
+ if (!tracker)
+ {
+ tracker = g_module_open ("libtrackerclient.so.0", flags);
+ version = TRACKER_0_6;
+ }
+
+ if (!tracker)
+ {
+ tracker = g_module_open ("libtracker.so.0", flags);
+ version = TRACKER_0_6;
+ }
+
+ if (!tracker)
+ return 0;
+
+ for (i = 0; i < G_N_ELEMENTS (tracker_dl_mapping); i++)
+ {
+ if ((tracker_dl_mapping[i].versions & version) == 0)
+ continue;
+
+ if (!g_module_symbol (tracker, tracker_dl_mapping[i].fn_name,
+ tracker_dl_mapping[i].fn_ptr_ref))
+ {
+ g_warning ("Missing symbol '%s' in libtracker\n",
+ tracker_dl_mapping[i].fn_name);
+ g_module_close (tracker);
+
+ for (i = 0; i < G_N_ELEMENTS (tracker_dl_mapping); i++)
+ tracker_dl_mapping[i].fn_ptr_ref = NULL;
+
+ return 0;
+ }
+ }
+ }
+
+ return version;
+}
+
+
+struct CajaSearchEngineTrackerDetails
+{
+ CajaQuery *query;
+ TrackerClient *client;
+ gboolean query_pending;
+ TrackerVersion version;
+};
+
+
+static void caja_search_engine_tracker_class_init (CajaSearchEngineTrackerClass *class);
+static void caja_search_engine_tracker_init (CajaSearchEngineTracker *engine);
+
+G_DEFINE_TYPE (CajaSearchEngineTracker,
+ caja_search_engine_tracker,
+ CAJA_TYPE_SEARCH_ENGINE);
+
+static CajaSearchEngineClass *parent_class = NULL;
+
+static void
+finalize (GObject *object)
+{
+ CajaSearchEngineTracker *tracker;
+
+ tracker = CAJA_SEARCH_ENGINE_TRACKER (object);
+
+ if (tracker->details->query)
+ {
+ g_object_unref (tracker->details->query);
+ tracker->details->query = NULL;
+ }
+
+ if (tracker->details->version == TRACKER_0_8)
+ {
+ g_object_unref (tracker->details->client);
+ }
+ else
+ {
+ tracker_disconnect (tracker->details->client);
+ }
+
+ g_free (tracker->details);
+
+ EEL_CALL_PARENT (G_OBJECT_CLASS, finalize, (object));
+}
+
+
+/* stolen from tracker sources, tracker.c */
+static void
+sparql_append_string_literal (GString *sparql,
+ const gchar *str)
+{
+ char *s;
+
+ s = tracker_sparql_escape (str);
+
+ g_string_append_c (sparql, '"');
+ g_string_append (sparql, s);
+ g_string_append_c (sparql, '"');
+
+ g_free (s);
+}
+
+
+static void
+search_callback (gpointer results, GError *error, gpointer user_data)
+{
+ CajaSearchEngineTracker *tracker;
+ char **results_p;
+ GPtrArray *OUT_result;
+ GList *hit_uris;
+ gint i;
+ char *uri;
+
+ tracker = CAJA_SEARCH_ENGINE_TRACKER (user_data);
+ hit_uris = NULL;
+
+ tracker->details->query_pending = FALSE;
+
+ if (error)
+ {
+ caja_search_engine_error (CAJA_SEARCH_ENGINE (tracker), error->message);
+ g_error_free (error);
+ return;
+ }
+
+ if (! results)
+ {
+ return;
+ }
+
+ if (tracker->details->version == TRACKER_0_8)
+ {
+ /* new tracker 0.8 API */
+ OUT_result = (GPtrArray*) results;
+
+ for (i = 0; i < OUT_result->len; i++)
+ {
+ uri = g_strdup (((gchar **) OUT_result->pdata[i])[0]);
+ if (uri)
+ {
+ hit_uris = g_list_prepend (hit_uris, (char *)uri);
+ }
+ }
+
+ g_ptr_array_foreach (OUT_result, (GFunc) g_free, NULL);
+ g_ptr_array_free (OUT_result, TRUE);
+
+ }
+ else
+ {
+ /* old tracker 0.6 API */
+ for (results_p = results; *results_p; results_p++)
+ {
+ if (tracker->details->version == TRACKER_0_6)
+ uri = g_filename_to_uri (*results_p, NULL, NULL);
+ else
+ uri = g_strdup (*results_p);
+
+ if (uri)
+ {
+ hit_uris = g_list_prepend (hit_uris, (char *)uri);
+ }
+ }
+ g_strfreev ((gchar **)results);
+ }
+
+ caja_search_engine_hits_added (CAJA_SEARCH_ENGINE (tracker), hit_uris);
+ caja_search_engine_finished (CAJA_SEARCH_ENGINE (tracker));
+ g_list_foreach (hit_uris, (GFunc) g_free, NULL);
+ g_list_free (hit_uris);
+}
+
+
+static void
+caja_search_engine_tracker_start (CajaSearchEngine *engine)
+{
+ CajaSearchEngineTracker *tracker;
+ GList *mimetypes, *l;
+ char *search_text, *location, *location_uri;
+ char **mimes;
+ int i, mime_count;
+ GString *sparql;
+
+ tracker = CAJA_SEARCH_ENGINE_TRACKER (engine);
+
+
+ if (tracker->details->query_pending)
+ {
+ return;
+ }
+
+ if (tracker->details->query == NULL)
+ {
+ return;
+ }
+
+ search_text = caja_query_get_text (tracker->details->query);
+
+ mimetypes = caja_query_get_mime_types (tracker->details->query);
+
+ location_uri = caja_query_get_location (tracker->details->query);
+
+ if (location_uri)
+ {
+ location = (tracker->details->version == TRACKER_0_6) ?
+ g_filename_from_uri (location_uri, NULL, NULL) :
+ g_strdup (location_uri);
+ g_free (location_uri);
+ }
+ else
+ {
+ location = NULL;
+ }
+
+ mime_count = g_list_length (mimetypes);
+
+ i = 0;
+ sparql = NULL;
+
+ if (tracker->details->version == TRACKER_0_8)
+ {
+ /* new tracker 0.8 API */
+ sparql = g_string_new ("SELECT ?url WHERE { ?file a nfo:FileDataObject ; nie:url ?url; ");
+ if (mime_count > 0)
+ g_string_append (sparql, "nie:mimeType ?mime ; ");
+ g_string_append (sparql, "fts:match ");
+ sparql_append_string_literal (sparql, search_text);
+
+ if (location || mime_count > 0)
+ {
+ g_string_append (sparql, " . FILTER (");
+
+ if (location)
+ {
+ g_string_append (sparql, "fn:starts-with(?url, ");
+ sparql_append_string_literal (sparql, location);
+ g_string_append (sparql, ")");
+ }
+
+ if (mime_count > 0)
+ {
+ if (location)
+ g_string_append (sparql, " && ");
+ g_string_append (sparql, "(");
+ for (l = mimetypes; l != NULL; l = l->next)
+ {
+ if (l != mimetypes)
+ g_string_append (sparql, " || ");
+ g_string_append (sparql, "?mime = ");
+ sparql_append_string_literal (sparql, l->data);
+ }
+ g_string_append (sparql, ")");
+ }
+
+ g_string_append (sparql, ")");
+ }
+ g_string_append (sparql, " }");
+
+ tracker_resources_sparql_query_async (tracker->details->client,
+ sparql->str,
+ (TrackerReplyGPtrArray) search_callback,
+ tracker);
+ g_string_free (sparql, TRUE);
+
+ }
+ else
+ {
+ /* old tracker 0.6 API */
+ if (mime_count > 0)
+ {
+ /* convert list into array */
+ mimes = g_new (char *, (mime_count + 1));
+
+ for (l = mimetypes; l != NULL; l = l->next)
+ {
+ mimes[i] = g_strdup (l->data);
+ i++;
+ }
+
+ mimes[mime_count] = NULL;
+
+ if (location)
+ {
+ tracker_search_metadata_by_text_and_mime_and_location_async (tracker->details->client,
+ search_text, (const char **)mimes, location,
+ (TrackerArrayReply) search_callback,
+ tracker);
+ }
+ else
+ {
+ tracker_search_metadata_by_text_and_mime_async (tracker->details->client,
+ search_text, (const char**)mimes,
+ (TrackerArrayReply) search_callback,
+ tracker);
+ }
+
+ g_strfreev (mimes);
+
+ }
+ else
+ {
+ if (location)
+ {
+ tracker_search_metadata_by_text_and_location_async (tracker->details->client,
+ search_text,
+ location,
+ (TrackerArrayReply) search_callback,
+ tracker);
+ }
+ else
+ {
+ tracker_search_metadata_by_text_async (tracker->details->client,
+ search_text,
+ (TrackerArrayReply) search_callback,
+ tracker);
+ }
+ }
+ }
+
+ g_free (location);
+
+ tracker->details->query_pending = TRUE;
+ g_free (search_text);
+ eel_g_list_free_deep (mimetypes);
+}
+
+static void
+caja_search_engine_tracker_stop (CajaSearchEngine *engine)
+{
+ CajaSearchEngineTracker *tracker;
+
+ tracker = CAJA_SEARCH_ENGINE_TRACKER (engine);
+
+ if (tracker->details->query && tracker->details->query_pending)
+ {
+ tracker_cancel_last_call (tracker->details->client);
+ tracker->details->query_pending = FALSE;
+ }
+}
+
+static gboolean
+caja_search_engine_tracker_is_indexed (CajaSearchEngine *engine)
+{
+ return TRUE;
+}
+
+static void
+caja_search_engine_tracker_set_query (CajaSearchEngine *engine, CajaQuery *query)
+{
+ CajaSearchEngineTracker *tracker;
+
+ tracker = CAJA_SEARCH_ENGINE_TRACKER (engine);
+
+ if (query)
+ {
+ g_object_ref (query);
+ }
+
+ if (tracker->details->query)
+ {
+ g_object_unref (tracker->details->query);
+ }
+
+ tracker->details->query = query;
+}
+
+static void
+caja_search_engine_tracker_class_init (CajaSearchEngineTrackerClass *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_tracker_set_query;
+ engine_class->start = caja_search_engine_tracker_start;
+ engine_class->stop = caja_search_engine_tracker_stop;
+ engine_class->is_indexed = caja_search_engine_tracker_is_indexed;
+}
+
+static void
+caja_search_engine_tracker_init (CajaSearchEngineTracker *engine)
+{
+ engine->details = g_new0 (CajaSearchEngineTrackerDetails, 1);
+}
+
+
+CajaSearchEngine *
+caja_search_engine_tracker_new (void)
+{
+ CajaSearchEngineTracker *engine;
+ TrackerClient *tracker_client;
+ TrackerVersion version;
+
+ version = open_libtracker ();
+
+ if (version == TRACKER_0_8)
+ {
+ tracker_client = tracker_client_new (TRACKER_CLIENT_ENABLE_WARNINGS, G_MAXINT);
+ }
+ else
+ {
+ if (! tracker_connect)
+ return NULL;
+
+ tracker_client = tracker_connect (FALSE, -1);
+ }
+
+ if (!tracker_client)
+ {
+ return NULL;
+ }
+
+ if (version == TRACKER_0_6)
+ {
+ GError *err = NULL;
+
+ tracker_get_version (tracker_client, &err);
+
+ if (err != NULL)
+ {
+ g_error_free (err);
+ tracker_disconnect (tracker_client);
+ return NULL;
+ }
+ }
+
+ engine = g_object_new (CAJA_TYPE_SEARCH_ENGINE_TRACKER, NULL);
+
+ engine->details->client = tracker_client;
+ engine->details->query_pending = FALSE;
+ engine->details->version = version;
+
+ return CAJA_SEARCH_ENGINE (engine);
+}