75 lines
2 KiB
TypeScript
75 lines
2 KiB
TypeScript
|
import { MtArgumentError, ParsedUpdate } from '@mtcute/client'
|
||
|
|
import { I18nValue, MtcuteI18nFunction, OtherLanguageWrap } from './types'
|
||
|
|
import { createI18nStringsIndex, extractLanguageFromUpdate } from './utils'
|
||
|
|
|
||
|
|
export interface MtcuteI18nParameters<Strings> {
|
||
|
|
/**
|
||
|
|
* Primary language which will also be used as a fallback
|
||
|
|
*/
|
||
|
|
primaryLanguage: {
|
||
|
|
/**
|
||
|
|
* Two letter language code.
|
||
|
|
*/
|
||
|
|
name: string
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Strings for the language.
|
||
|
|
*/
|
||
|
|
strings: Strings
|
||
|
|
}
|
||
|
|
|
||
|
|
otherLanguages?: Record<string, OtherLanguageWrap<Strings>>
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Language that will be used if no language is specified
|
||
|
|
*
|
||
|
|
* Defaults to {@link primaryLanguage}
|
||
|
|
*/
|
||
|
|
defaultLanguage?: string
|
||
|
|
}
|
||
|
|
|
||
|
|
export function createMtcuteI18n<Strings>(
|
||
|
|
params: MtcuteI18nParameters<Strings>
|
||
|
|
): MtcuteI18nFunction<Strings> {
|
||
|
|
const {
|
||
|
|
primaryLanguage,
|
||
|
|
otherLanguages,
|
||
|
|
defaultLanguage = primaryLanguage.name,
|
||
|
|
} = params
|
||
|
|
|
||
|
|
const indexes: Record<string, Record<string, I18nValue>> = {}
|
||
|
|
const fallbackIndex = (indexes[primaryLanguage.name] =
|
||
|
|
createI18nStringsIndex(primaryLanguage.strings))
|
||
|
|
if (otherLanguages) {
|
||
|
|
Object.keys(otherLanguages).forEach((lang) => {
|
||
|
|
indexes[lang] = createI18nStringsIndex(otherLanguages[lang])
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!(defaultLanguage in indexes)) {
|
||
|
|
throw new MtArgumentError(
|
||
|
|
'defaultLanguage is not a registered language'
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
const tr = (lang: ParsedUpdate['data'] | string | null, key: string, ...params: any[]) => {
|
||
|
|
if (lang === null) lang = defaultLanguage
|
||
|
|
|
||
|
|
if (typeof lang === 'object') {
|
||
|
|
lang = extractLanguageFromUpdate(lang) ?? defaultLanguage
|
||
|
|
}
|
||
|
|
|
||
|
|
const strings = indexes[lang] ?? fallbackIndex
|
||
|
|
|
||
|
|
let val = strings[key] ?? fallbackIndex[key] ?? `[missing: ${key}]`
|
||
|
|
|
||
|
|
if (typeof val === 'function') {
|
||
|
|
val = val(...params)
|
||
|
|
}
|
||
|
|
|
||
|
|
return val
|
||
|
|
}
|
||
|
|
|
||
|
|
return tr as MtcuteI18nFunction<Strings>
|
||
|
|
}
|