1 files changed, 125 insertions, 0 deletions
diff --git a/src/Filesystem.cpp b/src/Filesystem.cpp
new file mode 100644
index 0000000..09e5146
--- /dev/null
+++ b/src/Filesystem.cpp
@@ -0,0 +1,125 @@
+#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.resize(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 (m_Files.count(relpath) > 0)
+    {
+        std::cout << "Adding file: " << path << " (" << size << " bytes) and overwriting " << relpath << " to " << this
+                  << std::endl;
+    }
+    else
+    {
+        std::cout << "Adding file: " << path << " (" << size << " bytes) as " << relpath << " to " << this << std::endl;
+    }
+
+    m_Files[relpath] = fd;
+
+    return true;
+}
+
+bool Filesystem::Scan(std::string root)
+{
+    bool retval = true;
+
+    for (const auto & entry : std::filesystem::directory_iterator(root))
+    {
+        if (AddSingleFile(entry.path(), root) == false)
+        {
+            retval = true;
+        }
+    }
+
+    return retval;
+}
+
+bool Filesystem::Scan(std::string root, std::vector<std::string> extensions, bool exclude_extensions)
+{
+    bool retval = true;
+
+    for (const auto & entry : std::filesystem::directory_iterator(root))
+    {
+        std::string ext = std::filesystem::path(entry).extension();
+        bool found_extension = false;
+
+        for (const auto & extension : extensions)
+        {
+            if (ext == "." + extension)
+            {
+                found_extension = true;
+                break;
+            }
+        }
+
+        if (found_extension == true && exclude_extensions == false)
+        {
+            if (AddSingleFile(entry.path(), root) == false)
+            {
+                retval = false;
+            }
+        }
+        if (found_extension == false && exclude_extensions == true)
+        {
+            if (AddSingleFile(entry.path(), root) == false)
+            {
+                retval = false;
+            }
+        }
+    }
+
+    return retval;
+}
+
+const std::unordered_map<std::string, struct file_data> & Filesystem::GetFiles() const
+{
+    return m_Files;
+}