Overhaul figures

layouts/shortcodes/paige/code.html

@@ -1,6 +1,27 @@
-<figure class="align-items-center d-flex flex-column">
-    {{ highlight (.Inner | replaceRE "^\n" "") (.Get 0 | default (.Get "lang")) (.Get 2 | default (.Get "options")) }}
-    {{ with (.Get 1 | default (.Get "caption")) }}
-    <figcaption class="figure-caption">{{ . }}</figcaption>
-    {{ end }}
-</figure>
+{{ $caption := .Get "caption" }}
+{{ $captionclass := .Get "captionclass" }}
+{{ $captionstyle := .Get "captionstyle" }}
+{{ $class := .Get "class" }}
+{{ $contentclass := .Get "contentclass" | default "mb-2" }}
+{{ $contentstyle := .Get "contentstyle" }}
+{{ $style := .Get "style" }}
+
+{{ $lang := .Get 0 | default (.Get "lang") }}
+{{ $options := .Get "options" }}
+
+{{ if $caption }}
+    {{ $contentclass = $contentclass | default "mb-2" }}
+{{ end }}
+
+{{ $content := highlight (.Inner | replaceRE "^\n" "") $lang $options }}
+
+{{ partial "paige/figure.html" (dict
+    "caption" $caption
+    "captionclass" $captionclass
+    "captionstyle" $captionstyle
+    "class" $class
+    "content" $content
+    "contentclass" $contentclass
+    "contentstyle" $contentstyle
+    "style" $style
+) }}