anki/ts/lib/i18n/bundles.ts

// Copyright: Ankitects Pty Ltd and contributors
// License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html

import type { FluentBundle, FluentVariable } from "@fluent/bundle";
import { FluentNumber } from "@fluent/bundle";

let bundles: FluentBundle[] = [];

export function setBundles(newBundles: FluentBundle[]): void {
    bundles = newBundles;
}

export function firstLanguage(): string {
    return bundles[0].locales[0];
}

function toFluentNumber(num: number): FluentNumber {
    return new FluentNumber(num, {
        maximumFractionDigits: 2,
    });
}

function formatArgs(
    args: Record<string, FluentVariable>,
): Record<string, FluentVariable> {
    const entries: [string, FluentVariable][] = Object.entries(args).map(
        ([key, value]) => [
            key,
            typeof value === "number" ? toFluentNumber(value) : value,
        ],
    );
    const out: Record<string, FluentVariable> = {};
    for (const [key, value] of entries) {
        out[key] = value;
    }
    return out;
}

export function getMessage(
    key: string,
    args: Record<string, FluentVariable> = {},
): string | null {
    for (const bundle of bundles) {
        const msg = bundle.getMessage(key);
        if (msg && msg.value) {
            return bundle.formatPattern(msg.value, formatArgs(args));
        }
    }

    return null;
}