gtk2/gtk/gtksearchenginebeagle.c

413 lines
12 KiB
C
Raw Normal View History

/*
* Copyright (C) 2005 Novell, 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
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
*
* Author: Anders Carlsson <andersca@imendio.com>
*
* Based on nautilus-search-engine-beagle.c
*/
#include "config.h"
#include <gmodule.h>
#include "gtksearchenginebeagle.h"
#if 0
#include <beagle/beagle.h>
#endif
/* We dlopen() all the following from libbeagle at runtime */
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 _BeagleHitsSubtractedResponse BeagleHitsSubtractedResponse;
typedef struct _BeagleQueryPartProperty BeagleQueryPartProperty;
typedef struct _BeagleQueryPart BeagleQueryPart;
#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 G_CONST_RETURN char *(*beagle_hit_get_uri) (BeagleHit *hit) = NULL;
static GSList *(*beagle_hits_added_response_get_hits) (BeagleHitsAddedResponse *response) = NULL;
static GSList *(*beagle_hits_subtracted_response_get_uris) (BeagleHitsSubtractedResponse *response) = NULL;
static BeagleQuery *(*beagle_query_new) (void) = NULL;
static void (*beagle_query_add_text) (BeagleQuery *query,
const char *str) = NULL;
static void (*beagle_query_set_max_hits) (BeagleQuery *query,
gint max_hits) = 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) (const char *client_name) = 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_hits_subtracted_response_get_uris),
MAP (beagle_query_new),
MAP (beagle_query_add_text),
MAP (beagle_query_set_max_hits),
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_client_new)
#undef MAP
};
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)
beagle = g_module_open ("libbeagle.so.0", 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;
}
}
}
}
struct _GtkSearchEngineBeaglePrivate
{
BeagleClient *client;
GtkQuery *query;
BeagleQuery *current_query;
char *current_query_uri_prefix;
gboolean query_finished;
};
G_DEFINE_TYPE (GtkSearchEngineBeagle, _gtk_search_engine_beagle, GTK_TYPE_SEARCH_ENGINE);
static void
finalize (GObject *object)
{
GtkSearchEngineBeagle *beagle;
beagle = GTK_SEARCH_ENGINE_BEAGLE (object);
if (beagle->priv->current_query)
{
g_object_unref (beagle->priv->current_query);
beagle->priv->current_query = NULL;
g_free (beagle->priv->current_query_uri_prefix);
beagle->priv->current_query_uri_prefix = NULL;
}
if (beagle->priv->query)
{
g_object_unref (beagle->priv->query);
beagle->priv->query = NULL;
}
if (beagle->priv->client)
{
g_object_unref (beagle->priv->client);
beagle->priv->client = NULL;
}
G_OBJECT_CLASS (_gtk_search_engine_beagle_parent_class)->finalize (object);
}
static void
beagle_hits_added (BeagleQuery *query,
BeagleHitsAddedResponse *response,
GtkSearchEngineBeagle *engine)
{
GSList *hits, *list;
GList *hit_uris;
const gchar *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->priv->current_query_uri_prefix &&
!g_str_has_prefix (uri, engine->priv->current_query_uri_prefix))
continue;
hit_uris = g_list_prepend (hit_uris, (char *)uri);
}
_gtk_search_engine_hits_added (GTK_SEARCH_ENGINE (engine), hit_uris);
g_list_free (hit_uris);
}
static void
beagle_hits_subtracted (BeagleQuery *query,
BeagleHitsSubtractedResponse *response,
GtkSearchEngineBeagle *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);
}
_gtk_search_engine_hits_subtracted (GTK_SEARCH_ENGINE (engine), hit_uris);
g_list_free (hit_uris);
}
static void
beagle_finished (BeagleQuery *query,
BeagleFinishedResponse *response,
GtkSearchEngineBeagle *engine)
{
/* For some reason we keep getting finished events,
* only emit finished once */
if (engine->priv->query_finished)
return;
engine->priv->query_finished = TRUE;
_gtk_search_engine_finished (GTK_SEARCH_ENGINE (engine));
}
static void
beagle_error (BeagleQuery *query,
GError *error,
GtkSearchEngineBeagle *engine)
{
_gtk_search_engine_error (GTK_SEARCH_ENGINE (engine), error->message);
}
static void
gtk_search_engine_beagle_start (GtkSearchEngine *engine)
{
GtkSearchEngineBeagle *beagle;
GError *error;
gchar *text;
gchar *query;
error = NULL;
beagle = GTK_SEARCH_ENGINE_BEAGLE (engine);
g_return_if_fail (beagle->priv->query != NULL);
if (beagle->priv->current_query)
return;
beagle->priv->query_finished = FALSE;
beagle->priv->current_query = beagle_query_new ();
g_signal_connect (beagle->priv->current_query,
"hits-added", G_CALLBACK (beagle_hits_added), engine);
g_signal_connect (beagle->priv->current_query,
"hits-subtracted", G_CALLBACK (beagle_hits_subtracted), engine);
g_signal_connect (beagle->priv->current_query,
"finished", G_CALLBACK (beagle_finished), engine);
g_signal_connect (beagle->priv->current_query,
"error", G_CALLBACK (beagle_error), engine);
/* We only want files */
text = _gtk_query_get_text (beagle->priv->query);
query = g_strconcat (text, " type:File", NULL);
beagle_query_set_max_hits (beagle->priv->current_query, 1000);
beagle_query_add_text (beagle->priv->current_query, query);
beagle->priv->current_query_uri_prefix = _gtk_query_get_location (beagle->priv->query);
if (!beagle_client_send_request_async (beagle->priv->client,
BEAGLE_REQUEST (beagle->priv->current_query), &error))
{
_gtk_search_engine_error (engine, error->message);
g_error_free (error);
}
/* These must live during the lifetime of the query */
g_free (text);
g_free (query);
}
static void
gtk_search_engine_beagle_stop (GtkSearchEngine *engine)
{
GtkSearchEngineBeagle *beagle;
beagle = GTK_SEARCH_ENGINE_BEAGLE (engine);
if (beagle->priv->current_query)
{
g_object_unref (beagle->priv->current_query);
beagle->priv->current_query = NULL;
g_free (beagle->priv->current_query_uri_prefix);
beagle->priv->current_query_uri_prefix = NULL;
}
}
static gboolean
gtk_search_engine_beagle_is_indexed (GtkSearchEngine *engine)
{
return TRUE;
}
static void
gtk_search_engine_beagle_set_query (GtkSearchEngine *engine,
GtkQuery *query)
{
GtkSearchEngineBeagle *beagle;
beagle = GTK_SEARCH_ENGINE_BEAGLE (engine);
if (query)
g_object_ref (query);
if (beagle->priv->query)
g_object_unref (beagle->priv->query);
beagle->priv->query = query;
}
static void
_gtk_search_engine_beagle_class_init (GtkSearchEngineBeagleClass *class)
{
GObjectClass *gobject_class;
GtkSearchEngineClass *engine_class;
gobject_class = G_OBJECT_CLASS (class);
gobject_class->finalize = finalize;
engine_class = GTK_SEARCH_ENGINE_CLASS (class);
engine_class->set_query = gtk_search_engine_beagle_set_query;
engine_class->start = gtk_search_engine_beagle_start;
engine_class->stop = gtk_search_engine_beagle_stop;
engine_class->is_indexed = gtk_search_engine_beagle_is_indexed;
g_type_class_add_private (gobject_class, sizeof (GtkSearchEngineBeaglePrivate));
}
static void
_gtk_search_engine_beagle_init (GtkSearchEngineBeagle *engine)
{
engine->priv = G_TYPE_INSTANCE_GET_PRIVATE (engine, GTK_TYPE_SEARCH_ENGINE_BEAGLE, GtkSearchEngineBeaglePrivate);
}
GtkSearchEngine *
_gtk_search_engine_beagle_new (void)
{
GtkSearchEngineBeagle *engine;
BeagleClient *client;
open_libbeagle ();
if (!beagle_util_daemon_is_running)
return NULL;
/* check whether daemon is running as beagle_client_new
* doesn't fail when a stale socket file exists */
if (!beagle_util_daemon_is_running ())
return NULL;
client = beagle_client_new (NULL);
if (client == NULL)
return NULL;
engine = g_object_new (GTK_TYPE_SEARCH_ENGINE_BEAGLE, NULL);
engine->priv->client = client;
return GTK_SEARCH_ENGINE (engine);
}