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 assign from 'object-assign'; 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; const value = props.value || props.defaultValue; if (value && !moment.isMoment(value)) { throw new Error( 'The value/defaultValue of DatePicker or MonthPicker must be ' + 'a moment object after `antd@2.0`, see: http://u.ant.design/date-picker-value' ); } return { value, tempValue: undefined, }; }, componentWillReceiveProps(nextProps: PickerProps) { if ('value' in nextProps) { 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 (!('value' in props)) { 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 = ( ); // default width for showTime const pickerStyle = {} as any; if (props.showTime) { pickerStyle.minWidth = 154; } const clearIcon = (!props.disabled && props.allowClear && this.state.value) ? : null; const input = ({ value }) => ( {clearIcon} ); return ( {input} ); }, }); return CalenderWrapper; }