From 2ee313ab617dc1f47c0e178fac0f08d90ae77cbf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=99=93=E6=9E=AB?=
 <45890549+xiaofeng-bm@users.noreply.github.com>
Date: Fri, 16 Oct 2020 10:37:35 +0800
Subject: [PATCH] feat(input-number): add input-number component (#368)

---
 packages/element-plus/index.ts                |   3 +
 .../__tests__/input-number.spec.ts            |  32 ++
 packages/input-number/index.ts                |   5 +
 packages/input-number/package.json            |  12 +
 packages/input-number/src/index.vue           | 304 ++++++++++++++++++
 5 files changed, 356 insertions(+)
 create mode 100644 packages/input-number/__tests__/input-number.spec.ts
 create mode 100644 packages/input-number/index.ts
 create mode 100644 packages/input-number/package.json
 create mode 100644 packages/input-number/src/index.vue

diff --git a/packages/element-plus/index.ts b/packages/element-plus/index.ts
index 3653b55980..31454a03a7 100644
--- a/packages/element-plus/index.ts
+++ b/packages/element-plus/index.ts
@@ -47,6 +47,7 @@ import ElColorPicker from '@element-plus/color-picker'
 import ElSelect from '@element-plus/select'
 import ElTimeSelect from '@element-plus/time-select'
 import ElPagination from '@element-plus/pagination'
+import ElInputNumber from '@element-plus/input-number'
 
 export {
   ElAlert,
@@ -96,6 +97,7 @@ export {
   ElSelect,
   ElTimeSelect,
   ElPagination,
+  ElInputNumber,
 }
 
 const install = (app: App): void => {
@@ -147,6 +149,7 @@ const install = (app: App): void => {
   ElSelect(app)
   ElTimeSelect(app)
   ElPagination(app)
+  ElInputNumber(app)
 }
 
 const elementUI = {
diff --git a/packages/input-number/__tests__/input-number.spec.ts b/packages/input-number/__tests__/input-number.spec.ts
new file mode 100644
index 0000000000..09c81cfd4b
--- /dev/null
+++ b/packages/input-number/__tests__/input-number.spec.ts
@@ -0,0 +1,32 @@
+import { mount } from '@vue/test-utils'
+import InputNumber from '../src/index.vue'
+import { ref } from  'vue'
+
+const _mount = options => mount({
+  components: {
+    'el-input-number': InputNumber,
+  },
+  ...options,
+})
+describe('InputNumber.vue', () => {
+  test('create', async () => {
+    const wrapper = _mount({
+      template: `
+        <el-input-number
+          label="描述文字"
+          v-model="num">
+        </el-input-number>
+      `,
+      setup() {
+        const num = ref(1)
+
+        return {
+          num,
+        }
+      },
+    })
+
+    const InputNumberElm = wrapper.find('input')
+    expect(InputNumberElm.exists()).toBe(true)
+  })
+})
diff --git a/packages/input-number/index.ts b/packages/input-number/index.ts
new file mode 100644
index 0000000000..305fdf50a5
--- /dev/null
+++ b/packages/input-number/index.ts
@@ -0,0 +1,5 @@
+import { App } from 'vue'
+import InputNumber from './src/index.vue'
+export default (app: App): void => {
+  app.component(InputNumber.name, InputNumber)
+}
diff --git a/packages/input-number/package.json b/packages/input-number/package.json
new file mode 100644
index 0000000000..72b1029e42
--- /dev/null
+++ b/packages/input-number/package.json
@@ -0,0 +1,12 @@
+{
+  "name": "@element-plus/input-number",
+  "version": "0.0.0",
+  "main": "dist/index.js",
+  "license": "MIT",
+  "peerDependencies": {
+    "vue": "^3.0.0"
+  },
+  "devDependencies": {
+    "@vue/test-utils": "^2.0.0-beta.3"
+  }
+}
diff --git a/packages/input-number/src/index.vue b/packages/input-number/src/index.vue
new file mode 100644
index 0000000000..77a0111be3
--- /dev/null
+++ b/packages/input-number/src/index.vue
@@ -0,0 +1,304 @@
+<template>
+  <div
+    :class="[
+      'el-input-number',
+      inputNumberSize ? 'el-input-number--' + inputNumberSize : '',
+      { 'is-disabled': inputNumberDisabled },
+      { 'is-without-controls': !controls },
+      { 'is-controls-right': controlsAtRight },
+    ]"
+    @dragstart.prevent
+  >
+    <span
+      v-if="controls"
+      v-repeat-click="decrease"
+      class="el-input-number__decrease"
+      role="button"
+      :class="{ 'is-disabled': minDisabled }"
+      @keydown.enter="decrease"
+    >
+      <i :class="`el-icon-${controlsAtRight ? 'arrow-down' : 'minus'}`"></i>
+    </span>
+    <span
+      v-if="controls"
+      v-repeat-click="increase"
+      class="el-input-number__increase"
+      role="button"
+      :class="{ 'is-disabled': maxDisabled }"
+      @keydown.enter="increase"
+    >
+      <i :class="`el-icon-${controlsAtRight ? 'arrow-up' : 'plus'}`"></i>
+    </span>
+    <el-input
+      ref="input"
+      :model-value="displayValue"
+      :placeholder="placeholder"
+      :disabled="inputNumberDisabled"
+      :size="inputNumberSize"
+      :max="max"
+      :min="min"
+      :name="name"
+      :label="label"
+      @keydown.up.prevent="increase"
+      @keydown.down.prevent="decrease"
+      @blur="event => $emit('blur', event)"
+      @focus="event => $emit('focus', event)"
+      @input="handleInput"
+      @change="handleInputChange"
+    />
+  </div>
+</template>
+<script lang="ts">
+import {
+  computed,
+  defineComponent,
+  reactive,
+  ref,
+  watch,
+  inject,
+  onMounted,
+  onUpdated,
+} from 'vue'
+import { RepeatClick } from '@element-plus/directives'
+import ElInput from '@element-plus/input/src/index.vue'
+import type { PropType } from 'vue'
+import { parseInt } from 'lodash'
+const ELEMENT: {
+  size?: number
+} = {}
+interface ElForm {
+  disabled: boolean
+  statusIcon: string
+}
+interface ElFormItem {
+  elFormItemSize: number
+  validateState: string
+}
+interface IData {
+  currentValue: number | string
+  userInput: null | number | string
+}
+export default defineComponent({
+  name: 'ElInputNumber',
+  components: {
+    ElInput,
+  },
+  directives: {
+    RepeatClick,
+  },
+  props: {
+    step: {
+      type: Number,
+      default: 1,
+    },
+    stepStrictly: {
+      type: Boolean,
+      default: false,
+    },
+    max: {
+      type: Number,
+      default: Infinity,
+    },
+    min: {
+      type: Number,
+      default: -Infinity,
+    },
+    modelValue: Number,
+    disabled: {
+      type: Boolean,
+      default: false,
+    },
+    size: {
+      type: String as PropType<'large' | 'small'>,
+      validator: (val: string) => ['large', 'small'].includes(val),
+    },
+    controls: {
+      type: Boolean,
+      default: true,
+    },
+    controlsPosition: {
+      type: String,
+      default: '',
+    },
+    name: String,
+    label: String,
+    placeholder: String,
+    precision: {
+      type: Number,
+      validator: (val: number) => val >= 0 && val === parseInt(val + '', 10),
+    },
+  },
+  emits: ['update:modelValue', 'change', 'input', 'blur', 'focus'],
+  setup(props, { emit }) {
+    const input = ref(null)
+    const data = reactive<IData>({
+      currentValue: 0,
+      userInput: null,
+    })
+    const elForm = inject<ElForm>('elForm', {} as any)
+    const elFormItem = inject<ElFormItem>('elFormItem', {} as any)
+    const minDisabled = computed(() => {
+      return _decrease(props.modelValue) < props.min
+    })
+    const maxDisabled = computed(() => {
+      return _increase(props.modelValue) > props.max
+    })
+    const numPrecision = computed(() => {
+      const stepPrecision = getPrecision(props.step)
+      if (props.precision !== undefined) {
+        if (stepPrecision > props.precision) {
+          console.warn(
+            '[Element Warn][InputNumber]precision should not be less than the decimal places of step',
+          )
+        }
+        return props.precision
+      } else {
+        return Math.max(getPrecision(props.modelValue), stepPrecision)
+      }
+    })
+    const controlsAtRight = computed(() => {
+      return props.controls && props.controlsPosition === 'right'
+    })
+    const inputNumberSize = computed(() => {
+      return props.size || elFormItem.elFormItemSize || ELEMENT?.size
+    })
+    const inputNumberDisabled = computed(() => {
+      return props.disabled || !!(elForm || {}).disabled
+    })
+    const displayValue = computed(() => {
+      if (data.userInput !== null) {
+        return data.userInput
+      }
+      let currentValue = data.currentValue
+      if (typeof currentValue === 'number') {
+        if (props.precision !== undefined) {
+          currentValue = currentValue.toFixed(props.precision)
+        }
+      }
+      return currentValue
+    })
+    const toPrecision = (num, pre?) => {
+      if (pre === undefined) pre = numPrecision.value
+      return parseFloat(
+        Math.round(num * Math.pow(10, pre)) / Math.pow(10, pre) + '',
+      )
+    }
+    const getPrecision = value => {
+      if (value === undefined) return 0
+      const valueString = value.toString()
+      const dotPosition = valueString.indexOf('.')
+      let precision = 0
+      if (dotPosition !== -1) {
+        precision = valueString.length - dotPosition - 1
+      }
+      return precision
+    }
+    const _increase = val => {
+      if (typeof val !== 'number' && val !== undefined) return data.currentValue
+      const precisionFactor = Math.pow(10, numPrecision.value)
+      // Solve the accuracy problem of JS decimal calculation by converting the value to integer.
+      return toPrecision(
+        (precisionFactor * val + precisionFactor * props.step) / precisionFactor,
+      )
+    }
+    const _decrease = val => {
+      if (typeof val !== 'number' && val !== undefined) return data.currentValue
+      const precisionFactor = Math.pow(10, numPrecision.value)
+      // Solve the accuracy problem of JS decimal calculation by converting the value to integer.
+      return toPrecision(
+        (precisionFactor * val - precisionFactor * props.step) / precisionFactor,
+      )
+    }
+    const increase = () => {
+      if (inputNumberDisabled.value || maxDisabled.value) return
+      const value = props.modelValue || 0
+      const newVal = _increase(value)
+      setCurrentValue(newVal)
+    }
+    const decrease = () => {
+      if (inputNumberDisabled.value || minDisabled.value) return
+      const value = props.modelValue || 0
+      const newVal = _decrease(value)
+      setCurrentValue(newVal)
+    }
+    const setCurrentValue = newVal => {
+      const oldVal = data.currentValue
+      if (
+        typeof newVal === 'number' &&
+        props.precision !== undefined
+      ) {
+        newVal = toPrecision(newVal, props.precision)
+      }
+      if (newVal >= props.max) newVal = props.max
+      if (newVal <= props.min) newVal = props.min
+      if (oldVal === newVal) return
+      data.userInput = null
+      emit('update:modelValue', newVal)
+      emit('input', newVal)
+      emit('change', newVal)
+      data.currentValue = newVal
+    }
+    const handleInput = value => {
+      return (data.userInput = value)
+    }
+    const handleInputChange = value => {
+      const newVal = value === '' ? undefined : Number(value)
+      if (!isNaN(newVal) || value === '') {
+        setCurrentValue(newVal)
+      }
+      data.userInput = null
+    }
+    watch(
+      () => props.modelValue,
+      value => {
+        let newVal = value === undefined ? value : Number(value)
+        if (newVal !== undefined) {
+          if (isNaN(newVal)) return
+          if (props.stepStrictly) {
+            const stepPrecision = getPrecision(props.step)
+            const precisionFactor = Math.pow(10, stepPrecision)
+            newVal =
+              (Math.round(newVal / props.step) * precisionFactor * props.step) /
+              precisionFactor
+          }
+          if (props.precision !== undefined) {
+            newVal = toPrecision(newVal, props.precision)
+          }
+        }
+        if (newVal >= props.max) newVal = props.max
+        if (newVal <= props.min) newVal = props.min
+        data.currentValue = newVal
+        data.userInput = null
+        emit('update:modelValue', newVal)
+        emit('input', newVal)
+      },
+      { immediate: true },
+    )
+    onMounted(() => {
+      let innerInput = input.value.input
+      innerInput.setAttribute('role', 'spinbutton')
+      innerInput.setAttribute('aria-valuemax', props.max)
+      innerInput.setAttribute('aria-valuemin', props.min)
+      innerInput.setAttribute('aria-valuenow', data.currentValue)
+      innerInput.setAttribute('aria-disabled', inputNumberDisabled.value)
+    })
+    onUpdated(() => {
+      let innerInput = input.value.input
+      innerInput.setAttribute('aria-valuenow', data.currentValue)
+    })
+    return {
+      input,
+      displayValue,
+      handleInput,
+      handleInputChange,
+      controlsAtRight,
+      decrease,
+      increase,
+      inputNumberSize,
+      inputNumberDisabled,
+      maxDisabled,
+      minDisabled,
+    }
+  },
+})
+</script>