yomichan/ext/bg/js/util.js

/*
 * Copyright (C) 2016-2020  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 <https://www.gnu.org/licenses/>.
 */

function utilIsolate(value) {
    if (value === null) { return null; }

    switch (typeof value) {
        case 'boolean':
        case 'number':
        case 'string':
        case 'bigint':
        case 'symbol':
            return value;
    }

    const stringValue = JSON.stringify(value);
    return typeof stringValue === 'string' ? JSON.parse(stringValue) : null;
}

function utilFunctionIsolate(func) {
    return function (...args) {
        try {
            args = args.map((v) => utilIsolate(v));
            return func.call(this, ...args);
        } catch (e) {
            try {
                String(func);
            } catch (e2) {
                // Dead object
                return;
            }
            throw e;
        }
    };
}

function utilBackgroundIsolate(data) {
    const backgroundPage = chrome.extension.getBackgroundPage();
    return backgroundPage.utilIsolate(data);
}

function utilBackgroundFunctionIsolate(func) {
    const backgroundPage = chrome.extension.getBackgroundPage();
    return backgroundPage.utilFunctionIsolate(func);
}

function utilSetEqual(setA, setB) {
    if (setA.size !== setB.size) {
        return false;
    }

    for (const value of setA) {
        if (!setB.has(value)) {
            return false;
        }
    }

    return true;
}

function utilSetIntersection(setA, setB) {
    return new Set(
        [...setA].filter((value) => setB.has(value))
    );
}

function utilSetDifference(setA, setB) {
    return new Set(
        [...setA].filter((value) => !setB.has(value))
    );
}

function utilStringHashCode(string) {
    let hashCode = 0;

    if (typeof string !== 'string') { return hashCode; }

    for (let i = 0, charCode = string.charCodeAt(i); i < string.length; charCode = string.charCodeAt(++i)) {
        hashCode = ((hashCode << 5) - hashCode) + charCode;
        hashCode |= 0;
    }

    return hashCode;
}

function utilBackend() {
    return chrome.extension.getBackgroundPage().yomichanBackend;
}

async function utilAnkiGetModelNames() {
    return utilIsolate(await utilBackend().anki.getModelNames());
}

async function utilAnkiGetDeckNames() {
    return utilIsolate(await utilBackend().anki.getDeckNames());
}

async function utilDatabaseGetDictionaryInfo() {
    return utilIsolate(await utilBackend().translator.database.getDictionaryInfo());
}

async function utilDatabaseGetDictionaryCounts(dictionaryNames, getTotal) {
    return utilIsolate(await utilBackend().translator.database.getDictionaryCounts(
        utilBackgroundIsolate(dictionaryNames),
        utilBackgroundIsolate(getTotal)
    ));
}

async function utilAnkiGetModelFieldNames(modelName) {
    return utilIsolate(await utilBackend().anki.getModelFieldNames(
        utilBackgroundIsolate(modelName)
    ));
}

async function utilDatabasePurge() {
    return utilIsolate(await utilBackend().translator.purgeDatabase());
}

async function utilDatabaseDeleteDictionary(dictionaryName, onProgress) {
    return utilIsolate(await utilBackend().translator.database.deleteDictionary(
        utilBackgroundIsolate(dictionaryName),
        utilBackgroundFunctionIsolate(onProgress)
    ));
}

async function utilDatabaseImport(data, onProgress, details) {
    data = await utilReadFile(data);
    return utilIsolate(await utilBackend().translator.database.importDictionary(
        utilBackgroundIsolate(data),
        utilBackgroundFunctionIsolate(onProgress),
        utilBackgroundIsolate(details)
    ));
}

function utilReadFile(file) {
    return new Promise((resolve, reject) => {
        const reader = new FileReader();
        reader.onload = () => resolve(reader.result);
        reader.onerror = () => reject(reader.error);
        reader.readAsBinaryString(file);
    });
}

function utilReadFileArrayBuffer(file) {
    return new Promise((resolve, reject) => {
        const reader = new FileReader();
        reader.onload = () => resolve(reader.result);
        reader.onerror = () => reject(reader.error);
        reader.readAsArrayBuffer(file);
    });
}