yomichan/ext/mixed/js/display.js

/*
 * Copyright (C) 2017  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 Display {
    constructor(spinner, container) {
        this.spinner = spinner;
        this.container = container;
        this.definitions = [];
        this.options = null;
        this.context = null;
        this.sequence = 0;
        this.index = 0;
        this.audioPlaying = null;
        this.audioFallback = null;
        this.audioCache = {};
        this.styleNode = null;

        this.eventListeners = [];
        this.persistentEventListeners = [];
        this.interactive = false;
        this.eventListenersActive = false;
        this.clickScanPrevent = false;

        this.poppedContextHistory = [];

        this.windowScroll = new WindowScroll();

        this.setInteractive(true);
    }

    onError(_error) {
        throw new Error('Override me');
    }

    onSearchClear() {
        throw new Error('Override me');
    }

    onSourceTermView(e) {
        e.preventDefault();
        this.sourceTermView();
    }

    onHistoryTermView(e) {
        e.preventDefault();
        this.poppedTermView();
    }

    async onKanjiLookup(e) {
        try {
            e.preventDefault();
            if (!this.context) { return; }
            this.poppedContextHistory = [];

            const link = e.target;
            const context = {
                source: {
                    type: 'terms',
                    details: {
                        definitions: this.definitions,
                        context: Object.assign({}, this.context, {
                            index: this.entryIndexFind(link),
                            scroll: this.windowScroll.y
                        })
                    }
                },
                type: 'kanji',
                sentence: this.context.sentence,
                url: this.context.url
            };

            this.windowScroll.toY(0);

            const definitions = await apiKanjiFind(link.textContent, this.getOptionsContext());
            this.setContentKanji(definitions, context);
        } catch (error) {
            this.onError(error);
        }
    }

    onGlossaryMouseDown(e) {
        if (DOM.isMouseButtonPressed(e, 'primary')) {
            this.clickScanPrevent = false;
        }
    }

    onGlossaryMouseMove() {
        this.clickScanPrevent = true;
    }

    onGlossaryMouseUp(e) {
        if (!this.clickScanPrevent && DOM.isMouseButtonPressed(e, 'primary')) {
            this.onTermLookup(e);
        }
    }

    async onTermLookup(e, {disableScroll, selectText, disableHistory}={}) {
        try {
            if (!this.context) { return; }
            this.poppedContextHistory = [];
            const termLookupResults = await this.termLookup(e);
            if (!termLookupResults) { return; }
            const {textSource, definitions} = termLookupResults;

            const scannedElement = e.target;
            const sentence = docSentenceExtract(textSource, this.options.anki.sentenceExt);

            const context = {
                source: disableHistory ? this.context.source : {
                    type: 'terms',
                    details: {
                        definitions: this.definitions,
                        context: Object.assign({}, this.context, {
                            index: this.entryIndexFind(scannedElement),
                            scroll: this.windowScroll.y
                        })
                    }
                },
                type: 'terms',
                disableScroll,
                sentence,
                url: this.context.url
            };

            this.setContentTerms(definitions, context);

            if (!disableScroll) {
                this.windowScroll.toY(0);
            }

            if (selectText) {
                textSource.select();
            }
        } catch (error) {
            this.onError(error);
        }
    }

    async termLookup(e) {
        try {
            e.preventDefault();

            const textSource = docRangeFromPoint(e.clientX, e.clientY, this.options);
            if (textSource === null) {
                return false;
            }

            let definitions, length;
            try {
                textSource.setEndOffset(this.options.scanning.length);

                ({definitions, length} = await apiTermsFind(textSource.text(), {}, this.getOptionsContext()));
                if (definitions.length === 0) {
                    return false;
                }

                textSource.setEndOffset(length);
            } finally {
                textSource.cleanup();
            }

            return {textSource, definitions};
        } catch (error) {
            this.onError(error);
        }
    }

    onAudioPlay(e) {
        e.preventDefault();
        const link = e.currentTarget;
        const entry = link.closest('.entry');
        const definitionIndex = this.entryIndexFind(entry);
        const expressionIndex = Display.indexOf(entry.querySelectorAll('.expression .action-play-audio'), link);
        this.audioPlay(this.definitions[definitionIndex], expressionIndex, definitionIndex);
    }

    onNoteAdd(e) {
        e.preventDefault();
        const link = e.currentTarget;
        const index = this.entryIndexFind(link);
        this.noteAdd(this.definitions[index], link.dataset.mode);
    }

    onNoteView(e) {
        e.preventDefault();
        const link = e.currentTarget;
        apiNoteView(link.dataset.noteId);
    }

    onKeyDown(e) {
        const key = Display.getKeyFromEvent(e);
        const handlers = Display.onKeyDownHandlers;
        if (hasOwn(handlers, key)) {
            const handler = handlers[key];
            if (handler(this, e)) {
                e.preventDefault();
                return true;
            }
        }
        return false;
    }

    onWheel(e) {
        if (e.altKey) {
            if (e.deltaY !== 0) {
                this.entryScrollIntoView(this.index + (e.deltaY > 0 ? 1 : -1), null, true);
                e.preventDefault();
            }
        } else if (e.shiftKey) {
            const delta = -e.deltaX || e.deltaY;
            if (delta > 0) {
                this.sourceTermView();
                e.preventDefault();
            } else if (delta < 0) {
                this.poppedTermView();
                e.preventDefault();
            }
        }
    }

    onRuntimeMessage({action, params}, sender, callback) {
        const handlers = Display.runtimeMessageHandlers;
        if (hasOwn(handlers, action)) {
            const handler = handlers[action];
            const result = handler(this, params);
            callback(result);
        }
    }

    getOptionsContext() {
        throw new Error('Override me');
    }

    isInitialized() {
        return this.options !== null;
    }

    async initialize(options=null) {
        await this.updateOptions(options);
        chrome.runtime.onMessage.addListener(this.onRuntimeMessage.bind(this));
    }

    async updateOptions(options) {
        this.options = options ? options : await apiOptionsGet(this.getOptionsContext());
        this.updateTheme(this.options.general.popupTheme);
        this.setCustomCss(this.options.general.customPopupCss);
        audioPrepareTextToSpeech(this.options);
    }

    updateTheme(themeName) {
        document.documentElement.dataset.yomichanTheme = themeName;

        const stylesheets = document.querySelectorAll('link[data-yomichan-theme-name]');
        for (const stylesheet of stylesheets) {
            const match = (stylesheet.dataset.yomichanThemeName === themeName);
            stylesheet.rel = (match ? 'stylesheet' : 'stylesheet alternate');
        }
    }

    setCustomCss(css) {
        if (this.styleNode === null) {
            if (css.length === 0) { return; }
            this.styleNode = document.createElement('style');
        }

        this.styleNode.textContent = css;

        const parent = document.head;
        if (this.styleNode.parentNode !== parent) {
            parent.appendChild(this.styleNode);
        }
    }

    setInteractive(interactive) {
        interactive = !!interactive;
        if (this.interactive === interactive) { return; }
        this.interactive = interactive;

        if (interactive) {
            Display.addEventListener(this.persistentEventListeners, document, 'keydown', this.onKeyDown.bind(this), false);
            Display.addEventListener(this.persistentEventListeners, document, 'wheel', this.onWheel.bind(this), {passive: false});
        } else {
            Display.clearEventListeners(this.persistentEventListeners);
        }
        this.setEventListenersActive(this.eventListenersActive);
    }

    setEventListenersActive(active) {
        active = !!active && this.interactive;
        if (this.eventListenersActive === active) { return; }
        this.eventListenersActive = active;

        if (active) {
            this.addEventListeners('.action-add-note', 'click', this.onNoteAdd.bind(this));
            this.addEventListeners('.action-view-note', 'click', this.onNoteView.bind(this));
            this.addEventListeners('.action-play-audio', 'click', this.onAudioPlay.bind(this));
            this.addEventListeners('.kanji-link', 'click', this.onKanjiLookup.bind(this));
            this.addEventListeners('.source-term', 'click', this.onSourceTermView.bind(this));
            this.addEventListeners('.history-term', 'click', this.onHistoryTermView.bind(this));
            if (this.options.scanning.enablePopupSearch) {
                this.addEventListeners('.glossary-item', 'mouseup', this.onGlossaryMouseUp.bind(this));
                this.addEventListeners('.glossary-item', 'mousedown', this.onGlossaryMouseDown.bind(this));
                this.addEventListeners('.glossary-item', 'mousemove', this.onGlossaryMouseMove.bind(this));
            }
        } else {
            Display.clearEventListeners(this.eventListeners);
        }
    }

    addEventListeners(selector, type, listener, options) {
        for (const node of this.container.querySelectorAll(selector)) {
            Display.addEventListener(this.eventListeners, node, type, listener, options);
        }
    }

    setContent(type, details) {
        switch (type) {
            case 'terms':
                return this.setContentTerms(details.definitions, details.context);
            case 'kanji':
                return this.setContentKanji(details.definitions, details.context);
            case 'orphaned':
                return this.setContentOrphaned();
            default:
                return Promise.resolve();
        }
    }

    async setContentTerms(definitions, context) {
        if (!context) { throw new Error('Context expected'); }
        if (!this.isInitialized()) { return; }

        try {
            const options = this.options;

            this.setEventListenersActive(false);

            if (context.focus !== false) {
                window.focus();
            }

            this.definitions = definitions;
            this.context = context;

            const sequence = ++this.sequence;
            const params = {
                definitions,
                source: context.source,
                history: this.poppedContextHistory.length > 0,
                addable: options.anki.enable,
                grouped: options.general.resultOutputMode === 'group',
                merged: options.general.resultOutputMode === 'merge',
                playback: options.audio.enabled,
                compactGlossaries: options.general.compactGlossaries,
                debug: options.general.debugInfo
            };

            for (const definition of definitions) {
                definition.cloze = Display.clozeBuild(context.sentence, definition.source);
                definition.url = context.url;
            }

            const content = await apiTemplateRender('terms.html', params);
            this.container.innerHTML = content;
            const {index, scroll, disableScroll} = context;
            if (!disableScroll) {
                this.entryScrollIntoView(index || 0, scroll);
            } else {
                this.entrySetCurrent(index || 0);
            }

            if (options.audio.enabled && options.audio.autoPlay) {
                this.autoPlayAudio();
            }

            this.setEventListenersActive(true);

            await this.adderButtonUpdate(['term-kanji', 'term-kana'], sequence);
        } catch (e) {
            this.onError(e);
        }
    }

    async setContentKanji(definitions, context) {
        if (!context) { throw new Error('Context expected'); }
        if (!this.isInitialized()) { return; }

        try {
            const options = this.options;

            this.setEventListenersActive(false);

            if (context.focus !== false) {
                window.focus();
            }

            this.definitions = definitions;
            this.context = context;

            const sequence = ++this.sequence;
            const params = {
                definitions,
                source: context.source,
                history: this.poppedContextHistory.length > 0,
                addable: options.anki.enable,
                debug: options.general.debugInfo
            };

            for (const definition of definitions) {
                definition.cloze = Display.clozeBuild(context.sentence);
                definition.url = context.url;
            }

            const content = await apiTemplateRender('kanji.html', params);
            this.container.innerHTML = content;
            const {index, scroll} = context;
            this.entryScrollIntoView(index || 0, scroll);

            this.setEventListenersActive(true);

            await this.adderButtonUpdate(['kanji'], sequence);
        } catch (e) {
            this.onError(e);
        }
    }

    async setContentOrphaned() {
        const definitions = document.querySelector('#definitions');
        const errorOrphaned = document.querySelector('#error-orphaned');

        if (definitions !== null) {
            definitions.style.setProperty('display', 'none', 'important');
        }

        if (errorOrphaned !== null) {
            errorOrphaned.style.setProperty('display', 'block', 'important');
        }
    }

    autoPlayAudio() {
        this.audioPlay(this.definitions[0], this.firstExpressionIndex, 0);
    }

    async adderButtonUpdate(modes, sequence) {
        try {
            const states = await apiDefinitionsAddable(this.definitions, modes, this.getOptionsContext());
            if (!states || sequence !== this.sequence) {
                return;
            }

            for (let i = 0; i < states.length; ++i) {
                const state = states[i];
                let noteId = null;
                for (const mode in state) {
                    const button = this.adderButtonFind(i, mode);
                    if (button === null) {
                        continue;
                    }

                    const info = state[mode];
                    if (!info.canAdd && noteId === null && info.noteId) {
                        noteId = info.noteId;
                    }
                    button.classList.toggle('disabled', !info.canAdd);
                    button.classList.remove('pending');
                }
                if (noteId !== null) {
                    this.viewerButtonShow(i, noteId);
                }
            }
        } catch (e) {
            this.onError(e);
        }
    }

    entrySetCurrent(index) {
        index = Math.min(index, this.definitions.length - 1);
        index = Math.max(index, 0);

        const entryPre = this.getEntry(this.index);
        if (entryPre !== null) {
            entryPre.classList.remove('entry-current');
        }

        const entry = this.getEntry(index);
        if (entry !== null) {
            entry.classList.add('entry-current');
        }

        this.index = index;

        return entry;
    }

    entryScrollIntoView(index, scroll, smooth) {
        this.windowScroll.stop();

        const entry = this.entrySetCurrent(index);
        let target;
        if (scroll !== null) {
            target = scroll;
        } else {
            target = index === 0 || entry === null ? 0 : Display.getElementTop(entry);
        }

        if (smooth) {
            this.windowScroll.animate(this.windowScroll.x, target, 200);
        } else {
            this.windowScroll.toY(target);
        }
    }

    sourceTermView() {
        if (!this.context || !this.context.source) { return; }
        this.poppedContextHistory.push({
            type: this.context.type,
            details: {
                definitions: this.definitions,
                context: Object.assign({}, this.context, {
                    index: this.index,
                    scroll: this.windowScroll.y
                })
            }
        });
        const {type, details} = this.context.source;
        this.setContent(type, details);
    }

    poppedTermView() {
        if (this.poppedContextHistory.length === 0) { return; }
        this.context.index = this.index;
        this.context.scroll = this.windowScroll.y;
        const {type, details} = this.poppedContextHistory.pop();
        this.setContent(type, details);
    }

    noteTryAdd(mode) {
        const button = this.adderButtonFind(this.index, mode);
        if (button !== null && !button.classList.contains('disabled')) {
            this.noteAdd(this.definitions[this.index], mode);
        }
    }

    noteTryView() {
        const button = this.viewerButtonFind(this.index);
        if (button !== null && !button.classList.contains('disabled')) {
            apiNoteView(button.dataset.noteId);
        }
    }

    async noteAdd(definition, mode) {
        try {
            this.setSpinnerVisible(true);

            const context = {};
            if (this.noteUsesScreenshot()) {
                const screenshot = await this.getScreenshot();
                if (screenshot) {
                    context.screenshot = screenshot;
                }
            }

            const noteId = await apiDefinitionAdd(definition, mode, context, this.getOptionsContext());
            if (noteId) {
                const index = this.definitions.indexOf(definition);
                const adderButton = this.adderButtonFind(index, mode);
                if (adderButton !== null) {
                    adderButton.classList.add('disabled');
                }
                this.viewerButtonShow(index, noteId);
            } else {
                throw new Error('Note could not be added');
            }
        } catch (e) {
            this.onError(e);
        } finally {
            this.setSpinnerVisible(false);
        }
    }

    async audioPlay(definition, expressionIndex, entryIndex) {
        try {
            this.setSpinnerVisible(true);

            const expression = expressionIndex === -1 ? definition : definition.expressions[expressionIndex];

            if (this.audioPlaying !== null) {
                this.audioPlaying.pause();
                this.audioPlaying = null;
            }

            const sources = this.options.audio.sources;
            let {audio, source} = await audioGetFromSources(expression, sources, this.getOptionsContext(), false, this.audioCache);
            let info;
            if (audio === null) {
                if (this.audioFallback === null) {
                    this.audioFallback = new Audio('/mixed/mp3/button.mp3');
                }
                audio = this.audioFallback;
                info = 'Could not find audio';
            } else {
                info = `From source ${1 + sources.indexOf(source)}: ${source}`;
            }

            const button = this.audioButtonFindImage(entryIndex);
            if (button !== null) {
                let titleDefault = button.dataset.titleDefault;
                if (!titleDefault) {
                    titleDefault = button.title || '';
                    button.dataset.titleDefault = titleDefault;
                }
                button.title = `${titleDefault}\n${info}`;
            }

            this.audioPlaying = audio;
            audio.currentTime = 0;
            audio.volume = this.options.audio.volume / 100.0;
            audio.play();
        } catch (e) {
            this.onError(e);
        } finally {
            this.setSpinnerVisible(false);
        }
    }

    noteUsesScreenshot() {
        const fields = this.options.anki.terms.fields;
        for (const name in fields) {
            if (fields[name].includes('{screenshot}')) {
                return true;
            }
        }
        return false;
    }

    async getScreenshot() {
        try {
            await this.setPopupVisibleOverride(false);
            await Display.delay(1); // Wait for popup to be hidden.

            const {format, quality} = this.options.anki.screenshot;
            const dataUrl = await apiScreenshotGet({format, quality});
            if (!dataUrl || dataUrl.error) { return; }

            return {dataUrl, format};
        } finally {
            await this.setPopupVisibleOverride(null);
        }
    }

    get firstExpressionIndex() {
        return this.options.general.resultOutputMode === 'merge' ? 0 : -1;
    }

    setPopupVisibleOverride(visible) {
        return apiForward('popupSetVisibleOverride', {visible});
    }

    setSpinnerVisible(visible) {
        this.spinner.style.display = visible ? 'block' : '';
    }

    getEntry(index) {
        const entries = this.container.querySelectorAll('.entry');
        return index >= 0 && index < entries.length ? entries[index] : null;
    }

    static clozeBuild(sentence, source) {
        const result = {
            sentence: sentence.text.trim()
        };

        if (source) {
            result.prefix = sentence.text.substring(0, sentence.offset).trim();
            result.body = source.trim();
            result.suffix = sentence.text.substring(sentence.offset + source.length).trim();
        }

        return result;
    }

    entryIndexFind(element) {
        const entry = element.closest('.entry');
        return entry !== null ? Display.indexOf(this.container.querySelectorAll('.entry'), entry) : -1;
    }

    adderButtonFind(index, mode) {
        const entry = this.getEntry(index);
        return entry !== null ? entry.querySelector(`.action-add-note[data-mode="${mode}"]`) : null;
    }

    viewerButtonFind(index) {
        const entry = this.getEntry(index);
        return entry !== null ? entry.querySelector('.action-view-note') : null;
    }

    viewerButtonShow(index, noteId) {
        const viewerButton = this.viewerButtonFind(index);
        if (viewerButton === null) {
            return;
        }
        viewerButton.classList.remove('pending', 'disabled');
        viewerButton.dataset.noteId = noteId;
    }

    audioButtonFindImage(index) {
        const entry = this.getEntry(index);
        return entry !== null ? entry.querySelector('.action-play-audio>img') : null;
    }

    static delay(time) {
        return new Promise((resolve) => setTimeout(resolve, time));
    }

    static indexOf(nodeList, node) {
        for (let i = 0, ii = nodeList.length; i < ii; ++i) {
            if (nodeList[i] === node) {
                return i;
            }
        }
        return -1;
    }

    static addEventListener(eventListeners, object, type, listener, options) {
        object.addEventListener(type, listener, options);
        eventListeners.push([object, type, listener, options]);
    }

    static clearEventListeners(eventListeners) {
        for (const [object, type, listener, options] of eventListeners) {
            object.removeEventListener(type, listener, options);
        }
        eventListeners.length = 0;
    }

    static getElementTop(element) {
        const elementRect = element.getBoundingClientRect();
        const documentRect = document.documentElement.getBoundingClientRect();
        return elementRect.top - documentRect.top;
    }

    static getKeyFromEvent(event) {
        const key = event.key;
        return (typeof key === 'string' ? (key.length === 1 ? key.toUpperCase() : key) : '');
    }
}

Display.onKeyDownHandlers = {
    'Escape': (self) => {
        self.onSearchClear();
        return true;
    },

    'PageUp': (self, e) => {
        if (e.altKey) {
            self.entryScrollIntoView(self.index - 3, null, true);
            return true;
        }
        return false;
    },

    'PageDown': (self, e) => {
        if (e.altKey) {
            self.entryScrollIntoView(self.index + 3, null, true);
            return true;
        }
        return false;
    },

    'End': (self, e) => {
        if (e.altKey) {
            self.entryScrollIntoView(self.definitions.length - 1, null, true);
            return true;
        }
        return false;
    },

    'Home': (self, e) => {
        if (e.altKey) {
            self.entryScrollIntoView(0, null, true);
            return true;
        }
        return false;
    },

    'ArrowUp': (self, e) => {
        if (e.altKey) {
            self.entryScrollIntoView(self.index - 1, null, true);
            return true;
        }
        return false;
    },

    'ArrowDown': (self, e) => {
        if (e.altKey) {
            self.entryScrollIntoView(self.index + 1, null, true);
            return true;
        }
        return false;
    },

    'B': (self, e) => {
        if (e.altKey) {
            self.sourceTermView();
            return true;
        }
        return false;
    },

    'F': (self, e) => {
        if (e.altKey) {
            self.poppedTermView();
            return true;
        }
        return false;
    },

    'E': (self, e) => {
        if (e.altKey) {
            self.noteTryAdd('term-kanji');
            return true;
        }
        return false;
    },

    'K': (self, e) => {
        if (e.altKey) {
            self.noteTryAdd('kanji');
            return true;
        }
        return false;
    },

    'R': (self, e) => {
        if (e.altKey) {
            self.noteTryAdd('term-kana');
            return true;
        }
        return false;
    },

    'P': (self, e) => {
        if (e.altKey) {
            const entry = self.getEntry(self.index);
            if (entry !== null && entry.dataset.type === 'term') {
                self.audioPlay(self.definitions[self.index], self.firstExpressionIndex, self.index);
            }
            return true;
        }
        return false;
    },

    'V': (self, e) => {
        if (e.altKey) {
            self.noteTryView();
            return true;
        }
        return false;
    }
};

Display.runtimeMessageHandlers = {
    optionsUpdate: (self) => self.updateOptions(null)
};