Introduce Manager::addBooksFromDirectory()

Added a function to load books from a directory. Requires rootPath to iterate over.

Author: juuz0

include/manager.h

@@ -155,6 +155,17 @@ class Manager
                        const std::string& url = "",
                        const bool checkMetaData = false);
 
+  /**
+   * Add books from a directory into the libary.
+   * 
+   * @param path          The path of the directory to scan.
+   * @param path          If the function should stop for an invalid file.
+   * @param verboseFlag   Verbose logs flag.
+   */
+  void addBooksFromDirectory(const std::string& path,
+                             const bool skipInvalid = true,
+                             const bool verboseFlag = false);
+
   std::string writableLibraryPath;
 
   bool m_hasSearchResult = false;

src/manager.cpp

@@ -23,6 +23,11 @@
 #include "tools/pathTools.h"
 
 #include <pugixml.hpp>
+#include <filesystem>
+#include <iostream>
+#include <set>
+
+namespace fs = std::filesystem;
 
 namespace kiwix
 {
@@ -251,6 +256,46 @@ bool Manager::addBookFromPath(const std::string& pathToOpen,
           .empty());
 }
 
+void Manager::addBooksFromDirectory(const std::string& path,
+                                    const bool skipInvalid,
+                                    const bool verboseFlag)
+{
+  std::set<std::string> iteratedDirs;
+  std::queue<std::string> dirQueue;
+  dirQueue.push(fs::absolute(path).u8string());
+
+  while (!dirQueue.empty()) {
+    const auto currentPath = dirQueue.front();
+    dirQueue.pop();
+    if (verboseFlag)
+      std::cout << "Iterating over directory: " << currentPath << std::endl;
+    for (const auto& dirEntry : fs::directory_iterator(currentPath)) {
+      auto resolvedPath = dirEntry.path();
+      if (fs::is_symlink(dirEntry)) {
+        resolvedPath = fs::canonical(dirEntry.path());
+      }
+      const std::string pathString = resolvedPath.u8string();
+      if (fs::is_directory(resolvedPath)) {
+        if (iteratedDirs.find(pathString) == iteratedDirs.end())
+          dirQueue.push(pathString);
+        else if (verboseFlag)
+          std::cout << "Already iterated over " << pathString << ". Skipping..." << std::endl;
+      } else {
+        if (!this->addBookFromPath(pathString, pathString, "", false)) {
+          if (skipInvalid)
+            std::cerr << "Skipping invalid file: " << pathString << std::endl;
+          else {
+            throw std::runtime_error("Unable to add file: " + pathString + " into the library.");
+          }
+        }
+      }
+    }
+    if (verboseFlag)
+      std::cout << "Completed iterating over directory: " << currentPath << std::endl;
+    iteratedDirs.insert(currentPath);
+  }
+}
+
 bool Manager::readBookFromPath(const std::string& path, kiwix::Book* book)
 {
   std::string tmp_path = path;