83 lines
2.4 KiB
TypeScript
83 lines
2.4 KiB
TypeScript
// Copyright 2024 Signal Messenger, LLC
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
|
|
|
import { TYPE, parse } from '@formatjs/icu-messageformat-parser';
|
|
import type {
|
|
MessageFormatElement,
|
|
PluralOrSelectOption,
|
|
} from '@formatjs/icu-messageformat-parser';
|
|
import { missingCaseError } from './missingCaseError';
|
|
|
|
export type ICUMessageParamType = Readonly<
|
|
| {
|
|
type: 'string' | 'date' | 'number' | 'jsx' | 'time';
|
|
}
|
|
| {
|
|
type: 'select';
|
|
validOptions: ReadonlyArray<string>;
|
|
}
|
|
>;
|
|
|
|
export function getICUMessageParams(
|
|
message: string,
|
|
defaultRichTextElementNames: Array<string> = []
|
|
): Map<string, ICUMessageParamType> {
|
|
const params = new Map();
|
|
|
|
function visitOptions(options: Record<string, PluralOrSelectOption>) {
|
|
for (const option of Object.values(options)) {
|
|
visit(option.value);
|
|
}
|
|
}
|
|
|
|
function visit(elements: ReadonlyArray<MessageFormatElement>) {
|
|
for (const element of elements) {
|
|
switch (element.type) {
|
|
case TYPE.argument:
|
|
params.set(element.value, { type: 'string' });
|
|
break;
|
|
case TYPE.date:
|
|
params.set(element.value, { type: 'Date' });
|
|
break;
|
|
case TYPE.literal:
|
|
break;
|
|
case TYPE.number:
|
|
params.set(element.value, { type: 'number' });
|
|
break;
|
|
case TYPE.plural:
|
|
params.set(element.value, { type: 'number' });
|
|
visitOptions(element.options);
|
|
break;
|
|
case TYPE.pound:
|
|
break;
|
|
case TYPE.select: {
|
|
const validOptions = Object.entries(element.options)
|
|
// We use empty {other ...} to satisfy smartling, but don't allow
|
|
// it in the app.
|
|
.filter(([key, { value }]) => key !== 'other' || value.length)
|
|
.map(([key]) => key);
|
|
params.set(element.value, { type: 'select', validOptions });
|
|
visitOptions(element.options);
|
|
break;
|
|
}
|
|
case TYPE.tag:
|
|
params.set(element.value, { type: 'jsx' });
|
|
visit(element.children);
|
|
break;
|
|
case TYPE.time:
|
|
params.set(element.value, { type: 'time' });
|
|
break;
|
|
default:
|
|
throw missingCaseError(element);
|
|
}
|
|
}
|
|
}
|
|
|
|
visit(parse(message));
|
|
|
|
for (const defaultRichTextElementName of defaultRichTextElementNames) {
|
|
params.delete(defaultRichTextElementName);
|
|
}
|
|
|
|
return params;
|
|
}
|