1 files changed, 443 insertions, 0 deletions
diff --git a/libcaja-private/caja-search-engine-beagle.c b/libcaja-private/caja-search-engine-beagle.c
new file mode 100644
index 00000000..de591977
--- /dev/null
+++ b/libcaja-private/caja-search-engine-beagle.c
@@ -0,0 +1,443 @@
+/* -*- Mode: C; indent-tabs-mode: t; c-basic-offset: 8; tab-width: 8 -*- */
+/*
+ * Copyright (C) 2005 Novell, 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: Anders Carlsson <[email protected]>
+ *
+ */
+
+#include <config.h>
+#include "caja-search-engine-beagle.h"
+
+#include <eel/eel-gtk-macros.h>
+#include <eel/eel-glib-extensions.h>
+#include <gmodule.h>
+
+typedef struct _BeagleHit BeagleHit;
+typedef struct _BeagleQuery BeagleQuery;
+typedef struct _BeagleClient BeagleClient;
+typedef struct _BeagleRequest BeagleRequest;
+typedef struct _BeagleFinishedResponse BeagleFinishedResponse;
+typedef struct _BeagleHitsAddedResponse BeagleHitsAddedResponse;
+typedef struct _BeagleQueryPartProperty BeagleQueryPartProperty;
+typedef struct _BeagleQueryPart BeagleQueryPart;
+typedef struct _BeagleHitsSubtractedResponse BeagleHitsSubtractedResponse;
+
+struct CajaSearchEngineBeagleDetails
+{
+ BeagleClient *client;
+ CajaQuery *query;
+
+ BeagleQuery *current_query;
+ char *current_query_uri_prefix;
+ gboolean query_finished;
+};
+
+/* We dlopen() all the following from libbeagle at runtime */
+#define BEAGLE_HIT(x) ((BeagleHit *)(x))
+#define BEAGLE_REQUEST(obj) (G_TYPE_CHECK_INSTANCE_CAST ((obj), beagle_request_get_type(), BeagleRequest))
+#define BEAGLE_QUERY_PART(obj) (G_TYPE_CHECK_INSTANCE_CAST ((obj), beagle_query_part_get_type(), BeagleQueryPart))
+
+typedef enum
+{
+ BEAGLE_QUERY_PART_LOGIC_REQUIRED = 1,
+ BEAGLE_QUERY_PART_LOGIC_PROHIBITED = 2
+} BeagleQueryPartLogic;
+
+typedef enum
+{
+ BEAGLE_PROPERTY_TYPE_UNKNOWN = 0,
+ BEAGLE_PROPERTY_TYPE_TEXT = 1,
+ BEAGLE_PROPERTY_TYPE_KEYWORD = 2,
+ BEAGLE_PROPERTY_TYPE_DATE = 3,
+ BEAGLE_PROPERTY_TYPE_LAST = 4
+} BeaglePropertyType;
+
+/* *static* wrapper function pointers */
+static gboolean (*beagle_client_send_request_async) (BeagleClient *client,
+ BeagleRequest *request,
+ GError **err) = NULL;
+static const char* (*beagle_hit_get_uri)(BeagleHit* hit) = NULL;
+static GSList *(*beagle_hits_added_response_get_hits) (BeagleHitsAddedResponse *response) = NULL;
+static BeagleQuery *(*beagle_query_new) (void) = NULL;
+static void (*beagle_query_add_text) (BeagleQuery *query,
+ const char *str) = NULL;
+static BeagleQueryPartProperty *(*beagle_query_part_property_new) (void) = NULL;
+static void (*beagle_query_part_set_logic) (BeagleQueryPart *part,
+ BeagleQueryPartLogic logic) = NULL;
+static void (*beagle_query_part_property_set_key) (BeagleQueryPartProperty *part,
+ const char *key) = NULL;
+static void (*beagle_query_part_property_set_value) (BeagleQueryPartProperty *part,
+ const char * value) = NULL;
+static void (*beagle_query_part_property_set_property_type) (BeagleQueryPartProperty *part,
+ BeaglePropertyType prop_type) = NULL;
+static void (*beagle_query_add_part) (BeagleQuery *query,
+ BeagleQueryPart *part) = NULL;
+static GType (*beagle_request_get_type) (void) = NULL;
+static GType (*beagle_query_part_get_type) (void) = NULL;
+static gboolean (*beagle_util_daemon_is_running) (void) = NULL;
+static BeagleClient *(*beagle_client_new_real) (const char *client_name) = NULL;
+static void (*beagle_query_set_max_hits) (BeagleQuery *query,
+ int max_hits) = NULL;
+static GSList *(*beagle_hits_subtracted_response_get_uris) (BeagleHitsSubtractedResponse *response) = NULL;
+
+static struct BeagleDlMapping
+{
+ const char *fn_name;
+ gpointer *fn_ptr_ref;
+} beagle_dl_mapping[] =
+{
+#define MAP(a) { #a, (gpointer *)&a }
+ MAP (beagle_client_send_request_async),
+ MAP (beagle_hit_get_uri),
+ MAP (beagle_hits_added_response_get_hits),
+ MAP (beagle_query_new),
+ MAP (beagle_query_add_text),
+ MAP (beagle_query_part_property_new),
+ MAP (beagle_query_part_set_logic),
+ MAP (beagle_query_part_property_set_key),
+ MAP (beagle_query_part_property_set_value),
+ MAP (beagle_query_part_property_set_property_type),
+ MAP (beagle_query_add_part),
+ MAP (beagle_request_get_type),
+ MAP (beagle_query_part_get_type),
+ MAP (beagle_util_daemon_is_running),
+ MAP (beagle_query_set_max_hits),
+ MAP (beagle_hits_subtracted_response_get_uris),
+#undef MAP
+ { "beagle_client_new", (gpointer *)&beagle_client_new_real },
+};
+
+static void
+open_libbeagle (void)
+{
+ static gboolean done = FALSE;
+
+ if (!done)
+ {
+ int i;
+ GModule *beagle;
+
+ done = TRUE;
+
+ beagle = g_module_open ("libbeagle.so.1", G_MODULE_BIND_LAZY | G_MODULE_BIND_LOCAL);
+ if (!beagle)
+ return;
+
+ for (i = 0; i < G_N_ELEMENTS (beagle_dl_mapping); i++)
+ {
+ if (!g_module_symbol (beagle, beagle_dl_mapping[i].fn_name,
+ beagle_dl_mapping[i].fn_ptr_ref))
+ {
+ g_warning ("Missing symbol '%s' in libbeagle\n",
+ beagle_dl_mapping[i].fn_name);
+ g_module_close (beagle);
+
+ for (i = 0; i < G_N_ELEMENTS (beagle_dl_mapping); i++)
+ beagle_dl_mapping[i].fn_ptr_ref = NULL;
+
+ return;
+ }
+ }
+ }
+}
+
+static BeagleClient *
+beagle_client_new (const char *client_name)
+{
+ if (beagle_client_new_real)
+ return beagle_client_new_real (client_name);
+
+ return NULL;
+}
+
+static void caja_search_engine_beagle_class_init (CajaSearchEngineBeagleClass *class);
+static void caja_search_engine_beagle_init (CajaSearchEngineBeagle *engine);
+
+G_DEFINE_TYPE (CajaSearchEngineBeagle,
+ caja_search_engine_beagle,
+ CAJA_TYPE_SEARCH_ENGINE);
+
+static CajaSearchEngineClass *parent_class = NULL;
+
+static void
+finalize (GObject *object)
+{
+ CajaSearchEngineBeagle *beagle;
+
+ beagle = CAJA_SEARCH_ENGINE_BEAGLE (object);
+
+ if (beagle->details->current_query)
+ {
+ g_object_unref (beagle->details->current_query);
+ beagle->details->current_query = NULL;
+ g_free (beagle->details->current_query_uri_prefix);
+ beagle->details->current_query_uri_prefix = NULL;
+ }
+
+ if (beagle->details->query)
+ {
+ g_object_unref (beagle->details->query);
+ beagle->details->query = NULL;
+ }
+
+ if (beagle->details->client)
+ {
+ g_object_unref (beagle->details->client);
+ beagle->details->client = NULL;
+ }
+
+ g_free (beagle->details);
+
+ EEL_CALL_PARENT (G_OBJECT_CLASS, finalize, (object));
+}
+
+static void
+beagle_hits_added (BeagleQuery *query,
+ BeagleHitsAddedResponse *response,
+ CajaSearchEngineBeagle *engine)
+{
+ GSList *hits, *list;
+ GList *hit_uris;
+ const char *uri;
+
+ hit_uris = NULL;
+
+ hits = beagle_hits_added_response_get_hits (response);
+
+ for (list = hits; list != NULL; list = list->next)
+ {
+ BeagleHit *hit = BEAGLE_HIT (list->data);
+
+ uri = beagle_hit_get_uri (hit);
+
+ if (engine->details->current_query_uri_prefix &&
+ !g_str_has_prefix (uri, engine->details->current_query_uri_prefix))
+ {
+ continue;
+ }
+
+ hit_uris = g_list_prepend (hit_uris, (char *)uri);
+ }
+
+ caja_search_engine_hits_added (CAJA_SEARCH_ENGINE (engine), hit_uris);
+ g_list_free (hit_uris);
+}
+
+static void
+beagle_hits_subtracted (BeagleQuery *query,
+ BeagleHitsSubtractedResponse *response,
+ CajaSearchEngineBeagle *engine)
+{
+ GSList *uris, *list;
+ GList *hit_uris;
+
+ hit_uris = NULL;
+
+ uris = beagle_hits_subtracted_response_get_uris (response);
+
+ for (list = uris; list != NULL; list = list->next)
+ {
+ hit_uris = g_list_prepend (hit_uris, (char *)list->data);
+ }
+
+ caja_search_engine_hits_subtracted (CAJA_SEARCH_ENGINE (engine), hit_uris);
+ g_list_free (hit_uris);
+}
+
+static void
+beagle_finished (BeagleQuery *query,
+ BeagleFinishedResponse *response,
+ CajaSearchEngineBeagle *engine)
+{
+ /* For some reason we keep getting finished events,
+ * only emit finished once */
+ if (engine->details->query_finished)
+ {
+ return;
+ }
+
+ engine->details->query_finished = TRUE;
+ caja_search_engine_finished (CAJA_SEARCH_ENGINE (engine));
+}
+
+static void
+beagle_error (BeagleQuery *query,
+ GError *error,
+ CajaSearchEngineBeagle *engine)
+{
+ caja_search_engine_error (CAJA_SEARCH_ENGINE (engine), error->message);
+}
+
+static void
+caja_search_engine_beagle_start (CajaSearchEngine *engine)
+{
+ CajaSearchEngineBeagle *beagle;
+ GError *error;
+ GList *mimetypes, *l;
+ char *text, *mimetype;
+
+ error = NULL;
+ beagle = CAJA_SEARCH_ENGINE_BEAGLE (engine);
+
+ if (beagle->details->current_query)
+ {
+ return;
+ }
+
+ beagle->details->query_finished = FALSE;
+ beagle->details->current_query = beagle_query_new ();
+ g_signal_connect (beagle->details->current_query,
+ "hits-added", G_CALLBACK (beagle_hits_added), engine);
+ g_signal_connect (beagle->details->current_query,
+ "hits-subtracted", G_CALLBACK (beagle_hits_subtracted), engine);
+ g_signal_connect (beagle->details->current_query,
+ "finished", G_CALLBACK (beagle_finished), engine);
+ g_signal_connect (beagle->details->current_query,
+ "error", G_CALLBACK (beagle_error), engine);
+
+ /* We only want files */
+ beagle_query_add_text (beagle->details->current_query," type:File");
+
+ beagle_query_set_max_hits (beagle->details->current_query,
+ 1000);
+
+ text = caja_query_get_text (beagle->details->query);
+ beagle_query_add_text (beagle->details->current_query,
+ text);
+
+ mimetypes = caja_query_get_mime_types (beagle->details->query);
+ for (l = mimetypes; l != NULL; l = l->next)
+ {
+ char* temp;
+ mimetype = l->data;
+ temp = g_strconcat (" mimetype:", mimetype, NULL);
+ beagle_query_add_text (beagle->details->current_query,temp);
+ g_free (temp);
+ }
+
+ beagle->details->current_query_uri_prefix = caja_query_get_location (beagle->details->query);
+
+ if (!beagle_client_send_request_async (beagle->details->client,
+ BEAGLE_REQUEST (beagle->details->current_query), &error))
+ {
+ caja_search_engine_error (engine, error->message);
+ g_error_free (error);
+ }
+
+ /* These must live during the lifetime of the query */
+ g_free (text);
+ eel_g_list_free_deep (mimetypes);
+}
+
+static void
+caja_search_engine_beagle_stop (CajaSearchEngine *engine)
+{
+ CajaSearchEngineBeagle *beagle;
+
+ beagle = CAJA_SEARCH_ENGINE_BEAGLE (engine);
+
+ if (beagle->details->current_query)
+ {
+ g_object_unref (beagle->details->current_query);
+ beagle->details->current_query = NULL;
+ g_free (beagle->details->current_query_uri_prefix);
+ beagle->details->current_query_uri_prefix = NULL;
+ }
+}
+
+static gboolean
+caja_search_engine_beagle_is_indexed (CajaSearchEngine *engine)
+{
+ return TRUE;
+}
+
+static void
+caja_search_engine_beagle_set_query (CajaSearchEngine *engine, CajaQuery *query)
+{
+ CajaSearchEngineBeagle *beagle;
+
+ beagle = CAJA_SEARCH_ENGINE_BEAGLE (engine);
+
+ if (query)
+ {
+ g_object_ref (query);
+ }
+
+ if (beagle->details->query)
+ {
+ g_object_unref (beagle->details->query);
+ }
+
+ beagle->details->query = query;
+}
+
+static void
+caja_search_engine_beagle_class_init (CajaSearchEngineBeagleClass *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_beagle_set_query;
+ engine_class->start = caja_search_engine_beagle_start;
+ engine_class->stop = caja_search_engine_beagle_stop;
+ engine_class->is_indexed = caja_search_engine_beagle_is_indexed;
+}
+
+static void
+caja_search_engine_beagle_init (CajaSearchEngineBeagle *engine)
+{
+ engine->details = g_new0 (CajaSearchEngineBeagleDetails, 1);
+}
+
+
+CajaSearchEngine *
+caja_search_engine_beagle_new (void)
+{
+ CajaSearchEngineBeagle *engine;
+ BeagleClient *client;
+
+ open_libbeagle ();
+
+ if (beagle_util_daemon_is_running == NULL ||
+ !beagle_util_daemon_is_running ())
+ {
+ /* check whether daemon is running as beagle_client_new
+ * doesn't fail when a stale socket file exists */
+ return NULL;
+ }
+
+ client = beagle_client_new (NULL);
+
+ if (client == NULL)
+ {
+ return NULL;
+ }
+
+ engine = g_object_new (CAJA_TYPE_SEARCH_ENGINE_BEAGLE, NULL);
+
+ engine->details->client = client;
+
+ return CAJA_SEARCH_ENGINE (engine);
+}