1 files changed, 17 insertions, 9 deletions

diff --git a/tpl/objects/button.pug b/tpl/objects/button.pug
index 0884c8d..c356bfe 100644
--- a/tpl/objects/button.pug
+++ b/tpl/objects/button.pug
@@ -1,22 +1,30 @@
-mixin a-button
+include icon.pug
+
+mixin button
     -
         let classes = {
-            'o-button':          true,
-            'o-button--block':   attributes.block,
-            'o-button--primary': attributes.primary,
-            'o-button--outline': attributes.outline,
-            'o-button--round':   !!attributes.icon && !block
+            'o-button':            true,
+            'u-d-block':           attributes.block,
+            'o-button--secondary': attributes.secondary,
+            'o-button--quiet':     attributes.quiet,
+            'o-button--pill':      attributes.pill,
+            'o-button--badge':     attributes.badge,
+            'o-button--icon':      !!attributes.icon && !block,
+            'is-selected':         attributes.selected,
         }
-        if (attributes.variant) {
-            classes['o-button--' + attributes.variant] = true
+        if (attributes.theme) {
+            classes['o-button--' + attributes.theme] = true
         }
         if (attributes.size) {
             classes['o-button--' + attributes.size] = true
         }
+        if (attributes.class) {
+            classes[attributes.class] = true;
+        }
 
         let href = attributes.disabled ? null : '#';
 
-    a(class=classes href=href aria-disabled=attributes.disabled && String(attributes.disabled))
+    a(class=classes href=href)
         if attributes.icon
             +icon(attributes.icon)
             = ' '