From d19c151669ad82390485146d24228b514b979b39 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Sun, 10 Sep 2023 18:22:39 +0300
Subject: [PATCH] Add DeepLX translation provider #1111

---
 public/scripts/extensions/translate/index.js | 41 +++++++++++--
 src/secrets.js                               |  1 +
 src/translate.js                             | 63 +++++++++++++++++++-
 3 files changed, 98 insertions(+), 7 deletions(-)

diff --git a/public/scripts/extensions/translate/index.js b/public/scripts/extensions/translate/index.js
index 37bf9e761..dd20a7b29 100644
--- a/public/scripts/extensions/translate/index.js
+++ b/public/scripts/extensions/translate/index.js
@@ -135,8 +135,8 @@ const languageCodes = {
     'Zulu': 'zu',
 };
 
-const KEY_REQUIRED = ['deepl','libre'];
-const LOCAL_URL = ['libre', 'oneringtranslator'];
+const KEY_REQUIRED = ['deepl', 'libre'];
+const LOCAL_URL = ['libre', 'oneringtranslator', 'deeplx'];
 
 function showKeysButton() {
     const providerRequiresKey = KEY_REQUIRED.includes(extension_settings.translate.provider);
@@ -271,6 +271,27 @@ async function translateProviderDeepl(text, lang) {
     throw new Error(response.statusText);
 }
 
+/**
+ * Translates text using the DeepLX API
+ * @param {string} text Text to translate
+ * @param {string} lang Target language code
+ * @returns {Promise<string>} Translated text
+ */
+async function translateProviderDeepLX(text, lang) {
+    const response = await fetch('/api/translate/deeplx', {
+        method: 'POST',
+        headers: getRequestHeaders(),
+        body: JSON.stringify({ text: text, lang: lang }),
+    });
+
+    if (response.ok) {
+        const result = await response.text();
+        return result;
+    }
+
+    throw new Error(response.statusText);
+}
+
 /**
  * Translates text using the selected translation provider
  * @param {string} text Text to translate
@@ -290,6 +311,8 @@ async function translate(text, lang) {
                 return await translateProviderGoogle(text, lang);
             case 'deepl':
                 return await translateProviderDeepl(text, lang);
+            case 'deeplx':
+                return await translateProviderDeepLX(text, lang);
             case 'oneringtranslator':
                 return await translateProviderOneRing(text, lang);
             default:
@@ -437,6 +460,7 @@ jQuery(() => {
                         <option value="libre">Libre</option>
                         <option value="google">Google</option>
                         <option value="deepl">DeepL</option>
+                        <option value="deeplx">DeepLX</option>
                         <option value="oneringtranslator">OneRingTranslator</option>
                     <select>
                     <div id="translate_key_button" class="menu_button fa-solid fa-key margin0"></div>
@@ -490,13 +514,17 @@ jQuery(() => {
 
         await writeSecret(extension_settings.translate.provider, key);
         toastr.success('API Key saved');
+        $("#translate_key_button").addClass('success');
     });
     $('#translate_url_button').on('click', async () => {
         const optionText = $('#translation_provider option:selected').text();
-        const exampleURLs = {};
-        exampleURLs['libre'] = 'http://127.0.0.1:5000/translate';
-        exampleURLs['oneringtranslator'] = 'http://127.0.0.1:4990/translate';
-        const url = await callPopup(`<h3>${optionText} API URL</h3><i>Example: <tt>` + exampleURLs[extension_settings.translate.provider] + `</tt></i>`, 'input');
+        const exampleURLs = {
+            'libre': 'http://127.0.0.1:5000/translate',
+            'oneringtranslator': 'http://127.0.0.1:4990/translate',
+            'deeplx': 'http://127.0.0.1:1188/translate',
+        };
+        const popupText = `<h3>${optionText} API URL</h3><i>Example: <tt>${String(exampleURLs[extension_settings.translate.provider])}</tt></i>`;
+        const url = await callPopup(popupText, 'input');
 
         if (url == false) {
             return;
@@ -504,6 +532,7 @@ jQuery(() => {
 
         await writeSecret(extension_settings.translate.provider + "_url", url);
         toastr.success('API URL saved');
+        $("#translate_url_button").addClass('success');
     });
 
     loadSettings();
diff --git a/src/secrets.js b/src/secrets.js
index 49c84ba20..b09a33f08 100644
--- a/src/secrets.js
+++ b/src/secrets.js
@@ -17,6 +17,7 @@ const SECRET_KEYS = {
     AI21: 'api_key_ai21',
     SCALE_COOKIE: 'scale_cookie',
     ONERING_URL: 'oneringtranslator_url',
+    DEEPLX_URL: 'deeplx_url',
 }
 
 /**
diff --git a/src/translate.js b/src/translate.js
index f70a13f2b..a8f6056f0 100644
--- a/src/translate.js
+++ b/src/translate.js
@@ -3,6 +3,9 @@ const https = require('https');
 const { readSecret, SECRET_KEYS } = require('./secrets');
 const { generateRequestUrl, normaliseResponse } = require('google-translate-api-browser');
 
+const DEEPLX_URL_DEFAULT = 'http://127.0.0.1:1188/translate';
+const ONERING_URL_DEFAULT = 'http://127.0.0.1:4990/translate';
+
 /**
  * @param {import("express").Express} app
  * @param {any} jsonParser
@@ -134,13 +137,18 @@ function registerEndpoints(app, jsonParser) {
     });
 
     app.post('/api/translate/onering', jsonParser, async (request, response) => {
-        const url = readSecret(SECRET_KEYS.ONERING_URL);
+        const secretUrl = readSecret(SECRET_KEYS.ONERING_URL);
+        const url = secretUrl || ONERING_URL_DEFAULT;
 
         if (!url) {
             console.log('OneRing URL is not configured.');
             return response.sendStatus(401);
         }
 
+        if (!secretUrl && url === ONERING_URL_DEFAULT) {
+            console.log('OneRing URL is using default value.', ONERING_URL_DEFAULT);
+        }
+
         const text = request.body.text;
         const from_lang = request.body.from_lang;
         const to_lang = request.body.to_lang;
@@ -180,6 +188,59 @@ function registerEndpoints(app, jsonParser) {
             return response.sendStatus(500);
         }
     });
+
+    app.post('/api/translate/deeplx', jsonParser, async (request, response) => {
+        const secretUrl = readSecret(SECRET_KEYS.DEEPLX_URL);
+        const url = secretUrl || DEEPLX_URL_DEFAULT;
+
+        if (!url) {
+            console.log('DeepLX URL is not configured.');
+            return response.sendStatus(401);
+        }
+
+        if (!secretUrl && url === DEEPLX_URL_DEFAULT) {
+            console.log('DeepLX URL is using default value.', DEEPLX_URL_DEFAULT);
+        }
+
+        const text = request.body.text;
+        const lang = request.body.lang;
+
+        if (!text || !lang) {
+            return response.sendStatus(400);
+        }
+
+        console.log('Input text: ' + text);
+
+        try {
+            const result = await fetch(url, {
+                method: 'POST',
+                body: JSON.stringify({
+                    text: text,
+                    source_lang: 'auto',
+                    target_lang: lang,
+                }),
+                headers: {
+                    'Accept': 'application/json',
+                    'Content-Type': 'application/json',
+                },
+                timeout: 0,
+            });
+
+            if (!result.ok) {
+                const error = await result.text();
+                console.log('DeepLX error: ', result.statusText, error);
+                return response.sendStatus(result.status);
+            }
+
+            const json = await result.json();
+            console.log('Translated text: ' + json.data);
+
+            return response.send(json.data);
+        } catch (error) {
+            console.log("DeepLX translation error: " + error.message);
+            return response.sendStatus(500);
+        }
+    });
 }
 
 module.exports = {