summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--dev/translator-vm.js200
1 files changed, 0 insertions, 200 deletions
diff --git a/dev/translator-vm.js b/dev/translator-vm.js
deleted file mode 100644
index f407e57e..00000000
--- a/dev/translator-vm.js
+++ /dev/null
@@ -1,200 +0,0 @@
-/*
- * Copyright (C) 2023 Yomitan Authors
- * Copyright (C) 2021-2022 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/>.
- */
-
-import fs from 'fs';
-import url, {fileURLToPath} from 'node:url';
-import path from 'path';
-import {expect, vi} from 'vitest';
-import {AnkiNoteDataCreator} from '../ext/js/data/sandbox/anki-note-data-creator.js';
-import {DictionaryDatabase} from '../ext/js/language/dictionary-database.js';
-import {DictionaryImporterMediaLoader} from '../ext/js/language/dictionary-importer-media-loader.js';
-import {DictionaryImporter} from '../ext/js/language/dictionary-importer.js';
-import {JapaneseUtil} from '../ext/js/language/sandbox/japanese-util.js';
-import {Translator} from '../ext/js/language/translator.js';
-import {createDictionaryArchive} from './util.js';
-
-vi.mock('../ext/js/language/dictionary-importer-media-loader.js', async () => await import('../test/mocks/dictionary-importer-media-loader.js'));
-
-const dirname = path.dirname(fileURLToPath(import.meta.url));
-
-/**
- * Translator Virtual Machine.
- */
-export class TranslatorVM {
- constructor() {
- /** @type {import('dev/vm').PseudoChrome} */
- const chrome = {
- runtime: {
- getURL: (path2) => {
- return url.pathToFileURL(path.join(dirname, '..', 'ext', path2.replace(/^\//, ''))).href;
- }
- }
- };
- // @ts-expect-error - Overwriting a global
- global.chrome = chrome;
-
- /** @type {?JapaneseUtil} */
- this._japaneseUtil = null;
- /** @type {?Translator} */
- this._translator = null;
- /** @type {?AnkiNoteDataCreator} */
- this._ankiNoteDataCreator = null;
- /** @type {?string} */
- this._dictionaryName = null;
- }
-
- /**
- * Returns this VM's translator.
- * @type {Translator}
- */
- get translator() {
- if (this._translator === null) { throw new Error('Not prepared'); }
- return this._translator;
- }
-
- /**
- * Initialize this translator VM from a dictionary.
- * @param {string} dictionaryDirectory Directory of the dictionary files.
- * @param {string} dictionaryName Name of the dictionary.
- */
- async prepare(dictionaryDirectory, dictionaryName) {
- // Dictionary
- this._dictionaryName = dictionaryName;
- const testDictionary = createDictionaryArchive(dictionaryDirectory, dictionaryName);
- const testDictionaryContent = await testDictionary.generateAsync({type: 'arraybuffer'});
-
- // Setup database
- const dictionaryImporterMediaLoader = new DictionaryImporterMediaLoader();
- const dictionaryImporter = new DictionaryImporter(dictionaryImporterMediaLoader);
- const dictionaryDatabase = new DictionaryDatabase();
- await dictionaryDatabase.prepare();
-
- const {errors} = await dictionaryImporter.importDictionary(
- dictionaryDatabase,
- testDictionaryContent,
- {prefixWildcardsSupported: true}
- );
-
- expect(errors.length).toEqual(0);
-
- // Setup translator
- this._japaneseUtil = new JapaneseUtil(null);
- this._translator = new Translator({
- japaneseUtil: this._japaneseUtil,
- database: dictionaryDatabase
- });
- const deinflectionReasons = JSON.parse(fs.readFileSync(path.join(dirname, '..', 'ext', 'data/deinflect.json'), {encoding: 'utf8'}));
- this._translator.prepare(deinflectionReasons);
-
- // Assign properties
- this._ankiNoteDataCreator = new AnkiNoteDataCreator(this._japaneseUtil);
- }
-
- /**
- * @param {import('dictionary').DictionaryEntry} dictionaryEntry
- * @param {import('settings').ResultOutputMode} mode
- * @returns {import('anki-templates').NoteData}
- * @throws {Error}
- */
- createTestAnkiNoteData(dictionaryEntry, mode) {
- if (this._ankiNoteDataCreator === null) {
- throw new Error('Not prepared');
- }
- const marker = '{marker}';
- /** @type {import('anki-templates-internal').CreateDetails} */
- const data = {
- dictionaryEntry,
- resultOutputMode: mode,
- mode: 'test',
- glossaryLayoutMode: 'default',
- compactTags: false,
- context: {
- url: 'url:',
- sentence: {text: '', offset: 0},
- documentTitle: 'title',
- query: 'query',
- fullQuery: 'fullQuery'
- },
- media: {}
- };
- return this._ankiNoteDataCreator.create(marker, data);
- }
-
- /**
- * @template {import('translation').FindTermsOptions|import('translation').FindKanjiOptions} T
- * @param {import('dev/vm').OptionsPresetObject} optionsPresets
- * @param {string|import('dev/vm').OptionsPresetObject|(string|import('dev/vm').OptionsPresetObject)[]} optionsArray
- * @returns {T}
- * @throws {Error}
- */
- buildOptions(optionsPresets, optionsArray) {
- const dictionaryName = this._dictionaryName;
- /** @type {import('core').UnknownObject} */
- const options = {};
- if (!Array.isArray(optionsArray)) { optionsArray = [optionsArray]; }
- for (const entry of optionsArray) {
- switch (typeof entry) {
- case 'string':
- if (!Object.prototype.hasOwnProperty.call(optionsPresets, entry)) {
- throw new Error('Invalid options preset');
- }
- Object.assign(options, structuredClone(optionsPresets[entry]));
- break;
- case 'object':
- Object.assign(options, structuredClone(entry));
- break;
- default:
- throw new Error('Invalid options type');
- }
- }
-
- // Construct regex
- if (Array.isArray(options.textReplacements)) {
- options.textReplacements = options.textReplacements.map((value) => {
- if (Array.isArray(value)) {
- value = value.map(({pattern, flags, replacement}) => ({pattern: new RegExp(pattern, flags), replacement}));
- }
- return value;
- });
- }
-
- // Update structure
- const placeholder = '${title}';
- if (options.mainDictionary === placeholder) {
- options.mainDictionary = dictionaryName;
- }
- let {enabledDictionaryMap} = options;
- if (Array.isArray(enabledDictionaryMap)) {
- for (const entry of enabledDictionaryMap) {
- if (entry[0] === placeholder) {
- entry[0] = dictionaryName;
- }
- }
- enabledDictionaryMap = new Map(enabledDictionaryMap);
- options.enabledDictionaryMap = enabledDictionaryMap;
- }
- const {excludeDictionaryDefinitions} = options;
- options.excludeDictionaryDefinitions = (
- Array.isArray(excludeDictionaryDefinitions) ?
- new Set(excludeDictionaryDefinitions) :
- null
- );
-
- return /** @type {T} */ (options);
- }
-}