/* * 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.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(); } onNextTermView(e) { e.preventDefault(); this.nextTermView(); } async onKanjiLookup(e) { try { e.preventDefault(); if (!this.context) { return; } const link = e.target; this.context.update({ index: this.entryIndexFind(link), scroll: this.windowScroll.y }); const context = { sentence: this.context.get('sentence'), url: this.context.get('url') }; 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; } 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); this.context.update({ index: this.entryIndexFind(scannedElement), scroll: this.windowScroll.y }); const context = { disableScroll, disableHistory, sentence, url: this.context.get('url') }; if (disableHistory) { Object.assign(context, { previous: this.context.previous, next: this.context.next }); } else { Object.assign(context, { previous: this.context }); } this.setContentTerms(definitions, context); 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.nextTermView(); 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('.next-term', 'click', this.onNextTermView.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; if (context.disableHistory) { delete context.disableHistory; this.context = new DisplayContext('terms', definitions, context); } else { this.context = DisplayContext.push(this.context, 'terms', definitions, context); } const sequence = ++this.sequence; const params = { definitions, source: this.context.previous, next: this.context.next, 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 { delete context.disableScroll; 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; if (context.disableHistory) { delete context.disableHistory; this.context = new DisplayContext('kanji', definitions, context); } else { this.context = DisplayContext.push(this.context, 'kanji', definitions, context); } const sequence = ++this.sequence; const params = { definitions, source: this.context.previous, next: this.context.next, 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.previous) { return; } this.context.update({ index: this.index, scroll: this.windowScroll.y }); const previousContext = this.context.previous; previousContext.set('disableHistory', true); const details = { definitions: previousContext.definitions, context: previousContext.context }; this.setContent(previousContext.type, details); } nextTermView() { if (!this.context || !this.context.next) { return; } this.context.update({ index: this.index, scroll: this.windowScroll.y }); const nextContext = this.context.next; nextContext.set('disableHistory', true); const details = { definitions: nextContext.definitions, context: nextContext.context }; this.setContent(nextContext.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.nextTermView(); 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) };