78 lines
2.8 KiB
TypeScript
78 lines
2.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 {
|
|
SourceDisplayNameInfo,
|
|
SourceIdInfo,
|
|
SourceLanguageInfo,
|
|
SourceNsfwInfo,
|
|
SourceRepoInfo,
|
|
} from '@/modules/source/Source.types.ts';
|
|
import { DefaultLanguage, langSortCmp } from '@/modules/core/utils/Languages.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((source) => Sources.getLanguage(source)))].toSorted(langSortCmp);
|
|
}
|
|
|
|
static groupByLanguage<Source extends SourceIdInfo & SourceLanguageInfo & SourceDisplayNameInfo>(
|
|
sources: Source[],
|
|
): Record<string, Source[]> {
|
|
const sourcesByLanguage = Object.groupBy(sources, (source) => Sources.getLanguage(source));
|
|
const sourcesBySortedLanguage = Object.entries(sourcesByLanguage).toSorted(([a], [b]) => langSortCmp(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[] {
|
|
return sources
|
|
.filter((source) => showNsfw || !source.isNsfw || (keepLocalSource && Sources.isLocalSource(source)))
|
|
.filter(
|
|
(source) =>
|
|
!languages ||
|
|
languages.includes(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));
|
|
}
|
|
}
|