All files / src/components/Calendar/Shadcn calendar.tsx

100% Statements 144/144
90% Branches 27/30
100% Functions 6/6
100% Lines 144/144

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 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171    1x 1x           1x 1x   1x   1x 1x   46x 46x 46x 46x 46x 46x 46x 46x 46x 46x 46x     46x 46x 46x 46x 46x 46x 46x 46x 46x 46x 46x 46x 46x 24x 46x 46x 46x 46x 46x 46x 46x 46x 91x 91x 91x 91x 91x 91x 91x   91x 46x 186x 91x 91x   103x 91x 91x   4x 186x 46x 46x 12x 12x 12x 12x 12x 12x   12x 46x 46x 46x 46x   46x   3038x 3038x 3038x 3038x 3038x 3038x 3038x   3038x 3038x 1542x 3038x   3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x   3038x 3038x 3015x 2946x 2923x 2920x 2907x 2900x 2900x 2900x   3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x   3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x 3038x   3038x      
'use client';
 
import * as React from 'react';
import {
  type DayButton,
  DayPicker,
  getDefaultClassNames,
} from 'react-day-picker';
 
import { calendarDayButtonClasses } from '@/constants/calendarClassNames';
import { calendarVariants } from '@/constants/calendarVariants';
 
import { combineClasses } from '@shared/utils';
 
import { Button } from '@/components/Calendar/Shadcn/button';
import { ChevronIcon } from '@/components/icons/reacts';
 
function Calendar({
  className,
  classNames,
  showOutsideDays = true,
  captionLayout = 'label',
  buttonVariant = 'ghost',
  formatters,
  components,
  variant = 'default', // <-- define default
  ...props
}: React.ComponentProps<typeof DayPicker> & {
  buttonVariant?: React.ComponentProps<typeof Button>['variant'];
  variant?: keyof typeof import('@/constants/calendarVariants').calendarVariants;
}) {
  return (
    <DayPicker
      showOutsideDays={showOutsideDays}
      className={combineClasses(
        'bg-background group/calendar p-0 [--cell-size:2rem] [[data-slot=card-content]_&]:bg-transparent [[data-slot=popover-content]_&]:bg-transparent',
        String.raw`rtl:**:[.rdp-button\_next>svg]:rotate-180`,
        String.raw`rtl:**:[.rdp-button\_previous>svg]:rotate-180`,
        className,
      )}
      captionLayout={captionLayout}
      formatters={{
        formatMonthDropdown: (date) =>
          date.toLocaleString('default', { month: 'short' }),
        ...formatters,
      }}
      classNames={{
        ...calendarVariants[variant],
        ...classNames,
      }}
      components={{
        Root: ({ className, rootRef, ...props }) => {
          return (
            <div
              data-slot="calendar"
              ref={rootRef}
              className={combineClasses(className)}
              {...props}
            />
          );
        },
        Chevron: ({ className, orientation, ...props }) => {
          if (orientation === 'left') {
            return <ChevronIcon direction="left" />;
          }
 
          if (orientation === 'right') {
            return <ChevronIcon direction="right" />;
          }
 
          return <ChevronIcon direction="down" />;
        },
        DayButton: CalendarDayButton,
        WeekNumber: ({ children, ...props }) => {
          return (
            <td {...props}>
              <div className="flex size-(--cell-size) items-center justify-center text-center">
                {children}
              </div>
            </td>
          );
        },
        ...components,
      }}
      {...props}
    />
  );
}
 
function CalendarDayButton({
  className,
  day,
  modifiers,
  ...props
}: React.ComponentProps<typeof DayButton>) {
  const defaultClassNames = getDefaultClassNames();
 
  const ref = React.useRef<HTMLButtonElement>(null);
  React.useEffect(() => {
    if (modifiers.focused) ref.current?.focus();
  }, [modifiers.focused]);
 
  const {
    firstWeekStart = false,
    firstWeekMiddle = false,
    firstWeekEnd = false,
    secondWeekStart = false,
    secondWeekMiddle = false,
    secondWeekEnd = false,
    thirdWeekStart = false,
    thirdWeekMiddle = false,
    thirdWeekEnd = false,
    hoverWeekStart = false,
    hoverWeekMiddle = false,
    hoverWeekEnd = false,
    range_start,
    range_end,
    range_middle,
    today,
    selected,
  } = modifiers;
 
  const isInSelectedWeek =
    firstWeekStart ||
    firstWeekMiddle ||
    firstWeekEnd ||
    secondWeekStart ||
    secondWeekMiddle ||
    secondWeekEnd ||
    thirdWeekStart ||
    thirdWeekMiddle ||
    thirdWeekEnd;
 
  return (
    <Button
      data-testid="calendar-day-button"
      ref={ref}
      variant="ghost"
      size="icon"
      data-day={day.date.toLocaleDateString()}
      data-selected-single={
        selected && !range_start && !range_end && !range_middle
      }
      data-range-start={range_start}
      data-range-end={range_end}
      data-range-middle={range_middle}
      data-today={today}
      className={combineClasses(
        calendarDayButtonClasses,
        defaultClassNames.day,
        className,
      )}
      data-first-week-start={firstWeekStart}
      data-first-week-middle={firstWeekMiddle}
      data-first-week-end={firstWeekEnd}
      data-second-week-start={secondWeekStart}
      data-second-week-middle={secondWeekMiddle}
      data-second-week-end={secondWeekEnd}
      data-third-week-start={thirdWeekStart}
      data-third-week-middle={thirdWeekMiddle}
      data-third-week-end={thirdWeekEnd}
      data-hover-week-start={isInSelectedWeek ? undefined : hoverWeekStart}
      data-hover-week-middle={isInSelectedWeek ? undefined : hoverWeekMiddle}
      data-hover-week-end={isInSelectedWeek ? undefined : hoverWeekEnd}
      {...props}
    />
  );
}
 
export { Calendar, CalendarDayButton };