Mathias JakobsenandCopybot
eddec90cb1
[web] Add command palette GitOrigin-RevId: 5bf1903836810ca5f0e2bc7f6c00a4b1da797ea2
78 lines
2.0 KiB
TypeScript
78 lines
2.0 KiB
TypeScript
import { useCallback, useMemo } from 'react'
|
|
import MiniSearch from 'minisearch'
|
|
import { CommandPaletteSearchResult, CommandPaletteSource } from '../types'
|
|
import {
|
|
Command,
|
|
useCommandRegistry,
|
|
} from '@/features/ide-react/context/command-registry-context'
|
|
|
|
const ENABLED_COMMANDS: string[] = [
|
|
'new_file',
|
|
'new_folder',
|
|
'upload_file',
|
|
'open-settings',
|
|
'show_version_history',
|
|
'word_count',
|
|
'view-pdf-presentation-mode',
|
|
'comment',
|
|
'compile',
|
|
'stop-compile',
|
|
'recompile-from-scratch',
|
|
]
|
|
|
|
const useCommandRegistrySource = (): CommandPaletteSource => {
|
|
const { registry } = useCommandRegistry()
|
|
|
|
const commands = useMemo(() => {
|
|
const enabled = new Set(ENABLED_COMMANDS)
|
|
return [...registry.values()].filter(
|
|
c => enabled.has(c.id) && !c.disabled && c.handler
|
|
)
|
|
}, [registry])
|
|
|
|
const defaults = useCallback((): CommandPaletteSearchResult[] => {
|
|
return commands.map(command => ({
|
|
title: command.label,
|
|
onSelect: () => command.handler!({ location: 'command-palette' }),
|
|
score: 1,
|
|
}))
|
|
}, [commands])
|
|
|
|
const index = useMemo(() => {
|
|
const miniSearch = new MiniSearch<Command>({
|
|
fields: ['label'],
|
|
storeFields: ['id'],
|
|
idField: 'id',
|
|
})
|
|
miniSearch.addAll(commands)
|
|
return miniSearch
|
|
}, [commands])
|
|
|
|
return useMemo<CommandPaletteSource>(
|
|
() => ({
|
|
id: 'command-registry',
|
|
search(query) {
|
|
const results = index.search(query, {
|
|
prefix: true,
|
|
fuzzy: term => (term.length > 3 ? 0.2 : false),
|
|
})
|
|
return results.flatMap(({ id, score }) => {
|
|
const command = registry.get(id)
|
|
if (!command?.handler) return []
|
|
return [
|
|
{
|
|
title: command.label,
|
|
onSelect: () => command.handler!({ location: 'command-palette' }),
|
|
score,
|
|
},
|
|
]
|
|
})
|
|
},
|
|
defaults,
|
|
}),
|
|
[index, registry, defaults]
|
|
)
|
|
}
|
|
|
|
export default useCommandRegistrySource
|