127 lines
4.5 KiB
JavaScript
127 lines
4.5 KiB
JavaScript
import _extends from "@babel/runtime/helpers/esm/extends";
|
|
import * as React from 'react';
|
|
import useEventCallback from '@mui/utils/useEventCallback';
|
|
import { useIsDateDisabled } from './useIsDateDisabled';
|
|
import { useUtils } from '../internals/hooks/useUtils';
|
|
import { singleItemValueManager } from '../internals/utils/valueManagers';
|
|
import { SECTION_TYPE_GRANULARITY } from '../internals/utils/getDefaultReferenceDate';
|
|
export const createCalendarStateReducer = (reduceAnimations, disableSwitchToMonthOnDayFocus, utils) => (state, action) => {
|
|
switch (action.type) {
|
|
case 'changeMonth':
|
|
return _extends({}, state, {
|
|
slideDirection: action.direction,
|
|
currentMonth: action.newMonth,
|
|
isMonthSwitchingAnimating: !reduceAnimations
|
|
});
|
|
case 'finishMonthSwitchingAnimation':
|
|
return _extends({}, state, {
|
|
isMonthSwitchingAnimating: false
|
|
});
|
|
case 'changeFocusedDay':
|
|
{
|
|
if (state.focusedDay != null && action.focusedDay != null && utils.isSameDay(action.focusedDay, state.focusedDay)) {
|
|
return state;
|
|
}
|
|
const needMonthSwitch = action.focusedDay != null && !disableSwitchToMonthOnDayFocus && !utils.isSameMonth(state.currentMonth, action.focusedDay);
|
|
return _extends({}, state, {
|
|
focusedDay: action.focusedDay,
|
|
isMonthSwitchingAnimating: needMonthSwitch && !reduceAnimations && !action.withoutMonthSwitchingAnimation,
|
|
currentMonth: needMonthSwitch ? utils.startOfMonth(action.focusedDay) : state.currentMonth,
|
|
slideDirection: action.focusedDay != null && utils.isAfterDay(action.focusedDay, state.currentMonth) ? 'left' : 'right'
|
|
});
|
|
}
|
|
default:
|
|
throw new Error('missing support');
|
|
}
|
|
};
|
|
export const useCalendarState = params => {
|
|
const {
|
|
value,
|
|
referenceDate: referenceDateProp,
|
|
defaultCalendarMonth,
|
|
disableFuture,
|
|
disablePast,
|
|
disableSwitchToMonthOnDayFocus = false,
|
|
maxDate,
|
|
minDate,
|
|
onMonthChange,
|
|
reduceAnimations,
|
|
shouldDisableDate,
|
|
timezone
|
|
} = params;
|
|
const utils = useUtils();
|
|
const reducerFn = React.useRef(createCalendarStateReducer(Boolean(reduceAnimations), disableSwitchToMonthOnDayFocus, utils)).current;
|
|
const referenceDate = React.useMemo(() => {
|
|
let externalReferenceDate = null;
|
|
if (referenceDateProp) {
|
|
externalReferenceDate = referenceDateProp;
|
|
} else if (defaultCalendarMonth) {
|
|
// For `defaultCalendarMonth`, we just want to keep the month and the year to avoid a behavior change.
|
|
externalReferenceDate = utils.startOfMonth(defaultCalendarMonth);
|
|
}
|
|
return singleItemValueManager.getInitialReferenceValue({
|
|
value,
|
|
utils,
|
|
timezone,
|
|
props: params,
|
|
referenceDate: externalReferenceDate,
|
|
granularity: SECTION_TYPE_GRANULARITY.day
|
|
});
|
|
}, [] // eslint-disable-line react-hooks/exhaustive-deps
|
|
);
|
|
const [calendarState, dispatch] = React.useReducer(reducerFn, {
|
|
isMonthSwitchingAnimating: false,
|
|
focusedDay: referenceDate,
|
|
currentMonth: utils.startOfMonth(referenceDate),
|
|
slideDirection: 'left'
|
|
});
|
|
const handleChangeMonth = React.useCallback(payload => {
|
|
dispatch(_extends({
|
|
type: 'changeMonth'
|
|
}, payload));
|
|
if (onMonthChange) {
|
|
onMonthChange(payload.newMonth);
|
|
}
|
|
}, [onMonthChange]);
|
|
const changeMonth = React.useCallback(newDate => {
|
|
const newDateRequested = newDate;
|
|
if (utils.isSameMonth(newDateRequested, calendarState.currentMonth)) {
|
|
return;
|
|
}
|
|
handleChangeMonth({
|
|
newMonth: utils.startOfMonth(newDateRequested),
|
|
direction: utils.isAfterDay(newDateRequested, calendarState.currentMonth) ? 'left' : 'right'
|
|
});
|
|
}, [calendarState.currentMonth, handleChangeMonth, utils]);
|
|
const isDateDisabled = useIsDateDisabled({
|
|
shouldDisableDate,
|
|
minDate,
|
|
maxDate,
|
|
disableFuture,
|
|
disablePast,
|
|
timezone
|
|
});
|
|
const onMonthSwitchingAnimationEnd = React.useCallback(() => {
|
|
dispatch({
|
|
type: 'finishMonthSwitchingAnimation'
|
|
});
|
|
}, []);
|
|
const changeFocusedDay = useEventCallback((newFocusedDate, withoutMonthSwitchingAnimation) => {
|
|
if (!isDateDisabled(newFocusedDate)) {
|
|
dispatch({
|
|
type: 'changeFocusedDay',
|
|
focusedDay: newFocusedDate,
|
|
withoutMonthSwitchingAnimation
|
|
});
|
|
}
|
|
});
|
|
return {
|
|
referenceDate,
|
|
calendarState,
|
|
changeMonth,
|
|
changeFocusedDay,
|
|
isDateDisabled,
|
|
onMonthSwitchingAnimationEnd,
|
|
handleChangeMonth
|
|
};
|
|
}; |