yomichan/ext/bg/js/yomichan.js

274 lines
8.0 KiB
JavaScript
Raw Normal View History

2016-03-26 21:16:21 +00:00
/*
* 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/>.
*/
class Yomichan {
constructor() {
2016-04-04 05:22:38 +00:00
Handlebars.partials = Handlebars.templates;
2016-04-18 02:06:57 +00:00
Handlebars.registerHelper('kanjiLinks', function(options) {
let result = '';
for (const c of options.fn(this)) {
if (Translator.isKanji(c)) {
2016-04-18 03:50:58 +00:00
result += Handlebars.templates['kanji-link.html']({kanji: c}).trim();
2016-04-18 02:06:57 +00:00
} else {
result += c;
}
}
return result;
});
2016-03-26 21:16:21 +00:00
2016-03-28 03:00:41 +00:00
this.translator = new Translator();
2016-05-07 04:34:06 +00:00
this.asyncPools = {};
2016-05-06 01:55:43 +00:00
this.setState('disabled');
2016-03-26 21:16:21 +00:00
2016-04-07 03:48:36 +00:00
loadOptions((opts) => {
2016-05-06 01:55:43 +00:00
this.setOptions(opts);
2016-04-06 05:18:55 +00:00
2016-04-07 03:48:36 +00:00
chrome.runtime.onMessage.addListener(this.onMessage.bind(this));
chrome.browserAction.onClicked.addListener(this.onBrowserAction.bind(this));
2016-04-07 04:04:38 +00:00
2016-05-08 01:24:31 +00:00
if (this.options.activateOnStartup) {
2016-05-06 01:55:43 +00:00
this.setState('loading');
2016-04-07 04:04:38 +00:00
}
2016-04-07 03:48:36 +00:00
});
2016-03-26 21:16:21 +00:00
}
onMessage(request, sender, callback) {
2016-05-22 01:07:57 +00:00
const {action, params} = request, method = this['api_' + action];
2016-03-26 21:16:21 +00:00
2016-05-22 00:44:02 +00:00
if (typeof(method) === 'function') {
2016-05-22 01:07:57 +00:00
params.callback = callback;
method.call(this, params);
2016-05-22 00:44:02 +00:00
}
2016-05-22 00:11:17 +00:00
2016-05-22 00:44:02 +00:00
return true;
2016-05-22 00:11:17 +00:00
}
2016-03-26 21:16:21 +00:00
onBrowserAction(tab) {
switch (this.state) {
case 'disabled':
2016-05-06 01:55:43 +00:00
this.setState('loading');
2016-03-26 21:16:21 +00:00
break;
case 'enabled':
2016-05-06 01:55:43 +00:00
this.setState('disabled');
2016-03-26 21:16:21 +00:00
break;
}
}
2016-05-06 01:55:43 +00:00
setState(state) {
2016-04-06 05:18:55 +00:00
if (this.state === state) {
return;
}
2016-03-26 21:16:21 +00:00
this.state = state;
2016-03-28 00:04:49 +00:00
switch (state) {
case 'disabled':
2016-05-21 22:40:08 +00:00
chrome.browserAction.setBadgeText({text: 'off'});
2016-03-28 00:04:49 +00:00
break;
case 'enabled':
2016-05-21 22:40:08 +00:00
chrome.browserAction.setBadgeText({text: ''});
2016-03-28 00:04:49 +00:00
break;
case 'loading':
chrome.browserAction.setBadgeText({text: '...'});
2016-05-22 00:44:02 +00:00
this.translator.loadData({loadEnamDict: this.options.loadEnamDict}, () => this.setState('enabled'));
2016-03-28 00:04:49 +00:00
break;
}
2016-05-29 19:46:37 +00:00
this.notifyTabs('state', this.state);
2016-04-06 05:18:55 +00:00
}
2016-05-06 01:55:43 +00:00
setOptions(options) {
2016-04-07 04:02:31 +00:00
this.options = options;
2016-05-29 19:46:37 +00:00
this.notifyTabs('options', this.options);
2016-04-06 05:18:55 +00:00
}
2016-05-29 21:20:13 +00:00
getApiVersion() {
return 1;
}
2016-05-29 20:26:09 +00:00
notifyTabs(name, value) {
chrome.tabs.query({}, (tabs) => {
for (const tab of tabs) {
chrome.tabs.sendMessage(tab.id, {name, value}, () => null);
}
});
}
2016-05-29 21:20:13 +00:00
ankiInvokeSafe(action, params, pool, callback) {
this.api_getVersion({callback: (version) => {
if (version === this.getApiVersion()) {
this.ankiInvoke(action, params, pool, callback);
} else {
callback(null);
}
}});
}
2016-05-07 04:34:06 +00:00
ankiInvoke(action, params, pool, callback) {
2016-05-07 04:42:15 +00:00
if (this.options.enableAnkiConnect) {
if (pool !== null && this.asyncPools.hasOwnProperty(pool)) {
this.asyncPools[pool].abort();
}
2016-05-06 03:06:13 +00:00
const xhr = new XMLHttpRequest();
xhr.addEventListener('loadend', () => {
2016-05-07 04:34:06 +00:00
if (pool !== null) {
delete this.asyncPools[pool];
}
2016-05-06 03:06:13 +00:00
const resp = xhr.responseText;
callback(resp ? JSON.parse(resp) : null);
});
xhr.open('POST', 'http://127.0.0.1:8765');
2016-05-06 03:06:13 +00:00
xhr.withCredentials = true;
xhr.setRequestHeader('Content-Type', 'text/json');
2016-05-29 19:46:37 +00:00
xhr.send(JSON.stringify({action, params}));
2016-05-06 03:06:13 +00:00
} else {
2016-05-03 00:27:26 +00:00
callback(null);
}
}
2016-05-29 22:00:19 +00:00
formatField(field, definition, mode) {
2016-05-29 19:46:37 +00:00
const supported = ['character', 'expression', 'glossary', 'kunyomi', 'onyomi', 'reading'];
for (const key in definition) {
if (supported.indexOf(key) === -1) {
continue;
}
let value = definition[key];
2016-05-29 22:00:19 +00:00
if (value !== null && typeof(value) !== 'string') {
value = value.join(', ');
}
if (mode === 'vocab_kana') {
2016-05-29 19:46:37 +00:00
if (key === 'expression') {
value = definition.reading;
} else if (key === 'reading') {
2016-05-29 22:00:19 +00:00
value = null;
2016-05-29 19:46:37 +00:00
}
}
2016-05-29 22:00:19 +00:00
if (mode !== 'kanji' && key === 'glossary') {
value = '<ol>';
for (const gloss of definition.glossary) {
value += `<li>${gloss}</li>`;
}
value += '</ol>';
2016-05-29 19:46:37 +00:00
}
2016-05-29 22:00:19 +00:00
field = field.replace(`{${key}}`, value || '');
2016-05-29 19:46:37 +00:00
}
return field;
}
formatNote(definition, mode) {
const note = {fields: {}, tags: []};
let fields = [];
if (mode === 'kanji') {
fields = this.options.ankiKanjiFields;
note.deckName = this.options.ankiKanjiDeck;
note.modelName = this.options.ankiKanjiModel;
} else {
fields = this.options.ankiVocabFields;
note.deckName = this.options.ankiVocabDeck;
note.modelName = this.options.ankiVocabModel;
}
for (const name in fields) {
2016-05-29 22:00:19 +00:00
note.fields[name] = this.formatField(fields[name], definition, mode);
2016-05-29 19:46:37 +00:00
}
return note;
}
2016-05-29 20:26:09 +00:00
api_addDefinition({definition, mode, callback}) {
2016-05-29 19:46:37 +00:00
const note = this.formatNote(definition, mode);
2016-05-29 21:20:13 +00:00
this.ankiInvokeSafe('addNote', {note}, null, callback);
2016-05-22 00:44:02 +00:00
}
2016-05-29 20:26:09 +00:00
api_canAddDefinitions({definitions, modes, callback}) {
2016-05-29 19:46:37 +00:00
let notes = [];
for (const definition of definitions) {
for (const mode of modes) {
notes.push(this.formatNote(definition, mode));
}
}
2016-05-29 21:20:13 +00:00
this.ankiInvokeSafe('canAddNotes', {notes}, 'notes', (results) => {
2016-05-29 19:46:37 +00:00
const states = [];
2016-05-29 20:26:09 +00:00
if (results !== null) {
for (let resultBase = 0; resultBase < results.length; resultBase += modes.length) {
const state = {};
for (let modeOffset = 0; modeOffset < modes.length; ++modeOffset) {
state[modes[modeOffset]] = results[resultBase + modeOffset];
}
states.push(state);
}
2016-05-29 19:46:37 +00:00
}
callback(states);
});
2016-05-22 00:44:02 +00:00
}
2016-05-22 01:07:57 +00:00
api_findKanji({text, callback}) {
2016-05-22 00:44:02 +00:00
callback(this.translator.findKanji(text));
}
2016-05-22 01:07:57 +00:00
api_findTerm({text, callback}) {
2016-05-22 00:44:02 +00:00
callback(this.translator.findTerm(text));
}
2016-05-22 01:07:57 +00:00
api_getDeckNames({callback}) {
2016-05-29 21:20:13 +00:00
this.ankiInvokeSafe('deckNames', {}, null, callback);
2016-05-22 00:44:02 +00:00
}
2016-05-22 01:07:57 +00:00
api_getModelNames({callback}) {
2016-05-29 21:20:13 +00:00
this.ankiInvokeSafe('modelNames', {}, null, callback);
2016-05-22 00:44:02 +00:00
}
2016-05-22 20:41:52 +00:00
api_getModelFieldNames({modelName, callback}) {
2016-05-29 21:20:13 +00:00
this.ankiInvokeSafe('modelFieldNames', {modelName}, null, callback);
}
api_getVersion({callback}) {
this.ankiInvoke('version', {}, null, callback);
2016-05-22 05:59:29 +00:00
}
2016-05-22 01:07:57 +00:00
api_getOptions({callback}) {
2016-05-22 00:44:02 +00:00
callback(this.options);
}
2016-05-22 01:07:57 +00:00
api_getState({callback}) {
2016-05-22 00:44:02 +00:00
callback(this.state);
}
2016-05-22 01:07:57 +00:00
api_renderText({template, data, callback}) {
2016-05-22 00:44:02 +00:00
callback(Handlebars.templates[template](data));
}
2016-03-26 21:16:21 +00:00
}
window.yomichan = new Yomichan();