106 lines
3.8 KiB
TypeScript
106 lines
3.8 KiB
TypeScript
/*
|
|
* 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 { t } from 'i18next';
|
|
import {
|
|
ExtensionAction,
|
|
ExtensionGroupState,
|
|
ExtensionState,
|
|
GroupedExtensionsResult,
|
|
InstalledState,
|
|
TExtension,
|
|
} from '@/modules/extension/Extensions.types.ts';
|
|
import { DefaultLanguage, langCodeToName, langSortCmp } from '@/modules/core/utils/Languages.ts';
|
|
import { extensionLanguageToTranslationKey } from '@/modules/extension/Extensions.constants.ts';
|
|
import { enhancedCleanup } from '@/util/Strings.ts';
|
|
|
|
export const getInstalledState = (
|
|
isInstalled: boolean,
|
|
isObsolete: boolean,
|
|
hasUpdate: boolean,
|
|
): ExtensionAction | ExtensionState.OBSOLETE => {
|
|
if (isObsolete) {
|
|
return InstalledState.OBSOLETE;
|
|
}
|
|
|
|
if (hasUpdate) {
|
|
return InstalledState.UPDATE;
|
|
}
|
|
|
|
return isInstalled ? InstalledState.UNINSTALL : InstalledState.INSTALL;
|
|
};
|
|
|
|
export const isExtensionState = (value: string): boolean =>
|
|
[ExtensionGroupState.INSTALLED, ExtensionGroupState.UPDATE_PENDING, ExtensionGroupState.OBSOLETE].includes(
|
|
value as ExtensionGroupState,
|
|
);
|
|
|
|
export const isExtensionStateOrLanguage = (languageCode: string): boolean =>
|
|
isExtensionState(languageCode) ||
|
|
[DefaultLanguage.ALL, DefaultLanguage.OTHER, DefaultLanguage.LOCAL_SOURCE].includes(
|
|
languageCode as DefaultLanguage,
|
|
);
|
|
|
|
export const translateExtensionLanguage = (languageCode: string): string =>
|
|
isExtensionStateOrLanguage(languageCode)
|
|
? t(extensionLanguageToTranslationKey[languageCode as ExtensionGroupState | DefaultLanguage])
|
|
: langCodeToName(languageCode);
|
|
|
|
export function groupExtensionsByLanguage(extensions: TExtension[]): GroupedExtensionsResult {
|
|
const extensionsByLanguage = Object.groupBy<ExtensionGroupState | string, TExtension>(extensions, (extension) => {
|
|
if (!extension.isInstalled) {
|
|
return extension.lang;
|
|
}
|
|
|
|
if (extension.hasUpdate) {
|
|
return ExtensionGroupState.UPDATE_PENDING;
|
|
}
|
|
|
|
if (extension.isObsolete) {
|
|
return ExtensionGroupState.OBSOLETE;
|
|
}
|
|
|
|
return ExtensionGroupState.INSTALLED;
|
|
});
|
|
|
|
// sort groups by language
|
|
const extensionsBySortedLanguage = Object.entries(extensionsByLanguage).toSorted(([a], [b]) => {
|
|
const extensionGroupStates = Object.values(ExtensionGroupState);
|
|
|
|
if (extensionGroupStates.includes(a as ExtensionGroupState)) {
|
|
return -1;
|
|
}
|
|
if (extensionGroupStates.includes(b as ExtensionGroupState)) {
|
|
return 1;
|
|
}
|
|
|
|
return langSortCmp(a, b);
|
|
});
|
|
|
|
const groupedExtensionsSortedByLanguage = extensionsBySortedLanguage.map(([language, extensionsOfLanguage]) => [
|
|
language,
|
|
(extensionsOfLanguage ?? []).toSorted((a, b) => langSortCmp(a.lang, b.lang)),
|
|
]) satisfies GroupedExtensionsResult;
|
|
|
|
return groupedExtensionsSortedByLanguage.filter(([, extensionsOfLanguage]) => !!extensionsOfLanguage.length);
|
|
}
|
|
|
|
export const getLanguagesFromExtensions = (extensions: TExtension[]): string[] =>
|
|
[...new Set(extensions.map((extension) => extension.lang))].toSorted(langSortCmp);
|
|
|
|
export const filterExtensions = (
|
|
extensions: TExtension[],
|
|
selectedLanguages: string[],
|
|
showNsfw: boolean,
|
|
query: string | null | undefined,
|
|
): TExtension[] =>
|
|
extensions
|
|
.filter((extension) => selectedLanguages.includes(extension.lang) || extension.isInstalled)
|
|
.filter((extension) => showNsfw || !extension.isNsfw)
|
|
.filter((extension) => !query || enhancedCleanup(extension.name).includes(enhancedCleanup(query)));
|