From 192ea848c4d6c07fcb6453aae6868df165b172a5 Mon Sep 17 00:00:00 2001
From: Rosia E Evans <willowro@outlook.com>
Date: Wed, 11 Oct 2023 11:38:56 +0100
Subject: [PATCH] Minor cleanup of NavBarGenerator class

---
 Src/BlogPageBuilder.cpp | 15 ++++++---------
 1 file changed, 6 insertions(+), 9 deletions(-)

diff --git a/Src/BlogPageBuilder.cpp b/Src/BlogPageBuilder.cpp
index a083299..80d4af0 100644
--- a/Src/BlogPageBuilder.cpp
+++ b/Src/BlogPageBuilder.cpp
@@ -71,25 +71,22 @@ void BlogPageBuilder::collectPages()
 std::string NavBarGenerator::generateCategories(std::string rootFolderUrl)
 {
     std::stringstream buffer;
-
-    std::set<std::string> categories;
+    std::set<std::string> folders;
 
     auto directoryIterator = std::filesystem::directory_iterator(rootFolderUrl);
     for (auto folder : directoryIterator)
     {
         if (folder.is_directory())
         {
-            std::string path = folder.path();
-            std::string folderName = path.substr(path.find_last_of("/")+1, path.length()-path.find_last_of("/"));
-
-            categories.insert(folderName);
+            folders.insert(folder.path());
         }
     }
 
-    for (std::string category : categories)
+    for (std::string folder : folders)
     {
-        buffer << "\n<h2>" << category << "</h2>\n";
-        buffer << generateCategories(rootFolderUrl + "/" + category);
+        std::string categoryName = folder.substr(folder.find_last_of("/")+1, folder.length()-folder.find_last_of("/"));
+        buffer << "\n<h2>" << categoryName << "</h2>\n";
+        buffer << generateCategories(folder);
         buffer << "\n";
     }
     return buffer.str();