messageProvider.ts

Go to the documentation of this file.

 
import * as vscode from 'vscode';
import { messages } from './messageReference';
 
class MessageProvider {
    private locale: string = vscode.env.language.toLowerCase();
 
    private messageNotFound: string = "messageNotFound";
 
    private messages: Record<string, Record<string, (...args: any[]) => string>> = messages;
 
    private getFirstAvailableLanguageNode(): Record<string, (...args: any[]) => string> {
        const langs = Object.keys(this.messages);
        if (langs.length === 0) {
            throw new Error("No messages available.");
        }
        return this.messages[langs[0]];
    }
 
    private returnCorrectMessage(
        node: Record<string, (...args: any[]) => string>,
        messageKey: string,
        ...args: any[]
    ): string {
        if (node[messageKey] === undefined) {
            return node[this.messageNotFound](messageKey, ...args);
        }
        return node[messageKey](...args);
    }
 
    public get(messageKey: string, ...args: any[]): string;
 
    public get(messageKey: string, options: { language: string }, ...args: any[]): string;
 
    public get(messageKey: string, optionsOrArg?: { language: string } | any, ...args: any[]): string {
        let lang: string;
        let finalArgs: any[];
 
        if (optionsOrArg && typeof optionsOrArg === 'object' && 'language' in optionsOrArg) {
            lang = optionsOrArg.language;
            finalArgs = args;
        } else {
            lang = this.locale;
            finalArgs = args;
            if (optionsOrArg !== undefined) {
                finalArgs = [optionsOrArg, ...args];
            }
        }
 
        const node = this.messages[lang] || this.messages['en'] || this.getFirstAvailableLanguageNode();
        return this.returnCorrectMessage(node, messageKey, ...finalArgs);
    }
}
 
const instance = new MessageProvider();
 
export const getMessage = instance.get.bind(instance);