Skip to content

Commit b5a193a

Browse files
author
Michael Jordan
authored
Implement doc search using SearchAutocomplete and the AlgoliaSearch API (#3383)
1 parent 950d45d commit b5a193a

File tree

5 files changed

+607
-5
lines changed

5 files changed

+607
-5
lines changed

packages/dev/docs/package.json

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,9 @@
2323
"@spectrum-icons/illustrations": "^3.1.0",
2424
"@spectrum-icons/ui": "^3.1.0",
2525
"@spectrum-icons/workflow": "^4.0.0",
26+
"algoliasearch": "^4.14.1",
2627
"clsx": "^1.1.1",
28+
"dompurify": "^2.3.10",
2729
"globals-docs": "^2.4.1",
2830
"highlight.js": "9.18.1",
2931
"markdown-to-jsx": "^6.11.0",

packages/dev/docs/src/DocSearch.js

Lines changed: 347 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,347 @@
1+
import algoliasearch from 'algoliasearch/lite';
2+
import docsStyle from './docs.css';
3+
import DocumentOutline from '@spectrum-icons/workflow/DocumentOutline';
4+
import DOMPurify from 'dompurify';
5+
import {Icon} from '@react-spectrum/icon';
6+
import {Item, SearchAutocomplete, Section} from '@react-spectrum/autocomplete';
7+
import News from '@spectrum-icons/workflow/News';
8+
import React, {useEffect, useRef, useState} from 'react';
9+
import {Text} from '@adobe/react-spectrum';
10+
import {ThemeProvider} from './ThemeSwitcher';
11+
12+
let client = algoliasearch('1V1Q59JVTR', '44a7e2e7508ff185f25ac64c0a675f98');
13+
let searchIndex = client.initIndex('react-spectrum');
14+
15+
const searchOptions = {
16+
distinct: 1,
17+
attributesToRetrieve: [
18+
'hierarchy.lvl0',
19+
'hierarchy.lvl1',
20+
'hierarchy.lvl2',
21+
'hierarchy.lvl3',
22+
'hierarchy.lvl4',
23+
'hierarchy.lvl5',
24+
'hierarchy.lvl6',
25+
'content',
26+
'type',
27+
'url'
28+
],
29+
attributesToSnippet: [
30+
'hierarchy.lvl1:10',
31+
'hierarchy.lvl2:10',
32+
'hierarchy.lvl3:10',
33+
'hierarchy.lvl4:10',
34+
'hierarchy.lvl5:10',
35+
'hierarchy.lvl6:10',
36+
'content:10'
37+
],
38+
snippetEllipsisText: '…',
39+
highlightPreTag: `<mark role="presentation" class="${docsStyle.docSearchBoxMark}">`,
40+
highlightPostTag: '</mark>',
41+
hitsPerPage: 20
42+
};
43+
44+
const sectionTitles = {
45+
'react-aria': 'React Aria',
46+
'react-spectrum': 'React Spectrum',
47+
'react-stately': 'React Stately',
48+
'internationalized': 'Internationalized',
49+
'blog': 'Blog',
50+
'architecture': 'Architecture',
51+
'contribute': 'Contribute',
52+
'releases': 'Releases',
53+
'support': 'Support'
54+
};
55+
56+
function sectionTitlePredicate(hit) {
57+
let sectionTitle;
58+
for (const [path, title] of Object.entries(sectionTitles)) {
59+
let regexp = new RegExp('^.+//.+/' + path + '[/.].+$', 'i');
60+
if (hit.url.match(regexp)) {
61+
sectionTitle = title;
62+
break;
63+
}
64+
}
65+
if (!sectionTitle) {
66+
sectionTitle = 'Documentation';
67+
}
68+
return sectionTitle;
69+
}
70+
71+
export default function DocSearch() {
72+
const [searchValue, setSearchValue] = useState('');
73+
const [loadingState, setLoadingState] = useState(null);
74+
const [predictions, setPredictions] = useState(null);
75+
const [suggestions, setSuggestions] = useState([]);
76+
77+
const searchAutocompleteRef = useRef();
78+
79+
useEffect(() => {
80+
// Focus search input when user presses Ctrl/Cmd + K
81+
let handleKeyDown = (e) => {
82+
if ((e.ctrlKey || e.metaKey) && e.key === 'k') {
83+
e.preventDefault();
84+
searchAutocompleteRef.current.focus();
85+
}
86+
};
87+
88+
document.addEventListener('keydown', handleKeyDown);
89+
90+
return () => {
91+
document.removeEventListener('keydown', handleKeyDown);
92+
};
93+
}, []);
94+
95+
let updatePredictions = ({hits}) => {
96+
setPredictions(hits);
97+
98+
const groupedBySection = groupBy(hits, (hit) => sectionTitlePredicate(hit));
99+
let sections = [];
100+
for (const [title, hits] of Object.entries(groupedBySection)) {
101+
const items = Object.values(
102+
groupBy(hits, (hit) => hit.hierarchy.lvl1)
103+
)
104+
.map(
105+
groupedHits =>
106+
groupedHits.map((hit) => {
107+
const hierarchy = hit.hierarchy;
108+
const objectID = hit.objectID;
109+
const docsearchParent = hit.type !== 'lvl1' &&
110+
groupedHits.find(
111+
(siblingItem) =>
112+
siblingItem.type === 'lvl1' &&
113+
siblingItem.hierarchy.lvl1 ===
114+
hit.hierarchy.lvl1
115+
);
116+
117+
return {
118+
key: objectID,
119+
hit,
120+
hierarchy,
121+
docsearchParent,
122+
textValue: hit.type === 'content' ? hit[hit.type] : hierarchy[hit.type]
123+
};
124+
}
125+
));
126+
127+
sections.push({title, items: items.map((item) => item[0])});
128+
}
129+
let newSuggestions = sections.map((section, index) => ({key: `${index}-${section.title}`, title: section.title, children: section.items}));
130+
newSuggestions.push({
131+
key: 'algolia-footer',
132+
title: 'Search by',
133+
children: [
134+
{
135+
key: 'algolia-footer-logo',
136+
textValue: 'Algolia'
137+
}
138+
]
139+
});
140+
setSuggestions(newSuggestions);
141+
setLoadingState(null);
142+
};
143+
144+
let onInputChange = (query) => {
145+
setSearchValue(query);
146+
if (!query && predictions) {
147+
setPredictions(null);
148+
setSuggestions([]);
149+
setLoadingState(null);
150+
return;
151+
}
152+
setLoadingState('filtering');
153+
searchIndex
154+
.search(
155+
query,
156+
searchOptions
157+
)
158+
.then(updatePredictions);
159+
};
160+
161+
let onSubmit = (value, key) => {
162+
if (key === 'algolia-footer-logo') {
163+
window.open('https://www.algolia.com/ref/docsearch/?utm_source=react-spectrum.adobe.com&amp;utm_medium=referral&amp;utm_content=powered_by&amp;utm_campaign=docsearch', '_blank');
164+
searchAutocompleteRef.current.UNSAFE_getDOMNode().querySelector('[role="button"]').click();
165+
} else if (key) {
166+
let prediction = predictions.find(prediction => key === prediction.objectID);
167+
let url = prediction.url;
168+
window.location.href = `${window.location.hostname === 'reactspectrum.blob.core.windows.net' ? window.location.href.replace(/(.+\/docs\/)(.+)/, '$1') : '/'}${url.replace('https://react-spectrum.adobe.com/', '')}`;
169+
}
170+
};
171+
172+
return (
173+
<ThemeProvider UNSAFE_className={docsStyle.docSearchBoxThemeProvider}>
174+
<span role="search">
175+
<SearchAutocomplete
176+
ref={searchAutocompleteRef}
177+
aria-label="Search"
178+
UNSAFE_className={docsStyle.docSearchBox}
179+
items={suggestions}
180+
loadingState={loadingState}
181+
value={searchValue}
182+
onInputChange={onInputChange}
183+
onOpenChange={(isOpen) => {
184+
if (isOpen) {
185+
document.body.classList.add(docsStyle['docsearch-open']);
186+
} else {
187+
document.body.classList.remove(docsStyle['docsearch-open']);
188+
}
189+
}}
190+
onSubmit={onSubmit}>
191+
{(section) => (
192+
<Section key={section.key} items={section.children} title={section.title}>
193+
{(item) => (
194+
<Item key={item.key} textValue={item.textValue}>
195+
{item.key === 'algolia-footer-logo' && (
196+
<Text>
197+
<svg width="77" height="19" aria-label="Algolia" role="img" style={{color: 'var(--docsearch-logo-color)'}}>
198+
<path d="M2.5067 0h14.0245c1.384.001 2.5058 1.1205 2.5068 2.5017V16.5c-.0014 1.3808-1.1232 2.4995-2.5068 2.5H2.5067C1.1232 18.9995.0014 17.8808 0 16.5V2.4958A2.495 2.495 0 01.735.7294 2.505 2.505 0 012.5068 0zM37.95 15.0695c-3.7068.0168-3.7068-2.986-3.7068-3.4634L34.2372.3576 36.498 0v11.1794c0 .2715 0 1.9889 1.452 1.994v1.8961zm-9.1666-1.8388c.694 0 1.2086-.0397 1.5678-.1088v-2.2934a5.3639 5.3639 0 00-1.3303-.1679 4.8283 4.8283 0 00-.758.0582 2.2845 2.2845 0 00-.688.2024c-.2029.0979-.371.2362-.4919.4142-.1268.1788-.185.2826-.185.5533 0 .5297.185.8359.5205 1.0375.3355.2016.7928.3053 1.365.3053v-.0008zm-.1969-8.1817c.7463 0 1.3768.092 1.8856.2767.5088.1838.9195.4428 1.2204.7717.3068.334.5147.7777.6423 1.251.1327.4723.196.991.196 1.5603v5.798c-.5235.1036-1.05.192-1.5787.2649-.7048.1037-1.4976.156-2.3774.156-.5832 0-1.1215-.0582-1.6016-.167a3.385 3.385 0 01-1.2432-.5364 2.6034 2.6034 0 01-.8037-.9565c-.191-.3922-.29-.9447-.29-1.5208 0-.5533.11-.905.3246-1.2863a2.7351 2.7351 0 01.8849-.9329c.376-.242.8029-.415 1.2948-.5187a7.4517 7.4517 0 011.5381-.156 7.1162 7.1162 0 011.6667.2024V8.886c0-.259-.0296-.5061-.093-.7372a1.5847 1.5847 0 00-.3245-.6158 1.5079 1.5079 0 00-.6119-.4158 2.6788 2.6788 0 00-.966-.173c-.5206 0-.9948.0634-1.4283.1384a6.5481 6.5481 0 00-1.065.259l-.2712-1.849c.2831-.0986.7048-.1964 1.2491-.2943a9.2979 9.2979 0 011.752-.1501v.0008zm44.6597 8.1193c.6947 0 1.2086-.0405 1.567-.1097v-2.2942a5.3743 5.3743 0 00-1.3303-.1679c-.2485 0-.503.0177-.7573.0582a2.2853 2.2853 0 00-.688.2024 1.2333 1.2333 0 00-.4918.4142c-.1268.1788-.1843.2826-.1843.5533 0 .5297.1843.8359.5198 1.0375.3414.2066.7927.3053 1.365.3053v.0009zm-.191-8.1767c.7463 0 1.3768.0912 1.8856.2759.5087.1847.9195.4436 1.2204.7717.3.329.5147.7786.6414 1.251a5.7248 5.7248 0 01.197 1.562v5.7972c-.3466.0742-.874.1602-1.5788.2648-.7049.1038-1.4976.1552-2.3774.1552-.5832 0-1.1215-.0573-1.6016-.167a3.385 3.385 0 01-1.2432-.5356 2.6034 2.6034 0 01-.8038-.9565c-.191-.3922-.2898-.9447-.2898-1.5216 0-.5533.1098-.905.3245-1.2854a2.7373 2.7373 0 01.8849-.9338c.376-.2412.8029-.4141 1.2947-.5178a7.4545 7.4545 0 012.325-.1097c.2781.0287.5672.081.879.156v-.3686a2.7781 2.7781 0 00-.092-.738 1.5788 1.5788 0 00-.3246-.6166 1.5079 1.5079 0 00-.612-.415 2.6797 2.6797 0 00-.966-.1729c-.5205 0-.9947.0633-1.4282.1384a6.5608 6.5608 0 00-1.065.259l-.2712-1.8498c.283-.0979.7048-.1957 1.2491-.2935a9.8597 9.8597 0 011.752-.1494zm-6.79-1.072c-.7576.001-1.373-.6103-1.3759-1.3664 0-.755.6128-1.3664 1.376-1.3664.764 0 1.3775.6115 1.3775 1.3664s-.6195 1.3664-1.3776 1.3664zm1.1393 11.1507h-2.2726V5.3409l2.2734-.3568v10.0845l-.0008.0017zm-3.984 0c-3.707.0168-3.707-2.986-3.707-3.4642L59.7069.3576 61.9685 0v11.1794c0 .2715 0 1.9889 1.452 1.994V15.0703zm-7.3512-4.979c0-.975-.2138-1.7873-.6305-2.3516-.4167-.571-.9998-.852-1.747-.852-.7454 0-1.3302.281-1.7452.852-.4166.5702-.6195 1.3765-.6195 2.3516 0 .9851.208 1.6473.6254 2.2183.4158.576.9998.8587 1.7461.8587.7454 0 1.3303-.2885 1.747-.8595.4158-.5761.6237-1.2315.6237-2.2184v.0009zm2.3132-.006c0 .7609-.1099 1.3361-.3356 1.9654a4.654 4.654 0 01-.9533 1.6076A4.214 4.214 0 0155.613 14.69c-.579.2412-1.4697.3795-1.9143.3795-.4462-.005-1.3303-.1324-1.9033-.3795a4.307 4.307 0 01-1.474-1.0316c-.4115-.4445-.7293-.9801-.9609-1.6076a5.3423 5.3423 0 01-.3465-1.9653c0-.7608.104-1.493.3356-2.1155a4.683 4.683 0 01.9719-1.5958 4.3383 4.3383 0 011.479-1.0257c.5739-.242 1.2043-.3567 1.8864-.3567.6829 0 1.3125.1197 1.8906.3567a4.1245 4.1245 0 011.4816 1.0257 4.7587 4.7587 0 01.9592 1.5958c.2426.6225.3643 1.3547.3643 2.1155zm-17.0198 0c0 .9448.208 1.9932.6238 2.431.4166.4386.955.6579 1.6142.6579.3584 0 .6998-.0523 1.0176-.1502.3186-.0978.5721-.2134.775-.3517V7.0784a8.8706 8.8706 0 00-1.4926-.1906c-.8206-.0236-1.4452.312-1.8847.8468-.4335.5365-.6533 1.476-.6533 2.3516v-.0008zm6.2863 4.4485c0 1.5385-.3938 2.662-1.1866 3.3773-.791.7136-2.0005 1.0712-3.6308 1.0712-.5958 0-1.834-.1156-2.8228-.334l.3643-1.7865c.8282.173 1.9202.2193 2.4932.2193.9077 0 1.555-.1847 1.943-.5533.388-.3686.578-.916.578-1.643v-.3687a6.8289 6.8289 0 01-.8848.3349c-.3634.1096-.786.167-1.261.167-.6246 0-1.1917-.0979-1.7055-.2944a3.5554 3.5554 0 01-1.3244-.8645c-.3642-.3796-.6541-.8579-.8561-1.4289-.2028-.571-.3068-1.59-.3068-2.339 0-.7034.1099-1.5856.3245-2.1735.2198-.5871.5316-1.0949.9542-1.515.4167-.42.9255-.743 1.5213-.98a5.5923 5.5923 0 012.052-.3855c.7353 0 1.4114.092 2.0707.2024.6592.1088 1.2204.2236 1.6776.35v8.945-.0008zM11.5026 4.2418v-.6511c-.0005-.4553-.3704-.8241-.8266-.8241H8.749c-.4561 0-.826.3688-.8265.824v.669c0 .0742.0693.1264.1445.1096a6.0346 6.0346 0 011.6768-.2362 6.125 6.125 0 011.6202.2185.1116.1116 0 00.1386-.1097zm-5.2806.852l-.3296-.3282a.8266.8266 0 00-1.168 0l-.393.3922a.8199.8199 0 000 1.164l.3237.323c.0524.0515.1268.0397.1733-.0117.191-.259.3989-.507.6305-.7372.2374-.2362.48-.4437.7462-.6335.0575-.0354.0634-.1155.017-.1687zm3.5159 2.069v2.818c0 .081.0879.1392.1622.0987l2.5102-1.2964c.0574-.0287.0752-.0987.0464-.1552a3.1237 3.1237 0 00-2.603-1.574c-.0575 0-.115.0456-.115.1097l-.0008-.0009zm.0008 6.789c-2.0933.0005-3.7915-1.6912-3.7947-3.7804C5.9468 8.0821 7.6452 6.39 9.7387 6.391c2.0932-.0005 3.7911 1.6914 3.794 3.7804a3.7783 3.7783 0 01-1.1124 2.675 3.7936 3.7936 0 01-2.6824 1.1054h.0008zM9.738 4.8002c-1.9218 0-3.6975 1.0232-4.6584 2.6841a5.359 5.359 0 000 5.3683c.9609 1.661 2.7366 2.6841 4.6584 2.6841a5.3891 5.3891 0 003.8073-1.5725 5.3675 5.3675 0 001.578-3.7987 5.3574 5.3574 0 00-1.5771-3.797A5.379 5.379 0 009.7387 4.801l-.0008-.0008z" fill="currentColor" fillRule="evenodd" />
199+
</svg>
200+
</Text>
201+
)}
202+
{
203+
item.hierarchy &&
204+
item.hierarchy[item.hit.type] &&
205+
item.hit.type === 'lvl1' && (
206+
<>
207+
{
208+
section.title === 'Blog' || section.title === 'Releases' ?
209+
<News aria-label="news" /> :
210+
<DocumentOutline aria-label="web page" />
211+
}
212+
<Text>
213+
<Snippet
214+
className="DocSearch-Hit-title"
215+
hit={item.hit}
216+
attribute="hierarchy.lvl1" />
217+
</Text>
218+
{item.hit.content && (
219+
<Text slot="description">
220+
<Snippet
221+
className="DocSearch-Hit-path"
222+
hit={item.hit}
223+
attribute="content" />
224+
</Text>
225+
)}
226+
</>
227+
)
228+
}
229+
230+
{
231+
item.hierarchy && item.hierarchy[item.hit.type] &&
232+
(
233+
item.hit.type === 'lvl2' ||
234+
item.hit.type === 'lvl3' ||
235+
item.hit.type === 'lvl4' ||
236+
item.hit.type === 'lvl5' ||
237+
item.hit.type === 'lvl6'
238+
) && (
239+
<>
240+
<Hash
241+
UNSAFE_className={item.docsearchParent ? docsStyle.docSearchItemIndent : undefined}
242+
aria-label="in-page link" />
243+
<Text>
244+
<Snippet
245+
className="DocSearch-Hit-title"
246+
hit={item.hit}
247+
attribute={`hierarchy.${item.hit.type}`} />
248+
</Text>
249+
<Text slot="description">
250+
<Snippet
251+
className="DocSearch-Hit-path"
252+
hit={item.hit}
253+
attribute="hierarchy.lvl1" />
254+
</Text>
255+
</>
256+
)
257+
}
258+
259+
{
260+
item.hit && item.hit.type === 'content' && (
261+
<>
262+
<Hash
263+
UNSAFE_className={item.docsearchParent ? docsStyle.docSearchItemIndent : undefined}
264+
aria-label="in-page link" />
265+
<Text>
266+
<Snippet
267+
className="DocSearch-Hit-title"
268+
hit={item.hit}
269+
attribute="content" />
270+
</Text>
271+
<Text slot="description">
272+
<Snippet
273+
className="DocSearch-Hit-path"
274+
hit={item.hit}
275+
attribute="hierarchy.lvl1" />
276+
</Text>
277+
</>
278+
)
279+
}
280+
</Item>
281+
)}
282+
</Section>
283+
)}
284+
</SearchAutocomplete>
285+
</span>
286+
</ThemeProvider>
287+
);
288+
}
289+
290+
function Hash(props) {
291+
return (
292+
<Icon {...props}>
293+
<svg viewBox="0 0 20 20">
294+
<path d="M13 13h4-4V8H7v5h6v4-4H7V8H3h4V3v5h6V3v5h4-4v5zm-6 0v4-4H3h4z" stroke="currentColor" fill="none" fillRule="evenodd" strokeLinecap="round" strokeLinejoin="round" />
295+
</svg>
296+
</Icon>
297+
);
298+
}
299+
300+
function groupBy(values, predicate = (value) => value) {
301+
return values.reduce((accumulator, item) => {
302+
const key = predicate(item);
303+
304+
if (!Object.prototype.hasOwnProperty.call(accumulator, key)) {
305+
accumulator[key] = [];
306+
}
307+
308+
// We limit each section to show 20 hits maximum.
309+
// This acts as a frontend alternative to `distinct`.
310+
if (accumulator[key].length < 21) {
311+
accumulator[key].push(item);
312+
}
313+
314+
return accumulator;
315+
}, {});
316+
}
317+
318+
function getPropertyByPath(object, path) {
319+
const parts = path.split('.');
320+
321+
return parts.reduce((prev, current) => {
322+
if (prev?.[current]) {
323+
return prev[current];
324+
}
325+
return null;
326+
}, object);
327+
}
328+
329+
function Snippet({
330+
hit,
331+
attribute,
332+
tagName = 'span',
333+
...rest
334+
}) {
335+
return React.createElement(tagName, {
336+
...rest,
337+
dangerouslySetInnerHTML: {
338+
__html:
339+
DOMPurify.sanitize(
340+
getPropertyByPath(hit, `_snippetResult.${attribute}.value`) ||
341+
getPropertyByPath(hit, attribute)
342+
)
343+
}
344+
});
345+
}
346+
347+

0 commit comments

Comments
 (0)