yomichan/ext/fg/js/util.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/>.
 */


/*
 * Background
 */

function bgInvoke(action, params) {
    return new Promise((resolve, reject) => {
        try {
            chrome.runtime.sendMessage({action, params}, ({result, error}) => {
                if (error) {
                    reject(error);
                } else {
                    resolve(result);
                }
            });
        } catch (e) {
            window.orphaned = true;
            reject(e.message);
        }
    });
}

function bgOptionsGet() {
    return bgInvoke('optionsGet', {});
}

function bgTermsFind(text) {
    return bgInvoke('termsFind', {text});
}

function bgKanjiFind(text) {
    return bgInvoke('kanjiFind', {text});
}

function bgTemplateRender(template, data) {
    return bgInvoke('templateRender', {data, template});
}

function bgDefinitionsAddable(definitions, modes) {
    return bgInvoke('definitionsAddable', {definitions, modes}).catch(() => null);
}

function bgDefinitionAdd(definition, mode) {
    return bgInvoke('definitionAdd', {definition, mode});
}


/*
 * Document
 */

function docOffsetCalc(element) {
    const scrollTop = window.pageYOffset || document.documentElement.scrollTop || document.body.scrollTop;
    const scrollLeft = window.pageXOffset || document.documentElement.scrollLeft || document.body.scrollLeft;

    const clientTop = document.documentElement.clientTop || document.body.clientTop || 0;
    const clientLeft = document.documentElement.clientLeft || document.body.clientLeft || 0;

    const rect = element.getBoundingClientRect();
    const top  = Math.round(rect.top +  scrollTop - clientTop);
    const left = Math.round(rect.left + scrollLeft - clientLeft);

    return {top, left};
}

function docImposterCreate(element) {
    const styleProps = window.getComputedStyle(element);
    const stylePairs = [];
    for (const key of styleProps) {
        stylePairs.push(`${key}: ${styleProps[key]};`);
    }

    const offset = docOffsetCalc(element);
    const imposter = document.createElement('div');
    imposter.className = 'yomichan-imposter';
    imposter.innerText = element.value;
    imposter.style.cssText = stylePairs.join('\n');
    imposter.style.position = 'absolute';
    imposter.style.top = `${offset.top}px`;
    imposter.style.left = `${offset.left}px`;
    imposter.style.zIndex = 2147483646;
    if (element.nodeName === 'TEXTAREA' && styleProps.overflow === 'visible') {
        imposter.style.overflow = 'auto';
    }

    document.body.appendChild(imposter);
    imposter.scrollTop = element.scrollTop;
    imposter.scrollLeft = element.scrollLeft;
}

function docImposterDestroy() {
    for (const element of document.getElementsByClassName('yomichan-imposter')) {
        element.parentNode.removeChild(element);
    }
}

function docImposterHide() {
    for (const element of document.getElementsByClassName('yomichan-imposter')) {
        element.style.visibility = 'hidden';
    }
}

function docRangeFromPoint(point, imposter) {
    const element = document.elementFromPoint(point.x, point.y);
    if (element !== null) {
        if (element.nodeName === 'IMG' || element.nodeName === 'BUTTON') {
            return new TextSourceElement(element);
        } else if (imposter && (element.nodeName === 'INPUT' || element.nodeName === 'TEXTAREA')) {
            docImposterCreate(element);
        }
    }

    if (!document.caretRangeFromPoint) {
        document.caretRangeFromPoint = (x, y) => {
            const position = document.caretPositionFromPoint(x,y);
            if (position === null) {
                return null;
            }

            const range = document.createRange();
            range.setStart(position.offsetNode, position.offset);
            range.setEnd(position.offsetNode, position.offset);
            return range;
        };
    }

    const range = document.caretRangeFromPoint(point.x, point.y);
    if (range !== null) {
        docImposterHide();
        return new TextSourceRange(range);
    }

    docImposterDestroy();
    return null;
}

function docClozeExtract(source, extent) {
    const quotesFwd = {'「': '」', '『': '』', "'": "'", '"': '"'};
    const quotesBwd = {'」': '「', '』': '『', "'": "'", '"': '"'};
    const terminators = '…。．.？?！!';

    const sourceLocal = source.clone();
    const position = sourceLocal.setStartOffset(extent);
    sourceLocal.setEndOffset(position + extent);
    const content = sourceLocal.text();

    let quoteStack = [];

    let startPos = 0;
    for (let i = position; i >= startPos; --i) {
        const c = content[i];

        if (quoteStack.length === 0 && (terminators.includes(c) || c in quotesFwd)) {
            startPos = i + 1;
            break;
        }

        if (quoteStack.length > 0 && c === quoteStack[0]) {
            quoteStack.pop();
        } else if (c in quotesBwd) {
            quoteStack = [quotesBwd[c]].concat(quoteStack);
        }
    }

    quoteStack = [];

    let endPos = content.length - 1;
    for (let i = position; i <= endPos; ++i) {
        const c = content[i];

        if (quoteStack.length === 0) {
            if (terminators.includes(c)) {
                endPos = i + 1;
                break;
            }
            else if (c in quotesBwd) {
                endPos = i;
                break;
            }
        }

        if (quoteStack.length > 0 && c === quoteStack[0]) {
            quoteStack.pop();
        } else if (c in quotesFwd) {
            quoteStack = [quotesFwd[c]].concat(quoteStack);
        }
    }

    const sentence = content.substring(startPos, endPos);
    const clozePrefix = sentence.substring(0, position - startPos);
    const clozeBody = source.text();
    const clozeSuffix = sentence.substring(position - startPos + clozeBody.length);

    return {
        sentence: sentence.trim(),
        prefix: clozePrefix.trim(),
        body: clozeBody.trim(),
        suffix: clozeSuffix.trim()
    };
}