import * as React from 'react'; import TimePickerPanel from 'rc-time-picker/lib/Panel'; import classNames from 'classnames'; import LocaleReceiver from '../locale-provider/LocaleReceiver'; import { generateShowHourMinuteSecond } from '../time-picker'; declare const require: Function; function getColumns({ showHour, showMinute, showSecond, use12Hours }: any) { let column = 0; if (showHour) { column += 1; } if (showMinute) { column += 1; } if (showSecond) { column += 1; } if (use12Hours) { column += 1; } return column; } export default function wrapPicker(Picker: React.ComponentClass, defaultFormat?: string): any { return class PickerWrapper extends React.Component { static defaultProps = { format: defaultFormat || 'YYYY-MM-DD', transitionName: 'slide-up', popupStyle: {}, onChange() { }, onOk() { }, onOpenChange() { }, locale: {}, prefixCls: 'ant-calendar', inputPrefixCls: 'ant-input', }; handleOpenChange = (open: boolean) => { const { onOpenChange } = this.props; onOpenChange(open); } getDefaultLocale() { const locale = require('./locale/en_US'); return locale.default || locale; } renderPicker = (locale: any, localeCode: string) => { const props = this.props; const { prefixCls, inputPrefixCls } = props; const pickerClass = classNames(`${prefixCls}-picker`, { [`${prefixCls}-picker-${props.size}`]: !!props.size, }); const pickerInputClass = classNames(`${prefixCls}-picker-input`, inputPrefixCls, { [`${inputPrefixCls}-lg`]: props.size === 'large', [`${inputPrefixCls}-sm`]: props.size === 'small', [`${inputPrefixCls}-disabled`]: props.disabled, }); const timeFormat = (props.showTime && props.showTime.format) || 'HH:mm:ss'; const rcTimePickerProps = { ...generateShowHourMinuteSecond(timeFormat), format: timeFormat, use12Hours: (props.showTime && props.showTime.use12Hours), }; const columns = getColumns(rcTimePickerProps); const timePickerCls = `${prefixCls}-time-picker-column-${columns}`; const timePicker = props.showTime ? ( ) : null; return ( ); } render() { return ( {this.renderPicker} ); } }; }