1 files changed, 7 insertions, 3 deletions
diff --git a/src/Filesystem.hpp b/src/Filesystem.hpp
index 139c214..7ebfb84 100644
--- a/src/Filesystem.hpp
+++ b/src/Filesystem.hpp
@@ -1,12 +1,12 @@
 #ifndef FILESYSTEM_H
 #define FILESYSTEM_H 1
 
-#include <map>
+#include <inja/inja.hpp>
+#include <unordered_map>
 #include <string>
 #include <vector>
 
 struct file_data {
-  bool inja_renderable;
   std::vector<unsigned char> data;
 };
 
@@ -17,9 +17,13 @@ public:
 
   bool AddSingleFile(std::string path, std::string root);
   bool Scan(std::string root = "./wwwroot");
+  void AddInjaCallback(std::string functionName, std::size_t numberOfArgs, inja::CallbackFunction function);
+  void AddVoidInjaCallback(std::string functionName, std::size_t numberOfArgs, inja::VoidCallbackFunction function);
 
 private:
-  std::map<std::string, struct file_data> files;
+  std::unordered_map<std::string, struct file_data> m_Files;
+  inja::TemplateStorage m_Templates;
+  inja::Environment m_Inja;
 };
 
 #endif