import { ActionIcon, Box, createStyles, Divider, Kbd, Menu, Popover, ScrollArea, TextInput, Tooltip, } from '@mantine/core'; import { IconSearch, IconBrandYoutube, IconDownload, IconMovie } from '@tabler/icons'; import React, { 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'; 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 queryUrl = config.settings.searchUrl; 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: queryUrl, 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 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={(event) => tryMatchSearchEngine(event.currentTarget.value, setSearchQuery)} // 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) { if (selectedSearchEngine.url.includes('%s')) { window.open(selectedSearchEngine.url.replace('%s', searchQuery), openInNewTab); } else { window.open(selectedSearchEngine.url + searchQuery, openInNewTab); } } }} />
{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`) }), }); } }