mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-16 01:41:02 +08:00
165 lines
5.5 KiB
TypeScript
165 lines
5.5 KiB
TypeScript
import type { GenerateConfig } from 'rc-picker/lib/generate/index';
|
|
import type { Locale as RcPickerLocale, PickerMode } from 'rc-picker/lib/interface';
|
|
import type { SharedTimeProps } from 'rc-picker/lib/panels/TimePanel';
|
|
import type {
|
|
PickerBaseProps as RCPickerBaseProps,
|
|
PickerDateProps as RCPickerDateProps,
|
|
PickerTimeProps as RCPickerTimeProps,
|
|
} from 'rc-picker/lib/Picker';
|
|
import type {
|
|
RangePickerBaseProps as RCRangePickerBaseProps,
|
|
RangePickerDateProps as RCRangePickerDateProps,
|
|
RangePickerTimeProps as RCRangePickerTimeProps,
|
|
} from 'rc-picker/lib/RangePicker';
|
|
import type { SizeType } from '../../config-provider/SizeContext';
|
|
import type { TimePickerLocale } from '../../time-picker';
|
|
import type { InputStatus } from '../../_util/statusUtils';
|
|
import PickerButton from '../PickerButton';
|
|
import generateRangePicker from './generateRangePicker';
|
|
import generateSinglePicker from './generateSinglePicker';
|
|
|
|
export const Components = { button: PickerButton };
|
|
|
|
function toArray<T>(list: T | T[]): T[] {
|
|
if (!list) {
|
|
return [];
|
|
}
|
|
return Array.isArray(list) ? list : [list];
|
|
}
|
|
|
|
export function getTimeProps<DateType, DisabledTime>(
|
|
props: { format?: string; picker?: PickerMode } & Omit<
|
|
SharedTimeProps<DateType>,
|
|
'disabledTime'
|
|
> & {
|
|
disabledTime?: DisabledTime;
|
|
},
|
|
) {
|
|
const { format, picker, showHour, showMinute, showSecond, use12Hours } = props;
|
|
|
|
const firstFormat = toArray(format)[0];
|
|
const showTimeObj = { ...props };
|
|
|
|
if (firstFormat && typeof firstFormat === 'string') {
|
|
if (!firstFormat.includes('s') && showSecond === undefined) {
|
|
showTimeObj.showSecond = false;
|
|
}
|
|
if (!firstFormat.includes('m') && showMinute === undefined) {
|
|
showTimeObj.showMinute = false;
|
|
}
|
|
if (!firstFormat.includes('H') && !firstFormat.includes('h') && showHour === undefined) {
|
|
showTimeObj.showHour = false;
|
|
}
|
|
|
|
if ((firstFormat.includes('a') || firstFormat.includes('A')) && use12Hours === undefined) {
|
|
showTimeObj.use12Hours = true;
|
|
}
|
|
}
|
|
|
|
if (picker === 'time') {
|
|
return showTimeObj;
|
|
}
|
|
|
|
if (typeof firstFormat === 'function') {
|
|
// format of showTime should use default when format is custom format function
|
|
delete showTimeObj.format;
|
|
}
|
|
|
|
return {
|
|
showTime: showTimeObj,
|
|
};
|
|
}
|
|
const DataPickerPlacements = ['bottomLeft', 'bottomRight', 'topLeft', 'topRight'] as const;
|
|
type DataPickerPlacement = typeof DataPickerPlacements[number];
|
|
|
|
const HourStep = [0.5, 1, 1.5, 2, 3, 4, 6, 8, 12] as const;
|
|
type THourStep = typeof HourStep[number];
|
|
|
|
type InjectDefaultProps<Props> = Omit<
|
|
Props,
|
|
'locale' | 'generateConfig' | 'hideHeader' | 'components' | 'hourStep'
|
|
> & {
|
|
locale?: PickerLocale;
|
|
size?: SizeType;
|
|
placement?: DataPickerPlacement;
|
|
bordered?: boolean;
|
|
status?: InputStatus;
|
|
hourStep?: THourStep;
|
|
};
|
|
|
|
export type PickerLocale = {
|
|
lang: RcPickerLocale & AdditionalPickerLocaleLangProps;
|
|
timePickerLocale: TimePickerLocale;
|
|
} & AdditionalPickerLocaleProps;
|
|
|
|
export type AdditionalPickerLocaleProps = {
|
|
dateFormat?: string;
|
|
dateTimeFormat?: string;
|
|
weekFormat?: string;
|
|
monthFormat?: string;
|
|
};
|
|
|
|
export type AdditionalPickerLocaleLangProps = {
|
|
placeholder: string;
|
|
yearPlaceholder?: string;
|
|
quarterPlaceholder?: string;
|
|
monthPlaceholder?: string;
|
|
weekPlaceholder?: string;
|
|
rangeYearPlaceholder?: [string, string];
|
|
rangeQuarterPlaceholder?: [string, string];
|
|
rangeMonthPlaceholder?: [string, string];
|
|
rangeWeekPlaceholder?: [string, string];
|
|
rangePlaceholder?: [string, string];
|
|
};
|
|
|
|
// Picker Props
|
|
export type PickerBaseProps<DateType> = InjectDefaultProps<RCPickerBaseProps<DateType>>;
|
|
export type PickerDateProps<DateType> = InjectDefaultProps<RCPickerDateProps<DateType>>;
|
|
export type PickerTimeProps<DateType> = InjectDefaultProps<RCPickerTimeProps<DateType>>;
|
|
|
|
export type PickerProps<DateType> =
|
|
| PickerBaseProps<DateType>
|
|
| PickerDateProps<DateType>
|
|
| PickerTimeProps<DateType>;
|
|
|
|
// Range Picker Props
|
|
export type RangePickerBaseProps<DateType> = InjectDefaultProps<RCRangePickerBaseProps<DateType>>;
|
|
export type RangePickerDateProps<DateType> = InjectDefaultProps<RCRangePickerDateProps<DateType>>;
|
|
export type RangePickerTimeProps<DateType> = InjectDefaultProps<RCRangePickerTimeProps<DateType>>;
|
|
|
|
export type RangePickerProps<DateType> =
|
|
| RangePickerBaseProps<DateType>
|
|
| RangePickerDateProps<DateType>
|
|
| RangePickerTimeProps<DateType>;
|
|
|
|
function generatePicker<DateType>(generateConfig: GenerateConfig<DateType>) {
|
|
// =========================== Picker ===========================
|
|
const { DatePicker, WeekPicker, MonthPicker, YearPicker, TimePicker, QuarterPicker } =
|
|
generateSinglePicker(generateConfig);
|
|
|
|
// ======================== Range Picker ========================
|
|
const RangePicker = generateRangePicker(generateConfig);
|
|
|
|
// =========================== Export ===========================
|
|
type MergedDatePickerType = typeof DatePicker & {
|
|
WeekPicker: typeof WeekPicker;
|
|
MonthPicker: typeof MonthPicker;
|
|
YearPicker: typeof YearPicker;
|
|
RangePicker: typeof RangePicker;
|
|
TimePicker: typeof TimePicker;
|
|
QuarterPicker: typeof QuarterPicker;
|
|
};
|
|
|
|
const MergedDatePicker = DatePicker as MergedDatePickerType;
|
|
MergedDatePicker.WeekPicker = WeekPicker;
|
|
MergedDatePicker.MonthPicker = MonthPicker;
|
|
MergedDatePicker.YearPicker = YearPicker;
|
|
MergedDatePicker.RangePicker = RangePicker;
|
|
MergedDatePicker.TimePicker = TimePicker;
|
|
MergedDatePicker.QuarterPicker = QuarterPicker;
|
|
|
|
return MergedDatePicker;
|
|
}
|
|
|
|
export default generatePicker;
|