From ed0817359cab54ffa5c7f099e1d5947bc63bba07 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A2=81=E7=81=8F?= <admin@aresn.com>
Date: Thu, 5 Sep 2019 11:17:05 +0800
Subject: [PATCH] Badge add count slot

---
 examples/routers/badge.vue       |  8 ++++++--
 src/components/badge/badge.vue   | 12 +++++++++++-
 src/styles/components/badge.less |  9 ++++++++-
 3 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/examples/routers/badge.vue b/examples/routers/badge.vue
index 967730d9..3c5c8606 100644
--- a/examples/routers/badge.vue
+++ b/examples/routers/badge.vue
@@ -20,10 +20,14 @@
         <Badge :count="0" showZero>
             <a href="#" class="demo-badge"></a>
         </Badge>
+        <Badge>
+            <Icon type="md-time" slot="count" size="16" color="#ff6600" />
+            <a href="#" class="demo-badge"></a>
+        </Badge>
         <Button @click="setCount">set count</Button>
         <br><br>
         <Badge text="hot">
-            <Button type="ghost">Hello</Button>
+            <Button type="primary" ghost>Hello</Button>
         </Badge>
         <br><br>
         <div>
@@ -91,4 +95,4 @@
             }
         }
     };
-</script>
\ No newline at end of file
+</script>
diff --git a/src/components/badge/badge.vue b/src/components/badge/badge.vue
index 9eb4bdab..7b9fea9d 100644
--- a/src/components/badge/badge.vue
+++ b/src/components/badge/badge.vue
@@ -9,7 +9,8 @@
     </span>
     <span v-else :class="classes" ref="badge">
         <slot></slot>
-        <sup v-if="hasCount" :style="styles" :class="countClasses" v-show="badge">{{ finalCount }}</sup>
+        <sup v-if="$slots.count" :style="styles" :class="customCountClasses"><slot name="count"></slot></sup>
+        <sup v-else-if="hasCount" :style="styles" :class="countClasses" v-show="badge">{{ finalCount }}</sup>
     </span>
 </template>
 <script>
@@ -68,6 +69,15 @@
                     }
                 ];
             },
+            customCountClasses () {
+                return [
+                    `${prefixCls}-count`,
+                    `${prefixCls}-count-custom`,
+                    {
+                        [`${this.className}`]: !!this.className,
+                    }
+                ];
+            },
             statusClasses () {
                 return [
                     `${prefixCls}-status-dot`,
diff --git a/src/styles/components/badge.less b/src/styles/components/badge.less
index 38553f73..1d016e1e 100644
--- a/src/styles/components/badge.less
+++ b/src/styles/components/badge.less
@@ -27,6 +27,13 @@
         z-index: 10;
         box-shadow: 0 0 0 1px #fff;
 
+        &-custom{
+            background: transparent;
+            color: inherit;
+            border-color: transparent;
+            box-shadow: none;
+        }
+
         a,
         a:hover {
             color: #fff;
@@ -132,4 +139,4 @@
         transform: scale(2.4);
         opacity: 0;
     }
-}
\ No newline at end of file
+}