i18n-llm-translate/engines/ml/deepl.js

Automatically translates namespace-based JSON translation files across multiple languages from any source language

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.createDeepLTranslateEngine = createDeepLTranslateEngine;
const util_1 = require("../../util");
function createDeepLTranslateEngine(config) {
    if (!config.apiKey) {
        throw new Error('DeepL > Missing apiKey');
    }
    const API_BASE = 'https://api.deepl.com/v2/';
    function fetchTranslations(targetLanguageCode, translations, options) {
        return __awaiter(this, void 0, void 0, function* () {
            const bodyParams = new URLSearchParams();
            bodyParams.append('auth_key', config.apiKey);
            bodyParams.append('source_lang', options.baseLanguageCode.toUpperCase());
            bodyParams.append('target_lang', targetLanguageCode.toUpperCase());
            translations.forEach((text) => bodyParams.append('text', text));
            const result = yield fetch(`${API_BASE}translate`, {
                body: bodyParams,
                headers: {
                    'Content-Type': 'application/x-www-form-urlencoded',
                },
                method: 'POST',
            });
            if (!result.ok) {
                const errorText = yield result.text();
                throw new Error(`DeepL API error: ${result.status}\n${errorText}`);
            }
            return JSON.parse(yield result.text());
        });
    }
    return {
        name: 'DeepL',
        translate(translations, options) {
            return __awaiter(this, void 0, void 0, function* () {
                // TODO: Change for 'walk' method in future
                const flatTranslations = (0, util_1.flattenObject)(translations);
                const languagesTranslations = {};
                for (let targetLanguageCode of options.targetLanguageCodes) {
                    if (options.debug)
                        console.log(`DeepL translate > Translating '${options.baseLanguageCode}'>'${targetLanguageCode}'...`);
                    const baseTranslationsValues = Object.values(flatTranslations);
                    const result = yield fetchTranslations(targetLanguageCode, baseTranslationsValues, options);
                    const languageTranslations = {};
                    const keys = Object.keys(flatTranslations);
                    for (let i = 0; i < result.translations.length; i++) {
                        languageTranslations[keys[i]] = result.translations[i].text;
                    }
                    languagesTranslations[targetLanguageCode] = (0, util_1.unflattenObject)(languageTranslations);
                }
                return languagesTranslations;
            });
        },
        translateMissed(missingTranslations, options) {
            return __awaiter(this, void 0, void 0, function* () {
                // TODO: Change for 'walk' method in future
                const flatBaseTranslations = (0, util_1.flattenObject)(missingTranslations.baseLanguageTranslations);
                const languagesTranslations = {};
                for (let targetLanguageCode in missingTranslations.targetLanguageTranslationsKeys) {
                    if (options.debug)
                        console.log(`DeepL translate > Translating '${options.baseLanguageCode}'>'${targetLanguageCode}'...`);
                    const missingKeys = Object.keys((0, util_1.flattenObject)(missingTranslations.targetLanguageTranslationsKeys[targetLanguageCode]));
                    const missingValues = [];
                    for (let missingKey of missingKeys) {
                        missingValues.push(flatBaseTranslations[missingKey]);
                    }
                    const result = yield fetchTranslations(targetLanguageCode, missingValues, options);
                    const languageTranslations = {};
                    for (let i = 0; i < result.translations.length; i++) {
                        languageTranslations[missingKeys[i]] = result.translations[i].text;
                    }
                    languagesTranslations[targetLanguageCode] = (0, util_1.unflattenObject)(languageTranslations);
                }
                return languagesTranslations;
            });
        }
    };
}