98 lines
3.5 KiB
TypeScript
98 lines
3.5 KiB
TypeScript
import { isInRange } from '../utils/dateUtil';
|
|
import type { GenerateConfig } from '../generate';
|
|
import type { RangeValue, NullableDateType } from '../interface';
|
|
import { getValue } from '../utils/miscUtil';
|
|
|
|
export default function useCellClassName<DateType>({
|
|
cellPrefixCls,
|
|
generateConfig,
|
|
rangedValue,
|
|
hoverRangedValue,
|
|
isInView,
|
|
isSameCell,
|
|
offsetCell,
|
|
today,
|
|
value,
|
|
}: {
|
|
cellPrefixCls: string;
|
|
generateConfig: GenerateConfig<DateType>;
|
|
isSameCell: (current: NullableDateType<DateType>, target: NullableDateType<DateType>) => boolean;
|
|
offsetCell: (date: DateType, offset: number) => DateType;
|
|
isInView: (date: DateType) => boolean;
|
|
rangedValue?: RangeValue<DateType>;
|
|
hoverRangedValue?: RangeValue<DateType>;
|
|
today?: NullableDateType<DateType>;
|
|
value?: NullableDateType<DateType>;
|
|
}) {
|
|
function getClassName(currentDate: DateType) {
|
|
const prevDate = offsetCell(currentDate, -1);
|
|
const nextDate = offsetCell(currentDate, 1);
|
|
|
|
const rangeStart = getValue(rangedValue, 0);
|
|
const rangeEnd = getValue(rangedValue, 1);
|
|
|
|
const hoverStart = getValue(hoverRangedValue, 0);
|
|
const hoverEnd = getValue(hoverRangedValue, 1);
|
|
|
|
const isRangeHovered = isInRange(generateConfig, hoverStart, hoverEnd, currentDate);
|
|
|
|
function isRangeStart(date: DateType) {
|
|
return isSameCell(rangeStart, date);
|
|
}
|
|
function isRangeEnd(date: DateType) {
|
|
return isSameCell(rangeEnd, date);
|
|
}
|
|
const isHoverStart = isSameCell(hoverStart, currentDate);
|
|
const isHoverEnd = isSameCell(hoverEnd, currentDate);
|
|
|
|
const isHoverEdgeStart =
|
|
(isRangeHovered || isHoverEnd) && (!isInView(prevDate) || isRangeEnd(prevDate));
|
|
const isHoverEdgeEnd =
|
|
(isRangeHovered || isHoverStart) && (!isInView(nextDate) || isRangeStart(nextDate));
|
|
|
|
return {
|
|
// In view
|
|
[`${cellPrefixCls}-in-view`]: isInView(currentDate),
|
|
|
|
// Range
|
|
[`${cellPrefixCls}-in-range`]: isInRange<DateType>(
|
|
generateConfig,
|
|
rangeStart,
|
|
rangeEnd,
|
|
currentDate,
|
|
),
|
|
[`${cellPrefixCls}-range-start`]: isRangeStart(currentDate),
|
|
[`${cellPrefixCls}-range-end`]: isRangeEnd(currentDate),
|
|
[`${cellPrefixCls}-range-start-single`]: isRangeStart(currentDate) && !rangeEnd,
|
|
[`${cellPrefixCls}-range-end-single`]: isRangeEnd(currentDate) && !rangeStart,
|
|
[`${cellPrefixCls}-range-start-near-hover`]:
|
|
isRangeStart(currentDate) &&
|
|
(isSameCell(prevDate, hoverStart) ||
|
|
isInRange(generateConfig, hoverStart, hoverEnd, prevDate)),
|
|
[`${cellPrefixCls}-range-end-near-hover`]:
|
|
isRangeEnd(currentDate) &&
|
|
(isSameCell(nextDate, hoverEnd) ||
|
|
isInRange(generateConfig, hoverStart, hoverEnd, nextDate)),
|
|
|
|
// Range Hover
|
|
[`${cellPrefixCls}-range-hover`]: isRangeHovered,
|
|
[`${cellPrefixCls}-range-hover-start`]: isHoverStart,
|
|
[`${cellPrefixCls}-range-hover-end`]: isHoverEnd,
|
|
|
|
// Range Edge
|
|
[`${cellPrefixCls}-range-hover-edge-start`]: isHoverEdgeStart,
|
|
[`${cellPrefixCls}-range-hover-edge-end`]: isHoverEdgeEnd,
|
|
[`${cellPrefixCls}-range-hover-edge-start-near-range`]:
|
|
isHoverEdgeStart && isSameCell(prevDate, rangeEnd),
|
|
[`${cellPrefixCls}-range-hover-edge-end-near-range`]:
|
|
isHoverEdgeEnd && isSameCell(nextDate, rangeStart),
|
|
|
|
// Others
|
|
[`${cellPrefixCls}-today`]: isSameCell(today, currentDate),
|
|
[`${cellPrefixCls}-selected`]: isSameCell(value, currentDate),
|
|
};
|
|
}
|
|
|
|
return getClassName;
|
|
}
|