bsky-social-app/src/components/forms/DateField/index.tsx

import {useCallback, useImperativeHandle, useState} from 'react'
import {Keyboard, View} from 'react-native'
import DatePicker from 'react-native-date-picker'
import {msg} from '@lingui/core/macro'
import {useLingui} from '@lingui/react'
import {Trans} from '@lingui/react/macro'

import {atoms as a, useTheme} from '#/alf'
import {Button, ButtonText} from '#/components/Button'
import * as Dialog from '#/components/Dialog'
import {type DateFieldProps} from '#/components/forms/DateField/types'
import {toSimpleDateString} from '#/components/forms/DateField/utils'
import * as TextField from '#/components/forms/TextField'
import {DateFieldButton} from './index.shared'

export * as utils from '#/components/forms/DateField/utils'
export const LabelText = TextField.LabelText

/**
 * Date-only input. Accepts a string in the format YYYY-MM-DD, or a Date object.
 * Date objects are converted to strings in the format YYYY-MM-DD.
 * Returns a string in the format YYYY-MM-DD.
 *
 * To generate a string in the format YYYY-MM-DD from a Date object, use the
 * `utils.toSimpleDateString(Date)` export of this file.
 */
export function DateField({
  value,
  inputRef,
  onChangeDate,
  onConfirm,
  placeholder,
  testID,
  label,
  isInvalid,
  accessibilityHint,
  maximumDate,
}: DateFieldProps) {
  const {_, i18n} = useLingui()
  const t = useTheme()
  const control = Dialog.useDialogControl()

  /*
   * The picker requires a valid date, so when value is empty we fall back to
   * maximumDate (if set) or today. Draft state lets the picker scroll even when
   * the parent does not echo value back (e.g. a clearable field).
   */
  const fallbackDate = maximumDate
    ? toSimpleDateString(maximumDate)
    : toSimpleDateString(new Date())
  const [draft, setDraft] = useState(() =>
    value === '' ? fallbackDate : toSimpleDateString(value),
  )

  const onChangeInternal = useCallback(
    (date: Date | undefined) => {
      if (date) {
        const formatted = toSimpleDateString(date)
        setDraft(formatted)
        onChangeDate(formatted)
      }
    },
    [onChangeDate],
  )

  useImperativeHandle(
    inputRef,
    () => ({
      focus: () => {
        Keyboard.dismiss()
        setDraft(value === '' ? fallbackDate : toSimpleDateString(value))
        control.open()
      },
      blur: () => {
        control.close()
      },
    }),
    [control, value, fallbackDate],
  )

  return (
    <>
      <DateFieldButton
        label={label}
        value={value}
        placeholder={placeholder}
        onPress={() => {
          Keyboard.dismiss()
          setDraft(value === '' ? fallbackDate : toSimpleDateString(value))
          control.open()
        }}
        isInvalid={isInvalid}
        accessibilityHint={accessibilityHint}
      />
      <Dialog.Outer
        control={control}
        testID={testID}
        nativeOptions={{preventExpansion: true}}>
        <Dialog.Handle />
        <Dialog.ScrollableInner label={label}>
          <View style={a.gap_lg}>
            <View style={[a.relative, a.w_full, a.align_center]}>
              <DatePicker
                timeZoneOffsetInMinutes={0}
                theme={t.scheme}
                date={new Date(draft)}
                onDateChange={onChangeInternal}
                mode="date"
                locale={i18n.locale}
                testID={`${testID}-datepicker`}
                aria-label={label}
                accessibilityLabel={label}
                accessibilityHint={accessibilityHint}
                maximumDate={
                  maximumDate
                    ? new Date(toSimpleDateString(maximumDate))
                    : undefined
                }
              />
            </View>
            <Button
              label={_(msg`Done`)}
              onPress={() => {
                /*
                 * Commit the currently shown date even if the user never
                 * scrolled (onDateChange only fires on scroll). This keeps
                 * onChangeDate firing alongside onConfirm, matching Android and
                 * web, so an empty field confirmed without scrolling does not
                 * report a date via onConfirm while onChangeDate stays silent.
                 */
                onChangeDate(draft)
                onConfirm?.(draft)
                control.close()
              }}
              size="large"
              color="primary"
              variant="solid">
              <ButtonText>
                <Trans>Done</Trans>
              </ButtonText>
            </Button>
          </View>
        </Dialog.ScrollableInner>
      </Dialog.Outer>
    </>
  )
}