From e11fa34c097e8884b708b981d965508b0dc1cfd5 Mon Sep 17 00:00:00 2001
From: Guillaume Jacquemin <w.jcm59@gmail.com>
Date: Thu, 11 Jul 2024 19:48:02 +0200
Subject: [PATCH] Managers: add a VFS to help with #36.

---
 src/CMakeLists.txt                   |  2 +
 src/Managers/Vfs/VfsDirectory.h      | 88 ++++++++++++++++++++++++++++
 src/Managers/Vfs/VirtualFileSystem.h | 47 +++++++++++++++
 3 files changed, 137 insertions(+)
 create mode 100644 src/Managers/Vfs/VfsDirectory.h
 create mode 100644 src/Managers/Vfs/VirtualFileSystem.h

diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 89f4ac5..2523730 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -199,6 +199,8 @@ add_executable(MassBuilderSaveTool
     Managers/MassManager.cpp
     Managers/ProfileManager.h
     Managers/ProfileManager.cpp
+    Managers/Vfs/VirtualFileSystem.h
+    Managers/Vfs/VfsDirectory.h
     Maps/ArmourSlots.hpp
     Maps/BulletLauncherAttachmentStyles.hpp
     Maps/BulletLauncherSockets.hpp
diff --git a/src/Managers/Vfs/VfsDirectory.h b/src/Managers/Vfs/VfsDirectory.h
new file mode 100644
index 0000000..e0c203f
--- /dev/null
+++ b/src/Managers/Vfs/VfsDirectory.h
@@ -0,0 +1,88 @@
+#pragma once
+
+// MassBuilderSaveTool
+// Copyright (C) 2021-2024 Guillaume Jacquemin
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+#include <Corrade/Containers/Array.h>
+#include <Corrade/Containers/String.h>
+#include <Corrade/Containers/StringView.h>
+
+using namespace Corrade;
+
+namespace mbst { namespace Managers { namespace Vfs {
+
+CORRADE_HAS_TYPE(HasFilename, decltype(T::filename));
+
+template<typename FileType>
+class VfsDirectory {
+    public:
+        static_assert(HasFilename<FileType>::value && (std::is_same<decltype(FileType::filename), Containers::String>::value || std::is_same<decltype(FileType::filename), Containers::StringView>::value),
+                      "FileType has no string-like member named filename.");
+
+        using DirType = VfsDirectory<FileType>;
+
+        explicit VfsDirectory(Containers::StringView name): _name{name} {
+            // ctor
+        }
+
+        VfsDirectory(const VfsDirectory<FileType>& other) = delete;
+        VfsDirectory& operator=(const VfsDirectory<FileType>& other) = delete;
+
+        VfsDirectory(VfsDirectory<FileType>&& other) = default;
+        VfsDirectory& operator=(VfsDirectory<FileType>&& other) = default;
+
+        auto name() const -> Containers::StringView {
+            return _name;
+        }
+
+        auto subdirs() const -> Containers::ArrayView<const DirType> {
+            return _subdirs;
+        }
+
+        auto files() const -> Containers::ArrayView<const FileType*> {
+            return _files;
+        }
+
+    protected:
+        void buildNestedPath(VfsDirectory<FileType>& root, Containers::ArrayView<const Containers::MutableStringView> components,
+                             FileType& file)
+        {
+            if(components.size() > 1) {
+                bool found = false;
+                for(auto& subdir : root._subdirs) {
+                    if(subdir._name == components[0]) {
+                        found = true;
+                        buildNestedPath(subdir, components.exceptPrefix(1), file);
+                        break;
+                    }
+                }
+
+                if(!found) {
+                    arrayAppend(root._subdirs, InPlaceInit, components[0]);
+                    buildNestedPath(root._subdirs.back(), components.exceptPrefix(1), file);
+                }
+            }
+            else if(components.size() == 1) {
+                arrayAppend(root._files, &file);
+            }
+        }
+
+        Containers::String _name;
+        Containers::Array<DirType> _subdirs;
+        Containers::Array<FileType*> _files;
+};
+
+}}}
diff --git a/src/Managers/Vfs/VirtualFileSystem.h b/src/Managers/Vfs/VirtualFileSystem.h
new file mode 100644
index 0000000..0f3333c
--- /dev/null
+++ b/src/Managers/Vfs/VirtualFileSystem.h
@@ -0,0 +1,47 @@
+#pragma once
+
+// MassBuilderSaveTool
+// Copyright (C) 2021-2024 Guillaume Jacquemin
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+#include "VfsDirectory.h"
+
+namespace mbst { namespace Managers { namespace Vfs {
+
+template<typename FileType>
+class VirtualFileSystem: public VfsDirectory<FileType> {
+    public:
+        explicit VirtualFileSystem():
+            VfsDirectory<FileType>("")
+        {
+            // ctor
+        }
+        explicit VirtualFileSystem(Containers::ArrayView<FileType> files):
+            VfsDirectory<FileType>("")
+        {
+            build(files);
+        }
+
+        void build(Containers::ArrayView<FileType> files) {
+            for(auto& file : files) {
+                auto components = file.filename.split('/');
+                CORRADE_INTERNAL_ASSERT(components.size() == 0);
+
+                VfsDirectory<FileType>::buildNestedPath(*this, components, file);
+            }
+        }
+};
+
+}}}