567 lines
15 KiB
JavaScript
567 lines
15 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/>.
|
|
*/
|
|
|
|
|
|
/*
|
|
* Promise
|
|
*/
|
|
|
|
function promiseCallback(promise, callback) {
|
|
return promise.then(result => {
|
|
callback({result});
|
|
}).catch(error => {
|
|
/* eslint-disable */
|
|
console.log(error);
|
|
/* eslint-enable */
|
|
callback({error});
|
|
});
|
|
}
|
|
|
|
|
|
/*
|
|
* Instance
|
|
*/
|
|
|
|
function instYomi() {
|
|
return chrome.extension.getBackgroundPage().yomichan;
|
|
}
|
|
|
|
function instDb() {
|
|
return instYomi().translator.database;
|
|
}
|
|
|
|
function instAnki() {
|
|
return instYomi().anki;
|
|
}
|
|
|
|
|
|
/*
|
|
* Foreground
|
|
*/
|
|
|
|
function fgBroadcast(action, params) {
|
|
chrome.tabs.query({}, tabs => {
|
|
for (const tab of tabs) {
|
|
chrome.tabs.sendMessage(tab.id, {action, params}, () => null);
|
|
}
|
|
});
|
|
}
|
|
|
|
function fgOptionsSet(options) {
|
|
fgBroadcast('optionsSet', options);
|
|
}
|
|
|
|
|
|
/*
|
|
* Options
|
|
*/
|
|
|
|
function optionsSetDefaults(options) {
|
|
const defaults = {
|
|
general: {
|
|
enable: true,
|
|
audioPlayback: true,
|
|
groupResults: true,
|
|
softKatakana: true,
|
|
maxResults: 32,
|
|
showAdvanced: false,
|
|
popupWidth: 400,
|
|
popupHeight: 250,
|
|
popupOffset: 10
|
|
},
|
|
|
|
scanning: {
|
|
requireShift: true,
|
|
middleMouse: true,
|
|
selectText: true,
|
|
imposter: true,
|
|
delay: 15,
|
|
length: 10
|
|
},
|
|
|
|
dictionaries: {},
|
|
|
|
anki: {
|
|
enable: false,
|
|
server: 'http://127.0.0.1:8765',
|
|
tags: ['yomichan'],
|
|
htmlCards: true,
|
|
sentenceExt: 200,
|
|
terms: {deck: '', model: '', fields: {}},
|
|
kanji: {deck: '', model: '', fields: {}}
|
|
}
|
|
};
|
|
|
|
const combine = (target, source) => {
|
|
for (const key in source) {
|
|
if (!target.hasOwnProperty(key)) {
|
|
target[key] = source[key];
|
|
}
|
|
}
|
|
};
|
|
|
|
combine(options, defaults);
|
|
combine(options.general, defaults.general);
|
|
combine(options.scanning, defaults.scanning);
|
|
combine(options.anki, defaults.anki);
|
|
combine(options.anki.terms, defaults.anki.terms);
|
|
combine(options.anki.kanji, defaults.anki.kanji);
|
|
|
|
return options;
|
|
}
|
|
|
|
function optionsVersion(options) {
|
|
const fixups = [
|
|
() => {
|
|
const copy = (targetDict, targetKey, sourceDict, sourceKey) => {
|
|
targetDict[targetKey] = sourceDict.hasOwnProperty(sourceKey) ? sourceDict[sourceKey] : targetDict[targetKey];
|
|
};
|
|
|
|
copy(options.general, 'autoStart', options, 'activateOnStartup');
|
|
copy(options.general, 'audioPlayback', options, 'enableAudioPlayback');
|
|
copy(options.general, 'softKatakana', options, 'enableSoftKatakanaSearch');
|
|
copy(options.general, 'groupResults', options, 'groupTermResults');
|
|
copy(options.general, 'showAdvanced', options, 'showAdvancedOptions');
|
|
|
|
copy(options.scanning, 'requireShift', options, 'holdShiftToScan');
|
|
copy(options.scanning, 'selectText', options, 'selectMatchedText');
|
|
copy(options.scanning, 'delay', options, 'scanDelay');
|
|
copy(options.scanning, 'length', options, 'scanLength');
|
|
|
|
options.anki.enable = options.ankiMethod === 'ankiconnect';
|
|
|
|
copy(options.anki, 'tags', options, 'ankiCardTags');
|
|
copy(options.anki, 'sentenceExt', options, 'sentenceExtent');
|
|
copy(options.anki.terms, 'deck', options, 'ankiTermDeck');
|
|
copy(options.anki.terms, 'model', options, 'ankiTermModel');
|
|
copy(options.anki.terms, 'fields', options, 'ankiTermFields');
|
|
copy(options.anki.kanji, 'deck', options, 'ankiKanjiDeck');
|
|
copy(options.anki.kanji, 'model', options, 'ankiKanjiModel');
|
|
copy(options.anki.kanji, 'fields', options, 'ankiKanjiFields');
|
|
|
|
for (const title in options.dictionaries) {
|
|
const dictionary = options.dictionaries[title];
|
|
dictionary.enabled = dictionary.enableTerms || dictionary.enableKanji;
|
|
dictionary.priority = 0;
|
|
}
|
|
},
|
|
() => {
|
|
const fixupFields = fields => {
|
|
const fixups = {
|
|
'{expression-furigana}': '{furigana}',
|
|
'{glossary-list}': '{glossary}'
|
|
};
|
|
|
|
for (const name in fields) {
|
|
for (const fixup in fixups) {
|
|
fields[name] = fields[name].replace(fixup, fixups[fixup]);
|
|
}
|
|
}
|
|
};
|
|
|
|
fixupFields(options.anki.terms.fields);
|
|
fixupFields(options.anki.kanji.fields);
|
|
},
|
|
() => {
|
|
let hasEnabledDict = false;
|
|
for (const title in options.dictionaries) {
|
|
if (options.dictionaries[title].enabled) {
|
|
hasEnabledDict = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!hasEnabledDict) {
|
|
for (const title in options.dictionaries) {
|
|
options.dictionaries[title].enabled = true;
|
|
}
|
|
}
|
|
},
|
|
() => {
|
|
let hasEnabledDict = false;
|
|
for (const title in options.dictionaries) {
|
|
if (options.dictionaries[title].enabled) {
|
|
hasEnabledDict = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!hasEnabledDict) {
|
|
for (const title in options.dictionaries) {
|
|
options.dictionaries[title].enabled = true;
|
|
}
|
|
}
|
|
}
|
|
];
|
|
|
|
optionsSetDefaults(options);
|
|
if (!options.hasOwnProperty('version')) {
|
|
options.version = fixups.length;
|
|
}
|
|
|
|
while (options.version < fixups.length) {
|
|
fixups[options.version++]();
|
|
}
|
|
|
|
return options;
|
|
}
|
|
|
|
function optionsLoad() {
|
|
return new Promise((resolve, reject) => {
|
|
chrome.storage.local.get(null, store => resolve(store.options));
|
|
}).then(optionsStr => {
|
|
return optionsStr ? JSON.parse(optionsStr) : {};
|
|
}).catch(error => {
|
|
return {};
|
|
}).then(options => {
|
|
return optionsVersion(options);
|
|
});
|
|
}
|
|
|
|
function optionsSave(options) {
|
|
return new Promise((resolve, reject) => {
|
|
chrome.storage.local.set({options: JSON.stringify(options)}, resolve);
|
|
}).then(() => {
|
|
instYomi().optionsSet(options);
|
|
fgOptionsSet(options);
|
|
});
|
|
}
|
|
|
|
|
|
/*
|
|
* Dictionary
|
|
*/
|
|
|
|
function dictEnabled(options) {
|
|
const dictionaries = {};
|
|
for (const title in options.dictionaries) {
|
|
const dictionary = options.dictionaries[title];
|
|
if (dictionary.enabled) {
|
|
dictionaries[title] = dictionary;
|
|
}
|
|
}
|
|
|
|
return dictionaries;
|
|
}
|
|
|
|
function dictRowsSort(rows, options) {
|
|
return rows.sort((ra, rb) => {
|
|
const pa = (options.dictionaries[ra.title] || {}).priority || 0;
|
|
const pb = (options.dictionaries[rb.title] || {}).priority || 0;
|
|
if (pa > pb) {
|
|
return -1;
|
|
} else if (pa < pb) {
|
|
return 1;
|
|
} else {
|
|
return 0;
|
|
}
|
|
});
|
|
}
|
|
|
|
function dictTermsSort(definitions, dictionaries=null) {
|
|
return definitions.sort((v1, v2) => {
|
|
const sl1 = v1.source.length;
|
|
const sl2 = v2.source.length;
|
|
if (sl1 > sl2) {
|
|
return -1;
|
|
} else if (sl1 < sl2) {
|
|
return 1;
|
|
}
|
|
|
|
if (dictionaries !== null) {
|
|
const p1 = (dictionaries[v1.dictionary] || {}).priority || 0;
|
|
const p2 = (dictionaries[v2.dictionary] || {}).priority || 0;
|
|
if (p1 > p2) {
|
|
return -1;
|
|
} else if (p1 < p2) {
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
const s1 = v1.score;
|
|
const s2 = v2.score;
|
|
if (s1 > s2) {
|
|
return -1;
|
|
} else if (s1 < s2) {
|
|
return 1;
|
|
}
|
|
|
|
const rl1 = v1.reasons.length;
|
|
const rl2 = v2.reasons.length;
|
|
if (rl1 < rl2) {
|
|
return -1;
|
|
} else if (rl1 > rl2) {
|
|
return 1;
|
|
}
|
|
|
|
return v2.expression.localeCompare(v1.expression);
|
|
});
|
|
}
|
|
|
|
function dictTermsUndupe(definitions) {
|
|
const definitionGroups = {};
|
|
for (const definition of definitions) {
|
|
const definitionExisting = definitionGroups[definition.id];
|
|
if (!definitionGroups.hasOwnProperty(definition.id) || definition.expression.length > definitionExisting.expression.length) {
|
|
definitionGroups[definition.id] = definition;
|
|
}
|
|
}
|
|
|
|
const definitionsUnique = [];
|
|
for (const key in definitionGroups) {
|
|
definitionsUnique.push(definitionGroups[key]);
|
|
}
|
|
|
|
return definitionsUnique;
|
|
}
|
|
|
|
function dictTermsGroup(definitions, dictionaries) {
|
|
const groups = {};
|
|
for (const definition of definitions) {
|
|
const key = [definition.source, definition.expression].concat(definition.reasons);
|
|
if (definition.reading) {
|
|
key.push(definition.reading);
|
|
}
|
|
|
|
const group = groups[key];
|
|
if (group) {
|
|
group.push(definition);
|
|
} else {
|
|
groups[key] = [definition];
|
|
}
|
|
}
|
|
|
|
const results = [];
|
|
for (const key in groups) {
|
|
const groupDefs = groups[key];
|
|
const firstDef = groupDefs[0];
|
|
dictTermsSort(groupDefs, dictionaries);
|
|
results.push({
|
|
definitions: groupDefs,
|
|
expression: firstDef.expression,
|
|
reading: firstDef.reading,
|
|
reasons: firstDef.reasons,
|
|
score: groupDefs.reduce((x, y) => x.score > y.score ? x.score : y.score, Number.MIN_SAFE_INTEGER),
|
|
source: firstDef.source
|
|
});
|
|
}
|
|
|
|
return dictTermsSort(results);
|
|
}
|
|
|
|
function dictTagBuildSource(name) {
|
|
return dictTagSanitize({name, category: 'dictionary', order: 100});
|
|
}
|
|
|
|
function dictTagBuild(name, meta) {
|
|
const tag = {name};
|
|
const symbol = name.split(':')[0];
|
|
for (const prop in meta[symbol] || {}) {
|
|
tag[prop] = meta[symbol][prop];
|
|
}
|
|
|
|
return dictTagSanitize(tag);
|
|
}
|
|
|
|
function dictTagSanitize(tag) {
|
|
tag.name = tag.name || 'untitled';
|
|
tag.category = tag.category || 'default';
|
|
tag.notes = tag.notes || '';
|
|
tag.order = tag.order || 0;
|
|
return tag;
|
|
}
|
|
|
|
function dictTagsSort(tags) {
|
|
return tags.sort((v1, v2) => {
|
|
const order1 = v1.order;
|
|
const order2 = v2.order;
|
|
if (order1 < order2) {
|
|
return -1;
|
|
} else if (order1 > order2) {
|
|
return 1;
|
|
}
|
|
|
|
const name1 = v1.name;
|
|
const name2 = v2.name;
|
|
if (name1 < name2) {
|
|
return -1;
|
|
} else if (name1 > name2) {
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
});
|
|
}
|
|
|
|
function dictFieldSplit(field) {
|
|
return field.length === 0 ? [] : field.split(' ');
|
|
}
|
|
|
|
function dictFieldFormat(field, definition, mode, options) {
|
|
const markers = [
|
|
'audio',
|
|
'character',
|
|
'dictionary',
|
|
'expression',
|
|
'furigana',
|
|
'glossary',
|
|
'kunyomi',
|
|
'onyomi',
|
|
'reading',
|
|
'sentence',
|
|
'tags',
|
|
'url'
|
|
];
|
|
|
|
for (const marker of markers) {
|
|
const data = {
|
|
marker,
|
|
definition,
|
|
group: options.general.groupResults,
|
|
html: options.anki.htmlCards,
|
|
modeTermKanji: mode === 'term-kanji',
|
|
modeTermKana: mode === 'term-kana',
|
|
modeKanji: mode === 'kanji'
|
|
};
|
|
|
|
field = field.replace(
|
|
`{${marker}}`,
|
|
Handlebars.templates['fields.html'](data).trim()
|
|
);
|
|
}
|
|
|
|
return field;
|
|
}
|
|
|
|
|
|
/*
|
|
* Json
|
|
*/
|
|
|
|
function jsonLoad(url) {
|
|
return new Promise((resolve, reject) => {
|
|
const xhr = new XMLHttpRequest();
|
|
xhr.overrideMimeType('application/json');
|
|
xhr.addEventListener('load', () => resolve(xhr.responseText));
|
|
xhr.addEventListener('error', () => reject('failed to execute network request'));
|
|
xhr.open('GET', url);
|
|
xhr.send();
|
|
}).then(responseText => {
|
|
try {
|
|
return JSON.parse(responseText);
|
|
}
|
|
catch (e) {
|
|
return Promise.reject('invalid JSON response');
|
|
}
|
|
});
|
|
}
|
|
|
|
function jsonLoadInt(url) {
|
|
return jsonLoad(chrome.extension.getURL(url));
|
|
}
|
|
|
|
function jsonLoadDb(indexUrl, indexLoaded, termsLoaded, kanjiLoaded) {
|
|
const indexDir = indexUrl.slice(0, indexUrl.lastIndexOf('/'));
|
|
return jsonLoad(indexUrl).then(index => {
|
|
if (!index.title || !index.version || !index.revision) {
|
|
return Promise.reject('unrecognized dictionary format');
|
|
}
|
|
|
|
if (indexLoaded !== null) {
|
|
return indexLoaded(
|
|
index.title,
|
|
index.version,
|
|
index.revision,
|
|
index.tagMeta || {},
|
|
index.termBanks > 0,
|
|
index.kanjiBanks > 0
|
|
).then(() => index);
|
|
}
|
|
|
|
return index;
|
|
}).then(index => {
|
|
const loaders = [];
|
|
const banksTotal = index.termBanks + index.kanjiBanks;
|
|
let banksLoaded = 0;
|
|
|
|
for (let i = 1; i <= index.termBanks; ++i) {
|
|
const bankUrl = `${indexDir}/term_bank_${i}.json`;
|
|
loaders.push(() => jsonLoad(bankUrl).then(entries => termsLoaded(
|
|
index.title,
|
|
entries,
|
|
banksTotal,
|
|
banksLoaded++
|
|
)));
|
|
}
|
|
|
|
for (let i = 1; i <= index.kanjiBanks; ++i) {
|
|
const bankUrl = `${indexDir}/kanji_bank_${i}.json`;
|
|
loaders.push(() => jsonLoad(bankUrl).then(entries => kanjiLoaded(
|
|
index.title,
|
|
entries,
|
|
banksTotal,
|
|
banksLoaded++
|
|
)));
|
|
}
|
|
|
|
let chain = Promise.resolve();
|
|
for (const loader of loaders) {
|
|
chain = chain.then(loader);
|
|
}
|
|
|
|
return chain;
|
|
});
|
|
}
|
|
|
|
|
|
/*
|
|
* Helpers
|
|
*/
|
|
|
|
function handlebarsKanjiLinks(options) {
|
|
const isKanji = c => {
|
|
const code = c.charCodeAt(0);
|
|
return code >= 0x4e00 && code < 0x9fb0 || code >= 0x3400 && code < 0x4dc0;
|
|
};
|
|
|
|
let result = '';
|
|
for (const c of options.fn(this)) {
|
|
if (isKanji(c)) {
|
|
result += `<a href="#" class="kanji-link">${c}</a>`;
|
|
} else {
|
|
result += c;
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
function handlebarsMultiLine(options) {
|
|
return options.fn(this).split('\n').join('<br>');
|
|
}
|
|
|
|
function handlebarsRegister() {
|
|
Handlebars.partials = Handlebars.templates;
|
|
Handlebars.registerHelper('kanjiLinks', handlebarsKanjiLinks);
|
|
Handlebars.registerHelper('multiLine', handlebarsMultiLine);
|
|
}
|
|
|
|
function handlebarsRender(template, data) {
|
|
return Handlebars.templates[template](data);
|
|
}
|