/*
 * 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');

const dirname = path.dirname(fileURLToPath(import.meta.url));

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;
    }

    /** @type {Translator} */
    get translator() {
        if (this._translator === null) { throw new Error('Not prepared'); }
        return this._translator;
    }

    /**
     * @param {string} dictionaryDirectory
     * @param {string} dictionaryName
     */
    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);
    }
}