eyo-kernel/dist/dictionary.js

Restoring the letter «ё» (yo) in russian texts

export class Dictionary {
    dict = {};
    constructor() {
        this.dict = {};
    }
    /**
     * Очищает словарь.
     */
    clear() {
        this.dict = {};
    }
    /**
     * Восстанавливает в слове букву «ё».
     */
    restoreWord(word) {
        return this.dict[this.replaceYo(word)] || word;
    }
    /**
     * Добавляет слово в словарь.
     */
    addWord(rawWord) {
        let word = rawWord;
        if (rawWord.search('#') > -1) {
            word = word.split('#')[0].trim();
        }
        if (word.search(/\(/) > -1) {
            const parts = word.split(/[(|)]/);
            for (let i = 1, len = parts.length - 1; i < len; i++) {
                this.addWordInner(parts[0] + parts[i]);
            }
        }
        else {
            this.addWordInner(word);
        }
    }
    /**
     * Удаляет слово из словаря.
     */
    removeWord(word) {
        const wordE = this.replaceYo(word);
        delete this.dict[wordE];
        if (word.search(/^[А-ЯЁ]/) === -1) {
            delete this.dict[this.capitalize(wordE)];
        }
    }
    /**
     * Установить словарь.
     */
    set(dict) {
        this.clear();
        if (!dict) {
            return;
        }
        const buffer = Array.isArray(dict) ? dict : dict.trim().split(/\r?\n/);
        buffer.forEach(word => this.addWord(word));
    }
    /**
     * Получить словарь.
     */
    get() {
        return this.dict;
    }
    addWordInner(word) {
        // Слово может использоваться только со строчной буквы.
        // Пример: _киёв. Киев и только киёв.
        const hasUnderscore = word.search('_') === 0;
        word = word.replace(/^_/, '');
        const key = this.replaceYo(word);
        this.dict[key] = word;
        if (word.search(/^[А-ЯЁ]/) === -1 && !hasUnderscore) {
            this.dict[this.capitalize(key)] = this.capitalize(word);
        }
    }
    capitalize(text) {
        return text.substr(0, 1).toUpperCase() + text.substr(1);
    }
    replaceYo(word) {
        return word.replace(/Ё/g, 'Е').replace(/ё/g, 'е');
    }
}
;