/*

 * 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

 * License along with this library; if not, write to the

 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,

 *

 * Author: Alexander Larsson <alexl@redhat.com>

 *

 * Based on nautilus-search-engine-simple.c

 */

#include "config.h"

/* these must be defined even when HAVE_GNU_FTW is not defined

 * because (really) old versions of GNU libc have ftw.h but do

 * export ftw() and friends only if _XOPEN_SOURCE and _GNU_SOURCE

 * are defined. see bug #444097.

 */

#define _XOPEN_SOURCE 500

#define _GNU_SOURCE 

#ifdef HAVE_FTW_H

#include <ftw.h>

#endif

#include "gtksearchenginesimple.h"

#include "gtkprivate.h"

#include <string.h>

#define BATCH_SIZE 500

typedef struct 

{

  GtkSearchEngineSimple *engine;

  gchar *path;

  gchar **words;

  GList *found_list;

  gint n_processed_files;

  GList *uri_hits;

  /* accessed on both threads: */

  volatile gboolean cancelled;

} SearchThreadData;

struct _GtkSearchEngineSimplePrivate 

{

  GtkQuery *query;

  SearchThreadData *active_search;

  gboolean query_finished;

};

G_DEFINE_TYPE (GtkSearchEngineSimple, _gtk_search_engine_simple, GTK_TYPE_SEARCH_ENGINE);

static void

gtk_search_engine_simple_dispose (GObject *object)

{

  GtkSearchEngineSimple *simple;

  GtkSearchEngineSimplePrivate *priv;

  simple = GTK_SEARCH_ENGINE_SIMPLE (object);

  priv = simple->priv;

  if (priv->query) 

    {

      g_object_unref (priv->query);

      priv->query = NULL;

    }

  if (priv->active_search)

    {

      priv->active_search->cancelled = TRUE;

      priv->active_search = NULL;

    }

  G_OBJECT_CLASS (_gtk_search_engine_simple_parent_class)->dispose (object);

}

static SearchThreadData *

search_thread_data_new (GtkSearchEngineSimple *engine,

			GtkQuery              *query)

{

  SearchThreadData *data;

  char *text, *lower, *uri;

  data = g_new0 (SearchThreadData, 1);

  data->engine = g_object_ref (engine);

  uri = _gtk_query_get_location (query);

  if (uri != NULL) 

    {

      data->path = g_filename_from_uri (uri, NULL, NULL);

      g_free (uri);

    }

  if (data->path == NULL)

    data->path = g_strdup (g_get_home_dir ());

  text = _gtk_query_get_text (query);

  lower = g_ascii_strdown (text, -1);

  data->words = g_strsplit (lower, " ", -1);

  g_free (text);

  g_free (lower);

  return data;

}

static void 

search_thread_data_free (SearchThreadData *data)

{

  g_object_unref (data->engine);

  g_free (data->path);

  g_strfreev (data->words);

  g_free (data);

}

static gboolean

search_thread_done_idle (gpointer user_data)

{

  SearchThreadData *data;

  data = user_data;

  if (!data->cancelled)

    _gtk_search_engine_finished (GTK_SEARCH_ENGINE (data->engine));

  data->engine->priv->active_search = NULL;

  search_thread_data_free (data);

  return FALSE;

}

typedef struct 

{

  GList *uris;

  SearchThreadData *thread_data;

} SearchHits;

static gboolean

search_thread_add_hits_idle (gpointer user_data)

{

  SearchHits *hits;

  hits = user_data;

  if (!hits->thread_data->cancelled) 

    {

      _gtk_search_engine_hits_added (GTK_SEARCH_ENGINE (hits->thread_data->engine),

				    hits->uris);

    }

  g_list_foreach (hits->uris, (GFunc)g_free, NULL);

  g_list_free (hits->uris);

  g_free (hits);

  return FALSE;

}

static void

send_batch (SearchThreadData *data)

{

  SearchHits *hits;

  data->n_processed_files = 0;

  if (data->uri_hits) 

    {

      hits = g_new (SearchHits, 1);

      hits->uris = data->uri_hits;

      hits->thread_data = data;

      gdk_threads_add_idle (search_thread_add_hits_idle, hits);

    }

  data->uri_hits = NULL;

}

static GStaticPrivate search_thread_data = G_STATIC_PRIVATE_INIT;

#ifdef HAVE_FTW_H

static int

search_visit_func (const char        *fpath,

		   const struct stat *sb,

		   int                typeflag,

		   struct FTW        *ftwbuf)

{

  SearchThreadData *data;

  gint i;

  const gchar *name; 

  gchar *lower_name;

  gchar *uri;

  gboolean hit;

  gboolean is_hidden;

  data = (SearchThreadData*)g_static_private_get (&search_thread_data);

  if (data->cancelled)

#ifdef HAVE_GNU_FTW

    return FTW_STOP;

#else

    return 1;

#endif /* HAVE_GNU_FTW */

  name = strrchr (fpath, '/');

  if (name)

    name++;

  else

    name = fpath;

  is_hidden = *name == '.';

  hit = FALSE;

  if (!is_hidden) 

    {

      lower_name = g_ascii_strdown (name, -1);

      hit = TRUE;

      for (i = 0; data->words[i] != NULL; i++) 

	{

	  if (strstr (lower_name, data->words[i]) == NULL) 

	    {

	      hit = FALSE;

	      break;

	    }

	}

      g_free (lower_name);

    }

  if (hit) 

    {

      uri = g_filename_to_uri (fpath, NULL, NULL);

      data->uri_hits = g_list_prepend (data->uri_hits, uri);

    }

  data->n_processed_files++;

  if (data->n_processed_files > BATCH_SIZE)

    send_batch (data);

#ifdef HAVE_GNU_FTW

  if (is_hidden)

    return FTW_SKIP_SUBTREE;

  else

    return FTW_CONTINUE;

#else

  return 0;

#endif /* HAVE_GNU_FTW */

}

#endif /* HAVE_FTW_H */

static gpointer 

search_thread_func (gpointer user_data)

{

#ifdef HAVE_FTW_H

  SearchThreadData *data;

  data = user_data;

  g_static_private_set (&search_thread_data, data, NULL);

  nftw (data->path, search_visit_func, 20,

#ifdef HAVE_GNU_FTW

        FTW_ACTIONRETVAL |

#endif

        FTW_PHYS);

  send_batch (data);

  gdk_threads_add_idle (search_thread_done_idle, data);

#endif /* HAVE_FTW_H */

  return NULL;

}

static void

gtk_search_engine_simple_start (GtkSearchEngine *engine)

{

  GtkSearchEngineSimple *simple;

  SearchThreadData *data;

  simple = GTK_SEARCH_ENGINE_SIMPLE (engine);

  if (simple->priv->active_search != NULL)

    return;

  if (simple->priv->query == NULL)

    return;

  data = search_thread_data_new (simple, simple->priv->query);

  g_thread_create (search_thread_func, data, FALSE, NULL);

  simple->priv->active_search = data;

}

static void

gtk_search_engine_simple_stop (GtkSearchEngine *engine)

{

  GtkSearchEngineSimple *simple;

  simple = GTK_SEARCH_ENGINE_SIMPLE (engine);

  if (simple->priv->active_search != NULL) 

    {

      simple->priv->active_search->cancelled = TRUE;

      simple->priv->active_search = NULL;

    }

}

static gboolean

gtk_search_engine_simple_is_indexed (GtkSearchEngine *engine)

{

  return FALSE;

}

static void

gtk_search_engine_simple_set_query (GtkSearchEngine *engine, 

				    GtkQuery        *query)

{

  GtkSearchEngineSimple *simple;

  simple = GTK_SEARCH_ENGINE_SIMPLE (engine);

  if (query)

    g_object_ref (query);

  if (simple->priv->query) 

    g_object_unref (simple->priv->query);

  simple->priv->query = query;

}

static void

_gtk_search_engine_simple_class_init (GtkSearchEngineSimpleClass *class)

{

  GObjectClass *gobject_class;

  GtkSearchEngineClass *engine_class;

  gobject_class = G_OBJECT_CLASS (class);

  gobject_class->dispose = gtk_search_engine_simple_dispose;

  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;

  engine_class->is_indexed = gtk_search_engine_simple_is_indexed;

  g_type_class_add_private (gobject_class, sizeof (GtkSearchEngineSimplePrivate));

}

static void

_gtk_search_engine_simple_init (GtkSearchEngineSimple *engine)

{

  engine->priv = G_TYPE_INSTANCE_GET_PRIVATE (engine, GTK_TYPE_SEARCH_ENGINE_SIMPLE, GtkSearchEngineSimplePrivate);

}

GtkSearchEngine *

_gtk_search_engine_simple_new (void)

{

#ifdef HAVE_FTW_H

  return g_object_new (GTK_TYPE_SEARCH_ENGINE_SIMPLE, NULL);

#else

  return NULL;

#endif

}