From c9d979514c009e8b28d99cff2614b92019ccd5c8 Mon Sep 17 00:00:00 2001
From: Alex Shpak <alex-shpak@users.noreply.github.com>
Date: Tue, 11 Jun 2019 13:40:50 +0200
Subject: Add custom class support for mermaid and button shortcodes

---
 layouts/shortcodes/button.html  | 4 +++-
 layouts/shortcodes/mermaid.html | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

(limited to 'layouts/shortcodes')

diff --git a/layouts/shortcodes/button.html b/layouts/shortcodes/button.html
index 88b0f20..98d2cdd 100644
--- a/layouts/shortcodes/button.html
+++ b/layouts/shortcodes/button.html
@@ -7,4 +7,6 @@
 {{ with .Get "relref" }}
   {{ $ref = relref $ . }}
 {{ end }}
-<a {{ with $ref }} href="{{.}}" {{ end }} {{ with $target }} target="{{.}}" {{ end }} class="book-btn">{{ $.Inner }}</a>
+<a {{ with $ref }} href="{{.}}" {{ end }} {{ with $target }} target="{{.}}" {{ end }} class="book-btn{{ with .Get "class" }} {{ . }}{{ end }}">
+  {{ $.Inner }}
+</a>
diff --git a/layouts/shortcodes/mermaid.html b/layouts/shortcodes/mermaid.html
index 66eb4a6..f569bc4 100644
--- a/layouts/shortcodes/mermaid.html
+++ b/layouts/shortcodes/mermaid.html
@@ -4,6 +4,6 @@
 {{ .Page.Scratch.Set "mermaid" true }}
 {{ end }}
 
-<p class="mermaid">
+<p class="mermaid{{ with .Get "class" }} {{ . }}{{ end }}">
   {{ .Inner }}
 </p>
-- 
cgit v1.2.3