ant-design/components/time-picker/index.tsx
龙风 1cec18e8eb
fix: component.displayName To distinguish the environment (#36126)
Co-authored-by: afc163 <afc163@gmail.com>
2022-06-21 10:24:52 +08:00

75 lines
2.0 KiB
TypeScript

import type { Moment } from 'moment';
import * as React from 'react';
import DatePicker from '../date-picker';
import type { PickerTimeProps, RangePickerTimeProps } from '../date-picker/generatePicker';
import type { InputStatus } from '../_util/statusUtils';
import warning from '../_util/warning';
const { TimePicker: InternalTimePicker, RangePicker: InternalRangePicker } = DatePicker;
export interface TimePickerLocale {
placeholder?: string;
rangePlaceholder?: [string, string];
}
export interface TimeRangePickerProps extends Omit<RangePickerTimeProps<Moment>, 'picker'> {
popupClassName?: string;
}
const RangePicker = React.forwardRef<any, TimeRangePickerProps>((props, ref) => (
<InternalRangePicker
{...props}
dropdownClassName={props.popupClassName}
picker="time"
mode={undefined}
ref={ref}
/>
));
export interface TimePickerProps extends Omit<PickerTimeProps<Moment>, 'picker'> {
addon?: () => React.ReactNode;
popupClassName?: string;
status?: InputStatus;
}
const TimePicker = React.forwardRef<any, TimePickerProps>(
({ addon, renderExtraFooter, popupClassName, ...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}
dropdownClassName={popupClassName}
mode={undefined}
ref={ref}
renderExtraFooter={internalRenderExtraFooter}
/>
);
},
);
if (process.env.NODE_ENV !== 'production') {
TimePicker.displayName = 'TimePicker';
}
type MergedTimePicker = typeof TimePicker & {
RangePicker: typeof RangePicker;
};
(TimePicker as MergedTimePicker).RangePicker = RangePicker;
export default TimePicker as MergedTimePicker;