import * as moment from 'moment'; import RangeCalendar from '../vc-calendar/src/RangeCalendar'; import VcDatePicker from '../vc-calendar/src/Picker'; import classNames from 'classnames'; import shallowequal from 'shallowequal'; import CloseCircleFilled from '@ant-design/icons-vue/CloseCircleFilled'; import Tag from '../tag'; import { ConfigConsumerProps } from '../config-provider'; import interopDefault from '../_util/interopDefault'; import { RangePickerProps } from './interface'; import { hasProp, getOptionProps, initDefaultProps, mergeProps, getComponentFromProp, getListeners, } from '../_util/props-util'; import BaseMixin from '../_util/BaseMixin'; import { formatDate } from './utils'; import InputIcon from './InputIcon'; function noop() {} function getShowDateFromValue(value, mode) { const [start, end] = value; // value could be an empty array, then we should not reset showDate if (!start && !end) { return; } if (mode && mode[0] === 'month') { return [start, end]; } const newEnd = end && end.isSame(start, 'month') ? end.clone().add(1, 'month') : end; return [start, newEnd]; } function pickerValueAdapter(value) { if (!value) { return; } if (Array.isArray(value)) { return value; } return [value, value.clone().add(1, 'month')]; } function isEmptyArray(arr) { if (Array.isArray(arr)) { return arr.length === 0 || arr.every(i => !i); } return false; } function fixLocale(value, localeCode) { if (!localeCode) { return; } if (!value || value.length === 0) { return; } const [start, end] = value; if (start) { start.locale(localeCode); } if (end) { end.locale(localeCode); } } export default { name: 'ARangePicker', mixins: [BaseMixin], model: { prop: 'value', event: 'change', }, props: initDefaultProps(RangePickerProps(), { allowClear: true, showToday: false, separator: '~', }), inject: { configProvider: { default: () => ConfigConsumerProps }, }, data() { const value = this.value || this.defaultValue || []; const [start, end] = value; if ( (start && !interopDefault(moment).isMoment(start)) || (end && !interopDefault(moment).isMoment(end)) ) { throw new Error( 'The value/defaultValue of RangePicker must be a moment object array after `antd@2.0`, ' + 'see: https://u.ant.design/date-picker-value', ); } const pickerValue = !value || isEmptyArray(value) ? this.defaultPickerValue : value; return { sValue: value, sShowDate: pickerValueAdapter(pickerValue || interopDefault(moment)()), sOpen: this.open, sHoverValue: [], }; }, watch: { value(val) { const value = val || []; let state = { sValue: value }; if (!shallowequal(val, this.sValue)) { state = { ...state, sShowDate: getShowDateFromValue(value, this.mode) || this.sShowDate, }; } this.setState(state); }, open(val) { const state = { sOpen: val }; this.setState(state); }, sOpen(val, oldVal) { this.$nextTick(() => { if (!hasProp(this, 'open') && oldVal && !val) { this.focus(); } }); }, }, methods: { setValue(value, hidePanel) { this.handleChange(value); if ((hidePanel || !this.showTime) && !hasProp(this, 'open')) { this.setState({ sOpen: false }); } }, clearSelection(e) { e.preventDefault(); e.stopPropagation(); this.setState({ sValue: [] }); this.handleChange([]); }, clearHoverValue() { this.setState({ sHoverValue: [] }); }, handleChange(value) { if (!hasProp(this, 'value')) { this.setState(({ sShowDate }) => ({ sValue: value, sShowDate: getShowDateFromValue(value) || sShowDate, })); } if (value[0] && value[1] && value[0].diff(value[1]) > 0) { value[1] = undefined; } const [start, end] = value; this.$emit('change', value, [formatDate(start, this.format), formatDate(end, this.format)]); }, handleOpenChange(open) { if (!hasProp(this, 'open')) { this.setState({ sOpen: open }); } if (open === false) { this.clearHoverValue(); } this.$emit('openChange', open); }, handleShowDateChange(showDate) { this.setState({ sShowDate: showDate }); }, handleHoverChange(hoverValue) { this.setState({ sHoverValue: hoverValue }); }, handleRangeMouseLeave() { if (this.sOpen) { this.clearHoverValue(); } }, handleCalendarInputSelect(value) { const [start] = value; if (!start) { return; } this.setState(({ sShowDate }) => ({ sValue: value, sShowDate: getShowDateFromValue(value) || sShowDate, })); }, handleRangeClick(value) { if (typeof value === 'function') { value = value(); } this.setValue(value, true); this.$emit('ok', value); this.$emit('openChange', false); }, onMouseEnter(e) { this.$emit('mouseenter', e); }, onMouseLeave(e) { this.$emit('mouseleave', e); }, focus() { this.$refs.picker.focus(); }, blur() { this.$refs.picker.blur(); }, renderFooter() { const { ranges, $scopedSlots, $slots } = this; const { _prefixCls: prefixCls, _tagPrefixCls: tagPrefixCls } = this; const renderExtraFooter = this.renderExtraFooter || $scopedSlots.renderExtraFooter || $slots.renderExtraFooter; if (!ranges && !renderExtraFooter) { return null; } const customFooter = renderExtraFooter ? (
) : null; const operations = ranges && Object.keys(ranges).map(range => { const value = ranges[range]; const hoverValue = typeof value === 'function' ? value.call(this) : value; return (