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 ( this.handleRangeClick(value)} onMouseenter={() => this.setState({ sHoverValue: hoverValue })} onMouseleave={this.handleRangeMouseLeave} > {range} ); }); const rangeNode = operations && operations.length > 0 ? ( ) : null; return [rangeNode, customFooter]; }, }, render() { const props = getOptionProps(this); let suffixIcon = getComponentFromProp(this, 'suffixIcon'); suffixIcon = Array.isArray(suffixIcon) ? suffixIcon[0] : suffixIcon; const { sValue: value, sShowDate: showDate, sHoverValue: hoverValue, sOpen: open, $scopedSlots, } = this; const listeners = getListeners(this); const { calendarChange = noop, ok = noop, focus = noop, blur = noop, panelChange = noop, } = listeners; const { prefixCls: customizePrefixCls, tagPrefixCls: customizeTagPrefixCls, popupStyle, disabledDate, disabledTime, showTime, showToday, ranges, locale, localeCode, format, separator, inputReadOnly, } = props; const getPrefixCls = this.configProvider.getPrefixCls; const prefixCls = getPrefixCls('calendar', customizePrefixCls); const tagPrefixCls = getPrefixCls('tag', customizeTagPrefixCls); this._prefixCls = prefixCls; this._tagPrefixCls = tagPrefixCls; const dateRender = props.dateRender || $scopedSlots.dateRender; fixLocale(value, localeCode); fixLocale(showDate, localeCode); const calendarClassName = classNames({ [`${prefixCls}-time`]: showTime, [`${prefixCls}-range-with-ranges`]: ranges, }); // 需要选择时间时,点击 ok 时才触发 onChange const pickerChangeHandler = { on: { change: this.handleChange, }, }; let calendarProps = { on: { ok: this.handleChange, }, props: {}, }; if (props.timePicker) { pickerChangeHandler.on.change = changedValue => this.handleChange(changedValue); } else { calendarProps = { on: {}, props: {} }; } if ('mode' in props) { calendarProps.props.mode = props.mode; } const startPlaceholder = Array.isArray(props.placeholder) ? props.placeholder[0] : locale.lang.rangePlaceholder[0]; const endPlaceholder = Array.isArray(props.placeholder) ? props.placeholder[1] : locale.lang.rangePlaceholder[1]; const rangeCalendarProps = mergeProps(calendarProps, { props: { separator, format, prefixCls, renderFooter: this.renderFooter, timePicker: props.timePicker, disabledDate, disabledTime, dateInputPlaceholder: [startPlaceholder, endPlaceholder], locale: locale.lang, dateRender, value: showDate, hoverValue, showToday, inputReadOnly, }, on: { change: calendarChange, ok, valueChange: this.handleShowDateChange, hoverChange: this.handleHoverChange, panelChange, inputSelect: this.handleCalendarInputSelect, }, class: calendarClassName, scopedSlots: $scopedSlots, }); const calendar = ; // default width for showTime const pickerStyle = {}; if (props.showTime) { pickerStyle.width = '350px'; } const [startValue, endValue] = value; const clearIcon = !props.disabled && props.allowClear && value && (startValue || endValue) ? ( ) : null; const inputIcon = ; const input = ({ value: inputValue }) => { const [start, end] = inputValue; return ( {separator} {clearIcon} {inputIcon} ); }; const vcDatePickerProps = mergeProps( { props, on: listeners, }, pickerChangeHandler, { props: { calendar, value, open, prefixCls: `${prefixCls}-picker-container`, }, on: { openChange: this.handleOpenChange, }, style: popupStyle, scopedSlots: { default: input, ...$scopedSlots }, }, ); return ( ); }, };