mirror of
https://github.com/oxen-io/lokinet.git
synced 2024-11-11 07:10:36 +00:00
273270916e
This commit reflects changes to clang-format rules. Unfortunately, these rule changes create a massive change to the codebase, which causes an apparent rewrite of git history. Git blame's --ignore-rev flag can be used to ignore this commit when attempting to `git blame` some code.
79 lines
1.7 KiB
C++
79 lines
1.7 KiB
C++
#ifndef LLARP_FS_HPP
|
|
#define LLARP_FS_HPP
|
|
|
|
#include <functional>
|
|
|
|
#if defined(WIN32) || defined(_WIN32)
|
|
#define PATH_SEP "\\"
|
|
#else
|
|
#define PATH_SEP "/"
|
|
#endif
|
|
|
|
#include <ghc/filesystem.hpp>
|
|
namespace fs = ghc::filesystem;
|
|
|
|
#ifndef _MSC_VER
|
|
#include <dirent.h>
|
|
#endif
|
|
|
|
#include <nonstd/optional.hpp>
|
|
|
|
namespace llarp
|
|
{
|
|
namespace util
|
|
{
|
|
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>
|
|
nonstd::optional<T>
|
|
OpenFileStream(fs::path pathname, std::ios::openmode mode)
|
|
{
|
|
if (EnsurePrivateFile(pathname))
|
|
return {};
|
|
|
|
std::string f = pathname.string();
|
|
return T{pathname, mode};
|
|
}
|
|
|
|
using PathVisitor = std::function<bool(const fs::path&)>;
|
|
using PathIter = std::function<void(const fs::path&, PathVisitor)>;
|
|
|
|
static PathIter IterDir = [](const fs::path& path, PathVisitor visit) {
|
|
#ifdef _MSC_VER
|
|
for (auto& p : fs::directory_iterator(path))
|
|
{
|
|
if (!visit(p.path()))
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
#else
|
|
DIR* d = opendir(path.string().c_str());
|
|
if (d == nullptr)
|
|
return;
|
|
struct dirent* ent = nullptr;
|
|
do
|
|
{
|
|
ent = readdir(d);
|
|
if (!ent)
|
|
break;
|
|
if (ent->d_name[0] == '.')
|
|
continue;
|
|
fs::path p = path / fs::path(ent->d_name);
|
|
if (!visit(p))
|
|
break;
|
|
} while (ent);
|
|
closedir(d);
|
|
#endif
|
|
};
|
|
} // namespace util
|
|
} // namespace llarp
|
|
#endif // end LLARP_FS_HPP
|