diff options
Diffstat (limited to 'src/Filesystem.cpp')
-rw-r--r-- | src/Filesystem.cpp | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/src/Filesystem.cpp b/src/Filesystem.cpp new file mode 100644 index 0000000..18b8725 --- /dev/null +++ b/src/Filesystem.cpp @@ -0,0 +1,61 @@ +#include "Filesystem.hpp" + +#include <cstring> +#include <filesystem> +#include <fstream> +#include <iostream> + +static std::string make_path_relative(std::string &path, std::string &root) { + return std::filesystem::relative(path, root); +} + +bool Filesystem::AddSingleFile(std::string path, std::string root) { + std::ifstream ifs(path, std::ios::binary | std::ios::ate); + + if (!ifs) { + return false; + } + + auto end = ifs.tellg(); + if (end <= 0) { + return false; + } + if (!ifs.seekg(0, std::ios::beg)) { + return false; + } + + auto size = std::size_t(end - ifs.tellg()); + + if (size == 0) { + return false; + } + + struct file_data fd = {}; + try { + fd.data.reserve(size); + } catch (const std::exception &e) { + return false; + } + + if (!ifs.read((char *)fd.data.data(), fd.data.size())) { + return false; + } + + std::string relpath = make_path_relative(path, root); + if (files.count(relpath) > 0) { + std::cout << "Adding file: " << path << " and overwriting " << relpath + << std::endl; + } else { + std::cout << "Adding file: " << path << " as " << relpath << std::endl; + } + files[relpath] = fd; + + return true; +} + +bool Filesystem::Scan(std::string root) { + for (const auto &entry : std::filesystem::directory_iterator(root)) { + AddSingleFile(entry.path(), root); + } + return true; +} |