import React, { useId, useRef, type HTMLAttributes } from 'react'
import { mergeRefs } from '@react-aria/utils'
import classnames from 'classnames'
import { type ReactFocusOnProps } from 'react-focus-on/dist/es5/types'
import { FieldMessage, type FieldMessageProps } from '~components/FieldMessage'
import { Heading } from '~components/Heading'
import { type OverrideClassName } from '~components/types/OverrideClassName'
import {
  MultiSelectOptions,
  type MultiSelectOptionsProps,
} from './subcomponents/MultiSelectOptions'
import { MultiSelectToggle } from './subcomponents/MultiSelectToggle'
import { Popover, useFloating } from './subcomponents/Popover'
import { type MultiSelectOption, type ValidationMessage } from './types'
import styles from './MultiSelect.module.scss'

export type MultiSelectProps = {
  inputRef?: React.Ref<HTMLButtonElement>
  label: string
  items: MultiSelectOptionsProps['options']
  selectedValues: Set<MultiSelectOption['value']>
  /**
   * A description that provides context for the field
   */
  description?: FieldMessageProps['message']
  onSelectedValuesChange: MultiSelectOptionsProps['onChange']
  isOpen: boolean
  onOpenChange: (isOpen: boolean) => void
  /** A status and message to provide context to the validation issue  */
  validationMessage?: ValidationMessage
} & OverrideClassName<HTMLAttributes<HTMLDivElement>>

export const MultiSelect = ({
  id: propsId,
  label,
  items,
  selectedValues,
  description,
  onSelectedValuesChange,
  isOpen,
  onOpenChange,
  classNameOverride,
  validationMessage,
  inputRef,
  ...restProps
}: MultiSelectProps): JSX.Element => {
  const fallbackId = useId()
  const id = propsId ?? fallbackId
  const descriptionId = `${id}-description`
  const validationId = `${id}-validation-message`

  const toggleButtonRef = useRef<HTMLButtonElement>(null)
  const { refs } = useFloating()

  const handleToggleClick = (): void => onOpenChange(!isOpen)
  const handleClose = (): void => onOpenChange(false)

  const onClickOutside: ReactFocusOnProps['onClickOutside'] = (e) => {
    const toggle = refs.reference.current as Node
    const isInToggle = toggle.contains(e.target as HTMLElement)
    if (!isInToggle) {
      e.stopPropagation()
      handleClose()
    }
  }

  const itemsMap = items.reduce(
    (acc, item) => {
      acc[item.value] = item
      return acc
    },
    {} as Record<MultiSelectOption['value'], MultiSelectOption>,
  )

  const handleOnRemoveOption = (optionValue: MultiSelectOption['value']): void => {
    const newValues = new Set(selectedValues.values())
    newValues.delete(optionValue)
    onSelectedValuesChange(newValues)
  }

  const handleRemoveAllOptions = (): void => {
    const newValues = new Set([])
    onSelectedValuesChange(newValues)
  }

  return (
    <div id={id} className={classnames(classNameOverride)} {...restProps}>
      <Heading tag="span" variant="heading-6" id={`${id}--label`}>
        {label}
      </Heading>

      <div ref={refs.setReference} className={styles.toggleContainer}>
        <MultiSelectToggle
          ref={mergeRefs(toggleButtonRef, inputRef)}
          id={`${id}--toggle`}
          aria-labelledby={`${id}--label`}
          aria-describedby={`${validationId} ${descriptionId}`}
          aria-controls={`${id}--popover`}
          onClick={handleToggleClick}
          isOpen={isOpen}
          selectedOptions={Array.from(selectedValues).map((value) => itemsMap[value])}
          status={validationMessage?.status}
          onRemoveOption={handleOnRemoveOption}
          onRemoveAllOptions={handleRemoveAllOptions}
        />
      </div>

      {validationMessage && <FieldMessage id={validationId} {...validationMessage} />}
      {description && <FieldMessage id={descriptionId} message={description} />}

      {isOpen && (
        <Popover
          refs={refs}
          id={`${id}--popover`}
          role="dialog"
          aria-modal="true"
          tabIndex={-1}
          focusOnProps={{
            enabled: true,
            onClickOutside,
            onEscapeKey: handleClose,
            shards: [toggleButtonRef],
            noIsolation: true,
            onActivation: (): void => refs.floating?.current?.focus(),
          }}
          classNameOverride={styles.popover}
        >
          <MultiSelectOptions
            id={`${id}--options`}
            options={items}
            onChange={onSelectedValuesChange}
            selectedValues={selectedValues}
          />
        </Popover>
      )}
    </div>
  )
}

MultiSelect.displayName = 'MultiSelect'