File size: 2,183 Bytes
cf86710 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 | import type React from "react";
import type { DateLib } from "../classes/DateLib.js";
import { useControlledValue } from "../helpers/useControlledValue.js";
import type {
DayPickerProps,
Modifiers,
PropsRange,
Selection,
} from "../types/index.js";
import { addToRange, rangeContainsModifiers } from "../utils/index.js";
import { rangeIncludesDate } from "../utils/rangeIncludesDate.js";
/**
* Hook to manage range selection in the DayPicker component.
*
* @template T - The type of DayPicker props.
* @param props - The DayPicker props.
* @param dateLib - The date utility library instance.
* @returns An object containing the selected range, a function to select a
* range, and a function to check if a date is within the range.
*/
export function useRange<T extends DayPickerProps>(
props: T,
dateLib: DateLib,
): Selection<T> {
const {
disabled,
excludeDisabled,
selected: initiallySelected,
required,
onSelect,
} = props as PropsRange;
const [internallySelected, setSelected] = useControlledValue(
initiallySelected,
onSelect ? initiallySelected : undefined,
);
const selected = !onSelect ? internallySelected : initiallySelected;
const isSelected = (date: Date) =>
selected && rangeIncludesDate(selected, date, false, dateLib);
const select = (
triggerDate: Date,
modifiers: Modifiers,
e: React.MouseEvent | React.KeyboardEvent,
) => {
const { min, max } = props as PropsRange;
const newRange = triggerDate
? addToRange(triggerDate, selected, min, max, required, dateLib)
: undefined;
if (excludeDisabled && disabled && newRange?.from && newRange.to) {
if (
rangeContainsModifiers(
{ from: newRange.from, to: newRange.to },
disabled,
dateLib,
)
) {
// if a disabled days is found, the range is reset
newRange.from = triggerDate;
newRange.to = undefined;
}
}
if (!onSelect) {
setSelected(newRange);
}
onSelect?.(newRange, triggerDate, modifiers, e);
return newRange;
};
return {
selected,
select,
isSelected,
} as Selection<T>;
}
|