lokinet/llarp/util/fs.hpp

#pragma once

#include <functional>
#include <set>

#ifdef USE_GHC_FILESYSTEM
#include <ghc/filesystem.hpp>
namespace fs = ghc::filesystem;
#else
#include <filesystem>
namespace fs
{
  using namespace std::filesystem;
  using ifstream = std::ifstream;
  using ofstream = std::ofstream;
  using fstream = std::fstream;
}  // namespace fs

#endif

#ifndef _MSC_VER
#include <dirent.h>
#endif

#include <optional>

namespace llarp
{
  namespace util
  {
    struct FileHash
    {
      size_t
      operator()(const fs::path& f) const
      {
        std::hash<std::string> h;
        return h(f.string());
      }
    };

    using error_code_t = std::error_code;

    /// Ensure that a file exists and has correct permissions
    /// return any error code or success
    error_code_t
    EnsurePrivateFile(fs::path pathname);

    /// open a stream to a file and ensure it exists before open
    /// sets any permissions on creation
    template <typename T>
    std::optional<T>
    OpenFileStream(fs::path pathname, std::ios::openmode mode)
    {
      if (EnsurePrivateFile(pathname))
        return {};
      return std::make_optional<T>(pathname, mode);
    }

    template <typename PathVisitor>
    static void
    IterDir(const fs::path& path, PathVisitor visit)
    {
      DIR* d = opendir(path.string().c_str());
      if (d == nullptr)
        return;
      struct dirent* ent = nullptr;
      std::set<fs::path> entries;
      do
      {
        ent = readdir(d);
        if (not ent)
          break;
        if (ent->d_name[0] == '.')
          continue;
        entries.emplace(path / fs::path{ent->d_name});
      } while (ent);
      closedir(d);

      for (const auto& p : entries)
      {
        if (not visit(p))
          return;
      }
    };
  }  // namespace util
}  // namespace llarp