|
| 1 | +import React from 'react'; |
| 2 | + |
| 3 | +import type {DateTime} from '@gravity-ui/date-utils'; |
| 4 | +import {useControlledState} from '@gravity-ui/uikit'; |
| 5 | + |
| 6 | +import type {DateFieldState} from '../../DateField'; |
| 7 | +import type {DateFieldBase} from '../../types'; |
| 8 | +import {useDefaultTimeZone} from '../../utils/useDefaultTimeZone'; |
| 9 | + |
| 10 | +export interface DatePickerState<T = DateTime> { |
| 11 | + /** The currently selected date. */ |
| 12 | + value: T | null; |
| 13 | + /** Sets the selected date. */ |
| 14 | + setValue: (value: T | null) => void; |
| 15 | + /** |
| 16 | + * The date portion of the value. This may be set prior to `value` if the user has |
| 17 | + * selected a date but has not yet selected a time. |
| 18 | + */ |
| 19 | + dateValue: T | null; |
| 20 | + /** Sets the date portion of the value. */ |
| 21 | + setDateValue: (value: T) => void; |
| 22 | + /** |
| 23 | + * The time portion of the value. This may be set prior to `value` if the user has |
| 24 | + * selected a time but has not yet selected a date. |
| 25 | + */ |
| 26 | + timeValue: T | null; |
| 27 | + /** Sets the time portion of the value. */ |
| 28 | + setTimeValue: (value: T | null) => void; |
| 29 | + /** Whether the field is read only. */ |
| 30 | + readOnly?: boolean; |
| 31 | + /** Whether the field is disabled. */ |
| 32 | + disabled?: boolean; |
| 33 | + /** Format of the date when rendered in the input. */ |
| 34 | + format: string; |
| 35 | + /** Whether the date picker supports selecting a date. */ |
| 36 | + hasDate: boolean; |
| 37 | + /** Whether the date picker supports selecting a time. */ |
| 38 | + hasTime: boolean; |
| 39 | + /** Format of the time when rendered in the input. */ |
| 40 | + timeFormat?: string; |
| 41 | + timeZone: string; |
| 42 | + /** Whether the calendar popover is currently open. */ |
| 43 | + isOpen: boolean; |
| 44 | + /** Sets whether the calendar popover is open. */ |
| 45 | + setOpen: (isOpen: boolean) => void; |
| 46 | + dateFieldState: DateFieldState<T>; |
| 47 | +} |
| 48 | + |
| 49 | +export interface DatePickerStateFactoryOptions<T, O extends DateFieldBase<T>> { |
| 50 | + getPlaceholderTime: (placeholderValue: DateTime | undefined, timeZone?: string) => T; |
| 51 | + mergeDateTime: (date: T, time: T) => T; |
| 52 | + setTimezone: (date: T, timeZone: string) => T; |
| 53 | + getDateTime: (date: T | null | undefined) => DateTime | undefined; |
| 54 | + useDateFieldState: (props: O) => DateFieldState<T>; |
| 55 | +} |
| 56 | + |
| 57 | +export function datePickerStateFactory<T, O extends DateFieldBase<T>>({ |
| 58 | + getPlaceholderTime, |
| 59 | + mergeDateTime, |
| 60 | + setTimezone, |
| 61 | + getDateTime, |
| 62 | + useDateFieldState, |
| 63 | +}: DatePickerStateFactoryOptions<T, O>) { |
| 64 | + return function useDatePickerState(props: O): DatePickerState<T> { |
| 65 | + const {disabled, readOnly} = props; |
| 66 | + const [isOpen, setOpen] = React.useState(false); |
| 67 | + |
| 68 | + const [value, setValue] = useControlledState( |
| 69 | + props.value as never, |
| 70 | + props.defaultValue ?? null, |
| 71 | + props.onUpdate, |
| 72 | + ); |
| 73 | + const [selectedDateInner, setSelectedDate] = React.useState<T | null>(null); |
| 74 | + const [selectedTimeInner, setSelectedTime] = React.useState<T | null>(null); |
| 75 | + |
| 76 | + const inputTimeZone = useDefaultTimeZone( |
| 77 | + getDateTime(props.value) || getDateTime(props.defaultValue) || props.placeholderValue, |
| 78 | + ); |
| 79 | + const timeZone = props.timeZone || inputTimeZone; |
| 80 | + |
| 81 | + let selectedDate = selectedDateInner; |
| 82 | + let selectedTime = selectedTimeInner; |
| 83 | + |
| 84 | + const format = props.format || 'L'; |
| 85 | + |
| 86 | + const commitValue = (date: T, time: T) => { |
| 87 | + if (disabled || readOnly) { |
| 88 | + return; |
| 89 | + } |
| 90 | + |
| 91 | + setValue(setTimezone(mergeDateTime(date, time), inputTimeZone)); |
| 92 | + setSelectedDate(null); |
| 93 | + setSelectedTime(null); |
| 94 | + }; |
| 95 | + |
| 96 | + const dateFieldState = useDateFieldState({ |
| 97 | + ...props, |
| 98 | + value, |
| 99 | + onUpdate(date: T | null) { |
| 100 | + if (date) { |
| 101 | + commitValue(date, date); |
| 102 | + } else { |
| 103 | + setValue(null); |
| 104 | + } |
| 105 | + }, |
| 106 | + disabled, |
| 107 | + readOnly, |
| 108 | + validationState: props.validationState, |
| 109 | + minValue: props.minValue, |
| 110 | + maxValue: props.maxValue, |
| 111 | + isDateUnavailable: props.isDateUnavailable, |
| 112 | + format, |
| 113 | + placeholderValue: props.placeholderValue, |
| 114 | + timeZone, |
| 115 | + }); |
| 116 | + |
| 117 | + const timeFormat = React.useMemo(() => { |
| 118 | + const hasSeconds = dateFieldState.sections.some((s) => s.type === 'second'); |
| 119 | + return hasSeconds ? 'LTS' : 'LT'; |
| 120 | + }, [dateFieldState.sections]); |
| 121 | + |
| 122 | + if (value) { |
| 123 | + selectedDate = setTimezone(value, timeZone); |
| 124 | + if (dateFieldState.hasTime) { |
| 125 | + selectedTime = setTimezone(value, timeZone); |
| 126 | + } |
| 127 | + } |
| 128 | + |
| 129 | + // Intercept setValue to make sure the Time section is not changed by date selection in Calendar |
| 130 | + const selectDate = (newValue: T) => { |
| 131 | + if (disabled || readOnly) { |
| 132 | + return; |
| 133 | + } |
| 134 | + |
| 135 | + const shouldClose = !dateFieldState.hasTime; |
| 136 | + if (dateFieldState.hasTime) { |
| 137 | + if (selectedTime || shouldClose) { |
| 138 | + commitValue(newValue, selectedTime || newValue); |
| 139 | + } else { |
| 140 | + setSelectedDate(newValue); |
| 141 | + } |
| 142 | + } else { |
| 143 | + commitValue(newValue, newValue); |
| 144 | + } |
| 145 | + |
| 146 | + if (shouldClose) { |
| 147 | + setOpen(false); |
| 148 | + } |
| 149 | + }; |
| 150 | + |
| 151 | + const selectTime = (newValue: T | null) => { |
| 152 | + if (disabled || readOnly) { |
| 153 | + return; |
| 154 | + } |
| 155 | + |
| 156 | + const newTime = newValue ?? getPlaceholderTime(props.placeholderValue, timeZone); |
| 157 | + |
| 158 | + if (selectedDate) { |
| 159 | + commitValue(selectedDate, newTime); |
| 160 | + } else { |
| 161 | + setSelectedTime(newTime); |
| 162 | + } |
| 163 | + }; |
| 164 | + |
| 165 | + if (dateFieldState.hasTime && !selectedTime) { |
| 166 | + selectedTime = dateFieldState.displayValue; |
| 167 | + } |
| 168 | + |
| 169 | + return { |
| 170 | + value, |
| 171 | + setValue(newDate: T | null) { |
| 172 | + if (props.readOnly || props.disabled) { |
| 173 | + return; |
| 174 | + } |
| 175 | + |
| 176 | + if (newDate) { |
| 177 | + setValue(setTimezone(newDate, inputTimeZone)); |
| 178 | + } else { |
| 179 | + setValue(null); |
| 180 | + } |
| 181 | + }, |
| 182 | + dateValue: selectedDate, |
| 183 | + timeValue: selectedTime, |
| 184 | + setDateValue: selectDate, |
| 185 | + setTimeValue: selectTime, |
| 186 | + disabled, |
| 187 | + readOnly, |
| 188 | + format, |
| 189 | + hasDate: dateFieldState.hasDate, |
| 190 | + hasTime: dateFieldState.hasTime, |
| 191 | + timeFormat, |
| 192 | + timeZone, |
| 193 | + isOpen, |
| 194 | + setOpen(newIsOpen) { |
| 195 | + if (!newIsOpen && !value && selectedDate && dateFieldState.hasTime) { |
| 196 | + commitValue( |
| 197 | + selectedDate, |
| 198 | + selectedTime || getPlaceholderTime(props.placeholderValue, props.timeZone), |
| 199 | + ); |
| 200 | + } |
| 201 | + |
| 202 | + setOpen(newIsOpen); |
| 203 | + }, |
| 204 | + dateFieldState, |
| 205 | + }; |
| 206 | + }; |
| 207 | +} |
0 commit comments