import { VantComponent } from '../common/component'; import { isDef } from '../common/utils'; const currentYear = new Date().getFullYear(); function isValidDate(date) { return isDef(date) && !isNaN(new Date(date).getTime()) }; function range(num, min, max) { return Math.min(Math.max(num, min), max); } function padZero(val: string | number): string { return `00${val}`.slice(-2); } function times(n: number, iteratee: (number) => string): string[] { let index = -1; const result = Array(n); while (++index < n) { result[index] = iteratee(index); } return result; } function getTrueValue(formattedValue: string): number { if (!formattedValue) return; while (isNaN(parseInt(formattedValue, 10))) { formattedValue = formattedValue.slice(1); } return parseInt(formattedValue, 10); } function getMonthEndDay(year, month): number { return 32 - new Date(year, month - 1, 32).getDate(); } VantComponent({ props: { value: null, title: String, loading: Boolean, itemHeight: { type: Number, value: 44 }, visibleItemCount: { type: Number, value: 5 }, confirmButtonText: { type: String, value: '确认' }, cancelButtonText: { type: String, value: '取消' }, type: { type: String, value: 'datetime' }, showToolbar: { type: Boolean, value: true }, minDate: { type: Number, value: new Date(currentYear - 10, 0, 1).getTime() }, maxDate: { type: Number, value: new Date(currentYear + 10, 11, 31).getTime() }, minHour: { type: Number, value: 0 }, maxHour: { type: Number, value: 23 }, minMinute: { type: Number, value: 0 }, maxMinute: { type: Number, value: 59 } }, data: { pickerValue: [], innerValue: Date.now() }, computed: { columns() { const results = this.getRanges().map(({ type, range }) => { const values = times(range[1] - range[0] + 1, index => { let value = range[0] + index; value = type === 'year' ? `${value}` : padZero(value); return value; }); return values; }); return results; } }, watch: { value(val) { const { data } = this; val = this.correctValue(val); const isEqual = val === data.innerValue; if (!isEqual) { this.set({ innerValue: val }, () => { this.updateColumnValue(val); this.$emit('input', val); }); } } }, methods: { getRanges(): object[] { const { data } = this; if (data.type === 'time') { return [ { type: 'hour', range: [data.minHour, data.maxHour] }, { type: 'minute', range: [data.minMinute, data.maxMinute] } ]; } const { maxYear, maxDate, maxMonth, maxHour, maxMinute } = this.getBoundary('max', data.innerValue); const { minYear, minDate, minMonth, minHour, minMinute } = this.getBoundary('min', data.innerValue); const result = [ { type: 'year', range: [minYear, maxYear] }, { type: 'month', range: [minMonth, maxMonth] }, { type: 'day', range: [minDate, maxDate] }, { type: 'hour', range: [minHour, maxHour] }, { type: 'minute', range: [minMinute, maxMinute] } ]; if (data.type === 'date') result.splice(3, 2); if (data.type === 'year-month') result.splice(2, 3); return result; }, correctValue(value) { const { data } = this; // validate value const isDateType = data.type !== 'time'; if (isDateType && !isValidDate(value)) { value = data.minDate; } else if (!isDateType && !value) { const { minHour } = data; value = `${padZero(minHour)}:00`; } // time type if (!isDateType) { let [hour, minute] = value.split(':'); hour = padZero(range(hour, data.minHour, data.maxHour)); minute = padZero(range(minute, data.minMinute, data.maxMinute)); return `${hour}:${minute}`; } // date type value = Math.max(value, data.minDate); value = Math.min(value, data.maxDate); return value; }, getBoundary(type: string, innerValue: number): object { const value = new Date(innerValue); const boundary = new Date(this.data[`${type}Date`]); const year = boundary.getFullYear(); let month = 1; let date = 1; let hour = 0; let minute = 0; if (type === 'max') { month = 12; date = getMonthEndDay(value.getFullYear(), value.getMonth() + 1); hour = 23; minute = 59; } if (value.getFullYear() === year) { month = boundary.getMonth() + 1; if (value.getMonth() + 1 === month) { date = boundary.getDate(); if (value.getDate() === date) { hour = boundary.getHours(); if (value.getHours() === hour) { minute = boundary.getMinutes(); } } } } return { [`${type}Year`]: year, [`${type}Month`]: month, [`${type}Date`]: date, [`${type}Hour`]: hour, [`${type}Minute`]: minute }; }, onCancel() { this.$emit('cancel'); }, onConfirm(): void { this.$emit('confirm', this.data.innerValue); }, onChange(event: Weapp.Event): void { const { data } = this; const pickerValue = event.detail.value; const values = pickerValue .slice(0, data.columns.length) .map((value, index) => data.columns[index][value]); let value; if (data.type === 'time') { value = values.join(':'); } else { const year = getTrueValue(values[0]); const month = getTrueValue(values[1]); const maxDate = getMonthEndDay(year, month); let date = getTrueValue(values[2]); if (data.type === 'year-month') { date = 1; } date = date > maxDate ? maxDate : date; let hour = 0; let minute = 0; if (data.type === 'datetime') { hour = getTrueValue(values[3]); minute = getTrueValue(values[4]); } value = new Date(year, month - 1, date, hour, minute); } value = this.correctValue(value); this.set({ innerValue: value }, () => { this.updateColumnValue(value); this.$emit('input', value); this.$emit('change', this); }); }, getColumnValue(index) { return this.getValues()[index]; }, setColumnValue(index, value) { const { pickerValue, columns } = this.data; pickerValue[index] = columns[index].indexOf(value); this.set({ pickerValue }); }, getColumnValues(index) { return this.data.columns[index]; }, setColumnValues(index, values) { const { columns } = this.data; columns[index] = values; this.set({ columns }); }, getValues() { const { pickerValue, columns } = this.data; return pickerValue.map((value, index) => columns[index][value]); }, setValues(values) { const { columns } = this.data; this.set({ pickerValue: values.map((value, index) => columns[index].indexOf(value)) }); }, updateColumnValue(value): void { let values = []; const { data } = this; const { columns } = data; if (data.type === 'time') { const currentValue = value.split(':'); values = [ columns[0].indexOf(currentValue[0]), columns[1].indexOf(currentValue[1]) ]; } else { const date = new Date(value); values = [ columns[0].indexOf(`${date.getFullYear()}`), columns[1].indexOf(padZero(date.getMonth() + 1)) ]; if (data.type === 'date') { values.push(columns[2].indexOf(padZero(date.getDate()))); } if (data.type === 'datetime') { values.push( columns[2].indexOf(padZero(date.getDate())), columns[3].indexOf(padZero(date.getHours())), columns[4].indexOf(padZero(date.getMinutes())) ); } } this.set({ pickerValue: values }); } }, created() { const innerValue = this.correctValue(this.data.value); this.set({ innerValue }, () => { this.updateColumnValue(innerValue); this.$emit('input', innerValue); }); } });