From e3930b7acef2e32d41271f52c7294dc0ac4110e3 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:45:00 +0800
Subject: [PATCH] Badge add new prop: color

---
 examples/routers/badge.vue       | 27 +++++++++++++++++++++++++++
 src/components/badge/badge.vue   | 14 +++++++++++---
 src/styles/components/badge.less | 12 ++++++++++++
 3 files changed, 50 insertions(+), 3 deletions(-)

diff --git a/examples/routers/badge.vue b/examples/routers/badge.vue
index 83834a16..69d62a71 100644
--- a/examples/routers/badge.vue
+++ b/examples/routers/badge.vue
@@ -90,6 +90,33 @@
                 <a href="#" class="demo-badge"></a>
             </Badge>
         </div>
+        <Divider></Divider>
+        <Badge status="success" text="Success" />
+        <br />
+        <Badge status="error" text="Error" />
+        <br />
+        <Badge status="default" text="Default" />
+        <br />
+        <Badge status="processing" text="Processing" />
+        <br />
+        <Badge status="warning" text="Warning" />
+        <br />
+        <Badge color="blue" text="blue" />
+        <Badge color="green" text="green" />
+        <Badge color="red" text="red" />
+        <Badge color="yellow" text="yellow" />
+        <Badge color="pink" text="pink" />
+        <Badge color="magenta" text="magenta" />
+        <Badge color="volcano" text="volcano" />
+        <Badge color="orange" text="orange" />
+        <Badge color="gold" text="gold" />
+        <Badge color="lime" text="lime" />
+        <Badge color="cyan" text="cyan" />
+        <Badge color="geekblue" text="geekblue" />
+        <Badge color="purple" text="purple" />
+        <br />
+        <Badge color="#2db7f5" text="#2db7f5" />
+        <Badge color="#f50" text="#f50" />
     </div>
 </template>
 <script>
diff --git a/src/components/badge/badge.vue b/src/components/badge/badge.vue
index 77eb2ce1..2d30e1d0 100644
--- a/src/components/badge/badge.vue
+++ b/src/components/badge/badge.vue
@@ -3,8 +3,8 @@
         <slot></slot>
         <sup :class="dotClasses" :style="styles" v-show="badge"></sup>
     </span>
-    <span v-else-if="status" :class="classes" class="ivu-badge-status" ref="badge">
-        <span :class="statusClasses"></span>
+    <span v-else-if="status || color" :class="classes" class="ivu-badge-status" ref="badge">
+        <span :class="statusClasses" :style="statusStyles"></span>
         <span class="ivu-badge-status-text"><slot name="text">{{ text }}</slot></span>
     </span>
     <span v-else :class="classes" ref="badge">
@@ -15,6 +15,7 @@
 </template>
 <script>
     import { oneOf } from '../../utils/assist';
+    const initColorList = ['blue', 'green', 'red', 'yellow', 'pink', 'magenta', 'volcano', 'orange', 'gold', 'lime', 'cyan', 'geekblue', 'purple'];
     const prefixCls = 'ivu-badge';
 
     export default {
@@ -50,6 +51,9 @@
             },
             offset: {
                 type: Array
+            },
+            color: {
+                type: String
             }
         },
         computed: {
@@ -82,10 +86,14 @@
                 return [
                     `${prefixCls}-status-dot`,
                     {
-                        [`${prefixCls}-status-${this.status}`]: !!this.status
+                        [`${prefixCls}-status-${this.status}`]: !!this.status,
+                        [`${prefixCls}-status-${this.color}`]: !!this.color && oneOf(this.color, initColorList)
                     }
                 ];
             },
+            statusStyles () {
+                return oneOf(this.color, initColorList) ? {} : { backgroundColor: this.color};
+            },
             styles () {
                 const style = {};
                 if (this.offset && this.offset.length === 2) {
diff --git a/src/styles/components/badge.less b/src/styles/components/badge.less
index 1d016e1e..0dca5caf 100644
--- a/src/styles/components/badge.less
+++ b/src/styles/components/badge.less
@@ -127,6 +127,18 @@
             font-size: @font-size-small;
             margin-left: 6px;
         }
+
+        @colors: pink, magenta, red, volcano, orange, yellow, gold, cyan, lime, green, blue, geekblue, purple;
+
+        .make-color-classes(@i: length(@colors)) when (@i > 0) {
+            .make-color-classes(@i - 1);
+            @color: extract(@colors, @i);
+            @darkColor: '@{color}-6';
+            &-@{color} {
+                background-color: @@darkColor;
+            }
+        }
+        .make-color-classes();
     }
 }