113 lines
3.7 KiB
TypeScript
113 lines
3.7 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 {
|
|
SourceDisplayNameInfo,
|
|
SourceIdInfo,
|
|
SourceLanguageInfo,
|
|
SourceNsfwInfo,
|
|
SourceMetaInfo,
|
|
SourceRepoInfo,
|
|
} from '@/modules/source/Source.types.ts';
|
|
import {
|
|
DefaultLanguage,
|
|
languageSpecialSortComparator,
|
|
toComparableLanguage,
|
|
toComparableLanguages,
|
|
toUniqueLanguageCodes,
|
|
} from '@/modules/core/utils/Languages.ts';
|
|
import { getSourceMetadata } from '@/modules/source/services/SourceMetadata.ts';
|
|
|
|
export class Sources {
|
|
static readonly LOCAL_SOURCE_ID = '0';
|
|
|
|
static isLocalSource(source: SourceIdInfo): boolean {
|
|
return source.id === Sources.LOCAL_SOURCE_ID;
|
|
}
|
|
|
|
static getLanguage(source: SourceIdInfo & SourceLanguageInfo): string {
|
|
if (Sources.isLocalSource(source)) {
|
|
return DefaultLanguage.OTHER;
|
|
}
|
|
|
|
return source.lang;
|
|
}
|
|
|
|
static getLanguages(sources: (SourceIdInfo & SourceLanguageInfo)[]): string[] {
|
|
return [...new Set(sources.map(Sources.getLanguage))];
|
|
}
|
|
|
|
static groupByLanguage<Source extends SourceIdInfo & SourceLanguageInfo & SourceDisplayNameInfo & SourceMetaInfo>(
|
|
sources: Source[],
|
|
): Record<string, Source[]> {
|
|
const sourcesByLanguage = Object.groupBy(sources, (source) => {
|
|
if (getSourceMetadata(source).isPinned) {
|
|
return DefaultLanguage.PINNED;
|
|
}
|
|
|
|
return Sources.getLanguage(source);
|
|
});
|
|
const sourcesBySortedLanguage = Object.entries(sourcesByLanguage).toSorted(([a], [b]) => {
|
|
const isAPinned = a === DefaultLanguage.PINNED;
|
|
const isBPinned = b === DefaultLanguage.PINNED;
|
|
|
|
if (isAPinned) {
|
|
return -1;
|
|
}
|
|
|
|
if (isBPinned) {
|
|
return 1;
|
|
}
|
|
|
|
return languageSpecialSortComparator(a, b);
|
|
});
|
|
const sortedSourcesBySortedLanguage = sourcesBySortedLanguage.map(([language, sourcesOfLanguage]) => [
|
|
language,
|
|
(sourcesOfLanguage ?? []).toSorted((a, b) => a.displayName.localeCompare(b.displayName)),
|
|
]);
|
|
|
|
return Object.fromEntries(sortedSourcesBySortedLanguage);
|
|
}
|
|
|
|
static filter<Source extends SourceIdInfo & SourceLanguageInfo & SourceNsfwInfo>(
|
|
sources: Source[],
|
|
{
|
|
showNsfw,
|
|
languages,
|
|
keepLocalSource,
|
|
}: { showNsfw?: boolean; languages?: string[]; keepLocalSource?: boolean } = {},
|
|
): Source[] {
|
|
const normalizedLanguages = toComparableLanguages(toUniqueLanguageCodes(languages ?? []));
|
|
|
|
return sources
|
|
.filter(
|
|
(source) =>
|
|
showNsfw === undefined ||
|
|
showNsfw ||
|
|
!source.isNsfw ||
|
|
(keepLocalSource && Sources.isLocalSource(source)),
|
|
)
|
|
.filter(
|
|
(source) =>
|
|
!languages ||
|
|
normalizedLanguages.includes(toComparableLanguage(Sources.getLanguage(source))) ||
|
|
(keepLocalSource && Sources.isLocalSource(source)),
|
|
);
|
|
}
|
|
|
|
static areFromMultipleRepos<Source extends SourceIdInfo & SourceRepoInfo>(sources: Source[]): boolean {
|
|
const repo = sources.find((source) => !!source.extension.repo)?.extension.repo;
|
|
|
|
if (!repo || !sources.length) {
|
|
return false;
|
|
}
|
|
|
|
return sources.some((source) => source.extension.repo !== repo && !Sources.isLocalSource(source));
|
|
}
|
|
}
|