/*
 * 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 { TranslationKey } from '@/typings';
import { DefaultLanguage, langCodeToName } from '@/util/language';
import { ExtensionType } from '@/lib/graphql/generated/graphql.ts';

export enum ExtensionState {
    INSTALLED = 'INSTALLED',
    UPDATE_PENDING = 'UPDATE_PENDING',
    OBSOLETE = 'OBSOLETE',
}

export type TExtension = Pick<
    ExtensionType,
    | 'pkgName'
    | 'name'
    | 'lang'
    | 'versionCode'
    | 'versionName'
    | 'iconUrl'
    | 'repo'
    | 'isNsfw'
    | 'isInstalled'
    | 'isObsolete'
    | 'hasUpdate'
>;

export type GroupedExtensionsResult<KEY extends string = string> = [KEY, TExtension[]][];

export type GroupedByExtensionState = {
    [state in ExtensionState]: TExtension[];
};

export type GroupedByLanguage = {
    [language in DefaultLanguage]: TExtension[];
} & {
    [language: string]: TExtension[];
};

export type GroupedExtensions = GroupedByExtensionState & GroupedByLanguage;

export const extensionLanguageToTranslationKey: { [state in ExtensionState | DefaultLanguage]: TranslationKey } = {
    [ExtensionState.INSTALLED]: 'extension.state.label.installed',
    [ExtensionState.UPDATE_PENDING]: 'extension.state.label.update_pending',
    [ExtensionState.OBSOLETE]: 'extension.state.label.obsolete',
    [DefaultLanguage.ALL]: 'extension.language.all',
    [DefaultLanguage.OTHER]: 'extension.language.other',
    [DefaultLanguage.LOCAL_SOURCE]: 'extension.language.other',
};

export const isExtensionStateOrLanguage = (languageCode: string): boolean =>
    [
        ExtensionState.INSTALLED,
        ExtensionState.UPDATE_PENDING,
        ExtensionState.OBSOLETE,
        DefaultLanguage.ALL,
        DefaultLanguage.OTHER,
        DefaultLanguage.LOCAL_SOURCE,
    ].includes(languageCode as ExtensionState | DefaultLanguage);

export const translateExtensionLanguage = (languageCode: string): string =>
    isExtensionStateOrLanguage(languageCode)
        ? t(extensionLanguageToTranslationKey[languageCode as ExtensionState | DefaultLanguage])
        : langCodeToName(languageCode);