From 21698de271a6ca6906aa6a59dc5c664c88b91942 Mon Sep 17 00:00:00 2001
From: Wanli Song <songwanli2025@163.com>
Date: Thu, 1 Jul 2021 23:57:16 +0800
Subject: [PATCH] feat(spin): icon slot, rotate prop (#289)

* feat(spin): add icon slot

* doc(changelog

* fix(spin): icon slot logic

* test(spin): add icon slot test

* fix(spin): refact code

* feat(spin): add rotate prop

Co-authored-by: wanli.song@tusimple.ai <wanli.song@tusimple.ai>
Co-authored-by: 07akioni <07akioni2@gmail.com>
---
 CHANGELOG.en-US.md                            |  2 +
 CHANGELOG.zh-CN.md                            |  2 +
 src/_internal/loading/src/Loading.tsx         | 13 +++-
 .../loading/src/styles/index.cssr.ts          | 60 ++++++++++++-------
 src/spin/demos/enUS/customize-icon.demo.md    | 40 +++++++++++++
 src/spin/demos/enUS/index.demo-entry.md       |  3 +
 src/spin/demos/zhCN/customize-icon.demo.md    | 40 +++++++++++++
 src/spin/demos/zhCN/index.demo-entry.md       |  3 +
 src/spin/src/Spin.tsx                         | 14 ++++-
 src/spin/tests/Spin.spec.ts                   | 29 +++++++++
 10 files changed, 181 insertions(+), 25 deletions(-)
 create mode 100644 src/spin/demos/enUS/customize-icon.demo.md
 create mode 100644 src/spin/demos/zhCN/customize-icon.demo.md

diff --git a/CHANGELOG.en-US.md b/CHANGELOG.en-US.md
index 7e0940d79..5aa86530e 100644
--- a/CHANGELOG.en-US.md
+++ b/CHANGELOG.en-US.md
@@ -13,6 +13,8 @@
 - `n-upload` add `before-upload` prop.
 - `n-image` add `alt` prop.
 - Support the enter key on the numeric keypad.
+- `n-spin` support `icon` slot for icon customizing, closes[#260](https://github.com/TuSimple/naive-ui/issues/260).
+- `n-spin` add `rotate` prop fro slot icon to rotate.
 - `n-form` export `FormItemRule` & `FormRules` type.
 
 ### Fixes
diff --git a/CHANGELOG.zh-CN.md b/CHANGELOG.zh-CN.md
index 6e4c778db..494ea6c4d 100644
--- a/CHANGELOG.zh-CN.md
+++ b/CHANGELOG.zh-CN.md
@@ -13,6 +13,8 @@
 - `n-upload` 新增 `before-upload` 属性
 - `n-image` 新增 `alt` 属性.
 - 支持小键盘的 enter 键
+- `n-spin` 支持 `icon` 插槽为了自定义加载图标，closes[#260](https://github.com/TuSimple/naive-ui/issues/260)
+- `n-spin` 新增 `rotate` 属性控制自定义加载图标是否有旋转动画
 - `n-form` 导出 `FormItemRule` & `FormRules` 类型
 
 ### Fixes
diff --git a/src/_internal/loading/src/Loading.tsx b/src/_internal/loading/src/Loading.tsx
index cbca58346..b6857c406 100644
--- a/src/_internal/loading/src/Loading.tsx
+++ b/src/_internal/loading/src/Loading.tsx
@@ -33,13 +33,17 @@ export default defineComponent({
     show: {
       type: Boolean,
       default: true
+    },
+    rotate: {
+      type: Boolean,
+      default: true
     }
   },
   setup (props) {
     useStyle('BaseLoading', style, toRef(props, 'clsPrefix'))
   },
   render () {
-    const { clsPrefix, radius, strokeWidth, stroke, scale } = this
+    const { clsPrefix, radius, strokeWidth, stroke, scale, $slots } = this
     const scaledRadius = radius / scale
     return (
       <div class={`${clsPrefix}-base-loading`} role="img" aria-label="loading">
@@ -47,6 +51,13 @@ export default defineComponent({
           {{
             default: () =>
               this.show ? (
+                $slots.icon ?
+                  <div class={[
+                    `${clsPrefix}-base-loading__icon-slot`,
+                    this.rotate && `${clsPrefix}-base-loading__icon-slot--rotate`
+                    ]}>
+                   {$slots.icon()}
+                  </div> :
                 <svg
                   class={`${clsPrefix}-base-loading__icon`}
                   viewBox={`0 0 ${2 * scaledRadius} ${2 * scaledRadius}`}
diff --git a/src/_internal/loading/src/styles/index.cssr.ts b/src/_internal/loading/src/styles/index.cssr.ts
index 6daa3eecc..1e5cb3f8c 100644
--- a/src/_internal/loading/src/styles/index.cssr.ts
+++ b/src/_internal/loading/src/styles/index.cssr.ts
@@ -1,28 +1,44 @@
-import { cB, cE } from '../../../../_utils/cssr'
+import { c, cB, cE, cM } from '../../../../_utils/cssr'
 import iconSwitchTransition from '../../../../_styles/transitions/icon-switch.cssr'
 
-export default cB('base-loading', `
-  position: relative;
-  line-height: 0;
-  width: 1em;
-  height: 1em;
-`, [
-  cE('placeholder', {
-    position: 'absolute',
-    left: '50%',
-    top: '50%',
-    transform: 'translateX(-50%) translateY(-50%)'
-  }, [
-    iconSwitchTransition({
+export default  c([
+  c('@keyframes icon-rotate-animation', `
+    0% { 
+      transform:rotate(0deg);
+    }
+    100% { 
+      transform:rotate(360deg);
+    }
+  `),
+  cB('base-loading', `
+    position: relative;
+    line-height: 0;
+    width: 1em;
+    height: 1em;
+  `, [
+    cE('placeholder', {
+      position: 'absolute',
       left: '50%',
       top: '50%',
-      originalTransform: 'translateX(-50%) translateY(-50%)'
-    })
-  ]),
-  cE('icon', `
-    height: 1em;
-    width: 1em;
-  `, [
-    iconSwitchTransition()
+      transform: 'translateX(-50%) translateY(-50%)'
+    }, [
+      iconSwitchTransition({
+        left: '50%',
+        top: '50%',
+        originalTransform: 'translateX(-50%) translateY(-50%)'
+      })
+    ]),
+    cE('icon', `
+      height: 1em;
+      width: 1em;
+    `, [
+      iconSwitchTransition()
+    ]),
+    cE('icon-slot', [
+      cM('rotate', `
+        display: inline-block;
+        animation: icon-rotate-animation 1s linear infinite;
+      `)
+    ])
   ])
 ])
diff --git a/src/spin/demos/enUS/customize-icon.demo.md b/src/spin/demos/enUS/customize-icon.demo.md
new file mode 100644
index 000000000..5c2ec1f41
--- /dev/null
+++ b/src/spin/demos/enUS/customize-icon.demo.md
@@ -0,0 +1,40 @@
+# Customize Icon
+
+```html
+<n-space vertical>
+  <n-spin size="small">
+    <template #icon>
+      <n-icon>
+        <Reload />
+      </n-icon>
+    </template>
+  </n-spin>
+  <n-spin :show="show">
+    <n-alert title="La La La" type="success">
+      Leave it till tomorrow to unpack my case. Honey disconnect the phone.
+    </n-alert>
+    <template #icon>
+      <n-icon>
+        <Reload />
+      </n-icon>
+    </template>
+  </n-spin>
+  <n-button @click="show = !show">Click to Spin</n-button>
+</n-space>
+```
+
+```js
+import { ref, defineComponent } from 'vue'
+import { Reload } from '@vicons/ionicons5'
+
+export default defineComponent({
+  components: {
+    Reload
+  },
+  setup () {
+    return {
+      show: ref(false)
+    }
+  }
+})
+```
diff --git a/src/spin/demos/enUS/index.demo-entry.md b/src/spin/demos/enUS/index.demo-entry.md
index 9900ccdf2..12409b153 100644
--- a/src/spin/demos/enUS/index.demo-entry.md
+++ b/src/spin/demos/enUS/index.demo-entry.md
@@ -7,12 +7,14 @@ It can be called loading, but why it's called loading? Because there is another
 ```demo
 basic
 wrap
+customize-icon
 ```
 
 ## Props
 
 | Name | Type | Default | Description |
 | --- | --- | --- | --- |
+| rotate | `boolean` | `true` | If slot icon is rotate. |
 | size | `'small' \| 'medium' \| 'large' \| number` | `'medium'` |  |
 | show | `boolean` | `true` | If spin is active. |
 | stroke-width | `number` | `undefined` | Relative width of spin's stroke, you can assume the outer radius of spin is 100. |
@@ -23,3 +25,4 @@ wrap
 | Name    | Parameters | Description                        |
 | ------- | ---------- | ---------------------------------- |
 | default | `()`       | If set, spin will wrap the content |
+| icon | `()`       | Customize the spin icon |
diff --git a/src/spin/demos/zhCN/customize-icon.demo.md b/src/spin/demos/zhCN/customize-icon.demo.md
new file mode 100644
index 000000000..10d7c9a18
--- /dev/null
+++ b/src/spin/demos/zhCN/customize-icon.demo.md
@@ -0,0 +1,40 @@
+# 自定义图标
+
+```html
+<n-space vertical>
+  <n-spin size="small" >
+    <template #icon>
+      <n-icon>
+        <Reload />
+      </n-icon>
+    </template>
+  </n-spin>
+  <n-spin :show="show">
+    <n-alert title="La La La" type="success">
+    明天再打开行李箱。宝贝，挂电话啦。
+    </n-alert>
+    <template #icon>
+      <n-icon>
+        <Reload />
+      </n-icon>
+    </template>
+  </n-spin>
+  <n-button @click="show = !show">点出来加载</n-button>
+</n-space>
+```
+
+```js
+import { ref, defineComponent } from 'vue'
+import { Reload } from '@vicons/ionicons5'
+
+export default defineComponent({
+  components: {
+    Reload
+  },
+  setup () {
+    return {
+      show: ref(false)
+    }
+  }
+})
+```
diff --git a/src/spin/demos/zhCN/index.demo-entry.md b/src/spin/demos/zhCN/index.demo-entry.md
index f0e3d75af..4a3db375f 100644
--- a/src/spin/demos/zhCN/index.demo-entry.md
+++ b/src/spin/demos/zhCN/index.demo-entry.md
@@ -7,12 +7,14 @@
 ```demo
 basic
 wrap
+customize-icon
 ```
 
 ## Props
 
 | 名称 | 类型 | 默认值 | 说明 |
 | --- | --- | --- | --- |
+| rotate | `boolean` | `true` | 自定义加载图标是否有旋转动画 |
 | size | `'small' \| 'medium' \| 'large' \| number` | `'medium'` |  |
 | show | `boolean` | `true` | Spin 在填入内容的状态是否激活 |
 | stroke-width | `number` | `undefined` | Spin 边缘的相对宽度，假定 Spin 的外侧半径是 100 |
@@ -23,3 +25,4 @@ wrap
 | 名称    | 参数 | 说明                            |
 | ------- | ---- | ------------------------------- |
 | default | `()` | 如果填入，Spin 会包裹填入的内容 |
+| icon | `()`       | 自定义加载图标 |
diff --git a/src/spin/src/Spin.tsx b/src/spin/src/Spin.tsx
index 724fb7db6..b50fc8669 100644
--- a/src/spin/src/Spin.tsx
+++ b/src/spin/src/Spin.tsx
@@ -47,6 +47,10 @@ const spinProps = {
       return true
     },
     default: undefined
+  },
+  rotate: {
+    type: Boolean,
+    default: true
   }
 }
 
@@ -117,8 +121,11 @@ export default defineComponent({
                   clsPrefix={mergedClsPrefix}
                   stroke={this.stroke}
                   strokeWidth={this.mergedStrokeWidth}
+                  rotate={this.rotate}
                   class={`${mergedClsPrefix}-spin`}
-                />
+                >
+                  {{ icon: $slots.icon }}
+                </NBaseLoading>
               ) : null
           }}
         </Transition>
@@ -129,8 +136,11 @@ export default defineComponent({
         style={this.cssVars as CSSProperties}
         stroke={this.stroke}
         stroke-width={this.mergedStrokeWidth}
+        rotate={this.rotate}
         class={`${mergedClsPrefix}-spin`}
-      />
+      >
+        {{ icon: $slots.icon }}
+      </NBaseLoading>
     )
   }
 })
diff --git a/src/spin/tests/Spin.spec.ts b/src/spin/tests/Spin.spec.ts
index f30b262d2..e24f224a5 100644
--- a/src/spin/tests/Spin.spec.ts
+++ b/src/spin/tests/Spin.spec.ts
@@ -1,8 +1,37 @@
+import { h } from '@vue/runtime-core'
 import { mount } from '@vue/test-utils'
 import { NSpin } from '../index'
+import { Reload } from '@vicons/ionicons5'
+import { NIcon } from '../../icon'
 
 describe('n-spin', () => {
   it('should work with import on demand', () => {
     mount(NSpin)
   })
+  it('should work with icon slot', () => {
+    const wrapper = mount(NSpin, {
+      slots: {
+        icon: () => h(NIcon, null, {
+          default: () => h(Reload)
+        })
+      }
+    })
+
+    expect(wrapper.findComponent(NIcon).exists()).toBe(true)
+    expect(wrapper.findComponent(Reload).exists()).toBe(true)
+  })
+  it('rotate should work on icon slot', async () => {
+    const wrapper = mount(NSpin, {
+      slots: {
+        icon: () => h(NIcon, null, {
+          default: () => h(Reload)
+        })
+      }
+    })
+    expect(wrapper.find('.n-base-loading__icon-slot--rotate').exists()).toBe(true)
+    await wrapper.setProps({
+      rotate: false
+    })
+    expect(wrapper.find('.n-base-loading__icon-slot--rotate').exists()).toBe(false)
+  })
 })