/* GIO - GLib Input, Output and Streaming Library

 * 

 * Copyright (C) 2006-2007 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.1 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, see <http://www.gnu.org/licenses/>.

 *

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

 */

#include "config.h"

#include <glib.h>

#include <glocalfileenumerator.h>

#include <glocalfileinfo.h>

#include <glocalfile.h>

#include <gioerror.h>

#include <string.h>

#include <stdlib.h>

#include "glibintl.h"

#define CHUNK_SIZE 1000

#ifdef G_OS_WIN32

#define USE_GDIR

#endif

#ifndef USE_GDIR

#include <sys/types.h>

#include <dirent.h>

#include <errno.h>

typedef struct {

  char *name;

  long inode;

  GFileType type;

} DirEntry;

#endif

struct _GLocalFileEnumerator

{

  GFileEnumerator parent;

  GFileAttributeMatcher *matcher;

  GFileAttributeMatcher *reduced_matcher;

  char *filename;

  char *attributes;

  GFileQueryInfoFlags flags;

  gboolean got_parent_info;

  GLocalParentFileInfo parent_info;

#ifdef USE_GDIR

  GDir *dir;

#else

  DIR *dir;

  DirEntry *entries;

  int entries_pos;

  gboolean at_end;

#endif

  gboolean follow_symlinks;

};

#define g_local_file_enumerator_get_type _g_local_file_enumerator_get_type

G_DEFINE_TYPE (GLocalFileEnumerator, g_local_file_enumerator, G_TYPE_FILE_ENUMERATOR)

static GFileInfo *g_local_file_enumerator_next_file (GFileEnumerator  *enumerator,

                 GCancellable     *cancellable,

                 GError          **error);

static gboolean   g_local_file_enumerator_close     (GFileEnumerator  *enumerator,

                 GCancellable     *cancellable,

                 GError          **error);

static void

free_entries (GLocalFileEnumerator *local)

{

#ifndef USE_GDIR

  int i;

  if (local->entries != NULL)

    {

      for (i = 0; local->entries[i].name != NULL; i++)

  g_free (local->entries[i].name);

      g_free (local->entries);

    }

#endif

}

static void

g_local_file_enumerator_finalize (GObject *object)

{

  GLocalFileEnumerator *local;

  local = G_LOCAL_FILE_ENUMERATOR (object);

  if (local->got_parent_info)

    _g_local_file_info_free_parent_info (&local->parent_info);

  g_free (local->filename);

  g_file_attribute_matcher_unref (local->matcher);

  g_file_attribute_matcher_unref (local->reduced_matcher);

  if (local->dir)

    {

#ifdef USE_GDIR

      g_dir_close (local->dir);

#else

      closedir (local->dir);

#endif      

      local->dir = NULL;

    }

  free_entries (local);

  G_OBJECT_CLASS (g_local_file_enumerator_parent_class)->finalize (object);

}

static void

g_local_file_enumerator_class_init (GLocalFileEnumeratorClass *klass)

{

  GObjectClass *gobject_class = G_OBJECT_CLASS (klass);

  GFileEnumeratorClass *enumerator_class = G_FILE_ENUMERATOR_CLASS (klass);

  gobject_class->finalize = g_local_file_enumerator_finalize;

  enumerator_class->next_file = g_local_file_enumerator_next_file;

  enumerator_class->close_fn = g_local_file_enumerator_close;

}

static void

g_local_file_enumerator_init (GLocalFileEnumerator *local)

{

}

#ifdef USE_GDIR

static void

convert_file_to_io_error (GError **error,

        GError  *file_error)

{

  int new_code;

  if (file_error == NULL)

    return;

  new_code = G_IO_ERROR_FAILED;

  if (file_error->domain == G_FILE_ERROR) 

    {

      switch (file_error->code) 

        {

        case G_FILE_ERROR_NOENT:

          new_code = G_IO_ERROR_NOT_FOUND;

          break;

        case G_FILE_ERROR_ACCES:

          new_code = G_IO_ERROR_PERMISSION_DENIED;

          break;

        case G_FILE_ERROR_NOTDIR:

          new_code = G_IO_ERROR_NOT_DIRECTORY;

          break;

        case G_FILE_ERROR_MFILE:

          new_code = G_IO_ERROR_TOO_MANY_OPEN_FILES;

          break;

        default:

          break;

        }

    }

  g_set_error_literal (error, G_IO_ERROR,

                       new_code,

                       file_error->message);

}

#else

static GFileAttributeMatcher *

g_file_attribute_matcher_subtract_attributes (GFileAttributeMatcher *matcher,

                                              const char *           attributes)

{

  GFileAttributeMatcher *result, *tmp;

  tmp = g_file_attribute_matcher_new (attributes);

  result = g_file_attribute_matcher_subtract (matcher, tmp);

  g_file_attribute_matcher_unref (tmp);

  return result;

}

#endif

GFileEnumerator *

_g_local_file_enumerator_new (GLocalFile *file,

            const char           *attributes,

            GFileQueryInfoFlags   flags,

            GCancellable         *cancellable,

            GError              **error)

{

  GLocalFileEnumerator *local;

  char *filename = g_file_get_path (G_FILE (file));

#ifdef USE_GDIR

  GError *dir_error;

  GDir *dir;

  dir_error = NULL;

  dir = g_dir_open (filename, 0, error != NULL ? &dir_error : NULL);

  if (dir == NULL) 

    {

      if (error != NULL)

  {

    convert_file_to_io_error (error, dir_error);

    g_error_free (dir_error);

  }

      g_free (filename);

      return NULL;

    }

#else

  DIR *dir;

  int errsv;

  dir = opendir (filename);

  if (dir == NULL)

    {

      gchar *utf8_filename;

      errsv = errno;

      utf8_filename = g_filename_to_utf8 (filename, -1, NULL, NULL, NULL);

      g_set_error (error, G_IO_ERROR,

                   g_io_error_from_errno (errsv),

                   "Error opening directory '%s': %s",

                   utf8_filename, g_strerror (errsv));

      g_free (utf8_filename);

      g_free (filename);

      return NULL;

    }

#endif

  local = g_object_new (G_TYPE_LOCAL_FILE_ENUMERATOR,

                        "container", file,

                        NULL);

  local->dir = dir;

  local->filename = filename;

  local->matcher = g_file_attribute_matcher_new (attributes);

#ifndef USE_GDIR

  local->reduced_matcher = g_file_attribute_matcher_subtract_attributes (local->matcher,

                                                                         G_LOCAL_FILE_INFO_NOSTAT_ATTRIBUTES","

                                                                         "standard::type");

#endif

  local->flags = flags;

  return G_FILE_ENUMERATOR (local);

}

#ifndef USE_GDIR

static int

sort_by_inode (const void *_a, const void *_b)

{

  const DirEntry *a, *b;

  a = _a;

  b = _b;

  return a->inode - b->inode;

}

#ifdef HAVE_STRUCT_DIRENT_D_TYPE

static GFileType

file_type_from_dirent (char d_type)

{

  switch (d_type)

    {

    case DT_BLK:

    case DT_CHR:

    case DT_FIFO:

    case DT_SOCK:

      return G_FILE_TYPE_SPECIAL;

    case DT_DIR:

      return G_FILE_TYPE_DIRECTORY;

    case DT_LNK:

      return G_FILE_TYPE_SYMBOLIC_LINK;

    case DT_REG:

      return G_FILE_TYPE_REGULAR;

    case DT_UNKNOWN:

    default:

      return G_FILE_TYPE_UNKNOWN;

    }

}

#endif

static const char *

next_file_helper (GLocalFileEnumerator *local, GFileType *file_type)

{

  struct dirent *entry;

  const char *filename;

  int i;

  if (local->at_end)

    return NULL;

  if (local->entries == NULL ||

      (local->entries[local->entries_pos].name == NULL))

    {

      if (local->entries == NULL)

  local->entries = g_new (DirEntry, CHUNK_SIZE + 1);

      else

  {

    /* Restart by clearing old names */

    for (i = 0; local->entries[i].name != NULL; i++)

      g_free (local->entries[i].name);

  }

      for (i = 0; i < CHUNK_SIZE; i++)

  {

    entry = readdir (local->dir);

    while (entry 

     && (0 == strcmp (entry->d_name, ".") ||

         0 == strcmp (entry->d_name, "..")))

      entry = readdir (local->dir);

    if (entry)

      {

        local->entries[i].name = g_strdup (entry->d_name);

        local->entries[i].inode = entry->d_ino;

#if HAVE_STRUCT_DIRENT_D_TYPE

              local->entries[i].type = file_type_from_dirent (entry->d_type);

#else

              local->entries[i].type = G_FILE_TYPE_UNKNOWN;

#endif

      }

    else

      break;

  }

      local->entries[i].name = NULL;

      local->entries_pos = 0;

      qsort (local->entries, i, sizeof (DirEntry), sort_by_inode);

    }

  filename = local->entries[local->entries_pos].name;

  if (filename == NULL)

    local->at_end = TRUE;

  *file_type = local->entries[local->entries_pos].type;

  local->entries_pos++;

  return filename;

}

#endif

static GFileInfo *

g_local_file_enumerator_next_file (GFileEnumerator  *enumerator,

           GCancellable     *cancellable,

           GError          **error)

{

  GLocalFileEnumerator *local = G_LOCAL_FILE_ENUMERATOR (enumerator);

  const char *filename;

  char *path;

  GFileInfo *info;

  GError *my_error;

  GFileType file_type;

  if (!local->got_parent_info)

    {

      _g_local_file_info_get_parent_info (local->filename, local->matcher, &local->parent_info);

      local->got_parent_info = TRUE;

    }

 next_file:

#ifdef USE_GDIR

  filename = g_dir_read_name (local->dir);

  file_type = G_FILE_TYPE_UNKNOWN;

#else

  filename = next_file_helper (local, &file_type);

#endif

  if (filename == NULL)

    return NULL;

  my_error = NULL;

  path = g_build_filename (local->filename, filename, NULL);

  if (file_type == G_FILE_TYPE_UNKNOWN ||

      (file_type == G_FILE_TYPE_SYMBOLIC_LINK && !(local->flags & G_FILE_QUERY_INFO_NOFOLLOW_SYMLINKS)))

    {

      info = _g_local_file_info_get (filename, path,

                                     local->matcher,

                                     local->flags,

                                     &local->parent_info,

                                     &my_error); 

    }

  else

    {

      info = _g_local_file_info_get (filename, path,

                                     local->reduced_matcher,

                                     local->flags,

                                     &local->parent_info,

                                     &my_error); 

      if (info)

        {

          _g_local_file_info_get_nostat (info, filename, path, local->matcher);

          g_file_info_set_file_type (info, file_type);

          if (file_type == G_FILE_TYPE_SYMBOLIC_LINK)

            g_file_info_set_is_symlink (info, TRUE);

        }

    }

  g_free (path);

  if (info == NULL)

    {

      /* Failed to get info */

      /* If the file does not exist there might have been a race where

       * the file was removed between the readdir and the stat, so we

       * ignore the file. */

      if (g_error_matches (my_error, G_IO_ERROR, G_IO_ERROR_NOT_FOUND))

  {

    g_error_free (my_error);

    goto next_file;

  }

      else

  g_propagate_error (error, my_error);

    }

  return info;

}

static gboolean

g_local_file_enumerator_close (GFileEnumerator  *enumerator,

             GCancellable     *cancellable,

             GError          **error)

{

  GLocalFileEnumerator *local = G_LOCAL_FILE_ENUMERATOR (enumerator);

  if (local->dir)

    {

#ifdef USE_GDIR

      g_dir_close (local->dir);

#else

      closedir (local->dir);

#endif

      local->dir = NULL;

    }

  return TRUE;

}