yomichan/ext/bg/js/translator.js

/*
 * Copyright (C) 2016  Alex Yatskov <alex@foosoft.net>
 * Author: Alex Yatskov <alex@foosoft.net>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */


class Translator {
    constructor() {
        this.database = null;
        this.deinflector = null;
        this.tagCache = {};
    }

    async prepare() {
        if (!this.database) {
            this.database = new Database();
            await this.database.prepare();
        }

        if (!this.deinflector) {
            const url = chrome.runtime.getURL('/bg/lang/deinflect.json');
            const reasons = await requestJson(url, 'GET');
            this.deinflector = new Deinflector(reasons);
        }
    }

    async purgeDatabase() {
        this.tagCache = {};
        await this.database.purge();
    }

    async deleteDictionary(dictionaryName) {
        this.tagCache = {};
        await this.database.deleteDictionary(dictionaryName);
    }

    async getSequencedDefinitions(definitions, mainDictionary) {
        const definitionsBySequence = dictTermsMergeBySequence(definitions, mainDictionary);
        const defaultDefinitions = definitionsBySequence['-1'];

        const sequenceList = Object.keys(definitionsBySequence).map(v => Number(v)).filter(v => v >= 0);
        const sequencedDefinitions = sequenceList.map((key) => ({
            definitions: definitionsBySequence[key],
            rawDefinitions: []
        }));

        for (const definition of await this.database.findTermsBySequenceBulk(sequenceList, mainDictionary)) {
            sequencedDefinitions[definition.index].rawDefinitions.push(definition);
        }

        return {sequencedDefinitions, defaultDefinitions};
    }

    async getMergedSecondarySearchResults(text, expressionsMap, secondarySearchTitles) {
        if (secondarySearchTitles.length === 0) {
            return [];
        }

        const expressionList = [];
        const readingList = [];
        for (const expression of expressionsMap.keys()) {
            if (expression === text) { continue; }
            for (const reading of expressionsMap.get(expression).keys()) {
                expressionList.push(expression);
                readingList.push(reading);
            }
        }

        const definitions = await this.database.findTermsExactBulk(expressionList, readingList, secondarySearchTitles);
        for (const definition of definitions) {
            const definitionTags = await this.expandTags(definition.definitionTags, definition.dictionary);
            definitionTags.push(dictTagBuildSource(definition.dictionary));
            definition.definitionTags = definitionTags;
            const termTags = await this.expandTags(definition.termTags, definition.dictionary);
            definition.termTags = termTags;
        }

        if (definitions.length > 1) {
            definitions.sort((a, b) => a.index - b.index);
        }

        return definitions;
    }

    async getMergedDefinition(text, dictionaries, sequencedDefinition, defaultDefinitions, secondarySearchTitles, mergedByTermIndices) {
        const result = sequencedDefinition.definitions;
        const rawDefinitionsBySequence = sequencedDefinition.rawDefinitions;

        for (const definition of rawDefinitionsBySequence) {
            const definitionTags = await this.expandTags(definition.definitionTags, definition.dictionary);
            definitionTags.push(dictTagBuildSource(definition.dictionary));
            definition.definitionTags = definitionTags;
            const termTags = await this.expandTags(definition.termTags, definition.dictionary);
            definition.termTags = termTags;
        }

        const definitionsByGloss = dictTermsMergeByGloss(result, rawDefinitionsBySequence);
        const secondarySearchResults = await this.getMergedSecondarySearchResults(text, result.expressions, secondarySearchTitles);

        dictTermsMergeByGloss(result, defaultDefinitions.concat(secondarySearchResults), definitionsByGloss, mergedByTermIndices);

        for (const gloss in definitionsByGloss) {
            const definition = definitionsByGloss[gloss];
            dictTagsSort(definition.definitionTags);
            result.definitions.push(definition);
        }

        dictTermsSort(result.definitions, dictionaries);

        const expressions = [];
        for (const expression of result.expressions.keys()) {
            for (const reading of result.expressions.get(expression).keys()) {
                const termTags = result.expressions.get(expression).get(reading);
                const score = termTags.map(tag => tag.score).reduce((p, v) => p + v, 0);
                expressions.push({
                    expression: expression,
                    reading: reading,
                    termTags: dictTagsSort(termTags),
                    termFrequency: Translator.scoreToTermFrequency(score)
                });
            }
        }

        result.expressions = expressions;
        result.expression = Array.from(result.expression);
        result.reading = Array.from(result.reading);

        return result;
    }

    async findTerms(text, details, options) {
        switch (options.general.resultOutputMode) {
            case 'group':
                return await this.findTermsGrouped(text, details, options);
            case 'merge':
                return await this.findTermsMerged(text, details, options);
            case 'split':
                return await this.findTermsSplit(text, details, options);
            default:
                return [[], 0];
        }
    }

    async findTermsGrouped(text, details, options) {
        const dictionaries = dictEnabledSet(options);
        const titles = Object.keys(dictionaries);
        const [definitions, length] = await this.findTermsInternal(text, dictionaries, options.scanning.alphanumeric, details);

        const definitionsGrouped = dictTermsGroup(definitions, dictionaries);
        await this.buildTermFrequencies(definitionsGrouped, titles);

        if (options.general.compactTags) {
            for (const definition of definitionsGrouped) {
                dictTermsCompressTags(definition.definitions);
            }
        }

        return [definitionsGrouped, length];
    }

    async findTermsMerged(text, details, options) {
        const dictionaries = dictEnabledSet(options);
        const secondarySearchTitles = Object.keys(options.dictionaries).filter(dict => options.dictionaries[dict].allowSecondarySearches);
        const titles = Object.keys(dictionaries);
        const [definitions, length] = await this.findTermsInternal(text, dictionaries, options.scanning.alphanumeric, details);
        const {sequencedDefinitions, defaultDefinitions} = await this.getSequencedDefinitions(definitions, options.general.mainDictionary);
        const definitionsMerged = [];
        const mergedByTermIndices = new Set();

        for (const sequencedDefinition of sequencedDefinitions) {
            const result = await this.getMergedDefinition(
                text,
                dictionaries,
                sequencedDefinition,
                defaultDefinitions,
                secondarySearchTitles,
                mergedByTermIndices
            );
            definitionsMerged.push(result);
        }

        const strayDefinitions = defaultDefinitions.filter((definition, index) => !mergedByTermIndices.has(index));
        for (const groupedDefinition of dictTermsGroup(strayDefinitions, dictionaries)) {
            groupedDefinition.expressions = [{expression: groupedDefinition.expression, reading: groupedDefinition.reading}];
            definitionsMerged.push(groupedDefinition);
        }

        await this.buildTermFrequencies(definitionsMerged, titles);

        if (options.general.compactTags) {
            for (const definition of definitionsMerged) {
                dictTermsCompressTags(definition.definitions);
            }
        }

        return [dictTermsSort(definitionsMerged), length];
    }

    async findTermsSplit(text, details, options) {
        const dictionaries = dictEnabledSet(options);
        const titles = Object.keys(dictionaries);
        const [definitions, length] = await this.findTermsInternal(text, dictionaries, options.scanning.alphanumeric, details);

        await this.buildTermFrequencies(definitions, titles);

        return [definitions, length];
    }

    async findTermsInternal(text, dictionaries, alphanumeric, details) {
        if (!alphanumeric && text.length > 0) {
            const c = text[0];
            if (!jpIsKana(c) && !jpIsKanji(c)) {
                return [[], 0];
            }
        }

        const titles = Object.keys(dictionaries);
        const deinflections = (
            details.wildcard ?
            await this.findTermWildcard(text, titles) :
            await this.findTermDeinflections(text, titles)
        );

        let definitions = [];
        for (const deinflection of deinflections) {
            for (const definition of deinflection.definitions) {
                const definitionTags = await this.expandTags(definition.definitionTags, definition.dictionary);
                definitionTags.push(dictTagBuildSource(definition.dictionary));
                const termTags = await this.expandTags(definition.termTags, definition.dictionary);

                definitions.push({
                    source: deinflection.source,
                    reasons: deinflection.reasons,
                    score: definition.score,
                    id: definition.id,
                    dictionary: definition.dictionary,
                    expression: definition.expression,
                    reading: definition.reading,
                    glossary: definition.glossary,
                    definitionTags: dictTagsSort(definitionTags),
                    termTags: dictTagsSort(termTags),
                    sequence: definition.sequence
                });
            }
        }

        definitions = dictTermsUndupe(definitions);
        definitions = dictTermsSort(definitions, dictionaries);

        let length = 0;
        for (const definition of definitions) {
            length = Math.max(length, definition.source.length);
        }

        return [definitions, length];
    }

    async findTermWildcard(text, titles) {
        const definitions = await this.database.findTermsBulk([text], titles, true);
        if (definitions.length === 0) {
            return [];
        }

        return [{
            source: text,
            term: text,
            rules: 0,
            definitions,
            reasons: []
        }];
    }

    async findTermDeinflections(text, titles) {
        const text2 = jpKatakanaToHiragana(text);
        const deinflections = (text === text2 ? this.getDeinflections(text) : this.getDeinflections2(text, text2));

        if (deinflections.length === 0) {
            return [];
        }

        const uniqueDeinflectionTerms = [];
        const uniqueDeinflectionArrays = [];
        const uniqueDeinflectionsMap = {};
        for (const deinflection of deinflections) {
            const term = deinflection.term;
            let deinflectionArray;
            if (uniqueDeinflectionsMap.hasOwnProperty(term)) {
                deinflectionArray = uniqueDeinflectionsMap[term];
            } else {
                deinflectionArray = [];
                uniqueDeinflectionTerms.push(term);
                uniqueDeinflectionArrays.push(deinflectionArray);
                uniqueDeinflectionsMap[term] = deinflectionArray;
            }
            deinflectionArray.push(deinflection);
        }

        const definitions = await this.database.findTermsBulk(uniqueDeinflectionTerms, titles, false);

        for (const definition of definitions) {
            const definitionRules = Deinflector.rulesToRuleFlags(definition.rules);
            for (const deinflection of uniqueDeinflectionArrays[definition.index]) {
                const deinflectionRules = deinflection.rules;
                if (deinflectionRules === 0 || (definitionRules & deinflectionRules) !== 0) {
                    deinflection.definitions.push(definition);
                }
            }
        }

        return deinflections.filter(e => e.definitions.length > 0);
    }

    getDeinflections(text) {
        const deinflections = [];

        for (let i = text.length; i > 0; --i) {
            const textSubstring = text.substring(0, i);
            deinflections.push(...this.deinflector.deinflect(textSubstring));
        }

        return deinflections;
    }

    getDeinflections2(text1, text2) {
        const deinflections = [];

        for (let i = text1.length; i > 0; --i) {
            const text1Substring = text1.substring(0, i);
            const text2Substring = text2.substring(0, i);
            deinflections.push(...this.deinflector.deinflect(text1Substring));
            if (text1Substring !== text2Substring) {
                deinflections.push(...this.deinflector.deinflect(text2Substring));
            }
        }

        return deinflections;
    }

    async findKanji(text, options) {
        const dictionaries = dictEnabledSet(options);
        const titles = Object.keys(dictionaries);
        const kanjiUnique = {};
        const kanjiList = [];
        for (const c of text) {
            if (!kanjiUnique.hasOwnProperty(c)) {
                kanjiList.push(c);
                kanjiUnique[c] = true;
            }
        }

        const definitions = await this.database.findKanjiBulk(kanjiList, titles);
        if (definitions.length === 0) {
            return definitions;
        }

        if (definitions.length > 1) {
            definitions.sort((a, b) => a.index - b.index);
        }

        const kanjiList2 = [];
        for (const definition of definitions) {
            kanjiList2.push(definition.character);

            const tags = await this.expandTags(definition.tags, definition.dictionary);
            tags.push(dictTagBuildSource(definition.dictionary));

            definition.tags = dictTagsSort(tags);
            definition.stats = await this.expandStats(definition.stats, definition.dictionary);
            definition.frequencies = [];
        }

        for (const meta of await this.database.findKanjiMetaBulk(kanjiList2, titles)) {
            if (meta.mode !== 'freq') { continue; }
            definitions[meta.index].frequencies.push({
                character: meta.character,
                frequency: meta.data,
                dictionary: meta.dictionary
            });
        }

        return definitions;
    }

    async buildTermFrequencies(definitions, titles) {
        const terms = [];
        for (const definition of definitions) {
            if (definition.expressions) {
                terms.push(...definition.expressions);
            } else {
                terms.push(definition);
            }
        }

        if (terms.length === 0) {
            return;
        }

        // Create mapping of unique terms
        const expressionsUnique = [];
        const termsUnique = [];
        const termsUniqueMap = {};
        for (let i = 0, ii = terms.length; i < ii; ++i) {
            const term = terms[i];
            const expression = term.expression;
            term.frequencies = [];

            if (termsUniqueMap.hasOwnProperty(expression)) {
                termsUniqueMap[expression].push(term);
            } else {
                const termList = [term];
                expressionsUnique.push(expression);
                termsUnique.push(termList);
                termsUniqueMap[expression] = termList;
            }
        }

        const metas = await this.database.findTermMetaBulk(expressionsUnique, titles);
        for (const meta of metas) {
            if (meta.mode !== 'freq') {
                continue;
            }

            for (const term of termsUnique[meta.index]) {
                term.frequencies.push({
                    expression: meta.expression,
                    frequency: meta.data,
                    dictionary: meta.dictionary
                });
            }
        }
    }

    async expandTags(names, title) {
        const tagMetaList = await this.getTagMetaList(names, title);
        return tagMetaList.map((meta, index) => {
            const name = names[index];
            const tag = dictTagSanitize(Object.assign({}, meta !== null ? meta : {}, {name}));
            return dictTagSanitize(tag);
        });
    }

    async expandStats(items, title) {
        const names = Object.keys(items);
        const tagMetaList = await this.getTagMetaList(names, title);

        const stats = {};
        for (let i = 0; i < names.length; ++i) {
            const name = names[i];
            const meta = tagMetaList[i];
            if (meta === null) { continue; }

            const category = meta.category;
            const group = (
                stats.hasOwnProperty(category) ?
                stats[category] :
                (stats[category] = [])
            );

            const stat = Object.assign({}, meta, {name, value: items[name]});
            group.push(dictTagSanitize(stat));
        }

        const sortCompare = (a, b) => a.notes - b.notes;
        for (const category in stats) {
            stats[category].sort(sortCompare);
        }

        return stats;
    }

    async getTagMetaList(names, title) {
        const tagMetaList = [];
        const cache = (
            this.tagCache.hasOwnProperty(title) ?
            this.tagCache[title] :
            (this.tagCache[title] = {})
        );

        for (const name of names) {
            const base = Translator.getNameBase(name);

            if (cache.hasOwnProperty(base)) {
                tagMetaList.push(cache[base]);
            } else {
                const tagMeta = await this.database.findTagForTitle(base, title);
                cache[base] = tagMeta;
                tagMetaList.push(tagMeta);
            }
        }

        return tagMetaList;
    }

    static scoreToTermFrequency(score) {
        if (score > 0) {
            return 'popular';
        } else if (score < 0) {
            return 'rare';
        } else {
            return 'normal';
        }
    }

    static getNameBase(name) {
        const pos = name.indexOf(':');
        return (pos >= 0 ? name.substring(0, pos) : name);
    }
}