feat: Add Custom URL for Google Translate

#422 #423
pot-app · Sep 6, 2023 · bc5b372 · bc5b372
1 parent cdbaea9
commit bc5b372
Show file tree

Hide file tree

Showing 5 changed files with 83 additions and 22 deletions.
diff --git a/src/i18n/locales/en_US.json b/src/i18n/locales/en_US.json
@@ -185,7 +185,7 @@
                     "api": "Auth Key",
                     "deeplx": "DeepLX"
                 },
-                "google": { "title": "Google" },
+                "google": { "title": "Google", "custom_url": "Custom URL" },
                 "bing": { "title": "Bing" },
                 "openai": {
                     "title": "OpenAI",

diff --git a/src/i18n/locales/pt_BR.json b/src/i18n/locales/pt_BR.json
@@ -186,7 +186,8 @@
                     "deeplx": "DeepLX"
                 },
                 "google": {
-                    "title": "Google"
+                    "title": "Google",
+                    "custom_url": "URL personalizado"
                 },
                 "bing": {
                     "title": "Bing"

diff --git a/src/i18n/locales/zh_CN.json b/src/i18n/locales/zh_CN.json
@@ -184,7 +184,7 @@
                     "api": "Auth Key",
                     "deeplx": "DeepLX"
                 },
-                "google": { "title": "谷歌翻译" },
+                "google": { "title": "谷歌翻译", "custom_url": "自定义 URL" },
                 "bing": { "title": "必应翻译" },
                 "openai_summary": { "title": "OpenAI 总结" },
                 "openai_polish": { "title": "OpenAI 润色" },

diff --git a/src/services/translate/google/Config.jsx b/src/services/translate/google/Config.jsx
@@ -1,26 +1,70 @@
+import { Input, Button } from '@nextui-org/react';
+import toast, { Toaster } from 'react-hot-toast';
 import { useTranslation } from 'react-i18next';
-import { Button } from '@nextui-org/react';
-import React from 'react';
+import React, { useState } from 'react';
+
+import { useConfig } from '../../../hooks/useConfig';
+import { useToastStyle } from '../../../hooks';
+import { translate } from './index';
+import { Language } from './index';
 
 export function Config(props) {
     const { updateServiceList, onClose } = props;
+    const [config, setConfig] = useConfig(
+        'google',
+        {
+            custom_url: 'https://translate.google.com',
+        },
+        { sync: false }
+    );
+    const [isLoading, setIsLoading] = useState(false);
+
     const { t } = useTranslation();
+    const toastStyle = useToastStyle();
 
     return (
-        <>
-            <div>{t('services.no_need')}</div>
-            <div>
-                <Button
-                    fullWidth
-                    color='primary'
-                    onPress={() => {
-                        updateServiceList('google');
-                        onClose();
-                    }}
-                >
-                    {t('common.save')}
-                </Button>
-            </div>
-        </>
+        config !== null && (
+            <>
+                <Toaster />
+                <div className={'config-item'}>
+                    <h3 className='my-auto'>{t('services.translate.google.custom_url')}</h3>
+                    <Input
+                        value={config['custom_url']}
+                        variant='bordered'
+                        className='max-w-[50%]'
+                        onValueChange={(value) => {
+                            setConfig({
+                                ...config,
+                                custom_url: value,
+                            });
+                        }}
+                    />
+                </div>
+                <div>
+                    <Button
+                        isLoading={isLoading}
+                        color='primary'
+                        fullWidth
+                        onPress={() => {
+                            setIsLoading(true);
+                            translate('hello', Language.auto, Language.zh_cn, { config }).then(
+                                () => {
+                                    setIsLoading(false);
+                                    setConfig(config, true);
+                                    updateServiceList('google');
+                                    onClose();
+                                },
+                                (e) => {
+                                    setIsLoading(false);
+                                    toast.error(t('config.service.test_failed') + e.toString(), { style: toastStyle });
+                                }
+                            );
+                        }}
+                    >
+                        {t('common.save')}
+                    </Button>
+                </div>
+            </>
+        )
     );
 }
diff --git a/src/services/translate/google/index.jsx b/src/services/translate/google/index.jsx
@@ -1,8 +1,24 @@
 import { fetch } from '@tauri-apps/api/http';
 
-export async function translate(text, from, to) {
+export async function translate(text, from, to, options = {}) {
+    const { config } = options;
+
+    let translateConfig = (await store.get('google')) ?? {};
+    if (config !== undefined) {
+        translateConfig = config;
+    }
+
+    let { custom_url } = translateConfig;
+
+    if (custom_url === undefined || custom_url === '') {
+        custom_url = 'https://translate.google.com';
+    }
+    if (!custom_url.startsWith('http')) {
+        custom_url = 'https://' + custom_url;
+    }
+
     let res = await fetch(
-        `https://translate.google.com/translate_a/single?dt=at&dt=bd&dt=ex&dt=ld&dt=md&dt=qca&dt=rw&dt=rm&dt=ss&dt=t`,
+        `${custom_url}/translate_a/single?dt=at&dt=bd&dt=ex&dt=ld&dt=md&dt=qca&dt=rw&dt=rm&dt=ss&dt=t`,
         {
             method: 'GET',
             headers: { 'content-type': 'application/json' },