feat: implement maxResultsPerGroup prop

algolia · May 17, 2023 · 7f03b62 · 7f03b62
1 parent 2f21fd9
commit 7f03b62
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/packages/docsearch-react/src/DocSearchModal.tsx b/packages/docsearch-react/src/DocSearchModal.tsx
@@ -46,6 +46,7 @@ export function DocSearchModal({
   indexName,
   placeholder = 'Search docs',
   searchParameters,
+  maxResultsPerGroup,
   onClose = noop,
   transformItems = identity,
   hitComponent = Hit,
@@ -240,7 +241,7 @@ export function DocSearchModal({
             })
             .then(({ results }) => {
               const { hits, nbHits } = results[0];
-              const sources = groupBy(hits, (hit) => removeHighlightTags(hit));
+              const sources = groupBy(hits, (hit) => removeHighlightTags(hit), maxResultsPerGroup);
 
               // We store the `lvl0`s to display them as search suggestions
               // in the "no results" screen.
@@ -271,7 +272,7 @@ export function DocSearchModal({
                     },
                     getItems() {
                       return Object.values(
-                        groupBy(items, (item) => item.hierarchy.lvl1)
+                        groupBy(items, (item) => item.hierarchy.lvl1, maxResultsPerGroup)
                       )
                         .map(transformItems)
                         .map((groupedHits) =>
@@ -300,6 +301,7 @@ export function DocSearchModal({
     [
       indexName,
       searchParameters,
+      maxResultsPerGroup,
       searchClient,
       onClose,
       recentSearches,

diff --git a/packages/docsearch-react/src/utils/groupBy.ts b/packages/docsearch-react/src/utils/groupBy.ts
@@ -1,6 +1,7 @@
 export function groupBy<TValue extends Record<string, unknown>>(
   values: TValue[],
-  predicate: (value: TValue) => string
+  predicate: (value: TValue) => string,
+  maxResultsPerGroup?: number
 ): Record<string, TValue[]> {
   return values.reduce<Record<string, TValue[]>>((acc, item) => {
     const key = predicate(item);
@@ -11,7 +12,7 @@ export function groupBy<TValue extends Record<string, unknown>>(
 
     // We limit each section to show 5 hits maximum.
     // This acts as a frontend alternative to `distinct`.
-    if (acc[key].length < 5) {
+    if (acc[key].length < (maxResultsPerGroup || 5)) {
       acc[key].push(item);
     }