diff --git a/frontend/src/hooks/common/use-document-search.spec.ts b/frontend/src/hooks/common/use-document-search.spec.ts new file mode 100644 index 000000000..a0f514f54 --- /dev/null +++ b/frontend/src/hooks/common/use-document-search.spec.ts @@ -0,0 +1,64 @@ +/* + * SPDX-FileCopyrightText: 2023 The HedgeDoc developers (see AUTHORS file) + * + * SPDX-License-Identifier: AGPL-3.0-only + */ +import type { SearchIndexEntry } from './use-document-search' +import { useDocumentSearch } from './use-document-search' +import { renderHook } from '@testing-library/react' + +describe('useDocumentSearch', () => { + interface TestSearchIndexEntry extends SearchIndexEntry { + name: string + text: string + } + + const searchOptions = { + document: { + id: 'id', + field: ['name', 'text'] + } + } + const searchEntries: TestSearchIndexEntry[] = [ + { + id: 1, + name: 'Foo', + text: 'Lorem ipsum dolor sit amet, consectetur adipiscing elit. Aenean fermentum odio in bibendum venenatis. Cras aliquet ultrices finibus. Lorem ipsum dolor sit amet, consectetur adipiscing elit.' + }, + { + id: 2, + name: 'Bar', + text: 'Vivamus sed mauris eget magna sodales blandit. Aliquam tincidunt nunc et sapien scelerisque placerat. Pellentesque a orci ac risus molestie suscipit id vel arcu.' + }, + { + id: 3, + name: 'Cras', + text: 'Cras consectetur sit amet tortor eget sollicitudin. Ut convallis orci ipsum, eget dignissim nibh dignissim eget. Nunc commodo est neque, eget venenatis urna condimentum eget. Suspendisse dapibus ligula et enim venenatis hendrerit. ' + } + ] + it('results get populated', () => { + const { result, rerender } = renderHook( + (searchTerm: string) => useDocumentSearch(searchEntries, searchOptions, searchTerm), + { + initialProps: '' + } + ) + rerender('Foo') + + expect(result.current).toHaveLength(1) + expect(result.current[0]).toEqual({ field: 'name', result: [1] }) + }) + it('finds in multiple fields', () => { + const { result, rerender } = renderHook( + (searchTerm: string) => useDocumentSearch(searchEntries, searchOptions, searchTerm), + { + initialProps: '' + } + ) + rerender('Cras') + + expect(result.current).toHaveLength(2) + expect(result.current[0]).toEqual({ field: 'name', result: [3] }) + expect(result.current[1]).toEqual({ field: 'text', result: [3, 1] }) + }) +}) diff --git a/frontend/src/hooks/common/use-document-search.ts b/frontend/src/hooks/common/use-document-search.ts new file mode 100644 index 000000000..05d30aae6 --- /dev/null +++ b/frontend/src/hooks/common/use-document-search.ts @@ -0,0 +1,43 @@ +/* + * SPDX-FileCopyrightText: 2023 The HedgeDoc developers (see AUTHORS file) + * + * SPDX-License-Identifier: AGPL-3.0-only + */ +import type { IndexOptionsForDocumentSearch, Id, SimpleDocumentSearchResultSetUnit } from 'flexsearch-ts' +import { Document } from 'flexsearch-ts' +import { useEffect, useMemo, useState } from 'react' + +export interface SearchIndexEntry { + id: Id +} + +/** + * Generate document search index and provide functions to search. + * + * @param entries The list of entries to built the search index from + * @param options Options for the search index + * @param searchTerm What to search for + * @return An array of the search results + */ +export const useDocumentSearch = ( + entries: Array, + options: IndexOptionsForDocumentSearch, + searchTerm: string +): SimpleDocumentSearchResultSetUnit[] => { + const [results, setResults] = useState([]) + const searchIndex = useMemo(() => { + const index = new Document({ + tokenize: 'full', + ...options + }) + entries.forEach((entry) => { + index.add(entry) + }) + return index + }, [entries, options]) + useEffect(() => { + setResults(searchIndex.search(searchTerm)) + }, [searchIndex, searchTerm]) + + return results +}