"use client"
import { da, de, fi, nb, sv } from "date-fns/locale"
import { useState } from "react"
import { type DateRange, DayPicker } from "react-day-picker"
import { useIntl } from "react-intl"

import { Lang } from "@/constants/languages"
import { dt } from "@/lib/dt"

import Button from "@/components/TempDesignSystem/Button"
import Divider from "@/components/TempDesignSystem/Divider"
import Caption from "@/components/TempDesignSystem/Text/Caption"
import Subtitle from "@/components/TempDesignSystem/Text/Subtitle"
import useLang from "@/hooks/useLang"

import { ChevronLeftIcon } from "../Icons"

import styles from "./date-picker.module.css"
import classNames from "react-day-picker/style.module.css"

import type { DatePickerProps } from "@/types/components/datepicker"

const locales = {
  [Lang.da]: da,
  [Lang.de]: de,
  [Lang.fi]: fi,
  [Lang.no]: nb,
  [Lang.sv]: sv,
}

export default function DatePicker({
  close,
  handleOnSelect,
  initialSelected = {
    from: undefined,
    to: undefined,
  },
}: DatePickerProps) {
  const lang = useLang()
  const intl = useIntl()
  const [selectedDate, setSelectedDate] = useState<DateRange>(initialSelected)

  function handleSelectDate(selected: DateRange) {
    handleOnSelect(selected)
    setSelectedDate(selected)
  }

  /** English is default language and doesn't need to be imported */
  const locale = lang === Lang.en ? undefined : locales[lang]
  const currentDate = dt().toDate()
  const startOfMonth = dt(currentDate).set("date", 1).toDate()
  const yesterday = dt(currentDate).subtract(1, "day").toDate()
  return (
    <DayPicker
      classNames={{
        ...classNames,
        caption_label: `${classNames.caption_label} ${styles.captionLabel}`,
        day: `${classNames.day} ${styles.day}`,
        day_button: `${classNames.day_button} ${styles.dayButton}`,
        footer: styles.footer,
        month_caption: `${classNames.month_caption} ${styles.monthCaption}`,
        months: `${classNames.months} ${styles.months}`,
        range_end: styles.rangeEnd,
        range_middle: styles.rangeMiddle,
        range_start: styles.rangeStart,
        week: styles.week,
        weekday: `${classNames.weekday} ${styles.weekDay}`,
      }}
      disabled={{ from: startOfMonth, to: yesterday }}
      excludeDisabled
      footer
      formatters={{
        formatWeekdayName(weekday) {
          return dt(weekday).locale(lang).format("ddd")
        },
      }}
      lang={lang}
      locale={locale}
      mode="range"
      numberOfMonths={2}
      onSelect={handleSelectDate}
      pagedNavigation
      required
      selected={selectedDate}
      startMonth={currentDate}
      weekStartsOn={1}
      components={{
        Chevron(props) {
          return <ChevronLeftIcon {...props} height={20} width={20} />
        },
        Footer(props) {
          return (
            <>
              <Divider className={styles.divider} color="primaryLightSubtle" />
              <footer className={props.className}>
                <Button
                  intent="tertiary"
                  onPress={close}
                  size="small"
                  theme="base"
                >
                  <Caption color="white" textTransform="bold">
                    {intl.formatMessage({ id: "Select dates" })}
                  </Caption>
                </Button>
              </footer>
            </>
          )
        },
        MonthCaption(props) {
          return (
            <div className={props.className}>
              <Subtitle asChild type="two">
                {props.children}
              </Subtitle>
            </div>
          )
        },
        Nav(props) {
          if (Array.isArray(props.children)) {
            const prevButton = props.children?.[0]
            const nextButton = props.children?.[1]
            return (
              <>
                {prevButton ? (
                  <nav
                    className={`${props.className} ${styles.previousButton}`}
                  >
                    {prevButton}
                  </nav>
                ) : null}
                {nextButton ? (
                  <nav className={`${props.className} ${styles.nextButton}`}>
                    {nextButton}
                  </nav>
                ) : null}
              </>
            )
          }
          return <></>
        },
      }}
    />
  )
}