You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ant-design-vue/components/vc-picker/panels/YearPanel/YearBody.tsx

79 lines
2.5 KiB

import type { GenerateConfig } from '../../generate';
import { YEAR_DECADE_COUNT } from '.';
import type { Locale, NullableDateType } from '../../interface';
import useCellClassName from '../../hooks/useCellClassName';
import { formatValue, isSameYear } from '../../utils/dateUtil';
import { useInjectRange } from '../../RangeContext';
import PanelBody from '../PanelBody';
import useMergeProps from '../../hooks/useMergeProps';
export const YEAR_COL_COUNT = 3;
const YEAR_ROW_COUNT = 4;
export type YearBodyProps<DateType> = {
prefixCls: string;
locale: Locale;
generateConfig: GenerateConfig<DateType>;
value?: NullableDateType<DateType>;
viewDate: DateType;
disabledDate?: (date: DateType) => boolean;
onSelect: (value: DateType) => void;
};
function YearBody<DateType>(_props: YearBodyProps<DateType>) {
const props = useMergeProps(_props);
const { prefixCls, value, viewDate, locale, generateConfig } = props;
const { rangedValue, hoverRangedValue } = useInjectRange();
const yearPrefixCls = `${prefixCls}-cell`;
// =============================== Year ===============================
const yearNumber = generateConfig.getYear(viewDate);
const startYear = Math.floor(yearNumber / YEAR_DECADE_COUNT) * YEAR_DECADE_COUNT;
const endYear = startYear + YEAR_DECADE_COUNT - 1;
const baseYear = generateConfig.setYear(
viewDate,
startYear - Math.ceil((YEAR_COL_COUNT * YEAR_ROW_COUNT - YEAR_DECADE_COUNT) / 2),
);
const isInView = (date: DateType) => {
const currentYearNumber = generateConfig.getYear(date);
return startYear <= currentYearNumber && currentYearNumber <= endYear;
};
const getCellClassName = useCellClassName<DateType>({
cellPrefixCls: yearPrefixCls,
value,
generateConfig,
rangedValue: rangedValue.value,
hoverRangedValue: hoverRangedValue.value,
isSameCell: (current, target) => isSameYear(generateConfig, current, target),
isInView,
offsetCell: (date, offset) => generateConfig.addYear(date, offset),
});
return (
<PanelBody
{...props}
rowNum={YEAR_ROW_COUNT}
colNum={YEAR_COL_COUNT}
baseDate={baseYear}
getCellText={generateConfig.getYear}
getCellClassName={getCellClassName}
getCellDate={generateConfig.addYear}
titleCell={date =>
formatValue(date, {
locale,
format: 'YYYY',
generateConfig,
})
}
/>
);
}
YearBody.displayName = 'YearBody';
YearBody.inheritAttrs = false;
export default YearBody;