/* tslint:disable jsx-no-multiline-js */ import * as React from 'react'; import * as moment from 'moment'; import RangeCalendar from 'rc-calendar/lib/RangeCalendar'; import RcDatePicker from 'rc-calendar/lib/Picker'; import classNames from 'classnames'; import Icon from '../icon'; import warning from '../_util/warning'; import callMoment from '../_util/callMoment'; import { RangePickerValue, RangePickerPresetRange } from './interface'; export interface RangePickerState { value?: RangePickerValue; showDate?: RangePickerValue; open?: boolean; hoverValue?: RangePickerValue; } function getShowDateFromValue(value: RangePickerValue) { const [start, end] = value; // value could be an empty array, then we should not reset showDate if (!start && !end) { return; } const newEnd = end && end.isSame(start, 'month') ? end.clone().add(1, 'month') : end; return [start, newEnd] as RangePickerValue; } function formatValue(value: moment.Moment | undefined, format: string): string { return (value && value.format(format)) || ''; } function pickerValueAdapter(value?: moment.Moment | RangePickerValue): RangePickerValue | undefined { if (!value) { return; } if (Array.isArray(value)) { return value; } return [value, value.clone().add(1, 'month')]; } function isEmptyArray(arr: any) { if (Array.isArray(arr)) { return arr.length === 0 || arr.every(i => !i); } return false; } function fixLocale(value: RangePickerValue | undefined, localeCode: string) { if (!localeCode) { return; } if (!value || value.length === 0) { return; } if (value[0]) { value[0]!.locale(localeCode); } if (value[1]) { value[1]!.locale(localeCode); } } export default class RangePicker extends React.Component { static defaultProps = { prefixCls: 'ant-calendar', allowClear: true, showToday: false, }; private picker: HTMLSpanElement; constructor(props: any) { super(props); const value = props.value || props.defaultValue || []; if ( value[0] && !moment.isMoment(value[0]) || value[1] && !moment.isMoment(value[1]) ) { 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) ? props.defaultPickerValue : value; this.state = { value, showDate: pickerValueAdapter(pickerValue || callMoment(moment)), open: props.open, hoverValue: [], }; } componentWillReceiveProps(nextProps: any) { if ('value' in nextProps) { const state = this.state; const value = nextProps.value || []; this.setState({ value, showDate: getShowDateFromValue(value) || state.showDate, }); } if ('open' in nextProps) { this.setState({ open: nextProps.open, }); } } clearSelection = (e: React.MouseEvent) => { e.preventDefault(); e.stopPropagation(); this.setState({ value: [] }); this.handleChange([]); } clearHoverValue = () => this.setState({ hoverValue: [] }); handleChange = (value: RangePickerValue) => { const props = this.props; if (!('value' in props)) { this.setState(({ showDate }) => ({ value, showDate: getShowDateFromValue(value) || showDate, })); } props.onChange(value, [ formatValue(value[0], props.format), formatValue(value[1], props.format), ]); } handleOpenChange = (open: boolean) => { if (!('open' in this.props)) { this.setState({ open }); } if (open === false) { this.clearHoverValue(); } const { onOpenChange } = this.props; if (onOpenChange) { onOpenChange(open); } } handleShowDateChange = (showDate: RangePickerValue) => this.setState({ showDate }); handleHoverChange = (hoverValue: any) => this.setState({ hoverValue }); handleRangeMouseLeave = () => { if (this.state.open) { this.clearHoverValue(); } } handleCalendarInputSelect = (value: RangePickerValue) => { if (!value[0]) { return; } this.setState(({ showDate }) => ({ value, showDate: getShowDateFromValue(value) || showDate, })); } handleRangeClick = (value: RangePickerPresetRange) => { if (typeof value === 'function') { value = value(); } this.setValue(value, true); const { onOk } = this.props; if (onOk) { onOk(value); } } setValue(value: RangePickerValue, hidePanel?: boolean) { this.handleChange(value); if ((hidePanel || !this.props.showTime) && !('open' in this.props)) { this.setState({ open: false }); } } focus() { this.picker.focus(); } blur() { this.picker.blur(); } savePicker = (node: HTMLSpanElement) => { this.picker = node; } renderFooter = (...args: any[]) => { const { prefixCls, ranges, renderExtraFooter } = this.props; if (!ranges && !renderExtraFooter) { return null; } const customFooter = renderExtraFooter ? (
{renderExtraFooter(...args)}
) : null; const operations = Object.keys(ranges || {}).map((range) => { const value = ranges[range]; return ( this.handleRangeClick(value)} onMouseEnter={() => this.setState({ hoverValue: value })} onMouseLeave={this.handleRangeMouseLeave} > {range} ); }); const rangeNode = (
{operations}
); return [rangeNode, customFooter]; } render() { const { state, props } = this; const { value, showDate, hoverValue, open } = state; const { prefixCls, popupStyle, style, disabledDate, disabledTime, showTime, showToday, ranges, onOk, locale, localeCode, format, dateRender, onCalendarChange, } = props; fixLocale(value, localeCode); fixLocale(showDate, localeCode); warning(!('onOK' in props), 'It should be `RangePicker[onOk]`, instead of `onOK`!'); const calendarClassName = classNames({ [`${prefixCls}-time`]: showTime, [`${prefixCls}-range-with-ranges`]: ranges, }); // 需要选择时间时,点击 ok 时才触发 onChange let pickerChangeHandler = { onChange: this.handleChange, }; let calendarProps: any = { onOk: this.handleChange, }; if (props.timePicker) { pickerChangeHandler.onChange = changedValue => this.handleChange(changedValue); } else { calendarProps = {}; } if ('mode' in props) { calendarProps.mode = props.mode; } const startPlaceholder = ('placeholder' in props) ? props.placeholder[0] : locale.lang.rangePlaceholder[0]; const endPlaceholder = ('placeholder' in props) ? props.placeholder[1] : locale.lang.rangePlaceholder[1]; const calendar = ( ); // default width for showTime const pickerStyle = {} as any; if (props.showTime) { pickerStyle.width = (style && style.width) || 350; } const clearIcon = (!props.disabled && props.allowClear && value && (value[0] || value[1])) ? ( ) : null; const input = ({ value: inputValue }: { value: any }) => { const start = inputValue[0]; const end = inputValue[1]; return ( ~ {clearIcon} ); }; return ( {input} ); } }