Pileus Git - ~andy/gtk/blob - gtk/gtksearchenginetracker.c

   1 /*
   2  * Copyright (C) 2005 Mr Jamie McCracken
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, write to the
  16  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  17  *
  18  * Author: Jamie McCracken <jamiemcc@gnome.org>
  19  *
  20  * Based on nautilus-search-engine-tracker.c
  21  */
  22
  23 #include "config.h"
  24 #include <gmodule.h>
  25 #include "gtksearchenginetracker.h"
  26 #if 0
  27 #include <tracker.h>
  28 #endif
  29
  30 /* we dlopen() libtracker at runtime */
  31
  32 typedef struct _TrackerClient TrackerClient;
  33
  34 typedef enum
  35 {
  36   TRACKER_0_6 = 1 << 0,
  37   TRACKER_0_7 = 1 << 1
  38 } TrackerVersion;
  39
  40 typedef void (*TrackerArrayReply) (char **result, GError *error, gpointer user_data);
  41
  42 static TrackerClient * (*tracker_connect) (gboolean enable_warnings, gint timeout) = NULL;
  43 static void            (*tracker_disconnect) (TrackerClient *client) = NULL;
  44 static int             (*tracker_get_version) (TrackerClient *client, GError **error) = NULL;
  45 static void            (*tracker_cancel_last_call) (TrackerClient *client) = NULL;
  46
  47 static void (*tracker_search_metadata_by_text_async) (TrackerClient *client,
  48                                                       const char *query,
  49                                                       TrackerArrayReply callback,
  50                                                       gpointer user_data) = NULL;
  51 static void (*tracker_search_metadata_by_text_and_location_async) (TrackerClient *client,
  52                                                                    const char *query,
  53                                                                    const char *location,
  54                                                                    TrackerArrayReply callback,
  55                                                                    gpointer user_data) = NULL;
  56
  57 static struct TrackerDlMapping
  58 {
  59   const char *fn_name;
  60   gpointer *fn_ptr_ref;
  61   TrackerVersion versions;
  62 } tracker_dl_mapping[] =
  63 {
  64 #define MAP(a,v) { #a, (gpointer *)&a, v }
  65   MAP (tracker_connect, TRACKER_0_6 | TRACKER_0_7),
  66   MAP (tracker_disconnect, TRACKER_0_6 | TRACKER_0_7),
  67   MAP (tracker_get_version, TRACKER_0_6),
  68   MAP (tracker_cancel_last_call, TRACKER_0_6 | TRACKER_0_7),
  69   MAP (tracker_search_metadata_by_text_async, TRACKER_0_6 | TRACKER_0_7),
  70   MAP (tracker_search_metadata_by_text_and_location_async, TRACKER_0_6 | TRACKER_0_7),
  71 #undef MAP
  72 };
  73
  74 static TrackerVersion
  75 open_libtracker (void)
  76 {
  77   static gboolean done = FALSE;
  78   static TrackerVersion version = 0;
  79
  80   if (!done)
  81     {
  82       int i;
  83       GModule *tracker;
  84       GModuleFlags flags;
  85
  86       done = TRUE;
  87       flags = G_MODULE_BIND_LAZY | G_MODULE_BIND_LOCAL;
  88
  89       tracker = g_module_open ("libtracker-client-0.7.so.0", flags);
  90       version = TRACKER_0_7;
  91
  92       if (!tracker)
  93         {
  94           tracker = g_module_open ("libtrackerclient.so.0", flags);
  95           version = TRACKER_0_6;
  96         }
  97
  98       if (!tracker)
  99         {
 100           tracker = g_module_open ("libtracker.so.0", flags);
 101           version = TRACKER_0_6;
 102         }
 103
 104       if (!tracker)
 105         return 0;
 106
 107       for (i = 0; i < G_N_ELEMENTS (tracker_dl_mapping); i++)
 108         {
 109           if ((tracker_dl_mapping[i].versions & version) == 0)
 110             continue;
 111
 112           if (!g_module_symbol (tracker, tracker_dl_mapping[i].fn_name,
 113                                 tracker_dl_mapping[i].fn_ptr_ref))
 114             {
 115               g_warning ("Missing symbol '%s' in libtracker\n",
 116                          tracker_dl_mapping[i].fn_name);
 117               g_module_close (tracker);
 118
 119               for (i = 0; i < G_N_ELEMENTS (tracker_dl_mapping); i++)
 120                 tracker_dl_mapping[i].fn_ptr_ref = NULL;
 121
 122               return 0;
 123             }
 124         }
 125     }
 126
 127   return version;
 128 }
 129
 130 struct _GtkSearchEngineTrackerPrivate
 131 {
 132   GtkQuery      *query;
 133   TrackerClient *client;
 134   gboolean       query_pending;
 135   TrackerVersion version;
 136 };
 137
 138 G_DEFINE_TYPE (GtkSearchEngineTracker, _gtk_search_engine_tracker, GTK_TYPE_SEARCH_ENGINE);
 139
 140
 141 static void
 142 finalize (GObject *object)
 143 {
 144   GtkSearchEngineTracker *tracker;
 145
 146   tracker = GTK_SEARCH_ENGINE_TRACKER (object);
 147
 148   if (tracker->priv->query)
 149     {
 150       g_object_unref (tracker->priv->query);
 151       tracker->priv->query = NULL;
 152     }
 153
 154   tracker_disconnect (tracker->priv->client);
 155
 156   G_OBJECT_CLASS (_gtk_search_engine_tracker_parent_class)->finalize (object);
 157 }
 158
 159
 160 static void
 161 search_callback (gchar  **results,
 162                  GError  *error,
 163                  gpointer user_data)
 164 {
 165   GtkSearchEngineTracker *tracker;
 166   gchar **results_p;
 167   GList *hit_uris;
 168
 169   tracker = GTK_SEARCH_ENGINE_TRACKER (user_data);
 170   hit_uris = NULL;
 171
 172   tracker->priv->query_pending = FALSE;
 173
 174   if (error)
 175     {
 176       _gtk_search_engine_error ( GTK_SEARCH_ENGINE (tracker), error->message);
 177       g_error_free (error);
 178       return;
 179     }
 180
 181   if (!results)
 182     return;
 183
 184   for (results_p = results; *results_p; results_p++)
 185     {
 186       gchar *uri;
 187
 188       if (tracker->priv->version == TRACKER_0_6)
 189         uri = g_filename_to_uri (*results_p, NULL, NULL);
 190       else
 191         uri = *results_p;
 192
 193       if (uri)
 194         hit_uris = g_list_prepend (hit_uris, uri);
 195     }
 196
 197   _gtk_search_engine_hits_added (GTK_SEARCH_ENGINE (tracker), hit_uris);
 198   _gtk_search_engine_finished (GTK_SEARCH_ENGINE (tracker));
 199
 200   g_strfreev  (results);
 201   if (tracker->priv->version == TRACKER_0_6)
 202     g_list_foreach (hit_uris, (GFunc)g_free, NULL);
 203   g_list_free (hit_uris);
 204 }
 205
 206
 207 static void
 208 gtk_search_engine_tracker_start (GtkSearchEngine *engine)
 209 {
 210   GtkSearchEngineTracker *tracker;
 211   gchar *search_text, *location, *location_uri;
 212
 213   tracker = GTK_SEARCH_ENGINE_TRACKER (engine);
 214
 215   if (tracker->priv->query_pending)
 216     return;
 217
 218   if (tracker->priv->query == NULL)
 219     return;
 220
 221   search_text = _gtk_query_get_text (tracker->priv->query);
 222   location_uri = _gtk_query_get_location (tracker->priv->query);
 223
 224   location = NULL;
 225   if (location_uri)
 226     {
 227       if (tracker->priv->version == TRACKER_0_6)
 228         {
 229           location = g_filename_from_uri (location_uri, NULL, NULL);
 230           g_free (location_uri);
 231         }
 232       else
 233         location = location_uri;
 234     }
 235
 236   if (location)
 237     {
 238       tracker_search_metadata_by_text_and_location_async (tracker->priv->client,
 239                                                           search_text,
 240                                                           location,
 241                                                           search_callback,
 242                                                           tracker);
 243       g_free (location);
 244     }
 245   else
 246     {
 247       tracker_search_metadata_by_text_async (tracker->priv->client,
 248                                              search_text,
 249                                              search_callback,
 250                                              tracker);
 251     }
 252
 253   tracker->priv->query_pending = TRUE;
 254   g_free (search_text);
 255 }
 256
 257 static void
 258 gtk_search_engine_tracker_stop (GtkSearchEngine *engine)
 259 {
 260   GtkSearchEngineTracker *tracker;
 261
 262   tracker = GTK_SEARCH_ENGINE_TRACKER (engine);
 263
 264   if (tracker->priv->query && tracker->priv->query_pending)
 265     {
 266       tracker_cancel_last_call (tracker->priv->client);
 267       tracker->priv->query_pending = FALSE;
 268     }
 269 }
 270
 271 static gboolean
 272 gtk_search_engine_tracker_is_indexed (GtkSearchEngine *engine)
 273 {
 274   return TRUE;
 275 }
 276
 277 static void
 278 gtk_search_engine_tracker_set_query (GtkSearchEngine *engine,
 279                                      GtkQuery        *query)
 280 {
 281   GtkSearchEngineTracker *tracker;
 282
 283   tracker = GTK_SEARCH_ENGINE_TRACKER (engine);
 284
 285   if (query)
 286     g_object_ref (query);
 287
 288   if (tracker->priv->query)
 289     g_object_unref (tracker->priv->query);
 290
 291   tracker->priv->query = query;
 292 }
 293
 294 static void
 295 _gtk_search_engine_tracker_class_init (GtkSearchEngineTrackerClass *class)
 296 {
 297   GObjectClass *gobject_class;
 298   GtkSearchEngineClass *engine_class;
 299
 300   gobject_class = G_OBJECT_CLASS (class);
 301   gobject_class->finalize = finalize;
 302
 303   engine_class = GTK_SEARCH_ENGINE_CLASS (class);
 304   engine_class->set_query = gtk_search_engine_tracker_set_query;
 305   engine_class->start = gtk_search_engine_tracker_start;
 306   engine_class->stop = gtk_search_engine_tracker_stop;
 307   engine_class->is_indexed = gtk_search_engine_tracker_is_indexed;
 308
 309   g_type_class_add_private (gobject_class, sizeof (GtkSearchEngineTrackerPrivate));
 310 }
 311
 312 static void
 313 _gtk_search_engine_tracker_init (GtkSearchEngineTracker *engine)
 314 {
 315   engine->priv = G_TYPE_INSTANCE_GET_PRIVATE (engine, GTK_TYPE_SEARCH_ENGINE_TRACKER, GtkSearchEngineTrackerPrivate);
 316 }
 317
 318
 319 GtkSearchEngine *
 320 _gtk_search_engine_tracker_new (void)
 321 {
 322   GtkSearchEngineTracker *engine;
 323   TrackerClient *tracker_client;
 324   TrackerVersion version;
 325   GError *err = NULL;
 326
 327   version = open_libtracker ();
 328
 329   if (!tracker_connect)
 330     return NULL;
 331
 332   tracker_client = tracker_connect (FALSE, -1);
 333
 334   if (!tracker_client)
 335     return NULL;
 336
 337   if (version == TRACKER_0_6)
 338     {
 339       if (!tracker_get_version)
 340         return NULL;
 341
 342       tracker_get_version (tracker_client, &err);
 343
 344       if (err != NULL)
 345         {
 346           g_error_free (err);
 347           tracker_disconnect (tracker_client);
 348           return NULL;
 349         }
 350     }
 351
 352   engine = g_object_new (GTK_TYPE_SEARCH_ENGINE_TRACKER, NULL);
 353
 354   engine->priv->client = tracker_client;
 355   engine->priv->query_pending = FALSE;
 356   engine->priv->version = version;
 357
 358   return GTK_SEARCH_ENGINE (engine);
 359 }