perf(Builder): load template layouts at construction

cppalliance · Nov 25, 2024 · 9543810 · 9543810
1 parent 3d38f59
commit 9543810
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 2 deletions.
diff --git a/src/lib/Gen/hbs/Builder.cpp b/src/lib/Gen/hbs/Builder.cpp
@@ -143,6 +143,29 @@ Builder(
     helpers::registerAntoraHelpers(hbs_);
     helpers::registerLogicalHelpers(hbs_);
     helpers::registerContainerHelpers(hbs_);
+
+    // load templates
+    exp = forEachFile(layoutDir(), false,
+        [&](std::string_view pathName) -> Expected<void>
+        {
+            // Get template relative path
+            std::filesystem::path relPath = pathName;
+            relPath = relPath.lexically_relative(layoutDir());
+
+            // Skip non-handlebars files
+            MRDOCS_CHECK_OR(relPath.extension() == ".hbs", {});
+
+            // Load template contents
+            MRDOCS_TRY(std::string text, files::getFileText(pathName));
+
+            // Register template
+            this->templates_.emplace(relPath.generic_string(), text);
+            return {};
+        });
+    if (!exp)
+    {
+        exp.error().Throw();
+    }
 }
 
 //------------------------------------------------
@@ -154,8 +177,9 @@ callTemplate(
     std::string_view name,
     dom::Value const& context)
 {
-    auto pathName = files::appendPath(layoutDir(), name);
-    MRDOCS_TRY(auto fileText, files::getFileText(pathName));
+    auto it = templates_.find(name);
+    MRDOCS_CHECK(it != templates_.end(), formatError("Template {} not found", name));
+    std::string_view fileText = it->second;
     HandlebarsOptions options;
     options.escapeFunction = escapeFn_;
     OutputRef out(os);

diff --git a/src/lib/Gen/hbs/Builder.hpp b/src/lib/Gen/hbs/Builder.hpp
@@ -33,6 +33,7 @@ class Builder
 {
     js::Context ctx_;
     Handlebars hbs_;
+    std::map<std::string, std::string, std::less<>> templates_;
     std::function<void(OutputRef&, std::string_view)> escapeFn_;
 
     std::string