import * as React from 'react'; import * as moment from 'moment'; import { polyfill } from 'react-lifecycles-compat'; 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'; import warning from '../_util/warning'; import interopDefault from '../_util/interopDefault'; import getDataOrAriaProps from '../_util/getDataOrAriaProps'; export interface PickerProps { value?: moment.Moment; prefixCls: string; } export default function createPicker(TheCalendar: React.ComponentClass): any { class CalenderWrapper extends React.Component { static defaultProps = { prefixCls: 'ant-calendar', allowClear: true, showToday: true, }; static getDerivedStateFromProps(nextProps: PickerProps, prevState: any) { let state = null; if ('value' in nextProps) { state = { value: nextProps.value, }; if (nextProps.value !== prevState.value) { state = { ...state, showDate: nextProps.value, }; } } return state; } private input: any; constructor(props: any) { super(props); const value = props.value || props.defaultValue; if (value && !interopDefault(moment).isMoment(value)) { throw new Error( 'The value/defaultValue of DatePicker or MonthPicker must be ' + 'a moment object after `antd@2.0`, see: https://u.ant.design/date-picker-value', ); } this.state = { value, showDate: value, }; } renderFooter = (...args: any[]) => { const { prefixCls, renderExtraFooter } = this.props; return renderExtraFooter ? (
{renderExtraFooter(...args)}
) : null; } clearSelection = (e: React.MouseEvent) => { e.preventDefault(); e.stopPropagation(); this.handleChange(null); } handleChange = (value: moment.Moment | null) => { const props = this.props; if (!('value' in props)) { this.setState({ value, showDate: value, }); } props.onChange(value, (value && value.format(props.format)) || ''); } handleCalendarChange = (value: moment.Moment) => { this.setState({ showDate: value }); } focus() { this.input.focus(); } blur() { this.input.blur(); } saveInput = (node: any) => { this.input = node; } render() { const { value, showDate } = this.state; const props = omit(this.props, ['onChange']); const { prefixCls, locale, localeCode } = props; 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, }); if (value && localeCode) { value.locale(localeCode); } let pickerProps: Object = {}; let calendarProps: any = {}; if (props.showTime) { calendarProps = { // fix https://github.com/ant-design/ant-design/issues/1902 onSelect: this.handleChange, }; } else { pickerProps = { onChange: this.handleChange, }; } if ('mode' in props) { calendarProps.mode = props.mode; } warning(!('onOK' in props), 'It should be `DatePicker[onOk]` or `MonthPicker[onOk]`, instead of `onOK`!'); const calendar = ( ); const clearIcon = (!props.disabled && props.allowClear && value) ? ( ) : null; const dataOrAriaProps = getDataOrAriaProps(props); const input = ({ value: inputValue }: { value: moment.Moment | null }) => (
{clearIcon}
); return ( {input} ); } } polyfill(CalenderWrapper); return CalenderWrapper; }