"use client"

import * as React from "react"
import {
  DayPicker,
  getDefaultClassNames,
  type DayButton,
  type DropdownProps,
} from "react-day-picker"

import { cn } from "../../lib/utils"
import { Button } from "./button"
import {
  Select,
  SelectContent,
  SelectItem,
  SelectTrigger,
  SelectValue,
} from "./select"

function CalendarDropdown({
  options,
  value,
  onChange,
  disabled,
  "aria-label": ariaLabel,
}: DropdownProps) {
  const stringValue = value === undefined ? "" : String(value)

  return (
    <Select
      value={stringValue}
      onValueChange={(next) => {
        onChange?.({
          target: { value: next },
        } as React.ChangeEvent<HTMLSelectElement>)
      }}
      disabled={disabled}
    >
      <SelectTrigger
        size="sm"
        className="h-8 min-w-18 px-2"
        aria-label={ariaLabel}
      >
        <SelectValue />
      </SelectTrigger>
      <SelectContent className="max-h-60">
        {options?.map((option) => (
          <SelectItem
            key={option.value}
            value={String(option.value)}
            disabled={option.disabled}
          >
            {option.label}
          </SelectItem>
        ))}
      </SelectContent>
    </Select>
  )
}

function CalendarPreviousMonthButton({
  className,
  children,
  ...props
}: React.ComponentProps<"button">) {
  return (
    <Button
      type="button"
      variant="ghost"
      size="icon-sm"
      className={cn(
        "size-(--cell-size) shrink-0 bg-transparent p-0 hover:bg-interactive-hover-soft dark:bg-transparent dark:hover:bg-interactive-hover-soft",
        className,
      )}
      {...props}
    >
      {children ?? (
        <i className="fa-light fa-chevron-left text-sm text-foreground" aria-hidden="true" />
      )}
    </Button>
  )
}

function CalendarNextMonthButton({
  className,
  children,
  ...props
}: React.ComponentProps<"button">) {
  return (
    <Button
      type="button"
      variant="ghost"
      size="icon-sm"
      className={cn(
        "size-(--cell-size) shrink-0 bg-transparent p-0 hover:bg-interactive-hover-soft dark:bg-transparent dark:hover:bg-interactive-hover-soft",
        className,
      )}
      {...props}
    >
      {children ?? (
        <i className="fa-light fa-chevron-right text-sm text-foreground" aria-hidden="true" />
      )}
    </Button>
  )
}

function Calendar({
  className,
  classNames,
  showOutsideDays = true,
  captionLayout = "label",
  formatters,
  components,
  ...props
}: React.ComponentProps<typeof DayPicker>) {
  const defaultClassNames = getDefaultClassNames()

  return (
    <DayPicker
      showOutsideDays={showOutsideDays}
      className={cn(
        "group/calendar bg-popover p-3 text-foreground [--cell-size:--spacing(8)]",
        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={{
        root: cn("w-fit", defaultClassNames.root),
        months: cn(
          "relative flex flex-col gap-4 md:flex-row",
          defaultClassNames.months
        ),
        month: cn("flex w-full flex-col gap-4", defaultClassNames.month),
        nav: cn(
          "pointer-events-none absolute inset-x-0 top-0 z-30 flex w-full items-center justify-between gap-1",
          defaultClassNames.nav
        ),
        button_previous: cn("pointer-events-auto", defaultClassNames.button_previous),
        button_next: cn("pointer-events-auto", defaultClassNames.button_next),
        month_caption: cn(
          "relative z-10 flex h-(--cell-size) w-full items-center justify-center px-(--cell-size)",
          defaultClassNames.month_caption
        ),
        dropdowns: cn(
          "flex h-(--cell-size) w-full items-center justify-center gap-1.5 text-sm font-medium",
          defaultClassNames.dropdowns
        ),
        dropdown_root: cn("hidden", defaultClassNames.dropdown_root),
        dropdown: cn("hidden", defaultClassNames.dropdown),
        caption_label: cn(
          "text-sm font-medium select-none",
          defaultClassNames.caption_label
        ),
        table: "w-full border-collapse",
        weekdays: cn(defaultClassNames.weekdays),
        weekday: cn(
          "h-(--cell-size) w-(--cell-size) p-0 text-center text-xs font-normal text-muted-foreground select-none",
          defaultClassNames.weekday
        ),
        week: cn("mt-2", defaultClassNames.week),
        week_number_header: cn(
          "w-(--cell-size) select-none",
          defaultClassNames.week_number_header
        ),
        week_number: cn(
          "text-xs text-muted-foreground select-none",
          defaultClassNames.week_number
        ),
        day: cn(
          "group/day relative size-(--cell-size) p-0 text-center align-middle select-none",
          defaultClassNames.day
        ),
        range_start: cn(
          "rounded-l-md bg-accent",
          "[&_button]:!rounded-md [&_button]:!rounded-l-md [&_button]:!bg-primary [&_button]:!text-primary-foreground [&_button]:hover:!bg-primary/90 [&_button]:hover:!text-primary-foreground [&_button]:shadow-none",
          defaultClassNames.range_start
        ),
        range_middle: cn(
          "rounded-none bg-accent",
          "[&_button]:!rounded-none [&_button]:!bg-transparent [&_button]:!text-accent-foreground [&_button]:hover:!bg-transparent [&_button]:hover:!text-accent-foreground [&_button]:shadow-none",
          defaultClassNames.range_middle
        ),
        range_end: cn(
          "rounded-r-md bg-accent",
          "[&_button]:!rounded-md [&_button]:!rounded-r-md [&_button]:!bg-primary [&_button]:!text-primary-foreground [&_button]:hover:!bg-primary/90 [&_button]:hover:!text-primary-foreground [&_button]:shadow-none",
          defaultClassNames.range_end
        ),
        today: cn("rounded-md", defaultClassNames.today),
        outside: cn(
          "text-muted-foreground [&_button]:text-muted-foreground [&_button]:opacity-50",
          "aria-selected:bg-accent/50 aria-selected:[&_button]:opacity-100",
          defaultClassNames.outside
        ),
        disabled: cn(
          "text-muted-foreground opacity-50",
          defaultClassNames.disabled
        ),
        hidden: cn("invisible", defaultClassNames.hidden),
        ...classNames,
      }}
      components={{
        Root: ({ className, rootRef, ...props }) => {
          return (
            <div
              data-slot="calendar"
              ref={rootRef}
              className={cn(className)}
              {...props}
            />
          )
        },
        Dropdown: CalendarDropdown,
        PreviousMonthButton: CalendarPreviousMonthButton,
        NextMonthButton: CalendarNextMonthButton,
        Chevron: ({ className, orientation }) => {
          return (
            <i
              className={cn(
                "fa-light pointer-events-none text-[0.65rem] leading-none text-muted-foreground",
                orientation === "left" && "fa-chevron-left",
                orientation === "right" && "fa-chevron-right",
                orientation === "down" && "fa-chevron-down",
                className
              )}
              aria-hidden="true"
            />
          )
        },
        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 focusRef = React.useCallback(
    (node: HTMLButtonElement | null) => {
      if (modifiers.focused) node?.focus()
    },
    [modifiers.focused],
  )

  const isSelected =
    modifiers.selected &&
    !modifiers.range_start &&
    !modifiers.range_end &&
    !modifiers.range_middle

  const inRange =
    modifiers.range_start || modifiers.range_middle || modifiers.range_end

  return (
    <Button
      ref={focusRef}
      variant="ghost"
      size="icon"
      data-day={day.date.toLocaleDateString()}
      data-selected-single={isSelected}
      data-range-start={modifiers.range_start}
      data-range-end={modifiers.range_end}
      data-range-middle={modifiers.range_middle}
      className={cn(
        "size-(--cell-size) min-w-(--cell-size) flex-col gap-1 p-0 text-sm leading-none font-normal",
        "bg-transparent shadow-none hover:bg-interactive-hover-soft dark:bg-transparent dark:hover:bg-interactive-hover-soft",
        "group-data-[focused=true]/day:relative group-data-[focused=true]/day:z-10 group-data-[focused=true]/day:ring-3 group-data-[focused=true]/day:ring-ring/50",
        modifiers.outside &&
          !inRange &&
          !isSelected &&
          "text-muted-foreground opacity-50",
        modifiers.disabled && "text-muted-foreground opacity-50",
        modifiers.today &&
          !modifiers.selected &&
          "bg-accent font-semibold text-accent-foreground hover:bg-accent hover:text-accent-foreground",
        "data-[selected-single=true]:!bg-primary data-[selected-single=true]:!text-primary-foreground data-[selected-single=true]:hover:!bg-primary/90 data-[selected-single=true]:hover:!text-primary-foreground",
        "[&>span]:text-xs [&>span]:opacity-70",
        defaultClassNames.day_button,
        className
      )}
      {...props}
    />
  )
}

export { Calendar, CalendarDayButton, CalendarDropdown }