yomichan/ext/fg/js/driver.js
2017-03-31 21:48:10 -07:00

241 lines
6.9 KiB
JavaScript

/*
* 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/>.
*/
window.driver = new class {
constructor() {
this.popup = new Popup();
this.popupTimer = null;
this.lastMousePos = null;
this.mouseDownLeft = false;
this.mouseDownMiddle = false;
this.lastTextSource = null;
this.pendingLookup = false;
this.options = null;
bgOptionsGet().then(options => {
this.options = options;
window.addEventListener('mouseover', this.onMouseOver.bind(this));
window.addEventListener('mousedown', this.onMouseDown.bind(this));
window.addEventListener('mouseup', this.onMouseUp.bind(this));
window.addEventListener('mousemove', this.onMouseMove.bind(this));
window.addEventListener('resize', e => this.searchClear());
window.addEventListener('message', this.onFrameMessage.bind(this));
chrome.runtime.onMessage.addListener(this.onBgMessage.bind(this));
}).catch(this.handleError.bind(this));
}
popupTimerSet(callback) {
this.popupTimerClear();
this.popupTimer = window.setTimeout(callback, this.options.scanning.delay);
}
popupTimerClear() {
if (this.popupTimer) {
window.clearTimeout(this.popupTimer);
this.popupTimer = null;
}
}
onMouseOver(e) {
if (e.target === this.popup.container && this.popupTimer) {
this.popupTimerClear();
}
}
onMouseMove(e) {
this.lastMousePos = {x: e.clientX, y: e.clientY};
this.popupTimerClear();
if (!this.options.general.enable) {
return;
}
if (this.mouseDownLeft) {
return;
}
if (this.options.scanning.requireShift && !e.shiftKey && !(this.mouseDownMiddle && this.options.scanning.middleMouse)) {
return;
}
const searchFunc = () => this.searchAt(this.lastMousePos);
if (this.popup.isVisible()) {
searchFunc();
} else {
this.popupTimerSet(searchFunc);
}
}
onMouseDown(e) {
this.lastMousePos = {x: e.clientX, y: e.clientY};
this.popupTimerClear();
this.searchClear();
if (e.which === 1) {
this.mouseDownLeft = true;
} else if (e.which === 2) {
this.mouseDownMiddle = true;
}
}
onMouseUp(e) {
if (e.which === 1) {
this.mouseDownLeft = false;
} else if (e.which === 2) {
this.mouseDownMiddle = false;
}
}
onFrameMessage(e) {
const handlers = {
popupClose: () => {
this.searchClear();
},
selectionCopy: () => {
document.execCommand('copy');
}
};
const handler = handlers[e.data];
if (handler) {
handler();
}
}
onBgMessage({action, params}, sender, callback) {
const handlers = {
optionsSet: options => {
this.options = options;
if (!this.options.enable) {
this.searchClear();
}
}
};
const handler = handlers[action];
if (handler) {
handler(params);
}
callback();
}
searchAt(point) {
if (this.pendingLookup) {
return;
}
const textSource = docRangeFromPoint(point, this.options.scanning.imposter);
if (!textSource || !textSource.containsPoint(point)) {
return;
}
if (this.lastTextSource && this.lastTextSource.equals(textSource)) {
return;
}
this.pendingLookup = true;
this.searchTerms(textSource).then(found => {
if (!found) {
return this.searchKanji(textSource);
}
}).catch(error => {
this.handleError(error, textSource);
}).then(() => {
this.pendingLookup = false;
});
}
searchTerms(textSource) {
textSource.setEndOffset(this.options.scanning.length);
return bgTermsFind(textSource.text()).then(({definitions, length}) => {
if (definitions.length === 0) {
return false;
} else {
textSource.setEndOffset(length);
const cloze = docClozeExtract(textSource, this.options.anki.sentenceExt);
const url = window.location.href;
this.popup.showTermDefs(
textSource.getRect(),
definitions,
this.options,
{cloze, url}
);
this.lastTextSource = textSource;
if (this.options.scanning.selectText) {
textSource.select();
}
return true;
}
});
}
searchKanji(textSource) {
textSource.setEndOffset(1);
return bgKanjiFind(textSource.text()).then(definitions => {
if (definitions.length === 0) {
return false;
} else {
const cloze = docClozeExtract(textSource, this.options.anki.sentenceExt);
const url = window.location.href;
this.popup.showKanjiDefs(
textSource.getRect(),
definitions,
this.options,
{cloze, url}
);
this.lastTextSource = textSource;
if (this.options.scanning.selectText) {
textSource.select();
}
return true;
}
});
}
searchClear() {
docImposterDestroy();
this.popup.hide();
if (this.options.scanning.selectText && this.lastTextSource) {
this.lastTextSource.deselect();
}
this.lastTextSource = null;
}
handleError(error, textSource) {
if (window.orphaned) {
if (textSource && this.options.scanning.requireShift) {
this.popup.showOrphaned(textSource.getRect(), this.options);
}
} else {
window.alert(`Error: ${error}`);
}
}
};