feat(frontend): add useDocumentSearch hook

Signed-off-by: Philip Molares <philip.molares@udo.edu>
This commit is contained in:
Philip Molares 2023-06-24 15:14:28 +02:00
parent 1a3d655ca7
commit 1624ff9c3b
2 changed files with 107 additions and 0 deletions

View file

@ -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] })
})
})

View file

@ -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 = <T extends SearchIndexEntry>(
entries: Array<T>,
options: IndexOptionsForDocumentSearch<T>,
searchTerm: string
): SimpleDocumentSearchResultSetUnit[] => {
const [results, setResults] = useState<SimpleDocumentSearchResultSetUnit[]>([])
const searchIndex = useMemo(() => {
const index = new Document<T>({
tokenize: 'full',
...options
})
entries.forEach((entry) => {
index.add(entry)
})
return index
}, [entries, options])
useEffect(() => {
setResults(searchIndex.search(searchTerm))
}, [searchIndex, searchTerm])
return results
}