filesystem_utils.hpp

Go to the documentation of this file.

#pragma once
 
#include <filesystem>
#include <fstream>
#include <random>
#include <sstream>
 
#include "porytiles/utilities/panic/panic.hpp"
#include "porytiles/utilities/stream_digest.hpp"
 
namespace porytiles {
 
[[nodiscard]] inline std::filesystem::path create_tmpdir()
{
    int max_tries = 1000;
    auto tmp_dir = std::filesystem::temp_directory_path();
    int i = 0;
    std::random_device random_device;
    std::mt19937 mersenne_prng(random_device());
    std::uniform_int_distribution<uint64_t> uniform_int_distribution(0);
    std::filesystem::path path;
    while (true) {
        std::stringstream string_stream;
        string_stream << std::hex << uniform_int_distribution(mersenne_prng);
        path = tmp_dir / ("porytiles_" + string_stream.str());
        if (std::filesystem::create_directory(path)) {
            break;
        }
        if (i == max_tries) {
            panic("create_tmpdir: exceeded maximum retries");
        }
        i++;
    }
    return path;
}
 
[[nodiscard]] inline bool files_are_identical(const std::filesystem::path &file_a, const std::filesystem::path &file_b)
{
    if (!std::filesystem::exists(file_b)) {
        return false;
    }
    if (std::filesystem::file_size(file_a) != std::filesystem::file_size(file_b)) {
        return false;
    }
    StreamDigest digest;
    std::ifstream stream_a{file_a, std::ios::binary};
    std::ifstream stream_b{file_b, std::ios::binary};
    if (!stream_a || !stream_b) {
        return false;
    }
    return digest.digest(stream_a) == digest.digest(stream_b);
}
 
[[nodiscard]] inline std::filesystem::path strip_all_extensions(const std::filesystem::path &path)
{
    auto dir = path.parent_path();
    auto filename = path.filename();
    while (!filename.extension().empty()) {
        filename = filename.stem();
    }
    if (dir.empty()) {
        return filename;
    }
    return dir / filename;
}
 
} // namespace porytiles