mirror of
https://github.com/hedgedoc/hedgedoc.git
synced 2024-11-24 18:56:32 -05:00
feat(frontend): add useDocumentSearch hook
Signed-off-by: Philip Molares <philip.molares@udo.edu>
This commit is contained in:
parent
1a3d655ca7
commit
1624ff9c3b
2 changed files with 107 additions and 0 deletions
64
frontend/src/hooks/common/use-document-search.spec.ts
Normal file
64
frontend/src/hooks/common/use-document-search.spec.ts
Normal 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] })
|
||||
})
|
||||
})
|
43
frontend/src/hooks/common/use-document-search.ts
Normal file
43
frontend/src/hooks/common/use-document-search.ts
Normal 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
|
||||
}
|
Loading…
Reference in a new issue