@primer/components
Version:
Primer react components
203 lines (186 loc) • 7.84 kB
JavaScript
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 '../utils/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 && setInputValueFn(text);
};
}
const isItemSelected = (itemId, selectedItemIds) => selectedItemIds.includes(itemId);
function getItemById(itemId, items) {
return items.find(item => item.id === itemId);
}
;
// TODO: consider making 'aria-labelledby' required
function AutocompleteMenu(props) {
const {
activeDescendantRef,
id,
inputRef,
inputValue = '',
scrollContainerRef,
setAutocompleteSuggestion,
setShowMenu,
setInputValue,
setIsMenuDirectlyActivated,
setSelectedItemLength,
showMenu
} = useContext(AutocompleteContext);
const {
items,
selectedItemIds,
sortOnCloseFn,
emptyStateText,
addNewItem,
loading,
selectionVariant,
filterFn = getDefaultItemFilter(inputValue),
"aria-labelledby": ariaLabelledBy,
onOpenChange,
onSelectedChange = getDefaultOnSelectionChange(setInputValue),
customScrollContainerRef
} = props;
const listContainerRef = useRef(null);
const [highlightedItem, setHighlightedItem] = useState();
const [sortedItemIds, setSortedItemIds] = useState(items.map(({
id
}) => id));
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];
onSelectedChange && onSelectedChange(newSelectedItemIds.map(newSelectedItemId => getItemById(newSelectedItemId, items)));
if (selectionVariant === 'multiple') {
setInputValue && setInputValue('');
setAutocompleteSuggestion && setAutocompleteSuggestion('');
} else {
var _inputRef$current;
setShowMenu && setShowMenu(false);
inputRef === null || inputRef === void 0 ? void 0 : (_inputRef$current = inputRef.current) === null || _inputRef$current === void 0 ? void 0 : _inputRef$current.setSelectionRange(inputRef.current.value.length, inputRef.current.value.length);
}
}
};
}), [items, selectedItemIds]);
const itemSortOrderData = useMemo(() => sortedItemIds.reduce((acc, curr, i) => {
acc[curr] = i;
return acc;
}, {}), [sortedItemIds]);
const sortedAndFilteredItemsToRender = useMemo(() => selectableItems.filter((item, i) => filterFn(item, i)).sort((a, b) => itemSortOrderData[a.id] - itemSortOrderData[b.id]), [selectableItems]);
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, e) => {
// TODO: make it possible to pass a leadingVisual when using `addNewItem`
addNewItem.handleAddItem({ ...item,
id: item.id || uniqueId(),
leadingVisual: undefined
});
if (selectionVariant === 'multiple') {
setInputValue && setInputValue('');
setAutocompleteSuggestion && setAutocompleteSuggestion('');
}
}
}] : [])], [sortedAndFilteredItemsToRender, addNewItem]);
useFocusZone({
containerRef: listContainerRef,
focusOutBehavior: 'wrap',
focusableElementFilter: element => {
return !(element instanceof HTMLInputElement);
},
activeDescendantFocus: inputRef,
onActiveDescendantChanged: (current, _previous, directlyActivated) => {
if (activeDescendantRef) {
activeDescendantRef.current = current || null;
}
if (current) {
const selectedItem = selectableItems.find(item => item.id.toString() === (current === null || current === void 0 ? void 0 : current.dataset.id));
setHighlightedItem(selectedItem);
setIsMenuDirectlyActivated && setIsMenuDirectlyActivated(directlyActivated);
}
if (current && customScrollContainerRef && customScrollContainerRef.current && directlyActivated) {
scrollIntoViewingArea(current, customScrollContainerRef.current);
} else if (current && scrollContainerRef && scrollContainerRef.current && directlyActivated) {
scrollIntoViewingArea(current, scrollContainerRef.current);
}
}
}, [loading]);
useEffect(() => {
var _highlightedItem$text;
if (!setAutocompleteSuggestion) {
return;
}
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]);
useEffect(() => {
if (showMenu === false) {
setSortedItemIds([...sortedItemIds].sort(sortOnCloseFn ? sortOnCloseFn : getDefaultSortFn(itemId => isItemSelected(itemId, selectedItemIds))));
}
onOpenChange && onOpenChange(Boolean(showMenu));
}, [showMenu]);
useEffect(() => {
if (selectedItemIds.length) {
setSelectedItemLength && setSelectedItemLength(selectedItemIds.length);
}
}, [selectedItemIds]);
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;