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: { innerValue: Date.now(), columns: [] }, watch: { value(val) { const { data } = this; val = this.correctValue(val); const isEqual = val === data.innerValue; if (!isEqual) { this.updateColumnValue(val).then(() => { this.$emit('input', val); }); } }, type: 'updateColumns', minHour: 'updateColumns', maxHour: 'updateColumns', minMinute: 'updateColumns', maxMinute: 'updateColumns' }, methods: { asyncSet(data) { return new Promise(resolve => { this.set(data, resolve); }); }, getPicker() { if (this.picker == null) { const picker = (this.picker = this.selectComponent( '.van-datetime-picker' )); const { setColumnValues } = picker; picker.setColumnValues = (...args) => setColumnValues.apply(picker, [...args, false]); } return this.picker; }, updateColumns() { const results = this.getRanges().map(({ type, range }, index) => { 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 this.asyncSet({ columns: results }); }, 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(): void { const { data } = this; let value; const picker = this.getPicker(); if (data.type === 'time') { const indexes = picker.getIndexes(); value = `${indexes[0] + data.minHour}:${indexes[1] + data.minMinute}`; } else { const values = picker.getValues(); 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.updateColumnValue(value).then(() => { this.$emit('input', value); this.$emit('change', picker); }); }, updateColumnValue(value) { let values = []; const { data } = this; const picker = this.getPicker(); if (data.type === 'time') { const pair = value.split(':'); values = [pair[0], pair[1]]; } else { const date = new Date(value); values = [`${date.getFullYear()}`, padZero(date.getMonth() + 1)]; if (data.type === 'date') { values.push(padZero(date.getDate())); } if (data.type === 'datetime') { values.push( padZero(date.getDate()), padZero(date.getHours()), padZero(date.getMinutes()) ); } } return this.asyncSet({ innerValue: value }) .then(() => this.updateColumns()) .then(() => { picker.setValues(values); }); } }, created() { const innerValue = this.correctValue(this.data.value); this.updateColumnValue(innerValue).then(() => { this.$emit('input', innerValue); }); } });