Update

1 files changed, 20 insertions, 7 deletions

diff --git a/tpl/objects/lightbox.pug b/tpl/objects/lightbox.pug
index 58abdae..02a3200 100644
--- a/tpl/objects/lightbox.pug
+++ b/tpl/objects/lightbox.pug
@@ -1,20 +1,33 @@
 include icon.pug
 
 mixin lightbox(images)
-    .o-lightbox 
+    -
+        let classes = {
+            'o-lightbox': true,
+        }
+        let linksClasses = {
+            's-links': true,
+        }
+
+        if (attributes.theme) {
+            classes[`o-lightbox--${attributes.theme}`] = true;
+            linksClasses[`s-links--${attributes.theme}`] = true;
+        }
+
+    div(class=classes)
         header.o-lightbox__header
-            .s-links.s-links--colored
+            div(class=linksClasses)
                 block
-            +action-button(round=true quiet=true icon='x' class='o-lightbox__close-btn')
+            +action-button(theme=attributes.theme pill=true quiet=true icon='x' class='o-lightbox__close-btn')
         each img, i in images
             img.o-lightbox__img(src=images[i] id='image-' + i)
-            +action-button(round=true quiet=true icon='chevron-left' class='o-lightbox__nav-btn o-lightbox__nav-btn--prev')
-            +action-button(round=true quiet=true icon='chevron-right' class='o-lightbox__nav-btn o-lightbox__nav-btn--next')
+            +action-button(theme=attributes.theme pill=true quiet=true icon='chevron-left' class='o-lightbox__nav-btn o-lightbox__nav-btn--prev')
+            +action-button(theme=attributes.theme pill=true quiet=true icon='chevron-right' class='o-lightbox__nav-btn o-lightbox__nav-btn--next')
 
         img.o-lightbox__img.o-lightbox__img--default(src=images[0])
         if images.length > 1
-            +action-button(round=true quiet=true icon='chevron-left' class='o-lightbox__nav-btn o-lightbox__nav-btn--prev')
-            +action-button(round=true quiet=true icon='chevron-right' class='o-lightbox__nav-btn o-lightbox__nav-btn--next')
+            +action-button(theme=attributes.theme pill=true quiet=true icon='chevron-left' class='o-lightbox__nav-btn o-lightbox__nav-btn--prev')
+            +action-button(theme=attributes.theme pill=true quiet=true icon='chevron-right' class='o-lightbox__nav-btn o-lightbox__nav-btn--next')
         .o-lightbox__thumbnails
             each img, i in images
                 - classes = i === 0 ? 'is-selected' : ''