unity/crnlib/crn_find_files.cpp

// File: crn_win32_find_files.cpp
// See Copyright Notice and license at the end of inc/crnlib.h
#include "crn_core.h"
#include "crn_find_files.h"
#include "crn_file_utils.h"
#include "crn_strutils.h"

#ifdef CRNLIB_USE_WIN32_API
#include "crn_winhdr.h"

#elif defined(__GNUC__)
#include <fnmatch.h>
#include <dirent.h>
#endif

namespace crnlib {
#ifdef CRNLIB_USE_WIN32_API
bool find_files::find(const char* pBasepath, const char* pFilespec, uint flags) {
  m_last_error = S_OK;
  m_files.resize(0);

  return find_internal(pBasepath, "", pFilespec, flags, 0);
}

bool find_files::find(const char* pSpec, uint flags) {
  dynamic_string find_name(pSpec);

  if (!file_utils::full_path(find_name))
    return false;

  dynamic_string find_pathname, find_filename;
  if (!file_utils::split_path(find_name.get_ptr(), find_pathname, find_filename))
    return false;

  return find(find_pathname.get_ptr(), find_filename.get_ptr(), flags);
}

bool find_files::find_internal(const char* pBasepath, const char* pRelpath, const char* pFilespec, uint flags, int level) {
  WIN32_FIND_DATAA find_data;

  dynamic_string filename;

  dynamic_string_array child_paths;
  if (flags & cFlagRecursive) {
    if (strlen(pRelpath))
      file_utils::combine_path(filename, pBasepath, pRelpath, "*");
    else
      file_utils::combine_path(filename, pBasepath, "*");

    HANDLE handle = FindFirstFileA(filename.get_ptr(), &find_data);
    if (handle == INVALID_HANDLE_VALUE) {
      HRESULT hres = GetLastError();
      if ((level == 0) && (hres != NO_ERROR) && (hres != ERROR_FILE_NOT_FOUND)) {
        m_last_error = hres;
        return false;
      }
    } else {
      do {
        const bool is_dir = (find_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0;

        bool skip = !is_dir;
        if (is_dir)
          skip = (strcmp(find_data.cFileName, ".") == 0) || (strcmp(find_data.cFileName, "..") == 0);

        if (find_data.dwFileAttributes & (FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_TEMPORARY))
          skip = true;

        if (find_data.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN) {
          if ((flags & cFlagAllowHidden) == 0)
            skip = true;
        }

        if (!skip) {
          dynamic_string child_path(find_data.cFileName);
          if ((!child_path.count_char('?')) && (!child_path.count_char('*')))
            child_paths.push_back(child_path);
        }

      } while (FindNextFileA(handle, &find_data) != 0);

      HRESULT hres = GetLastError();

      FindClose(handle);
      handle = INVALID_HANDLE_VALUE;

      if (hres != ERROR_NO_MORE_FILES) {
        m_last_error = hres;
        return false;
      }
    }
  }

  if (strlen(pRelpath))
    file_utils::combine_path(filename, pBasepath, pRelpath, pFilespec);
  else
    file_utils::combine_path(filename, pBasepath, pFilespec);

  HANDLE handle = FindFirstFileA(filename.get_ptr(), &find_data);
  if (handle == INVALID_HANDLE_VALUE) {
    HRESULT hres = GetLastError();
    if ((level == 0) && (hres != NO_ERROR) && (hres != ERROR_FILE_NOT_FOUND)) {
      m_last_error = hres;
      return false;
    }
  } else {
    do {
      const bool is_dir = (find_data.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) != 0;

      bool skip = false;
      if (is_dir)
        skip = (strcmp(find_data.cFileName, ".") == 0) || (strcmp(find_data.cFileName, "..") == 0);

      if (find_data.dwFileAttributes & (FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_TEMPORARY))
        skip = true;

      if (find_data.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN) {
        if ((flags & cFlagAllowHidden) == 0)
          skip = true;
      }

      if (!skip) {
        if (((is_dir) && (flags & cFlagAllowDirs)) || ((!is_dir) && (flags & cFlagAllowFiles))) {
          m_files.resize(m_files.size() + 1);
          file_desc& file = m_files.back();
          file.m_is_dir = is_dir;
          file.m_base = pBasepath;
          file.m_name = find_data.cFileName;
          file.m_rel = pRelpath;
          if (strlen(pRelpath))
            file_utils::combine_path(file.m_fullname, pBasepath, pRelpath, find_data.cFileName);
          else
            file_utils::combine_path(file.m_fullname, pBasepath, find_data.cFileName);
        }
      }

    } while (FindNextFileA(handle, &find_data) != 0);

    HRESULT hres = GetLastError();

    FindClose(handle);

    if (hres != ERROR_NO_MORE_FILES) {
      m_last_error = hres;
      return false;
    }
  }

  for (uint i = 0; i < child_paths.size(); i++) {
    dynamic_string child_path;
    if (strlen(pRelpath))
      file_utils::combine_path(child_path, pRelpath, child_paths[i].get_ptr());
    else
      child_path = child_paths[i];

    if (!find_internal(pBasepath, child_path.get_ptr(), pFilespec, flags, level + 1))
      return false;
  }

  return true;
}
#elif defined(__GNUC__)
bool find_files::find(const char* pBasepath, const char* pFilespec, uint flags) {
  m_files.resize(0);
  return find_internal(pBasepath, "", pFilespec, flags, 0);
}

bool find_files::find(const char* pSpec, uint flags) {
  dynamic_string find_name(pSpec);

  if (!file_utils::full_path(find_name))
    return false;

  dynamic_string find_pathname, find_filename;
  if (!file_utils::split_path(find_name.get_ptr(), find_pathname, find_filename))
    return false;

  return find(find_pathname.get_ptr(), find_filename.get_ptr(), flags);
}

bool find_files::find_internal(const char* pBasepath, const char* pRelpath, const char* pFilespec, uint flags, int level) {
  dynamic_string pathname;
  if (strlen(pRelpath))
    file_utils::combine_path(pathname, pBasepath, pRelpath);
  else
    pathname = pBasepath;

  if (!pathname.is_empty()) {
    char c = pathname.back();
    if (c != '/')
      pathname += "/";
  }

  DIR* dp = opendir(pathname.get_ptr());

  if (!dp)
    return level ? true : false;

  dynamic_string_array paths;

  for (;;) {
    struct dirent* ep = readdir(dp);
    if (!ep)
      break;
    if ((strcmp(ep->d_name, ".") == 0) || (strcmp(ep->d_name, "..") == 0))
      continue;

    const bool is_directory = (ep->d_type & DT_DIR) != 0;
    const bool is_file = (ep->d_type & DT_REG) != 0;

    dynamic_string filename(ep->d_name);

    if (is_directory) {
      if (flags & cFlagRecursive) {
        paths.push_back(filename);
      }
    }

    if (((is_file) && (flags & cFlagAllowFiles)) || ((is_directory) && (flags & cFlagAllowDirs))) {
      if (0 == fnmatch(pFilespec, filename.get_ptr(), 0)) {
        m_files.resize(m_files.size() + 1);
        file_desc& file = m_files.back();
        file.m_is_dir = is_directory;
        file.m_base = pBasepath;
        file.m_rel = pRelpath;
        file.m_name = filename;
        file.m_fullname = pathname + filename;
      }
    }
  }

  closedir(dp);
  dp = NULL;

  if (flags & cFlagRecursive) {
    for (uint i = 0; i < paths.size(); i++) {
      dynamic_string childpath;
      if (strlen(pRelpath))
        file_utils::combine_path(childpath, pRelpath, paths[i].get_ptr());
      else
        childpath = paths[i];

      if (!find_internal(pBasepath, childpath.get_ptr(), pFilespec, flags, level + 1))
        return false;
    }
  }

  return true;
}
#else
#error Unimplemented
#endif

}  // namespace crnlib