/* * Copyright (C) Contributors to the Suwayomi project * * This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this * file, You can obtain one at https://mozilla.org/MPL/2.0/. */ import { StringParam, useQueryParam } from 'use-query-params'; import { useMemo } from 'react'; import { LibrarySortMode, NullAndUndefined, TManga } from '@/typings.ts'; import { useLibraryOptionsContext } from '@/components/context/LibraryOptionsContext.tsx'; import { useSearchSettings } from '@/util/searchSettings.ts'; const unreadFilter = (unread: NullAndUndefined, { unreadCount }: TManga): boolean => { switch (unread) { case true: return !!unreadCount && unreadCount >= 1; case false: return unreadCount === 0; default: return true; } }; const downloadedFilter = (downloaded: NullAndUndefined, { downloadCount }: TManga): boolean => { switch (downloaded) { case true: return !!downloadCount && downloadCount >= 1; case false: return downloadCount === 0; default: return true; } }; const queryFilter = (query: NullAndUndefined, { title }: TManga): boolean => { if (!query) return true; return title.toLowerCase().includes(query.toLowerCase()); }; const queryGenreFilter = (query: NullAndUndefined, { genre }: TManga): boolean => { if (!query) return true; const queries = query.split(',').map((str) => str.toLowerCase().trim()); return queries.every((element) => genre.map((el) => el.toLowerCase()).includes(element)); }; const filterManga = ( mangas: TManga[], query: NullAndUndefined, unread: NullAndUndefined, downloaded: NullAndUndefined, ignoreFilters: boolean, ): TManga[] => mangas.filter((manga) => { const ignoreFiltersWhileSearching = ignoreFilters && query?.length; const matchesSearch = queryFilter(query, manga) || queryGenreFilter(query, manga); const matchesFilters = ignoreFiltersWhileSearching || (downloadedFilter(downloaded, manga) && unreadFilter(unread, manga)); return matchesSearch && matchesFilters; }); const sortByUnread = (a: TManga, b: TManga): number => (a.unreadCount ?? 0) - (b.unreadCount ?? 0); const sortByTitle = (a: TManga, b: TManga): number => a.title.localeCompare(b.title); const sortByDateAdded = (a: TManga, b: TManga): number => Number(a.inLibraryAt) - Number(b.inLibraryAt); const sortByLastRead = (a: TManga, b: TManga): number => Number(b.lastReadChapter?.lastReadAt ?? 0) - Number(a.lastReadChapter?.lastReadAt ?? 0); const sortManga = ( manga: TManga[], sort: NullAndUndefined, desc: NullAndUndefined, ): TManga[] => { const result = [...manga]; switch (sort) { case 'sortAlph': result.sort(sortByTitle); break; case 'sortDateAdded': result.sort(sortByDateAdded); break; case 'sortToRead': result.sort(sortByUnread); break; case 'sortLastRead': result.sort(sortByLastRead); break; default: break; } if (desc === true) { result.reverse(); } return result; }; export const useGetVisibleLibraryMangas = (mangas: TManga[]) => { const [query] = useQueryParam('query', StringParam); const { options } = useLibraryOptionsContext(); const { unread, downloaded } = options; const { settings } = useSearchSettings(); const filteredMangas = useMemo( () => filterManga(mangas, query, unread, downloaded, settings.ignoreFilters), [mangas, query, unread, downloaded, settings.ignoreFilters], ); const sortedMangas = useMemo( () => sortManga(filteredMangas, options.sorts, options.sortDesc), [filteredMangas, options.sorts, options.sortDesc], ); const showFilteredOutMessage = (unread != null || downloaded != null || !!query) && filteredMangas.length === 0 && mangas.length > 0; return { visibleMangas: sortedMangas, showFilteredOutMessage, }; };