UNPKG

@primer/components

Version:
205 lines (186 loc) • 8.24 kB
import React, { useContext, useEffect, useMemo, useRef, useState } from 'react'; import { ActionList } from '../ActionList'; import { useFocusZone } from '../hooks/useFocusZone'; import { Box, Spinner } from '../'; import { AutocompleteContext } from './AutocompleteContext'; import { PlusIcon } from '@primer/octicons-react'; import { uniqueId } from '../utils/uniqueId'; import { scrollIntoViewingArea } from '../behaviors/scrollIntoViewingArea'; const getDefaultSortFn = isItemSelectedFn => (itemIdA, itemIdB) => isItemSelectedFn(itemIdA) === isItemSelectedFn(itemIdB) ? 0 : isItemSelectedFn(itemIdA) ? -1 : 1; function getDefaultItemFilter(filterValue) { return function (item, _i) { var _item$text; return Boolean((_item$text = item.text) === null || _item$text === void 0 ? void 0 : _item$text.toLowerCase().startsWith(filterValue.toLowerCase())); }; } function getDefaultOnSelectionChange(setInputValueFn) { return function (itemOrItems) { const { text = '' } = Array.isArray(itemOrItems) ? itemOrItems.slice(-1)[0] : itemOrItems; setInputValueFn(text); }; } const isItemSelected = (itemId, selectedItemIds) => selectedItemIds.includes(itemId); function getItemById(itemId, items) { return items.find(item => item.id === itemId); } // eslint-disable-next-line @typescript-eslint/no-explicit-any function AutocompleteMenu(props) { const autocompleteContext = useContext(AutocompleteContext); if (autocompleteContext === null) { throw new Error('AutocompleteContext returned null values'); } const { activeDescendantRef, id, inputRef, inputValue = '', scrollContainerRef, setAutocompleteSuggestion, setShowMenu, setInputValue, setIsMenuDirectlyActivated, setSelectedItemLength, showMenu } = autocompleteContext; const { items, selectedItemIds, sortOnCloseFn, emptyStateText, addNewItem, loading, selectionVariant, filterFn, 'aria-labelledby': ariaLabelledBy, onOpenChange, onSelectedChange, customScrollContainerRef } = props; const listContainerRef = useRef(null); const [highlightedItem, setHighlightedItem] = useState(); const [sortedItemIds, setSortedItemIds] = useState(items.map(({ id: itemId }) => itemId)); const selectableItems = useMemo(() => items.map(selectableItem => { return { ...selectableItem, role: 'option', id: selectableItem.id, selected: selectionVariant === 'multiple' ? selectedItemIds.includes(selectableItem.id) : undefined, onAction: item => { const otherSelectedItemIds = selectedItemIds.filter(selectedItemId => selectedItemId !== item.id); const newSelectedItemIds = selectedItemIds.includes(item.id) ? otherSelectedItemIds : [...otherSelectedItemIds, item.id]; const onSelectedChangeFn = onSelectedChange ? onSelectedChange : getDefaultOnSelectionChange(setInputValue); onSelectedChangeFn(newSelectedItemIds.map(newSelectedItemId => getItemById(newSelectedItemId, items))); if (selectionVariant === 'multiple') { setInputValue(''); setAutocompleteSuggestion(''); } else { var _inputRef$current; setShowMenu(false); (_inputRef$current = inputRef.current) === null || _inputRef$current === void 0 ? void 0 : _inputRef$current.setSelectionRange(inputRef.current.value.length, inputRef.current.value.length); } } }; }), [items, selectedItemIds, inputRef, onSelectedChange, selectionVariant, setAutocompleteSuggestion, setInputValue, setShowMenu]); const itemSortOrderData = useMemo(() => sortedItemIds.reduce((acc, curr, i) => { acc[curr] = i; return acc; }, {}), [sortedItemIds]); const sortedAndFilteredItemsToRender = useMemo(() => selectableItems.filter(filterFn ? filterFn : getDefaultItemFilter(inputValue)).sort((a, b) => itemSortOrderData[a.id] - itemSortOrderData[b.id]), [selectableItems, itemSortOrderData, filterFn, inputValue]); const allItemsToRender = useMemo(() => [// sorted and filtered selectable items ...sortedAndFilteredItemsToRender, // menu item used for creating a token from whatever is in the text input ...(addNewItem ? [{ ...addNewItem, leadingVisual: () => /*#__PURE__*/React.createElement(PlusIcon, null), onAction: item => { // TODO: make it possible to pass a leadingVisual when using `addNewItem` addNewItem.handleAddItem({ ...item, id: item.id || uniqueId(), leadingVisual: undefined }); if (selectionVariant === 'multiple') { setInputValue(''); setAutocompleteSuggestion(''); } } }] : [])], [sortedAndFilteredItemsToRender, addNewItem, setAutocompleteSuggestion, selectionVariant, setInputValue]); useFocusZone({ containerRef: listContainerRef, focusOutBehavior: 'wrap', focusableElementFilter: element => { return !(element instanceof HTMLInputElement); }, activeDescendantFocus: inputRef, onActiveDescendantChanged: (current, _previous, directlyActivated) => { activeDescendantRef.current = current || null; if (current) { const selectedItem = selectableItems.find(item => item.id.toString() === current.getAttribute('data-id')); setHighlightedItem(selectedItem); setIsMenuDirectlyActivated(directlyActivated); } if (current && customScrollContainerRef && customScrollContainerRef.current && directlyActivated) { scrollIntoViewingArea(current, customScrollContainerRef.current); } else if (current && scrollContainerRef.current && directlyActivated) { scrollIntoViewingArea(current, scrollContainerRef.current); } } }, [loading]); useEffect(() => { var _highlightedItem$text; if (highlightedItem !== null && highlightedItem !== void 0 && (_highlightedItem$text = highlightedItem.text) !== null && _highlightedItem$text !== void 0 && _highlightedItem$text.startsWith(inputValue) && !selectedItemIds.includes(highlightedItem.id)) { setAutocompleteSuggestion(highlightedItem.text); } else { setAutocompleteSuggestion(''); } }, [highlightedItem, inputValue, selectedItemIds, setAutocompleteSuggestion]); useEffect(() => { const itemIdSortResult = [...sortedItemIds].sort(sortOnCloseFn ? sortOnCloseFn : getDefaultSortFn(itemId => isItemSelected(itemId, selectedItemIds))); const sortResultMatchesState = itemIdSortResult.length === sortedItemIds.length && itemIdSortResult.every((element, index) => element === sortedItemIds[index]); if (showMenu === false && !sortResultMatchesState) { setSortedItemIds(itemIdSortResult); } onOpenChange && onOpenChange(Boolean(showMenu)); }, [showMenu, onOpenChange, selectedItemIds, sortOnCloseFn, sortedItemIds]); useEffect(() => { if (selectedItemIds.length) { setSelectedItemLength(selectedItemIds.length); } }, [selectedItemIds, setSelectedItemLength]); return /*#__PURE__*/React.createElement(Box, { sx: !showMenu ? { // visually hides this label for sighted users position: 'absolute', width: '1px', height: '1px', padding: '0', margin: '-1px', overflow: 'hidden', clip: 'rect(0, 0, 0, 0)', whiteSpace: 'nowrap', borderWidth: '0' } : {} }, loading ? /*#__PURE__*/React.createElement(Box, { p: 3, display: "flex", justifyContent: "center" }, /*#__PURE__*/React.createElement(Spinner, null)) : /*#__PURE__*/React.createElement("div", { ref: listContainerRef }, allItemsToRender.length ? /*#__PURE__*/React.createElement(ActionList, { selectionVariant: "multiple" // have to typecast to `ItemProps` because we have an extra property // on `items` for Autocomplete: `metadata` , items: allItemsToRender, role: "listbox", id: `${id}-listbox`, "aria-labelledby": ariaLabelledBy }) : /*#__PURE__*/React.createElement(Box, { p: 3 }, emptyStateText))); } AutocompleteMenu.displayName = "AutocompleteMenu"; AutocompleteMenu.defaultProps = { emptyStateText: 'No selectable options', selectionVariant: 'single' }; AutocompleteMenu.displayName = 'AutocompleteMenu'; export default AutocompleteMenu;