reka-ui/src/shared/date/comparators.ts

Vue port for Radix UI Primitives.

/*
  * Implementation ported from https://github.com/melt-ui/melt-ui/blob/develop/src/lib/internal/helpers/date/utils.ts
*/

import type { DateValue, ZonedDateTime } from '@internationalized/date'
import { CalendarDate, CalendarDateTime, createCalendar, DateFormatter, Time, toCalendar } from '@internationalized/date'

export type TimeValue = Time | CalendarDateTime | ZonedDateTime

export type Granularity = 'day' | 'hour' | 'minute' | 'second'
export type TimeGranularity = 'hour' | 'minute' | 'second'

type GetDefaultDateProps = {
  defaultValue?: DateValue | DateValue[] | undefined
  defaultPlaceholder?: DateValue | undefined
  granularity?: Granularity
  locale?: string
}

/**
 * A helper function used throughout the various date builders
 * to generate a default `DateValue` using the `defaultValue`,
 * `defaultPlaceholder`, and `granularity` props.
 *
 * It's important to match the `DateValue` type being used
 * elsewhere in the builder, so they behave according to the
 * behavior the user expects based on the props they've provided.
 *
 */
export function getDefaultDate(props: GetDefaultDateProps): DateValue {
  const { defaultValue, defaultPlaceholder, granularity = 'day', locale = 'en' } = props

  if (Array.isArray(defaultValue) && defaultValue.length)
    return defaultValue.at(-1)!.copy()

  if (defaultValue && !Array.isArray(defaultValue))
    return defaultValue.copy()

  if (defaultPlaceholder)
    return defaultPlaceholder.copy()

  const date = new Date()
  const year = date.getFullYear()
  const month = date.getMonth() + 1
  const day = date.getDate()
  const calendarDateTimeGranularities = ['hour', 'minute', 'second']

  const defaultFormatter = new DateFormatter(locale)
  const calendar = createCalendar(defaultFormatter.resolvedOptions().calendar)

  if (calendarDateTimeGranularities.includes(granularity ?? 'day'))
    return toCalendar(new CalendarDateTime(year, month, day, 0, 0, 0), calendar)

  return toCalendar(new CalendarDate(year, month, day), calendar)
}

type GetDefaultTimeProps = {
  defaultValue?: TimeValue | undefined
  defaultPlaceholder?: TimeValue | undefined
}

export function getDefaultTime(props: GetDefaultTimeProps): TimeValue {
  const { defaultValue, defaultPlaceholder } = props

  if (defaultValue) {
    return defaultValue.copy()
  }

  if (defaultPlaceholder) {
    return defaultPlaceholder.copy()
  }

  return new Time(0, 0, 0)
}