All files / src/components/TimeSelection/hooks useTimeRanges.ts

97.43% Statements 38/39
88.88% Branches 32/36
100% Functions 4/4
97.36% Lines 37/38

Press n or j to go to the next uncovered block, b, p or k for the previous block.

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 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 1441x                             1x           1x 2100x 9x   2091x 5x   2086x     1x                   51x   51x 2096x 2096x   2096x 2096x 2096x   2096x 14x     2096x 1x       2096x             51x     1x               17x 17x 17x   17x 17x 17x                     17x                     17x                     17x 2x 2x   2x                       15x             17x      
import React from 'react';
 
import type {DateTime} from '@gravity-ui/date-utils';
 
import type {TimeSelectionView, WheelValue} from '../TimeSelection.types';
 
interface UseTimeRangesProps {
    ampm?: boolean;
    timeSteps?: Partial<Record<TimeSelectionView, number>>;
    isTimeDisabled?: (value: DateTime, view: TimeSelectionView) => boolean;
    minValue?: DateTime;
    maxValue?: DateTime;
    value?: DateTime;
}
 
const DEFAULT_TIME_STEPS: Record<TimeSelectionView, number> = {
    hours: 1,
    minutes: 1,
    seconds: 1,
};
 
const isWithinRange = (testValue: DateTime, minValue?: DateTime, maxValue?: DateTime): boolean => {
    if (minValue && testValue.isBefore(minValue)) {
        return false;
    }
    if (maxValue && testValue.isAfter(maxValue)) {
        return false;
    }
    return true;
};
 
const generateRange = (
    start: number,
    end: number,
    step: number,
    baseValue: DateTime | undefined,
    view: TimeSelectionView,
    isTimeDisabled?: (value: DateTime, view: TimeSelectionView) => boolean,
    minValue?: DateTime,
    maxValue?: DateTime,
): WheelValue[] => {
    const range: WheelValue[] = [];
 
    for (let i = start; i <= end; i += step) {
        const label = String(i).padStart(2, '0');
        let disabled = false;
 
        Eif (baseValue) {
            const unitKey = view === 'hours' ? 'hour' : view === 'minutes' ? 'minute' : 'second';
            const testValue = baseValue.set(unitKey, i);
 
            if (!isWithinRange(testValue, minValue, maxValue)) {
                disabled = true;
            }
 
            if (!disabled && isTimeDisabled?.(testValue, view)) {
                disabled = true;
            }
        }
 
        range.push({
            label,
            value: label,
            disabled,
        });
    }
 
    return range;
};
 
export const useTimeRanges = ({
    ampm = false,
    timeSteps = {},
    isTimeDisabled,
    minValue,
    maxValue,
    value,
}: UseTimeRangesProps) => {
    return React.useMemo(() => {
        const steps = {...DEFAULT_TIME_STEPS, ...timeSteps};
        const ranges: Record<string, WheelValue[]> = {};
 
        const hoursMax = ampm ? 12 : 23;
        const hoursStart = ampm ? 1 : 0;
        ranges.hours = generateRange(
            hoursStart,
            hoursMax,
            steps.hours,
            value,
            'hours',
            isTimeDisabled,
            minValue,
            maxValue,
        );
 
        ranges.minutes = generateRange(
            0,
            59,
            steps.minutes,
            value,
            'minutes',
            isTimeDisabled,
            minValue,
            maxValue,
        );
 
        ranges.seconds = generateRange(
            0,
            59,
            steps.seconds,
            value,
            'seconds',
            isTimeDisabled,
            minValue,
            maxValue,
        );
 
        if (ampm && value) {
            const amValue = value.hour() < 12 ? value : value.set('hour', value.hour() - 12);
            const pmValue = value.hour() >= 12 ? value : value.set('hour', value.hour() + 12);
 
            ranges.ampm = [
                {
                    label: 'AM',
                    value: 'AM',
                    disabled: !isWithinRange(amValue, minValue, maxValue),
                },
                {
                    label: 'PM',
                    value: 'PM',
                    disabled: !isWithinRange(pmValue, minValue, maxValue),
                },
            ];
        I} else if (ampm) {
            ranges.ampm = [
                {label: 'AM', value: 'AM', disabled: false},
                {label: 'PM', value: 'PM', disabled: false},
            ];
        }
 
        return ranges;
    }, [ampm, timeSteps, isTimeDisabled, minValue, maxValue, value]);
};