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-09-13 05:14:58 +00:00
|
|
|
Handlebars.registerHelper('kanjiLinks', kanjiLinks);
|
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-08-21 02:15:30 +00:00
|
|
|
this.ankiConnectVer = 0;
|
2016-05-06 01:55:43 +00:00
|
|
|
this.setState('disabled');
|
2016-03-26 21:16:21 +00:00
|
|
|
|
2016-06-04 04:17:55 +00:00
|
|
|
chrome.runtime.onInstalled.addListener(this.onInstalled.bind(this));
|
|
|
|
chrome.runtime.onMessage.addListener(this.onMessage.bind(this));
|
|
|
|
chrome.browserAction.onClicked.addListener(this.onBrowserAction.bind(this));
|
2016-09-11 19:29:18 +00:00
|
|
|
chrome.tabs.onCreated.addListener(tab => this.onTabReady(tab.id));
|
2016-07-23 22:14:13 +00:00
|
|
|
chrome.tabs.onUpdated.addListener(this.onTabReady.bind(this));
|
2016-06-04 04:17:55 +00:00
|
|
|
|
2016-09-11 19:29:18 +00:00
|
|
|
loadOptions(opts => {
|
2016-05-06 01:55:43 +00:00
|
|
|
this.setOptions(opts);
|
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
|
|
|
}
|
|
|
|
|
2016-06-04 04:17:55 +00:00
|
|
|
onInstalled(details) {
|
|
|
|
if (details.reason === 'install') {
|
2016-08-14 19:08:06 +00:00
|
|
|
chrome.tabs.create({url: chrome.extension.getURL('bg/guide.html')});
|
2016-06-04 04:17:55 +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-07-24 06:02:55 +00:00
|
|
|
onTabReady(tabId) {
|
|
|
|
this.tabInvoke(tabId, 'setOptions', this.options);
|
|
|
|
this.tabInvoke(tabId, 'setEnabled', this.state === 'enabled');
|
2016-07-23 22:14:13 +00:00
|
|
|
}
|
|
|
|
|
2016-07-24 06:02:55 +00:00
|
|
|
onBrowserAction() {
|
2016-03-26 21:16:21 +00:00
|
|
|
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-09-11 19:40:45 +00:00
|
|
|
this.translator.loadData().then(() => this.setState('enabled'));
|
2016-03-28 00:04:49 +00:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2016-07-23 22:14:13 +00:00
|
|
|
this.tabInvokeAll('setEnabled', this.state === 'enabled');
|
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-07-23 22:14:13 +00:00
|
|
|
this.tabInvokeAll('setOptions', this.options);
|
2016-04-06 05:18:55 +00:00
|
|
|
}
|
|
|
|
|
2016-05-29 21:20:13 +00:00
|
|
|
getApiVersion() {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
2016-07-23 22:14:13 +00:00
|
|
|
tabInvokeAll(action, params) {
|
2016-09-11 19:29:18 +00:00
|
|
|
chrome.tabs.query({}, tabs => {
|
2016-08-14 19:08:06 +00:00
|
|
|
for (const tab of tabs) {
|
2016-07-23 22:14:13 +00:00
|
|
|
this.tabInvoke(tab.id, action, params);
|
2016-05-29 20:26:09 +00:00
|
|
|
}
|
|
|
|
});
|
|
|
|
}
|
|
|
|
|
2016-07-23 22:14:13 +00:00
|
|
|
tabInvoke(tabId, action, params) {
|
|
|
|
chrome.tabs.sendMessage(tabId, {action, params}, () => null);
|
|
|
|
}
|
|
|
|
|
2016-05-29 21:20:13 +00:00
|
|
|
ankiInvokeSafe(action, params, pool, callback) {
|
2016-08-21 02:15:30 +00:00
|
|
|
if (this.ankiConnectVer === this.getApiVersion()) {
|
|
|
|
this.ankiInvoke(action, params, pool, callback);
|
|
|
|
} else {
|
2016-09-11 19:29:18 +00:00
|
|
|
this.api_getVersion({callback: version => {
|
2016-08-21 02:15:30 +00:00
|
|
|
if (version === this.getApiVersion()) {
|
|
|
|
this.ankiConnectVer = version;
|
|
|
|
this.ankiInvoke(action, params, pool, callback);
|
|
|
|
} else {
|
|
|
|
callback(null);
|
|
|
|
}
|
|
|
|
}});
|
|
|
|
}
|
2016-05-29 21:20:13 +00:00
|
|
|
}
|
|
|
|
|
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);
|
|
|
|
});
|
|
|
|
|
2016-05-18 17:24:11 +00:00
|
|
|
xhr.open('POST', 'http://127.0.0.1:8765');
|
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-08-19 03:33:39 +00:00
|
|
|
const markers = [
|
2016-07-17 17:35:16 +00:00
|
|
|
'audio',
|
2016-07-11 01:16:51 +00:00
|
|
|
'character',
|
|
|
|
'expression',
|
|
|
|
'glossary',
|
|
|
|
'glossary-list',
|
|
|
|
'kunyomi',
|
|
|
|
'onyomi',
|
|
|
|
'reading',
|
2016-07-26 03:07:54 +00:00
|
|
|
'sentence',
|
2016-07-11 01:16:51 +00:00
|
|
|
'tags',
|
2016-07-26 03:07:54 +00:00
|
|
|
'url',
|
2016-07-11 01:16:51 +00:00
|
|
|
];
|
|
|
|
|
2016-08-19 03:33:39 +00:00
|
|
|
for (const marker of markers) {
|
|
|
|
let value = definition[marker] || null;
|
|
|
|
switch (marker) {
|
2016-07-17 17:35:16 +00:00
|
|
|
case 'audio':
|
|
|
|
value = '';
|
|
|
|
break;
|
2016-07-11 01:16:51 +00:00
|
|
|
case 'expression':
|
2016-08-10 15:44:52 +00:00
|
|
|
if (mode === 'term_kana' && definition.reading) {
|
2016-07-11 01:16:51 +00:00
|
|
|
value = definition.reading;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 'reading':
|
2016-08-10 15:44:52 +00:00
|
|
|
if (mode === 'term_kana') {
|
2016-07-11 01:16:51 +00:00
|
|
|
value = null;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 'glossary-list':
|
|
|
|
if (definition.glossary) {
|
|
|
|
value = '<ol>';
|
2016-08-14 19:08:06 +00:00
|
|
|
for (const gloss of definition.glossary) {
|
2016-07-11 01:16:51 +00:00
|
|
|
value += `<li>${gloss}</li>`;
|
|
|
|
}
|
|
|
|
value += '</ol>';
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case 'tags':
|
|
|
|
if (definition.tags) {
|
2016-09-11 19:29:18 +00:00
|
|
|
value = definition.tags.map(t => t.name);
|
2016-07-11 01:16:51 +00:00
|
|
|
}
|
|
|
|
break;
|
2016-05-29 19:46:37 +00:00
|
|
|
}
|
|
|
|
|
2016-05-29 22:00:19 +00:00
|
|
|
if (value !== null && typeof(value) !== 'string') {
|
|
|
|
value = value.join(', ');
|
|
|
|
}
|
|
|
|
|
2016-08-19 03:33:39 +00:00
|
|
|
field = field.replace(`{${marker}}`, value || '');
|
2016-05-29 19:46:37 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return field;
|
|
|
|
}
|
|
|
|
|
|
|
|
formatNote(definition, mode) {
|
2016-06-15 03:43:43 +00:00
|
|
|
const note = {fields: {}, tags: this.options.ankiCardTags};
|
2016-05-29 19:46:37 +00:00
|
|
|
|
|
|
|
let fields = [];
|
|
|
|
if (mode === 'kanji') {
|
2016-08-14 19:08:06 +00:00
|
|
|
fields = this.options.ankiKanjiFields;
|
|
|
|
note.deckName = this.options.ankiKanjiDeck;
|
2016-05-29 19:46:37 +00:00
|
|
|
note.modelName = this.options.ankiKanjiModel;
|
|
|
|
} else {
|
2016-08-14 19:08:06 +00:00
|
|
|
fields = this.options.ankiTermFields;
|
|
|
|
note.deckName = this.options.ankiTermDeck;
|
2016-08-10 15:44:52 +00:00
|
|
|
note.modelName = this.options.ankiTermModel;
|
2016-07-17 17:35:16 +00:00
|
|
|
|
|
|
|
const audio = {
|
|
|
|
kanji: definition.expression,
|
|
|
|
kana: definition.reading,
|
|
|
|
fields: []
|
|
|
|
};
|
|
|
|
|
2016-08-14 19:08:06 +00:00
|
|
|
for (const name in fields) {
|
2016-09-11 19:29:18 +00:00
|
|
|
if (fields[name].includes('{audio}')) {
|
2016-07-17 17:35:16 +00:00
|
|
|
audio.fields.push(name);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (audio.fields.length > 0) {
|
|
|
|
note.audio = audio;
|
|
|
|
}
|
2016-05-29 19:46:37 +00:00
|
|
|
}
|
|
|
|
|
2016-08-19 04:48:40 +00:00
|
|
|
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-08-14 19:08:06 +00:00
|
|
|
const notes = [];
|
|
|
|
for (const definition of definitions) {
|
|
|
|
for (const mode of modes) {
|
2016-05-29 19:46:37 +00:00
|
|
|
notes.push(this.formatNote(definition, mode));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-09-11 19:29:18 +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-09-11 19:29:18 +00:00
|
|
|
this.translator.findKanji(text).then(result => callback(result));
|
2016-05-22 00:44:02 +00:00
|
|
|
}
|
|
|
|
|
2016-05-22 01:07:57 +00:00
|
|
|
api_findTerm({text, callback}) {
|
2016-09-11 19:29:18 +00:00
|
|
|
this.translator.findTerm(text).then(result => callback(result));
|
2016-05-22 00:44:02 +00:00
|
|
|
}
|
|
|
|
|
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_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();
|