perf(hugopress): cache body-end hook

hugomods · Jun 16, 2024 · 2461a6d · 2461a6d
1 parent 979353b
commit 2461a6d
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 1 deletion.
diff --git a/go.mod b/go.mod
@@ -1,3 +1,5 @@
 module github.com/hugomods/mermaid
 
 go 1.19
+
+require github.com/hugomods/hugopress v0.5.0 // indirect
diff --git a/go.sum b/go.sum
@@ -0,0 +1,2 @@
+github.com/hugomods/hugopress v0.5.0 h1:eQHg1aKnc5StGFnV5H0BORhE0UmHYpaWHnX9QNseHrw=
+github.com/hugomods/hugopress v0.5.0/go.mod h1:CRYvr60xOkZOf4Atkuj+uEj2EH5pjJ39Ws3aJAkoMkA=
diff --git a/hugo.toml b/hugo.toml
@@ -2,6 +2,9 @@
 min = "0.110.0"
 
 [params.hugopress.modules.mermaid.hooks.body-end]
+partial = "mermaid/assets/js"
+cacheable = true
+cache_store_key = "hasMermaid"
 
 [params.mermaid]
 js_url = "https://cdn.jsdelivr.net/npm/mermaid@latest/dist/mermaid.esm.min.mjs"
diff --git a/layouts/partials/mermaid/assets/js.html b/layouts/partials/mermaid/assets/js.html
@@ -1,4 +1,4 @@
-{{- if or (.HasShortcode "mermaid") (.Store.Get "hasMermaid") }}
+{{- if .Store.Get "hasMermaid" }}
   <script type="module">
     import mermaid from "{{ .Site.Params.mermaid.js_url }}";
   </script>

diff --git a/layouts/shortcodes/mermaid.html b/layouts/shortcodes/mermaid.html
@@ -1,3 +1,4 @@
+{{ .Page.Store.Set "hasMermaid" true -}}
 <pre class="mermaid">
     {{- .Inner | safeHTML -}}
 </pre>