import { ActionIcon, Autocomplete, Box, createStyles, Divider, Kbd, Menu, Popover, ScrollArea, TextInput, Tooltip, } from '@mantine/core'; import { IconSearch, IconBrandYoutube, IconDownload, IconMovie } from '@tabler/icons'; import React, { forwardRef, useEffect, useRef, useState } from 'react'; import { useDebouncedValue, useHotkeys } from '@mantine/hooks'; import { showNotification } from '@mantine/notifications'; import { useTranslation } from 'next-i18next'; import axios from 'axios'; import { IModule } from '../ModuleTypes'; import { useConfig } from '../../tools/state'; import { OverseerrModule } from '../overseerr'; import Tip from '../../components/layout/Tip'; import { OverseerrMediaDisplay } from '../common'; import SmallServiceItem from '../../components/AppShelf/SmallServiceItem'; export const SearchModule: IModule = { title: 'Search', icon: IconSearch, component: SearchModuleComponent, id: 'search', }; interface ItemProps extends React.ComponentPropsWithoutRef<'div'> { label: string; disabled: boolean; value: string; description: string; icon: React.ReactNode; url: string; shortcut: string; } const useStyles = createStyles((theme) => ({ item: { '&[data-hovered]': { backgroundColor: theme.colors[theme.primaryColor][theme.fn.primaryShade()], color: theme.white, }, }, })); export function SearchModuleComponent() { const { config } = useConfig(); const { t } = useTranslation('modules/search'); const [searchQuery, setSearchQuery] = useState(''); const [debounced, cancel] = useDebouncedValue(searchQuery, 250); const isOverseerrEnabled = config.modules?.[OverseerrModule.id]?.enabled ?? false; const OverseerrService = config.services.find( (service) => service.type === 'Overseerr' || service.type === 'Jellyseerr' ); const searchEnginesList: ItemProps[] = [ { icon: , disabled: false, label: t('searchEngines.search.name'), value: 'search', description: t('searchEngines.search.description'), url: config.settings.searchUrl, shortcut: 's', }, { icon: , disabled: false, label: t('searchEngines.torrents.name'), value: 'torrents', description: t('searchEngines.torrents.description'), url: 'https://www.torrentdownloads.me/search/?search=', shortcut: 't', }, { icon: , disabled: false, label: t('searchEngines.youtube.name'), value: 'youtube', description: t('searchEngines.youtube.description'), url: 'https://www.youtube.com/results?search_query=', shortcut: 'y', }, { icon: , disabled: !(isOverseerrEnabled === true && OverseerrService !== undefined), label: t('searchEngines.overseerr.name'), value: 'overseerr', description: t('searchEngines.overseerr.description'), url: `${OverseerrService?.url}search?query=`, shortcut: 'm', }, ]; const [selectedSearchEngine, setSearchEngine] = useState(searchEnginesList[0]); const matchingServices = config.services.filter((service) => { if (searchQuery === '' || searchQuery === undefined) { return false; } return service.name.toLowerCase().includes(searchQuery.toLowerCase()); }); const autocompleteData = matchingServices.map((service) => ({ label: service.name, value: service.name, icon: service.icon, url: service.openedUrl ?? service.url, })); const AutoCompleteItem = forwardRef( ({ label, value, icon, url, ...others }: any, ref) => (
) ); useEffect(() => { // Refresh the default search engine every time the config for it changes #521 setSearchEngine(searchEnginesList[0]); }, [config.settings.searchUrl]); const textInput = useRef(null); useHotkeys([['mod+K', () => textInput.current && textInput.current.focus()]]); const { classes } = useStyles(); const openInNewTab = config.settings.searchNewTab ? '_blank' : '_self'; const [OverseerrResults, setOverseerrResults] = useState([]); const [opened, setOpened] = useState(false); useEffect(() => { if (debounced !== '' && selectedSearchEngine.value === 'overseerr' && searchQuery.length > 3) { axios.get(`/api/modules/overseerr?query=${searchQuery}`).then((res) => { setOverseerrResults(res.data.results ?? []); }); } }, [debounced]); const isModuleEnabled = config.modules?.[SearchModule.id]?.enabled ?? false; if (!isModuleEnabled) { return null; } //TODO: Fix the bug where clicking anything inside the Modal to ask for a movie // will close it (Because it closes the underlying Popover) return ( 0 && opened && searchQuery.length > 3} position="bottom" withinPortal shadow="md" radius="md" zIndex={100} transition="pop-top-right" > setOpened(true)} autoFocus rightSection={} placeholder={t(`searchEngines.${selectedSearchEngine.value}.description`)} value={searchQuery} onChange={(currentString) => tryMatchSearchEngine(currentString, setSearchQuery)} itemComponent={AutoCompleteItem} data={autocompleteData} onItemSubmit={(item) => { setOpened(false); if (item.url) { setSearchQuery(''); window.open(item.openedUrl ? item.openedUrl : item.url, openInNewTab); } }} // Replace %s if it is in selectedSearchEngine.url with searchQuery, otherwise append searchQuery at the end of it onKeyDown={(event) => { if ( event.key === 'Enter' && searchQuery.length > 0 && autocompleteData.length === 0 ) { if (selectedSearchEngine.url.includes('%s')) { window.open(selectedSearchEngine.url.replace('%s', searchQuery), openInNewTab); } else { window.open(selectedSearchEngine.url + searchQuery, openInNewTab); } } }} radius="md" size="md" />
{OverseerrResults.slice(0, 5).map((result, index) => ( {index < OverseerrResults.length - 1 && } ))}
); function tryMatchSearchEngine(query: string, setSearchQuery: (value: string) => void) { const foundSearchEngine = searchEnginesList.find( (engine) => query.includes(`!${engine.shortcut}`) && !engine.disabled ); if (foundSearchEngine) { setSearchQuery(query.replace(`!${foundSearchEngine.shortcut}`, '')); changeSearchEngine(foundSearchEngine); } else { setSearchQuery(query); } } function SearchModuleMenu() { return ( {selectedSearchEngine.icon} {searchEnginesList.map((item) => ( !{item.shortcut}} disabled={item.disabled} onClick={() => { changeSearchEngine(item); }} > {item.label} ))} {t('tip')} mod+k{' '} ); } function changeSearchEngine(item: ItemProps) { setSearchEngine(item); showNotification({ radius: 'lg', disallowClose: true, id: 'spotlight', autoClose: 1000, icon: {item.icon}, message: t('switchedSearchEngine', { searchEngine: t(`searchEngines.${item.value}.name`) }), }); } }