FileUtils.h

Go to the documentation of this file.

/*
 * FileUtils.h
 *
 *  Created on: 16 Nov 2017
 *      Author: Nicola Mori
 */

#ifndef FILEUTILS_H_
#define FILEUTILS_H_

#include <glob.h> // glob(), globfree()
#include <sstream>
#include <stdexcept>
#include <string.h> // memset()
#include <string>
#include <sys/stat.h>
#include <vector>

namespace EA {

class FileUtils {
public:
  template <class Container>
  static std::string FindFile(const std::string &fileName, const Container &folders, bool cwd = false);

  static std::string GetCWD();

  template <template <class, class> class Container = std::vector>
  static Container<std::string, std::allocator<std::string>> Glob(const std::string &pattern);
};

template <class Container>
std::string FileUtils::FindFile(const std::string &fileName, const Container &folders, bool cwd) {

  struct stat buffer;
  bool found = false;
  if (fileName[0] == '/') {
    // libName is an absolute path
    if (stat(fileName.c_str(), &buffer) == 0) {
      return fileName;
    }
  } else {
    for (auto &folder : folders) {
      std::string fullPath = folder + "/" + fileName;
      if (stat(fullPath.c_str(), &buffer) == 0) {
        return fullPath;
      }
    }
    if (cwd) {
      std::string fullPath = GetCWD() + "/" + fileName;
      if (stat(fullPath.c_str(), &buffer) == 0) {
        return fullPath;
      }
    }
  }

  return std::string("");
}

template <template <class, class> class Container>
Container<std::string, std::allocator<std::string>> FileUtils::Glob(const std::string &pattern) {
  const std::string routineName("FileUtils::Glob");

  // glob struct resides on the stack
  glob_t globResult;
  memset(&globResult, 0, sizeof(globResult));

  // do the glob operation
  int returnValue = glob(pattern.c_str(), GLOB_TILDE, NULL, &globResult);
  if (returnValue != 0 && returnValue != GLOB_NOMATCH) {
    globfree(&globResult);
    std::stringstream ss;
    ss << routineName << " failed with return value " << returnValue << std::endl;
    throw std::runtime_error(ss.str());
  }

  // collect all the fileNames into a std::list<std::string>
  Container<std::string, std::allocator<std::string>> fileNames;
  for (size_t i = 0; i < globResult.gl_pathc; ++i) {
    fileNames.push_back(std::string(globResult.gl_pathv[i]));
  }

  // cleanup
  globfree(&globResult);

  // done
  return fileNames;
}

} // namespace EA

#endif /* FILEUTILS_H_ */