import React from 'react'; import moment from 'moment'; import MonthCalendar from 'rc-calendar/lib/MonthCalendar'; import RcDatePicker from 'rc-calendar/lib/Picker'; import classNames from 'classnames'; import omit from 'omit.js'; import Icon from '../icon'; export interface PickerProps { value?: moment.Moment; prefixCls: string; } export default function createPicker(TheCalendar) { // use class typescript error const CalenderWrapper = React.createClass({ getDefaultProps() { return { prefixCls: 'ant-calendar', allowClear: true, showToday: true, }; }, getInitialState() { const props = this.props; return { value: props.value || props.defaultValue, tempValue: undefined, }; }, componentWillReceiveProps(nextProps: PickerProps) { if (nextProps.value !== undefined) { this.setState({ value: nextProps.value, tempValue: nextProps.value, }); } }, clearSelection(e) { e.preventDefault(); e.stopPropagation(); this.setState({ value: null }); this.handleChange(null); }, handleChange(value) { const props = this.props; if (props.value === undefined) { this.setState({ value }); } props.onChange(value, (value && value.format(props.format)) || ''); }, handleTempChange(tempValue) { this.setState({ tempValue }); }, // Clear temp value and trigger onChange when hide DatePicker[showTime] panel handleOpenChange(open) { const { showTime, onOpenChange, onChange, format } = this.props; if (!open) { // tricky code to avoid triggering onChange multiple times // when click `Now` button let tempValue; this.setState(prevState => { tempValue = prevState.tempValue; const nextState = { tempValue: undefined } as any; if (showTime && tempValue) { nextState.value = tempValue; onChange(tempValue, (tempValue && tempValue.format(format)) || ''); } return nextState; }); } if (onOpenChange) { onOpenChange(open); } }, render() { const props = omit(this.props, ['onChange']); const prefixCls = props.prefixCls; const locale = props.locale; const placeholder = ('placeholder' in props) ? props.placeholder : locale.lang.placeholder; const disabledTime = props.showTime ? props.disabledTime : null; const calendarClassName = classNames({ [`${prefixCls}-time`]: props.showTime, [`${prefixCls}-month`]: MonthCalendar === TheCalendar, }); // 需要选择时间时,点击 ok 时才触发 onChange let pickerChangeHandler: Object = {}; let calendarHandler: Object = {}; if (props.showTime) { calendarHandler = { // fix https://github.com/ant-design/ant-design/issues/1902 onSelect: (value) => { this.handleTempChange(value); }, }; } else { pickerChangeHandler = { onChange: this.handleChange, }; } const calendar = ( ); const clearIcon = (!props.disabled && props.allowClear && this.state.value) ? : null; return ( { ({ value }) => { return ( {clearIcon} ); } } ); }, }); return CalenderWrapper; }