import { type Key } from '@react-types/shared'
import { type SingleSelectItem, type SingleSelectOption } from '../types'
import { isSelectOptionGroup } from './isSelectOptionGroup'

export const getDisabledKeysFromItems = <Option extends SingleSelectOption = SingleSelectOption>(
  items: SingleSelectItem<Option>[],
): Key[] =>
  items.reduce((acc: Key[], item) => {
    if (isSelectOptionGroup(item)) {
      const keys = Array.from(item.options)
        .filter((groupItem) => groupItem.disabled)
        .map((disabledItems) => disabledItems.value)
      return [...acc, ...keys]
    }

    return item.disabled ? [...acc, item.value] : acc
  }, [])