import { ActionIcon, Autocomplete, Box, createStyles, Divider, Kbd, Menu, Popover, ScrollArea, Tooltip, } from '@mantine/core'; import { useDebouncedValue, useHotkeys } from '@mantine/hooks'; import { showNotification } from '@mantine/notifications'; import { IconBrandYoutube, IconDownload, IconMovie, IconSearch } from '@tabler/icons'; import { useQuery } from '@tanstack/react-query'; import axios from 'axios'; import { useTranslation } from 'next-i18next'; import React, { forwardRef, useEffect, useRef, useState } from 'react'; import { useConfigContext } from '../../../config/provider'; import { OverseerrMediaDisplay } from '../../../modules/common'; import { IModule } from '../../../modules/ModuleTypes'; import { ConfigType } from '../../../types/config'; import { searchUrls } from '../../Settings/Common/SearchEngine/SearchEngineSelector'; import Tip from '../Tip'; import { useCardStyles } from '../useCardStyles'; import SmallAppItem from './SmallAppItem'; export const SearchModule: IModule = { title: 'Search', icon: IconSearch, component: Search, 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 Search() { const { t } = useTranslation('modules/search'); const { config } = useConfigContext(); const [searchQuery, setSearchQuery] = useState(''); const [debounced] = useDebouncedValue(searchQuery, 250); const { classes: cardClasses } = useCardStyles(true); const isOverseerrEnabled = config?.apps.some( (x) => x.integration.type === 'overseerr' || x.integration.type === 'jellyseerr' ); const overseerrApp = config?.apps.find( (app) => app.integration?.type === 'overseerr' || app.integration?.type === 'jellyseerr' ); const searchEngineSettings = config?.settings.common.searchEngine; const searchEngineUrl = !searchEngineSettings ? searchUrls.google : searchEngineSettings.type === 'custom' ? searchEngineSettings.properties.template : searchUrls[searchEngineSettings.type]; const searchEnginesList: ItemProps[] = [ { icon: , disabled: false, label: t('searchEngines.search.name'), value: 'search', description: t('searchEngines.search.description'), url: searchEngineUrl, 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 && overseerrApp !== undefined), label: t('searchEngines.overseerr.name'), value: 'overseerr', description: t('searchEngines.overseerr.description'), url: `${overseerrApp?.url}search?query=`, shortcut: 'm', }, ]; const [selectedSearchEngine, setSearchEngine] = useState(searchEnginesList[0]); const matchingApps = config?.apps.filter((app) => { if (searchQuery === '' || searchQuery === undefined) { return false; } return app.name.toLowerCase().includes(searchQuery.toLowerCase()); }) ?? []; const autocompleteData = matchingApps.map((app) => ({ label: app.name, value: app.name, icon: app.appearance.iconUrl, url: app.behaviour.externalUrl ?? app.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]); }, [searchEngineUrl]); const textInput = useRef(null); useHotkeys([['mod+K', () => textInput.current?.focus()]]); const { classes } = useStyles(); const openTarget = getOpenTarget(config); const [opened, setOpened] = useState(false); const { data: OverseerrResults, isLoading, error, } = useQuery( ['overseerr', debounced], async () => { const res = await axios.get(`/api/modules/overseerr?query=${debounced}`); return res.data.results ?? []; }, { enabled: isOverseerrEnabled === true && selectedSearchEngine.value === 'overseerr' && debounced.length > 3, refetchOnWindowFocus: false, refetchOnMount: false, refetchInterval: false, } ); const isModuleEnabled = config?.settings.customization.layout.enabledSearchbar; 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, openTarget); } }} // 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), openTarget); } else { window.open(selectedSearchEngine.url + searchQuery, openTarget); } } }} classNames={{ input: cardClasses.card, }} radius="lg" size="md" /> {OverseerrResults && OverseerrResults.slice(0, 4).map((result: any, index: number) => ( {index < OverseerrResults.length - 1 && index < 3 && ( )} ))} ); 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`) }), }); } } const getOpenTarget = (config: ConfigType | undefined): '_blank' | '_self' => { if (!config || config.settings.common.searchEngine.properties.openInNewTab === undefined) { return '_blank'; } return config.settings.common.searchEngine.properties.openInNewTab ? '_blank' : '_self'; };