87fbb3c01c
* Add support for running getDictionaryCounts via DictionaryWorker * Run dictionary integrity checks on a separate thread * Remove api.getDictionaryCounts
108 lines
3.6 KiB
JavaScript
108 lines
3.6 KiB
JavaScript
/*
|
|
* Copyright (C) 2021 Yomichan Authors
|
|
*
|
|
* 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 <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
/* global
|
|
* DictionaryDatabase
|
|
* DictionaryImporter
|
|
* DictionaryWorkerMediaLoader
|
|
*/
|
|
|
|
class DictionaryWorkerHandler {
|
|
constructor() {
|
|
this._mediaLoader = new DictionaryWorkerMediaLoader();
|
|
}
|
|
|
|
prepare() {
|
|
self.addEventListener('message', this._onMessage.bind(this), false);
|
|
}
|
|
|
|
// Private
|
|
|
|
_onMessage(e) {
|
|
const {action, params} = e.data;
|
|
switch (action) {
|
|
case 'importDictionary':
|
|
this._onMessageWithProgress(params, this._importDictionary.bind(this));
|
|
break;
|
|
case 'deleteDictionary':
|
|
this._onMessageWithProgress(params, this._deleteDictionary.bind(this));
|
|
break;
|
|
case 'getDictionaryCounts':
|
|
this._onMessageWithProgress(params, this._getDictionaryCounts.bind(this));
|
|
break;
|
|
case 'getImageResolution.response':
|
|
this._mediaLoader.handleMessage(params);
|
|
break;
|
|
}
|
|
}
|
|
|
|
async _onMessageWithProgress(params, handler) {
|
|
const onProgress = (...args) => {
|
|
self.postMessage({
|
|
action: 'progress',
|
|
params: {args}
|
|
});
|
|
};
|
|
let response;
|
|
try {
|
|
const result = await handler(params, onProgress);
|
|
response = {result};
|
|
} catch (e) {
|
|
response = {error: serializeError(e)};
|
|
}
|
|
self.postMessage({action: 'complete', params: response});
|
|
}
|
|
|
|
async _importDictionary({details, archiveContent}, onProgress) {
|
|
const dictionaryDatabase = await this._getPreparedDictionaryDatabase();
|
|
try {
|
|
const dictionaryImporter = new DictionaryImporter(this._mediaLoader, onProgress);
|
|
const {result, errors} = await dictionaryImporter.importDictionary(dictionaryDatabase, archiveContent, details);
|
|
return {
|
|
result,
|
|
errors: errors.map((error) => serializeError(error))
|
|
};
|
|
} finally {
|
|
dictionaryDatabase.close();
|
|
}
|
|
}
|
|
|
|
async _deleteDictionary({dictionaryTitle}, onProgress) {
|
|
const dictionaryDatabase = await this._getPreparedDictionaryDatabase();
|
|
try {
|
|
return await dictionaryDatabase.deleteDictionary(dictionaryTitle, {rate: 1000}, onProgress);
|
|
} finally {
|
|
dictionaryDatabase.close();
|
|
}
|
|
}
|
|
|
|
async _getDictionaryCounts({dictionaryNames, getTotal}) {
|
|
const dictionaryDatabase = await this._getPreparedDictionaryDatabase();
|
|
try {
|
|
return await dictionaryDatabase.getDictionaryCounts(dictionaryNames, getTotal);
|
|
} finally {
|
|
dictionaryDatabase.close();
|
|
}
|
|
}
|
|
|
|
async _getPreparedDictionaryDatabase() {
|
|
const dictionaryDatabase = new DictionaryDatabase();
|
|
await dictionaryDatabase.prepare();
|
|
return dictionaryDatabase;
|
|
}
|
|
}
|