@atlaskit/editor-plugin-type-ahead 10.2.2 → 10.3.0

package/CHANGELOG.md

@@ -1,5 +1,25 @@
 # @atlaskit/editor-plugin-type-ahead
 
+## 10.3.0
+
+### Minor Changes
+
+- [`beab4fffee103`](https://bitbucket.org/atlassian/atlassian-frontend-monorepo/commits/beab4fffee103) -
+  EDITOR-7318: Add type-ahead sections support with getSections handler API.
+
+  Adds `TypeAheadSection` and `TypeAheadSectionUpdate` types and an optional `getSections()` method
+  to `TypeAheadHandler` in `editor-common`. Implements section grouping in
+  `editor-plugin-type-ahead` via `buildSectionedResult`. Adds People/Agents section split to
+  `editor-plugin-mentions`.
+
+  Removes the `updateSection` action and associated `sectionOverrides`/`sectionsRefreshKey`
+  pm-plugin state — section definitions are now derived directly from `getSections()` at load time
+  rather than being mutated at runtime.
+
+### Patch Changes
+
+- Updated dependencies
+
 ## 10.2.2
 
 ### Patch Changes

package/dist/cjs/pm-plugins/commands/update-list-items.js

@@ -7,12 +7,14 @@ exports.updateListItem = void 0;
 var _actions = require("../actions");
 var _key = require("../key");
 var updateListItem = exports.updateListItem = function updateListItem(items) {
+  var sections = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : [];
   return function (state, dispatch) {
     var tr = state.tr;
     tr.setMeta(_key.pluginKey, {
       action: _actions.ACTIONS.UPDATE_LIST_ITEMS,
       params: {
-        items: items
+        items: items,
+        sections: sections
       }
     });
     if (dispatch) {

package/dist/cjs/pm-plugins/main.js

@@ -59,6 +59,7 @@ function createPlugin(_ref) {
           decorationSet: _view.DecorationSet.empty,
           decorationElement: null,
           items: [],
+          sections: [],
           errorInfo: null,
           selectedIndex: -1,
           stats: null,

package/dist/cjs/pm-plugins/reducer.js

@@ -76,6 +76,7 @@ var createReducer = exports.createReducer = function createReducer(_ref) {
       inputMethod: inputMethod,
       selectedIndex: typeof selectedIndex === 'number' ? selectedIndex : -1,
       items: [],
+      sections: [],
       query: reopenQuery || '',
       removePrefixTriggerOnCancel: removePrefixTriggerOnCancel
     });
@@ -94,6 +95,7 @@ var createReducer = exports.createReducer = function createReducer(_ref) {
       stats: null,
       triggerHandler: undefined,
       items: [],
+      sections: [],
       removePrefixTriggerOnCancel: undefined
     });
   };
@@ -144,13 +146,17 @@ var createReducer = exports.createReducer = function createReducer(_ref) {
       return _objectSpread(_objectSpread({}, currentPluginState), {}, {
         errorInfo: errorInfo,
         items: [],
+        sections: [],
         selectedIndex: -1
       });
     } else if (shouldUpdateListItems) {
-      var items = params.items;
+      var items = params.items,
+        _params$sections = params.sections,
+        sections = _params$sections === void 0 ? [] : _params$sections;
       var selectedIndex = currentPluginState.selectedIndex;
       return _objectSpread(_objectSpread({}, currentPluginState), {}, {
         items: items,
+        sections: sections,
         selectedIndex: Math.max(selectedIndex >= items.length ? items.length - 1 : selectedIndex, -1)
       });
     } else if (shouldUpdateSelectedIndex) {

package/dist/cjs/typeAheadPlugin.js

@@ -214,7 +214,7 @@ var typeAheadPlugin = exports.typeAheadPlugin = function typeAheadPlugin(_ref) {
       }];
     },
     getSharedState: function getSharedState(editorState) {
-      var _state$decorationSet, _state$decorationElem, _state$items, _state$errorInfo, _state$selectedIndex;
+      var _state$decorationSet, _state$decorationElem, _state$items, _state$sections, _state$errorInfo, _state$selectedIndex;
       if (!editorState) {
         return {
           query: '',
@@ -225,6 +225,7 @@ var typeAheadPlugin = exports.typeAheadPlugin = function typeAheadPlugin(_ref) {
           decorationElement: null,
           triggerHandler: undefined,
           items: [],
+          sections: [],
           errorInfo: null,
           selectedIndex: 0
         };
@@ -240,6 +241,7 @@ var typeAheadPlugin = exports.typeAheadPlugin = function typeAheadPlugin(_ref) {
         decorationElement: (_state$decorationElem = state === null || state === void 0 ? void 0 : state.decorationElement) !== null && _state$decorationElem !== void 0 ? _state$decorationElem : null,
         triggerHandler: state === null || state === void 0 ? void 0 : state.triggerHandler,
         items: (_state$items = state === null || state === void 0 ? void 0 : state.items) !== null && _state$items !== void 0 ? _state$items : [],
+        sections: (_state$sections = state === null || state === void 0 ? void 0 : state.sections) !== null && _state$sections !== void 0 ? _state$sections : [],
         errorInfo: (_state$errorInfo = state === null || state === void 0 ? void 0 : state.errorInfo) !== null && _state$errorInfo !== void 0 ? _state$errorInfo : null,
         selectedIndex: (_state$selectedIndex = state === null || state === void 0 ? void 0 : state.selectedIndex) !== null && _state$selectedIndex !== void 0 ? _state$selectedIndex : 0
       };

package/dist/cjs/ui/ContentComponent.js

@@ -13,25 +13,27 @@ function ContentComponent(_ref) {
     editorView = _ref.editorView,
     popupMountRef = _ref.popupMountRef;
   var _useSharedPluginState = (0, _hooks.useSharedPluginStateWithSelector)(api, ['typeAhead'], function (states) {
-      var _states$typeAheadStat, _states$typeAheadStat2, _states$typeAheadStat3, _states$typeAheadStat4, _states$typeAheadStat5, _states$typeAheadStat6, _states$typeAheadStat7;
+      var _states$typeAheadStat, _states$typeAheadStat2, _states$typeAheadStat3, _states$typeAheadStat4, _states$typeAheadStat5, _states$typeAheadStat6, _states$typeAheadStat7, _states$typeAheadStat8;
       return {
         triggerHandler: (_states$typeAheadStat = states.typeAheadState) === null || _states$typeAheadStat === void 0 ? void 0 : _states$typeAheadStat.triggerHandler,
         items: (_states$typeAheadStat2 = states.typeAheadState) === null || _states$typeAheadStat2 === void 0 ? void 0 : _states$typeAheadStat2.items,
-        errorInfo: (_states$typeAheadStat3 = states.typeAheadState) === null || _states$typeAheadStat3 === void 0 ? void 0 : _states$typeAheadStat3.errorInfo,
-        decorationElement: (_states$typeAheadStat4 = states.typeAheadState) === null || _states$typeAheadStat4 === void 0 ? void 0 : _states$typeAheadStat4.decorationElement,
-        decorationSet: (_states$typeAheadStat5 = states.typeAheadState) === null || _states$typeAheadStat5 === void 0 ? void 0 : _states$typeAheadStat5.decorationSet,
-        query: (_states$typeAheadStat6 = states.typeAheadState) === null || _states$typeAheadStat6 === void 0 ? void 0 : _states$typeAheadStat6.query,
-        selectedIndex: (_states$typeAheadStat7 = states.typeAheadState) === null || _states$typeAheadStat7 === void 0 ? void 0 : _states$typeAheadStat7.selectedIndex
+        sections: (_states$typeAheadStat3 = states.typeAheadState) === null || _states$typeAheadStat3 === void 0 ? void 0 : _states$typeAheadStat3.sections,
+        errorInfo: (_states$typeAheadStat4 = states.typeAheadState) === null || _states$typeAheadStat4 === void 0 ? void 0 : _states$typeAheadStat4.errorInfo,
+        decorationElement: (_states$typeAheadStat5 = states.typeAheadState) === null || _states$typeAheadStat5 === void 0 ? void 0 : _states$typeAheadStat5.decorationElement,
+        decorationSet: (_states$typeAheadStat6 = states.typeAheadState) === null || _states$typeAheadStat6 === void 0 ? void 0 : _states$typeAheadStat6.decorationSet,
+        query: (_states$typeAheadStat7 = states.typeAheadState) === null || _states$typeAheadStat7 === void 0 ? void 0 : _states$typeAheadStat7.query,
+        selectedIndex: (_states$typeAheadStat8 = states.typeAheadState) === null || _states$typeAheadStat8 === void 0 ? void 0 : _states$typeAheadStat8.selectedIndex
       };
     }),
     triggerHandler = _useSharedPluginState.triggerHandler,
     items = _useSharedPluginState.items,
+    sections = _useSharedPluginState.sections,
     errorInfo = _useSharedPluginState.errorInfo,
     decorationElement = _useSharedPluginState.decorationElement,
     decorationSet = _useSharedPluginState.decorationSet,
     query = _useSharedPluginState.query,
     selectedIndex = _useSharedPluginState.selectedIndex;
-  if (items === undefined || decorationSet === undefined || errorInfo === undefined || decorationElement === undefined || query === undefined || selectedIndex === undefined) {
+  if (items === undefined || sections === undefined || decorationSet === undefined || errorInfo === undefined || decorationElement === undefined || query === undefined || selectedIndex === undefined) {
     return null;
   }
   return /*#__PURE__*/_react.default.createElement(_TypeAheadMenu.TypeAheadMenu, {
@@ -42,6 +44,7 @@ function ContentComponent(_ref) {
     typeAheadState: {
       triggerHandler: triggerHandler,
       items: items,
+      sections: sections,
       errorInfo: errorInfo,
       decorationElement: decorationElement,
       decorationSet: decorationSet,

package/dist/cjs/ui/TypeAheadList.js

@@ -7,6 +7,7 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.TypeAheadList = void 0;
 var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
+var _toConsumableArray2 = _interopRequireDefault(require("@babel/runtime/helpers/toConsumableArray"));
 var _react = _interopRequireWildcard(require("react"));
 var _react2 = require("@emotion/react");
 var _reactIntl = require("react-intl");
@@ -19,6 +20,7 @@ var _ui = require("@atlaskit/editor-common/ui");
 var _menu = require("@atlaskit/menu");
 var _compiled = require("@atlaskit/primitives/compiled");
 var _expValEquals = require("@atlaskit/tmp-editor-statsig/exp-val-equals");
+var _experiments = require("@atlaskit/tmp-editor-statsig/experiments");
 var _closeTypeAhead = require("../pm-plugins/commands/close-type-ahead");
 var _updateSelectedIndex = require("../pm-plugins/commands/update-selected-index");
 var _constants = require("../pm-plugins/constants");
@@ -42,8 +44,43 @@ var list = (0, _react2.css)({
     padding: "var(--ds-space-100, 8px)".concat(" ", "var(--ds-space-150, 12px)")
   }
 });
-var TypeaheadAssistiveTextPureComponent = /*#__PURE__*/_react.default.memo(function (_ref) {
-  var numberOfResults = _ref.numberOfResults;
+var buildTypeAheadRows = function buildTypeAheadRows(_ref) {
+  var itemsLength = _ref.itemsLength,
+    sections = _ref.sections;
+  if (sections.length === 0) {
+    return Array.from({
+      length: itemsLength
+    }, function (_, itemIndex) {
+      return {
+        type: 'item',
+        itemIndex: itemIndex
+      };
+    });
+  }
+  var sortedSections = (0, _toConsumableArray2.default)(sections).sort(function (left, right) {
+    return left.startIndex - right.startIndex;
+  });
+  var sectionsByStartIndex = new Map(sortedSections.map(function (section) {
+    return [section.startIndex, section];
+  }));
+  var rows = [];
+  for (var itemIndex = 0; itemIndex < itemsLength; itemIndex++) {
+    var section = sectionsByStartIndex.get(itemIndex);
+    if (section) {
+      rows.push({
+        type: 'section',
+        section: section
+      });
+    }
+    rows.push({
+      type: 'item',
+      itemIndex: itemIndex
+    });
+  }
+  return rows;
+};
+var TypeaheadAssistiveTextPureComponent = /*#__PURE__*/_react.default.memo(function (_ref2) {
+  var numberOfResults = _ref2.numberOfResults;
   var intl = (0, _reactIntl.useIntl)();
   return (0, _react2.jsx)(_ui.AssistiveText, {
     assistiveText: intl.formatMessage(_typeAhead.typeAheadListMessages.searchResultsLabel, {
@@ -55,21 +92,22 @@ var TypeaheadAssistiveTextPureComponent = /*#__PURE__*/_react.default.memo(funct
     id: _constants.TYPE_AHEAD_DECORATION_ELEMENT_ID + '__popup'
   });
 });
-var TypeAheadListComponent = /*#__PURE__*/_react.default.memo(function (_ref2) {
-  var _triggerHandler$getMo, _decorationElement$qu2;
-  var items = _ref2.items,
-    emptyItem = _ref2.emptyItem,
-    selectedIndex = _ref2.selectedIndex,
-    editorView = _ref2.editorView,
-    onItemClick = _ref2.onItemClick,
-    intl = _ref2.intl,
-    fitHeight = _ref2.fitHeight,
-    decorationElement = _ref2.decorationElement,
-    triggerHandler = _ref2.triggerHandler,
-    moreElementsInQuickInsertViewEnabled = _ref2.moreElementsInQuickInsertViewEnabled,
-    api = _ref2.api,
-    showMoreOptionsButton = _ref2.showMoreOptionsButton,
-    onMoreOptionsClicked = _ref2.onMoreOptionsClicked;
+var TypeAheadListComponent = /*#__PURE__*/_react.default.memo(function (_ref3) {
+  var _itemRowIndexByItemIn, _triggerHandler$getMo, _decorationElement$qu2;
+  var items = _ref3.items,
+    sections = _ref3.sections,
+    emptyItem = _ref3.emptyItem,
+    selectedIndex = _ref3.selectedIndex,
+    editorView = _ref3.editorView,
+    onItemClick = _ref3.onItemClick,
+    intl = _ref3.intl,
+    fitHeight = _ref3.fitHeight,
+    decorationElement = _ref3.decorationElement,
+    triggerHandler = _ref3.triggerHandler,
+    moreElementsInQuickInsertViewEnabled = _ref3.moreElementsInQuickInsertViewEnabled,
+    api = _ref3.api,
+    showMoreOptionsButton = _ref3.showMoreOptionsButton,
+    onMoreOptionsClicked = _ref3.onMoreOptionsClicked;
   var listRef = (0, _react.useRef)();
   var listContainerRef = (0, _react.useRef)(null);
   var lastInputMethodRef = (0, _react.useRef)('keyboard');
@@ -96,6 +134,27 @@ var TypeAheadListComponent = /*#__PURE__*/_react.default.memo(function (_ref2) {
     _useState4 = (0, _slicedToArray2.default)(_useState3, 2),
     cache = _useState4[0],
     setCache = _useState4[1];
+  var listRows = (0, _react.useMemo)(function () {
+    return buildTypeAheadRows({
+      itemsLength: itemsLength,
+      sections: sections || []
+    });
+  }, [itemsLength, sections]);
+  var itemRowIndexByItemIndex = (0, _react.useMemo)(function () {
+    var rowIndexByItemIndex = new Map();
+    listRows.forEach(function (row, rowIndex) {
+      if (row.type === 'item') {
+        rowIndexByItemIndex.set(row.itemIndex, rowIndex);
+      }
+    });
+    return rowIndexByItemIndex;
+  }, [listRows]);
+  var populatedSectionCount = (0, _react.useMemo)(function () {
+    return listRows.filter(function (row) {
+      return row.type === 'section';
+    }).length;
+  }, [listRows]);
+  var selectedItemRowIndex = selectedIndex >= 0 ? (_itemRowIndexByItemIn = itemRowIndexByItemIndex.get(selectedIndex)) !== null && _itemRowIndexByItemIn !== void 0 ? _itemRowIndexByItemIn : -1 : -1;
   var onItemsRendered = (0, _react.useCallback)(function (props) {
     lastVisibleIndexes.current = props;
   }, []);
@@ -132,8 +191,8 @@ var TypeAheadListComponent = /*#__PURE__*/_react.default.memo(function (_ref2) {
   var lastVisibleStopIndex = lastVisibleIndexes.current.stopIndex;
   var onScroll = (0, _react.useCallback)(
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  function (_ref3) {
-    var scrollUpdateWasRequested = _ref3.scrollUpdateWasRequested;
+  function (_ref4) {
+    var scrollUpdateWasRequested = _ref4.scrollUpdateWasRequested;
     if (!scrollUpdateWasRequested) {
       return;
     }
@@ -151,28 +210,32 @@ var TypeAheadListComponent = /*#__PURE__*/_react.default.memo(function (_ref2) {
     requestAnimationFrame(function () {
       requestAnimationFrame(function () {
         var isViewMoreSelected = showMoreOptionsButton && selectedIndex === itemsLength;
-        var isSelectedItemVisible = selectedIndex >= lastVisibleStartIndex && selectedIndex <= lastVisibleStopIndex ||
+        var isSelectedItemVisible = selectedItemRowIndex >= lastVisibleStartIndex && selectedItemRowIndex <= lastVisibleStopIndex ||
         // view more is always visible, hence no scrolling
         isViewMoreSelected;
 
         //Should scroll to the list item only when the selectedIndex >= 0 and item is not visible
         if (!isSelectedItemVisible && selectedIndex !== -1) {
-          listRef.current.scrollToRow(selectedIndex);
+          listRef.current.scrollToRow(selectedItemRowIndex);
         } else if (selectedIndex === -1) {
           listRef.current.scrollToRow(0);
         }
       });
     });
-  }, [selectedIndex, lastVisibleStartIndex, lastVisibleStopIndex, itemsLength, showMoreOptionsButton]);
-  var _onMouseMove = function onMouseMove(event, index) {
+  }, [selectedIndex, lastVisibleStartIndex, lastVisibleStopIndex, itemsLength, showMoreOptionsButton, selectedItemRowIndex]);
+  var _onMouseMove = function onMouseMove(event, row) {
+    if (row.type !== 'item') {
+      return;
+    }
+    var itemIndex = row.itemIndex;
     event.preventDefault();
     event.stopPropagation();
-    if (selectedIndex === index) {
+    if (selectedIndex === itemIndex) {
       return;
     }
     mouseMovedRef.current = true;
     lastInputMethodRef.current = 'mouse';
-    (0, _updateSelectedIndex.updateSelectedIndex)(index, api)(editorView.state, editorView.dispatch);
+    (0, _updateSelectedIndex.updateSelectedIndex)(itemIndex, api)(editorView.state, editorView.dispatch);
   };
   (0, _react.useLayoutEffect)(function () {
     if (mouseMovedRef.current) {
@@ -186,17 +249,17 @@ var TypeAheadListComponent = /*#__PURE__*/_react.default.memo(function (_ref2) {
       return;
     }
     var isViewMoreSelected = showMoreOptionsButton && selectedIndex === itemsLength;
-    var isSelectedItemVisible = selectedIndex >= lastVisibleStartIndex && selectedIndex <= lastVisibleStopIndex ||
+    var isSelectedItemVisible = selectedItemRowIndex >= lastVisibleStartIndex && selectedItemRowIndex <= lastVisibleStopIndex ||
     // view more is always visible, hence no scrolling
     isViewMoreSelected;
 
     //Should scroll to the list item only when the selectedIndex >= 0 and item is not visible
     if (!isSelectedItemVisible && selectedIndex !== -1) {
-      listRef.current.scrollToRow(selectedIndex);
+      listRef.current.scrollToRow(selectedItemRowIndex);
     } else if (selectedIndex === -1) {
       listRef.current.scrollToRow(0);
     }
-  }, [selectedIndex, lastVisibleStartIndex, lastVisibleStopIndex, itemsLength, showMoreOptionsButton]);
+  }, [selectedIndex, lastVisibleStartIndex, lastVisibleStopIndex, itemsLength, showMoreOptionsButton, selectedItemRowIndex]);
   (0, _react.useLayoutEffect)(function () {
     setCache(new _CellMeasurer.CellMeasurerCache({
       fixedWidth: true,
@@ -212,10 +275,10 @@ var TypeAheadListComponent = /*#__PURE__*/_react.default.memo(function (_ref2) {
         listContainerRef.current.firstChild.scrollTo(0, 0);
       }
     });
-  }, [items]);
+  }, [items, sections]);
   (0, _react.useLayoutEffect)(function () {
     // Exclude view more item from the count
-    var itemsToRender = showMoreOptionsButton ? items.slice(0, -1) : items;
+    var itemsToRender = (0, _experiments.editorExperiment)('platform_editor_agent_mentions', true) ? listRows : showMoreOptionsButton ? items.slice(0, -1) : items;
     var height = Math.min(
     // eslint-disable-next-line @atlassian/perf-linting/no-expensive-computations-in-render -- Ignored via go/ees017 (to be fixed)
     itemsToRender.reduce(function (prevValue, currentValue, index) {
@@ -224,7 +287,7 @@ var TypeAheadListComponent = /*#__PURE__*/_react.default.memo(function (_ref2) {
       });
     }, 0), fitHeight);
     setHeight(height);
-  }, [items, cache, fitHeight, showMoreOptionsButton]);
+  }, [listRows, items, cache, fitHeight, showMoreOptionsButton]);
   (0, _react.useLayoutEffect)(function () {
     if (!listContainerRef.current) {
       return;
@@ -287,8 +350,8 @@ var TypeAheadListComponent = /*#__PURE__*/_react.default.memo(function (_ref2) {
     if (onMoreOptionsClicked) {
       onMoreOptionsClicked();
     }
-    api === null || api === void 0 || api.core.actions.execute(function (_ref4) {
-      var tr = _ref4.tr;
+    api === null || api === void 0 || api.core.actions.execute(function (_ref5) {
+      var tr = _ref5.tr;
       (0, _closeTypeAhead.closeTypeAhead)(tr);
       config === null || config === void 0 || config.onClick({
         tr: tr
@@ -296,22 +359,25 @@ var TypeAheadListComponent = /*#__PURE__*/_react.default.memo(function (_ref2) {
       return tr;
     });
   };
-  var renderRow = function renderRow(_ref5) {
-    var index = _ref5.index,
-      key = _ref5.key,
-      style = _ref5.style,
-      parent = _ref5.parent,
-      isScrolling = _ref5.isScrolling,
-      isVisible = _ref5.isVisible;
-    var currentItem = items[index];
+  var renderRow = function renderRow(_ref6) {
+    var index = _ref6.index,
+      key = _ref6.key,
+      style = _ref6.style,
+      parent = _ref6.parent,
+      isScrolling = _ref6.isScrolling,
+      isVisible = _ref6.isVisible;
+    var currentRow = listRows[index];
+    if (!currentRow) {
+      return null;
+    }
     return (0, _react2.jsx)(_CellMeasurer.CellMeasurer, {
       key: key,
       cache: cache,
       parent: parent,
       columnIndex: 0,
       rowIndex: index
-    }, function (_ref6) {
-      var measure = _ref6.measure;
+    }, function (_ref7) {
+      var measure = _ref7.measure;
       return (0, _react2.jsx)(_ListRow.ListRow, {
         measure: measure,
         index: index
@@ -323,21 +389,29 @@ var TypeAheadListComponent = /*#__PURE__*/_react.default.memo(function (_ref2) {
         // eslint-disable-next-line @atlassian/perf-linting/no-unstable-inline-props -- Ignored via go/ees017 (to be fixed)
         ,
         onMouseMove: function onMouseMove(e) {
-          return _onMouseMove(e, index);
+          return _onMouseMove(e, currentRow);
         }
-      }, (0, _react2.jsx)(_TypeAheadListItem.TypeAheadListItem, {
-        key: items[index].title,
-        item: currentItem,
+      }, currentRow.type === 'section' ? populatedSectionCount === 1 ? null : (0, _react2.jsx)(_compiled.Box, {
+        paddingInline: "space.150",
+        paddingBlock: "space.050"
+      }, (0, _react2.jsx)(_compiled.Text, {
+        as: "span",
+        size: "small",
+        color: "color.text.subtle",
+        weight: "medium"
+      }, currentRow.section.title)) : (0, _react2.jsx)(_TypeAheadListItem.TypeAheadListItem, {
+        key: items[currentRow.itemIndex].title,
+        item: items[currentRow.itemIndex],
         firstOnlineSupportedIndex: firstOnlineSupportedRow,
         itemsLength: itemsLength,
-        itemIndex: index,
+        itemIndex: currentRow.itemIndex,
         selectedIndex: selectedIndex
         // eslint-disable-next-line @atlassian/perf-linting/no-unstable-inline-props -- Ignored via go/ees017 (to be fixed)
         ,
-        onItemClick: function onItemClick(mode, index) {
-          actions.onItemClick(mode, index, _analytics.INPUT_METHOD.MOUSE);
+        onItemClick: function onItemClick(mode, itemIndex) {
+          actions.onItemClick(mode, itemIndex, _analytics.INPUT_METHOD.MOUSE);
         },
-        ariaLabel: (0, _utils.getTypeAheadListAriaLabels)(triggerHandler === null || triggerHandler === void 0 ? void 0 : triggerHandler.trigger, intl, currentItem).listItemAriaLabel,
+        ariaLabel: (0, _utils.getTypeAheadListAriaLabels)(triggerHandler === null || triggerHandler === void 0 ? void 0 : triggerHandler.trigger, intl, items[currentRow.itemIndex]).listItemAriaLabel,
         moreElementsInQuickInsertViewEnabled: moreElementsInQuickInsertViewEnabled,
         api: api,
         lastInputMethodRef: lastInputMethodRef
@@ -371,7 +445,7 @@ var TypeAheadListComponent = /*#__PURE__*/_react.default.memo(function (_ref2) {
     ref: listRef
     // Skip rendering the view more button in the list
     ,
-    rowCount: itemsLength,
+    rowCount: listRows.length,
     rowHeight: cache.rowHeight,
     onRowsRendered: onItemsRendered,
     width: LIST_WIDTH,

package/dist/cjs/ui/TypeAheadMenu.js

@@ -23,6 +23,8 @@ var TypeAheadMenu = exports.TypeAheadMenu = /*#__PURE__*/_react.default.memo(fun
   var isOpen = typeAheadState.decorationSet.find().length > 0;
   var triggerHandler = typeAheadState.triggerHandler,
     items = typeAheadState.items,
+    _typeAheadState$secti = typeAheadState.sections,
+    sections = _typeAheadState$secti === void 0 ? [] : _typeAheadState$secti,
     errorInfo = typeAheadState.errorInfo,
     decorationElement = typeAheadState.decorationElement,
     decorationSet = typeAheadState.decorationSet,
@@ -103,6 +105,7 @@ var TypeAheadMenu = exports.TypeAheadMenu = /*#__PURE__*/_react.default.memo(fun
     anchorElement: decorationElement,
     triggerHandler: triggerHandler,
     items: items,
+    sections: sections,
     emptyItem: emptyItem,
     errorInfo: errorInfo,
     selectedIndex: selectedIndex,

package/dist/cjs/ui/TypeAheadPopup.js

@@ -70,6 +70,8 @@ var TypeAheadPopup = exports.TypeAheadPopup = /*#__PURE__*/_react.default.memo(f
     popupsBoundariesElement = props.popupsBoundariesElement,
     popupsScrollableElement = props.popupsScrollableElement,
     items = props.items,
+    _props$sections = props.sections,
+    sections = _props$sections === void 0 ? [] : _props$sections,
     emptyItem = props.emptyItem,
     errorInfo = props.errorInfo,
     selectedIndex = props.selectedIndex,
@@ -345,6 +347,7 @@ var TypeAheadPopup = exports.TypeAheadPopup = /*#__PURE__*/_react.default.memo(f
     triggerHandler: triggerHandler
   }), (0, _react2.jsx)(_TypeAheadList.TypeAheadList, {
     items: items,
+    sections: sections,
     emptyItem: emptyItem,
     selectedIndex: selectedIndex
     // eslint-disable-next-line @atlassian/perf-linting/no-unstable-inline-props -- Ignored via go/ees017 (to be fixed)

package/dist/cjs/ui/WrapperTypeAhead.js

@@ -8,6 +8,7 @@ Object.defineProperty(exports, "__esModule", {
 exports.WrapperTypeAhead = void 0;
 var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
 var _react = _interopRequireWildcard(require("react"));
+var _reactIntl = require("react-intl");
 var _analytics = require("@atlaskit/editor-common/analytics");
 var _typeAhead = require("@atlaskit/editor-common/type-ahead");
 var _platformFeatureFlags = require("@atlaskit/platform-feature-flags");
@@ -40,6 +41,7 @@ var WrapperTypeAhead = exports.WrapperTypeAhead = /*#__PURE__*/_react.default.me
   if ((0, _experiments.editorExperiment)('platform_editor_controls', 'variant1')) {
     showMoreOptionsButton = !!(triggerHandler !== null && triggerHandler !== void 0 && triggerHandler.getMoreOptionsButtonConfig);
   }
+  var intl = (0, _reactIntl.useIntl)();
   var _useState = (0, _react.useState)(false),
     _useState2 = (0, _slicedToArray2.default)(_useState, 2),
     closed = _useState2[0],
@@ -50,7 +52,7 @@ var WrapperTypeAhead = exports.WrapperTypeAhead = /*#__PURE__*/_react.default.me
     setQuery = _useState4[1];
   var queryRef = (0, _react.useRef)(query);
   var editorViewRef = (0, _react.useRef)(editorView);
-  var items = (0, _useLoadItems.useLoadItems)(triggerHandler, editorView, query, showMoreOptionsButton, api);
+  var items = (0, _useLoadItems.useLoadItems)(triggerHandler, editorView, query, showMoreOptionsButton, api, intl);
   (0, _react.useEffect)(function () {
     if (!closed && (0, _platformFeatureFlags.fg)('platform_editor_ease_of_use_metrics')) {
       var _api$metrics;

package/dist/cjs/ui/hooks/build-sectioned-result.js

@@ -0,0 +1,131 @@
+"use strict";
+
+var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.buildSectionedResult = void 0;
+var _toConsumableArray2 = _interopRequireDefault(require("@babel/runtime/helpers/toConsumableArray"));
+var _experiments = require("@atlaskit/tmp-editor-statsig/experiments");
+function _createForOfIteratorHelper(r, e) { var t = "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (!t) { if (Array.isArray(r) || (t = _unsupportedIterableToArray(r)) || e && r && "number" == typeof r.length) { t && (r = t); var _n = 0, F = function F() {}; return { s: F, n: function n() { return _n >= r.length ? { done: !0 } : { done: !1, value: r[_n++] }; }, e: function e(r) { throw r; }, f: F }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } var o, a = !0, u = !1; return { s: function s() { t = t.call(r); }, n: function n() { var r = t.next(); return a = r.done, r; }, e: function e(r) { u = !0, o = r; }, f: function f() { try { a || null == t.return || t.return(); } finally { if (u) throw o; } } }; }
+function _unsupportedIterableToArray(r, a) { if (r) { if ("string" == typeof r) return _arrayLikeToArray(r, a); var t = {}.toString.call(r).slice(8, -1); return "Object" === t && r.constructor && (t = r.constructor.name), "Map" === t || "Set" === t ? Array.from(r) : "Arguments" === t || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(t) ? _arrayLikeToArray(r, a) : void 0; } }
+function _arrayLikeToArray(r, a) { (null == a || a > r.length) && (a = r.length); for (var e = 0, n = Array(a); e < a; e++) n[e] = r[e]; return n; }
+var buildSectionedResult = exports.buildSectionedResult = function buildSectionedResult(_ref) {
+  var _triggerHandler$getSe;
+  var items = _ref.items,
+    triggerHandler = _ref.triggerHandler,
+    intl = _ref.intl;
+  if (!(0, _experiments.editorExperiment)('platform_editor_agent_mentions', true) || !intl) {
+    return {
+      items: items,
+      sections: []
+    };
+  }
+  var sectionDefinitions = (_triggerHandler$getSe = triggerHandler.getSections) === null || _triggerHandler$getSe === void 0 ? void 0 : _triggerHandler$getSe.call(triggerHandler, {
+    intl: intl
+  });
+  if (!sectionDefinitions || sectionDefinitions.length === 0) {
+    return {
+      items: items,
+      sections: []
+    };
+  }
+
+  // Track which item indexes have been claimed by a section, or excluded (matched a section but
+  // cut by that section's limit). Items excluded by limit should not appear anywhere in the output.
+  var assignedToSection = new Set();
+  var excludedByLimit = new Set();
+  var groupedBySection = new Map();
+  var _iterator = _createForOfIteratorHelper(sectionDefinitions),
+    _step;
+  try {
+    for (_iterator.s(); !(_step = _iterator.n()).done;) {
+      var section = _step.value;
+      var matchingIndexes = [];
+      for (var _i = 0; _i < items.length; _i++) {
+        if (assignedToSection.has(_i) || excludedByLimit.has(_i)) {
+          continue;
+        }
+        if (section.filter(items[_i])) {
+          matchingIndexes.push(_i);
+        }
+      }
+      var acceptedIndexes = section.limit !== undefined ? matchingIndexes.slice(0, section.limit) : matchingIndexes;
+      var rejectedByLimit = matchingIndexes.slice(acceptedIndexes.length);
+      var _iterator3 = _createForOfIteratorHelper(acceptedIndexes),
+        _step3;
+      try {
+        for (_iterator3.s(); !(_step3 = _iterator3.n()).done;) {
+          var _i2 = _step3.value;
+          assignedToSection.add(_i2);
+        }
+      } catch (err) {
+        _iterator3.e(err);
+      } finally {
+        _iterator3.f();
+      }
+      var _iterator4 = _createForOfIteratorHelper(rejectedByLimit),
+        _step4;
+      try {
+        for (_iterator4.s(); !(_step4 = _iterator4.n()).done;) {
+          var _i3 = _step4.value;
+          excludedByLimit.add(_i3);
+        }
+      } catch (err) {
+        _iterator4.e(err);
+      } finally {
+        _iterator4.f();
+      }
+      if (acceptedIndexes.length === 0) {
+        continue;
+      }
+      groupedBySection.set(section.id, {
+        indexes: acceptedIndexes,
+        section: section
+      });
+    }
+  } catch (err) {
+    _iterator.e(err);
+  } finally {
+    _iterator.f();
+  }
+  var flattenedItems = [];
+  var sections = [];
+  var _iterator2 = _createForOfIteratorHelper(sectionDefinitions),
+    _step2;
+  try {
+    for (_iterator2.s(); !(_step2 = _iterator2.n()).done;) {
+      var _section = _step2.value;
+      var grouped = groupedBySection.get(_section.id);
+      if (!grouped) {
+        continue;
+      }
+      var startIndex = flattenedItems.length;
+      flattenedItems.push.apply(flattenedItems, (0, _toConsumableArray2.default)(grouped.indexes.map(function (i) {
+        return items[i];
+      })));
+      var endIndex = flattenedItems.length - 1;
+      sections.push({
+        endIndex: endIndex,
+        id: _section.id,
+        startIndex: startIndex,
+        title: _section.title
+      });
+    }
+
+    // Append items not claimed by any section and not excluded by a limit
+  } catch (err) {
+    _iterator2.e(err);
+  } finally {
+    _iterator2.f();
+  }
+  for (var i = 0; i < items.length; i++) {
+    if (!assignedToSection.has(i) && !excludedByLimit.has(i)) {
+      flattenedItems.push(items[i]);
+    }
+  }
+  return {
+    items: flattenedItems,
+    sections: sections
+  };
+};