/*
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#include "hh_readdir.h"

#include <caml/alloc.h>
#include <caml/callback.h>
#include <caml/fail.h>
#include <caml/memory.h>
#include <caml/unixsupport.h>
#include <dirent.h>
#include <errno.h>

#ifdef __APPLE__
/* On glibc platforms (including mingw32) this is conditionally defined
 * as appropriate; MacOS doesn't have glibc, but does have d_type, so just
 * define a macro with the same name.
 *
 * If d_type is not available, 0 is returned.
 */
#define _DIRENT_HAVE_D_TYPE
#endif

CAMLprim value hh_readdir(value path) {
  CAMLparam1(path);
  CAMLlocal3(head, tail, list);

  if (Tag_val(path) != String_tag) {
    caml_invalid_argument("Path must be a string");
  }

  DIR* dir = opendir(String_val(path));
  if (!dir) {
    /* from caml/unixsupport.h */
    unix_error(errno, "opendir", path);
  }

  list = Val_emptylist;
  struct dirent* ent;
  while (1) {
    errno = 0;
    ent = readdir(dir);
    if (!ent) {
      if (errno) {
        unix_error(errno, "readdir", path);
      }
      break;
    }
    head = caml_alloc_tuple(2);
#ifdef __APPLE__
    Store_field(
        head, 0, caml_alloc_initialized_string(ent->d_namlen, ent->d_name));
#else
    Store_field(head, 0, caml_copy_string(ent->d_name));
#endif
#ifdef _DIRENT_HAVE_D_TYPE
    Store_field(head, 1, Val_int(ent->d_type));
#else
    // 0 is DT_UNKNOWN
    Store_field(head, 1, Val_int(0));
#endif

    tail = list;
    list = caml_alloc(2, 0);
    Store_field(list, 0, head);
    Store_field(list, 1, tail);
  }
  closedir(dir);
  CAMLreturn(list);
}
