gtksearchenginesimple.c 9.8 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/*
 * Copyright (C) 2005 Red Hat, 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
Javier Jardón's avatar
Javier Jardón committed
15
 * License along with this library. If not, see <http://www.gnu.org/licenses/>.
16 17 18 19 20 21
 *
 * Author: Alexander Larsson <alexl@redhat.com>
 *
 * Based on nautilus-search-engine-simple.c
 */

22
#include "config.h"
23

24
#include <gio/gio.h>
25

Emmanuele Bassi's avatar
Emmanuele Bassi committed
26 27
#include <gdk/gdk.h>

28
#include "gtksearchenginesimple.h"
29
#include "gtkfilesystem.h"
Matthias Clasen's avatar
Matthias Clasen committed
30
#include "gtkprivate.h"
31 32

#include <string.h>
33

34 35
#define BATCH_SIZE 500

Matthias Clasen's avatar
Matthias Clasen committed
36
typedef struct
37 38
{
  GtkSearchEngineSimple *engine;
39
  GCancellable *cancellable;
Matthias Clasen's avatar
Matthias Clasen committed
40

41
  GQueue *directories;
Matthias Clasen's avatar
Matthias Clasen committed
42

43
  gint n_processed_files;
44
  GList *hits;
Matthias Clasen's avatar
Matthias Clasen committed
45

46
  GtkQuery *query;
47
  gboolean recursive;
48 49 50
} SearchThreadData;


51
struct _GtkSearchEngineSimple
52
{
53 54
  GtkSearchEngine parent;

55
  GtkQuery *query;
Matthias Clasen's avatar
Matthias Clasen committed
56

57
  SearchThreadData *active_search;
Matthias Clasen's avatar
Matthias Clasen committed
58

59
  gboolean query_finished;
60 61 62

  GtkSearchEngineSimpleIsIndexed is_indexed_callback;
  gpointer                       is_indexed_data;
63
  GDestroyNotify                 is_indexed_data_destroy;
64 65
};

66 67 68 69
struct _GtkSearchEngineSimpleClass
{
  GtkSearchEngineClass parent_class;
};
70

71
G_DEFINE_TYPE (GtkSearchEngineSimple, _gtk_search_engine_simple, GTK_TYPE_SEARCH_ENGINE)
72 73

static void
Emmanuele Bassi's avatar
Emmanuele Bassi committed
74
gtk_search_engine_simple_dispose (GObject *object)
75
{
76
  GtkSearchEngineSimple *simple = GTK_SEARCH_ENGINE_SIMPLE (object);
Matthias Clasen's avatar
Matthias Clasen committed
77

78
  g_clear_object (&simple->query);
Matthias Clasen's avatar
Matthias Clasen committed
79

80
  if (simple->active_search)
Emmanuele Bassi's avatar
Emmanuele Bassi committed
81
    {
82 83
      g_cancellable_cancel (simple->active_search->cancellable);
      simple->active_search = NULL;
Emmanuele Bassi's avatar
Emmanuele Bassi committed
84 85
    }

86
  _gtk_search_engine_simple_set_indexed_cb (simple, NULL, NULL, NULL);
87

Emmanuele Bassi's avatar
Emmanuele Bassi committed
88
  G_OBJECT_CLASS (_gtk_search_engine_simple_parent_class)->dispose (object);
89 90
}

91 92 93 94 95 96 97 98
static void
queue_if_local (SearchThreadData *data,
                GFile            *file)
{
  if (!_gtk_file_consider_as_remote (file))
    g_queue_push_tail (data->directories, g_object_ref (file));
}

99 100 101 102 103
static SearchThreadData *
search_thread_data_new (GtkSearchEngineSimple *engine,
			GtkQuery              *query)
{
  SearchThreadData *data;
Matthias Clasen's avatar
Matthias Clasen committed
104

105
  data = g_new0 (SearchThreadData, 1);
Matthias Clasen's avatar
Matthias Clasen committed
106

107
  data->engine = g_object_ref (engine);
108 109
  data->directories = g_queue_new ();
  data->query = g_object_ref (query);
110
  data->recursive = _gtk_search_engine_get_recursive (GTK_SEARCH_ENGINE (engine));
111
  queue_if_local (data, gtk_query_get_location (query));
Matthias Clasen's avatar
Matthias Clasen committed
112

113
  data->cancellable = g_cancellable_new ();
Matthias Clasen's avatar
Matthias Clasen committed
114

115 116 117
  return data;
}

Matthias Clasen's avatar
Matthias Clasen committed
118
static void
119 120
search_thread_data_free (SearchThreadData *data)
{
121 122 123 124
  g_queue_foreach (data->directories, (GFunc)g_object_unref, NULL);
  g_queue_free (data->directories);
  g_object_unref (data->cancellable);
  g_object_unref (data->query);
125
  g_object_unref (data->engine);
126

127 128 129 130 131 132 133 134 135
  g_free (data);
}

static gboolean
search_thread_done_idle (gpointer user_data)
{
  SearchThreadData *data;

  data = user_data;
Matthias Clasen's avatar
Matthias Clasen committed
136

137
  if (!g_cancellable_is_cancelled (data->cancellable))
Emmanuele Bassi's avatar
Emmanuele Bassi committed
138
    _gtk_search_engine_finished (GTK_SEARCH_ENGINE (data->engine));
Matthias Clasen's avatar
Matthias Clasen committed
139

140
  data->engine->active_search = NULL;
141
  search_thread_data_free (data);
Matthias Clasen's avatar
Matthias Clasen committed
142

143 144 145
  return FALSE;
}

Matthias Clasen's avatar
Matthias Clasen committed
146
typedef struct
147
{
148
  GList *hits;
149
  SearchThreadData *thread_data;
150
} Batch;
151 152 153 154

static gboolean
search_thread_add_hits_idle (gpointer user_data)
{
155
  Batch *batch = user_data;
156

157 158
  if (!g_cancellable_is_cancelled (batch->thread_data->cancellable))
    _gtk_search_engine_hits_added (GTK_SEARCH_ENGINE (batch->thread_data->engine), batch->hits);
159

160 161
  g_list_free_full (batch->hits, (GDestroyNotify)_gtk_search_hit_free);
  g_free (batch);
Matthias Clasen's avatar
Matthias Clasen committed
162

163 164 165 166 167 168
  return FALSE;
}

static void
send_batch (SearchThreadData *data)
{
169
  Batch *batch;
Matthias Clasen's avatar
Matthias Clasen committed
170

171
  data->n_processed_files = 0;
Matthias Clasen's avatar
Matthias Clasen committed
172

173
  if (data->hits)
174
    {
175 176
      guint id;

177 178 179
      batch = g_new (Batch, 1);
      batch->hits = data->hits;
      batch->thread_data = data;
Matthias Clasen's avatar
Matthias Clasen committed
180

181
      id = gdk_threads_add_idle (search_thread_add_hits_idle, batch);
182
      g_source_set_name_by_id (id, "[gtk+] search_thread_add_hits_idle");
183
    }
Emmanuele Bassi's avatar
Emmanuele Bassi committed
184

185
  data->hits = NULL;
186 187
}

188 189 190 191
static gboolean
is_indexed (GtkSearchEngineSimple *engine,
            GFile                 *location)
{
192
  if (engine->is_indexed_callback)
193
    {
194
      if (engine->is_indexed_callback (location, engine->is_indexed_data))
195 196 197 198 199 200 201 202 203 204 205 206
        {
          gchar *uri = g_file_get_uri (location);
          g_debug ("Simple search engine: Skipping indexed location: %s\n", uri);
          g_free (uri);

          return TRUE;
        }
    }

  return FALSE;
}

207
static void
208
visit_directory (GFile *dir, SearchThreadData *data)
209
{
210 211 212 213 214 215 216 217 218
  GFileEnumerator *enumerator;
  GFileInfo *info;
  GFile *child;
  const gchar *display_name;

  enumerator = g_file_enumerate_children (dir,
                                          G_FILE_ATTRIBUTE_STANDARD_NAME ","
                                          G_FILE_ATTRIBUTE_STANDARD_DISPLAY_NAME ","
                                          G_FILE_ATTRIBUTE_STANDARD_TYPE ","
219 220 221 222
                                          G_FILE_ATTRIBUTE_STANDARD_IS_HIDDEN ","
                                          G_FILE_ATTRIBUTE_STANDARD_IS_BACKUP ","
                                          G_FILE_ATTRIBUTE_STANDARD_SIZE ","
                                          G_FILE_ATTRIBUTE_STANDARD_CONTENT_TYPE ","
223
                                          G_FILE_ATTRIBUTE_STANDARD_TARGET_URI ","
224
                                          G_FILE_ATTRIBUTE_TIME_MODIFIED ","
225
                                          G_FILE_ATTRIBUTE_TIME_ACCESS ","
226
                                          G_FILE_ATTRIBUTE_ACCESS_CAN_RENAME ","
227
                                          G_FILE_ATTRIBUTE_ACCESS_CAN_TRASH ","
228
                                          G_FILE_ATTRIBUTE_ACCESS_CAN_DELETE,
229 230 231 232
                                          G_FILE_QUERY_INFO_NOFOLLOW_SYMLINKS,
                                          data->cancellable, NULL);
  if (enumerator == NULL)
    return;
233

234
  while (g_file_enumerator_iterate (enumerator, &info, &child, data->cancellable, NULL))
235
    {
236 237
      if (info == NULL)
        break;
238

239 240 241
      display_name = g_file_info_get_display_name (info);
      if (display_name == NULL)
        continue;
242 243 244 245

      if (g_file_info_get_is_hidden (info))
        continue;

246
      if (gtk_query_matches_string (data->query, display_name))
247 248 249 250 251 252 253 254
        {
          GtkSearchHit *hit;

          hit = g_new (GtkSearchHit, 1);
          hit->uri = g_file_get_uri (child);
          hit->info = g_object_ref (info);
          data->hits = g_list_prepend (data->hits, hit);
        }
255

256 257 258
      data->n_processed_files++;
      if (data->n_processed_files > BATCH_SIZE)
        send_batch (data);
259

260 261 262
      if (data->recursive &&
          g_file_info_get_file_type (info) == G_FILE_TYPE_DIRECTORY &&
          !is_indexed (data->engine, child))
263
        queue_if_local (data, child);
264
    }
265

266
  g_object_unref (enumerator);
267 268
}

Matthias Clasen's avatar
Matthias Clasen committed
269
static gpointer
270 271 272
search_thread_func (gpointer user_data)
{
  SearchThreadData *data;
273 274
  GFile *dir;
  guint id;
Matthias Clasen's avatar
Matthias Clasen committed
275

276
  data = user_data;
Matthias Clasen's avatar
Matthias Clasen committed
277

278 279 280 281 282 283
  while (!g_cancellable_is_cancelled (data->cancellable) &&
         (dir = g_queue_pop_head (data->directories)) != NULL)
    {
      visit_directory (dir, data);
      g_object_unref (dir);
    }
284

285 286
  if (!g_cancellable_is_cancelled (data->cancellable))
    send_batch (data);
Matthias Clasen's avatar
Matthias Clasen committed
287

288 289
  id = gdk_threads_add_idle (search_thread_done_idle, data);
  g_source_set_name_by_id (id, "[gtk+] search_thread_done_idle");
Matthias Clasen's avatar
Matthias Clasen committed
290

291 292 293 294 295 296 297 298
  return NULL;
}

static void
gtk_search_engine_simple_start (GtkSearchEngine *engine)
{
  GtkSearchEngineSimple *simple;
  SearchThreadData *data;
Matthias Clasen's avatar
Matthias Clasen committed
299

300
  simple = GTK_SEARCH_ENGINE_SIMPLE (engine);
Matthias Clasen's avatar
Matthias Clasen committed
301

302
  if (simple->active_search != NULL)
303
    return;
Matthias Clasen's avatar
Matthias Clasen committed
304

305
  if (simple->query == NULL)
306
    return;
Matthias Clasen's avatar
Matthias Clasen committed
307

308
  data = search_thread_data_new (simple, simple->query);
Matthias Clasen's avatar
Matthias Clasen committed
309

310
  g_thread_unref (g_thread_new ("file-search", search_thread_func, data));
Matthias Clasen's avatar
Matthias Clasen committed
311

312
  simple->active_search = data;
313 314 315 316 317 318
}

static void
gtk_search_engine_simple_stop (GtkSearchEngine *engine)
{
  GtkSearchEngineSimple *simple;
Matthias Clasen's avatar
Matthias Clasen committed
319

320
  simple = GTK_SEARCH_ENGINE_SIMPLE (engine);
Matthias Clasen's avatar
Matthias Clasen committed
321

322
  if (simple->active_search != NULL)
323
    {
324 325
      g_cancellable_cancel (simple->active_search->cancellable);
      simple->active_search = NULL;
326 327 328 329
    }
}

static void
Matthias Clasen's avatar
Matthias Clasen committed
330
gtk_search_engine_simple_set_query (GtkSearchEngine *engine,
331 332 333
				    GtkQuery        *query)
{
  GtkSearchEngineSimple *simple;
Matthias Clasen's avatar
Matthias Clasen committed
334

335
  simple = GTK_SEARCH_ENGINE_SIMPLE (engine);
Matthias Clasen's avatar
Matthias Clasen committed
336

337 338 339
  if (query)
    g_object_ref (query);

340 341
  if (simple->query)
    g_object_unref (simple->query);
342

343
  simple->query = query;
344 345 346 347 348 349 350
}

static void
_gtk_search_engine_simple_class_init (GtkSearchEngineSimpleClass *class)
{
  GObjectClass *gobject_class;
  GtkSearchEngineClass *engine_class;
Matthias Clasen's avatar
Matthias Clasen committed
351

352
  gobject_class = G_OBJECT_CLASS (class);
Emmanuele Bassi's avatar
Emmanuele Bassi committed
353
  gobject_class->dispose = gtk_search_engine_simple_dispose;
Matthias Clasen's avatar
Matthias Clasen committed
354

355 356 357 358 359 360 361 362 363 364 365 366 367 368
  engine_class = GTK_SEARCH_ENGINE_CLASS (class);
  engine_class->set_query = gtk_search_engine_simple_set_query;
  engine_class->start = gtk_search_engine_simple_start;
  engine_class->stop = gtk_search_engine_simple_stop;
}

static void
_gtk_search_engine_simple_init (GtkSearchEngineSimple *engine)
{
}

GtkSearchEngine *
_gtk_search_engine_simple_new (void)
{
369
  return g_object_new (GTK_TYPE_SEARCH_ENGINE_SIMPLE, NULL);
370
}
371 372 373 374

void
_gtk_search_engine_simple_set_indexed_cb (GtkSearchEngineSimple          *engine,
                                          GtkSearchEngineSimpleIsIndexed  callback,
375 376
                                          gpointer                        data,
                                          GDestroyNotify                  destroy)
377
{
378 379
  if (engine->is_indexed_data_destroy)
    engine->is_indexed_data_destroy (engine->is_indexed_data);
380

381 382 383
  engine->is_indexed_callback = callback;
  engine->is_indexed_data = data;
  engine->is_indexed_data_destroy = destroy;
384
}