ant-design/components/time-picker/index.tsx
2019-12-25 16:23:14 +08:00

57 lines
1.6 KiB
TypeScript

import { Moment } from 'moment';
import * as React from 'react';
import DatePicker from '../date-picker';
import { PickerTimeProps, RangePickerTimeProps } from '../date-picker/generatePicker';
import warning from '../_util/warning';
import { Omit } from '../_util/type';
const { TimePicker: InternalTimePicker, RangePicker: InternalRangePicker } = DatePicker;
export interface TimeRangePickerProps extends RangePickerTimeProps<Moment> {}
const RangePicker = React.forwardRef<any, TimeRangePickerProps>((props, ref) => {
return <InternalRangePicker {...props} picker="time" mode={undefined} ref={ref} />;
});
export interface TimePickerProps extends Omit<PickerTimeProps<Moment>, 'picker'> {
addon?: () => React.ReactNode;
}
const TimePicker = React.forwardRef<any, TimePickerProps>(
({ addon, renderExtraFooter, ...restProps }, ref) => {
const internalRenderExtraFooter = React.useMemo(() => {
if (renderExtraFooter) {
return renderExtraFooter;
}
if (addon) {
warning(
false,
'TimePicker',
'`addon` is deprecated. Please use `renderExtraFooter` instead.',
);
return addon;
}
return undefined;
}, [addon, renderExtraFooter]);
return (
<InternalTimePicker
{...restProps}
mode={undefined}
ref={ref}
renderExtraFooter={internalRenderExtraFooter}
/>
);
},
);
TimePicker.displayName = 'TimePicker';
type MergedTimePicker = typeof TimePicker & {
RangePicker: typeof RangePicker;
};
(TimePicker as MergedTimePicker).RangePicker = RangePicker;
export default TimePicker as MergedTimePicker;