237 lines
7.0 KiB
JavaScript
237 lines
7.0 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/>.
|
||
|
*/
|
||
|
|
||
|
|
||
|
function utilMessageDispatch({action, params}, sender, callback) {
|
||
|
const forward = (promise, callback) => {
|
||
|
return promise.then(result => {
|
||
|
callback({result});
|
||
|
}).catch(error => {
|
||
|
callback({error});
|
||
|
});
|
||
|
};
|
||
|
|
||
|
const handlers = {
|
||
|
optionsGet: ({callback}) => {
|
||
|
forward(optionsLoad(), callback);
|
||
|
},
|
||
|
|
||
|
kanjiFind: ({text, callback}) => {
|
||
|
forward(apiKanjiFind(text), callback);
|
||
|
},
|
||
|
|
||
|
termsFind: ({text, callback}) => {
|
||
|
forward(apiTermsFind(text), callback);
|
||
|
},
|
||
|
|
||
|
templateRender: ({template, data, callback}) => {
|
||
|
forward(apiTemplateRender(template, data), callback);
|
||
|
},
|
||
|
|
||
|
definitionAdd: ({definition, mode, callback}) => {
|
||
|
forward(apiDefinitionAdd(definition, mode), callback);
|
||
|
},
|
||
|
|
||
|
definitionsAddable: ({definitions, modes, callback}) => {
|
||
|
forward(apiDefinitionsAddable(definitions, modes), callback);
|
||
|
},
|
||
|
|
||
|
noteView: ({noteId}) => {
|
||
|
forward(apiNoteView(noteId), callback);
|
||
|
}
|
||
|
};
|
||
|
|
||
|
const handler = handlers[action];
|
||
|
if (handler) {
|
||
|
params.callback = callback;
|
||
|
handler(params);
|
||
|
}
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
function utilCommandDispatch(command) {
|
||
|
const handlers = {
|
||
|
search: () => {
|
||
|
chrome.tabs.create({url: chrome.extension.getURL('/bg/search.html')});
|
||
|
},
|
||
|
|
||
|
help: () => {
|
||
|
chrome.tabs.create({url: 'https://foosoft.net/projects/yomichan/'});
|
||
|
},
|
||
|
|
||
|
options: () => {
|
||
|
chrome.runtime.openOptionsPage();
|
||
|
},
|
||
|
|
||
|
toggle: () => {
|
||
|
this.options.general.enable = !this.options.general.enable;
|
||
|
optionsSave(this.options).then(() => this.optionsSet(this.options));
|
||
|
}
|
||
|
};
|
||
|
|
||
|
const handler = handlers[command];
|
||
|
if (handler) {
|
||
|
handler();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
function utilNoteFormat(definition, mode) {
|
||
|
const options = Backend.instance().options;
|
||
|
const note = {fields: {}, tags: options.anki.tags};
|
||
|
let fields = [];
|
||
|
|
||
|
if (mode === 'kanji') {
|
||
|
fields = options.anki.kanji.fields;
|
||
|
note.deckName = options.anki.kanji.deck;
|
||
|
note.modelName = options.anki.kanji.model;
|
||
|
} else {
|
||
|
fields = options.anki.terms.fields;
|
||
|
note.deckName = options.anki.terms.deck;
|
||
|
note.modelName = options.anki.terms.model;
|
||
|
|
||
|
if (definition.audio) {
|
||
|
const audio = {
|
||
|
url: definition.audio.url,
|
||
|
filename: definition.audio.filename,
|
||
|
skipHash: '7e2c2f954ef6051373ba916f000168dc',
|
||
|
fields: []
|
||
|
};
|
||
|
|
||
|
for (const name in fields) {
|
||
|
if (fields[name].includes('{audio}')) {
|
||
|
audio.fields.push(name);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (audio.fields.length > 0) {
|
||
|
note.audio = audio;
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
for (const name in fields) {
|
||
|
note.fields[name] = dictFieldFormat(fields[name], definition, mode, options);
|
||
|
}
|
||
|
|
||
|
return note;
|
||
|
}
|
||
|
|
||
|
async function apiOptionsSet(options) {
|
||
|
// In Firefox, setting options from the options UI somehow carries references
|
||
|
// to the DOM across to the background page, causing the options object to
|
||
|
// become a "DeadObject" after the options page is closed. The workaround used
|
||
|
// here is to create a deep copy of the options object.
|
||
|
Backend.instance().options = JSON.parse(JSON.stringify(options));
|
||
|
|
||
|
if (!options.general.enable) {
|
||
|
chrome.browserAction.setBadgeBackgroundColor({color: '#d9534f'});
|
||
|
chrome.browserAction.setBadgeText({text: 'off'});
|
||
|
} else if (!dictConfigured(options)) {
|
||
|
chrome.browserAction.setBadgeBackgroundColor({color: '#f0ad4e'});
|
||
|
chrome.browserAction.setBadgeText({text: '!'});
|
||
|
} else {
|
||
|
chrome.browserAction.setBadgeText({text: ''});
|
||
|
}
|
||
|
|
||
|
if (options.anki.enable) {
|
||
|
Backend.instance().anki = new AnkiConnect(options.anki.server);
|
||
|
} else {
|
||
|
Backend.instance().anki = new AnkiNull();
|
||
|
}
|
||
|
|
||
|
chrome.tabs.query({}, tabs => {
|
||
|
for (const tab of tabs) {
|
||
|
chrome.tabs.sendMessage(tab.id, {action: 'optionsSet', params: options}, () => null);
|
||
|
}
|
||
|
});
|
||
|
}
|
||
|
|
||
|
async function apiOptionsGet() {
|
||
|
return Backend.instance().options;
|
||
|
}
|
||
|
|
||
|
async function apiTermsFind(text) {
|
||
|
const options = Backend.instance().options;
|
||
|
const translator = Backend.instance().translator;
|
||
|
|
||
|
const searcher = options.general.groupResults ?
|
||
|
translator.findTermsGrouped.bind(translator) :
|
||
|
translator.findTerms.bind(translator);
|
||
|
|
||
|
const {definitions, length} = await searcher(
|
||
|
text,
|
||
|
dictEnabledSet(options),
|
||
|
options.scanning.alphanumeric
|
||
|
);
|
||
|
|
||
|
return {
|
||
|
length,
|
||
|
definitions: definitions.slice(0, options.general.maxResults)
|
||
|
};
|
||
|
}
|
||
|
|
||
|
async function apiKanjiFind(text) {
|
||
|
const options = Backend.instance().options;
|
||
|
const definitions = await Backend.instance().translator.findKanji(text, dictEnabledSet(options));
|
||
|
return definitions.slice(0, options.general.maxResults);
|
||
|
}
|
||
|
|
||
|
async function apiDefinitionAdd(definition, mode) {
|
||
|
if (mode !== 'kanji') {
|
||
|
const options = Backend.instance().options;
|
||
|
await audioInject(
|
||
|
definition,
|
||
|
options.anki.terms.fields,
|
||
|
options.general.audioSource
|
||
|
);
|
||
|
}
|
||
|
|
||
|
return Backend.instance().anki.addNote(utilNoteFormat(definition, mode));
|
||
|
}
|
||
|
|
||
|
async function apiDefinitionsAddable(definitions, modes) {
|
||
|
const notes = [];
|
||
|
for (const definition of definitions) {
|
||
|
for (const mode of modes) {
|
||
|
notes.push(utilNoteFormat(definition, mode));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
const results = await Backend.instance().anki.canAddNotes(notes);
|
||
|
const states = [];
|
||
|
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);
|
||
|
}
|
||
|
|
||
|
return states;
|
||
|
}
|
||
|
|
||
|
async function apiNoteView(noteId) {
|
||
|
return Backend.instance().anki.guiBrowse(`nid:${noteId}`);
|
||
|
}
|
||
|
|
||
|
async function apiTemplateRender(template, data) {
|
||
|
return handlebarsRender(template, data);
|
||
|
}
|