feat: add LibreTranslate

LiuYi0526 · Jan 31, 2024 · c7df4ae · c7df4ae
1 parent 43353aa
commit c7df4ae
Show file tree

Hide file tree

Showing 6 changed files with 56 additions and 0 deletions.
diff --git a/TMessagesProj/src/main/java/tw/nekomimi/nekogram/settings/NekoGeneralSettingsActivity.java b/TMessagesProj/src/main/java/tw/nekomimi/nekogram/settings/NekoGeneralSettingsActivity.java
@@ -354,6 +354,7 @@ public void onItemClick(int id) {
                             LocaleController.getString("ProviderMicrosoftTranslator", R.string.ProviderDeepLTranslate),
                             LocaleController.getString("ProviderTelegramAPI", R.string.ProviderTelegramAPI),
                             LocaleController.getString("ProviderTranSmartTranslate", R.string.ProviderTranSmartTranslate),
+                            LocaleController.getString("ProviderLibreTranslateTranslate", R.string.ProviderLibreTranslateTranslate),
                     }, (i, __) -> {
                         boolean needReset = NekoConfig.translationProvider.Int() - 1 != i && (NekoConfig.translationProvider.Int() == 1 || i == 0);
                         NekoConfig.translationProvider.setConfigInt(i + 1);
@@ -769,6 +770,9 @@ public void onBindViewHolder(RecyclerView.ViewHolder holder, int position) {
                                 case Translator.providerTranSmart:
                                     value = LocaleController.getString("ProviderTranSmartTranslate", R.string.ProviderTranSmartTranslate);
                                     break;
+                                case Translator.providerLibreTranslate:
+                                    value = LocaleController.getString("ProviderLibreTranslateTranslate", R.string.ProviderLibreTranslateTranslate);
+                                    break;
                                 default:
                                     value = "Unknown";
                             }

diff --git a/TMessagesProj/src/main/java/tw/nekomimi/nekogram/transtale/Translator.kt b/TMessagesProj/src/main/java/tw/nekomimi/nekogram/transtale/Translator.kt
@@ -83,6 +83,7 @@ interface Translator {
         const val providerDeepL = 7
         const val providerTelegram = 8
         const val providerTranSmart = 9
+        const val providerLibreTranslate = 10
 
         @Throws(Exception::class)
         suspend fun translate(to: Locale, query: String): String {
@@ -121,6 +122,7 @@ interface Translator {
                 providerDeepL -> DeepLTranslator
                 providerTelegram -> TelegramAPITranslator
                 providerTranSmart -> TranSmartTranslator
+                providerLibreTranslate -> LibreTranslator
                 else -> throw IllegalArgumentException()
             }
 

diff --git a/TMessagesProj/src/main/java/tw/nekomimi/nekogram/transtale/source/LibreTranslator.kt b/TMessagesProj/src/main/java/tw/nekomimi/nekogram/transtale/source/LibreTranslator.kt
@@ -0,0 +1,47 @@
+package tw.nekomimi.nekogram.transtale.source
+
+import cn.hutool.http.HttpUtil
+import org.json.JSONArray
+import org.json.JSONObject
+import org.telegram.messenger.LocaleController
+import org.telegram.messenger.R
+import tw.nekomimi.nekogram.transtale.Translator
+import tw.nekomimi.nekogram.utils.applyUserAgent
+import java.util.Date
+import java.util.UUID
+
+object LibreTranslator : Translator {
+
+    private val targetLanguages = listOf(
+        "sq", "ar", "az", "bn", "bg", "ca", "zh", "zt", "cs", "da", "nl", "en", "eo", "et", "fi", "fr", "de", "el", "he", "hi", "hu", "id", "ga", "it", "ja", "ko", "lv", "lt", "ms", "nb", "fa", "pl", "pt", "ro", "ru", "sr", "sk", "sl", "es", "sv", "tl", "th", "tr", "uk", "ur", "vi"
+    )
+
+    override suspend fun doTranslate(from: String, to: String, query: String): String {
+
+        if (to !in targetLanguages) {
+            error(LocaleController.getString("TranslateApiUnsupported", R.string.TranslateApiUnsupported))
+        }
+
+        val response = HttpUtil.createPost("https://libretranslate.com/translate")
+            .header("Content-Type", "application/json")
+            .applyUserAgent()
+            .body(JSONObject().apply {
+                put("q", query)
+                put("source", if (targetLanguages.contains(from)) from else "en")
+                put("target", to)
+                put("format", "text")
+                put("api_key", "")
+            }.toString())
+            .execute()
+
+        if (response.status != 200) {
+            error("HTTP ${response.status} : ${response.body()}")
+        }
+
+        val target: JSONArray = JSONObject(response.body()).getJSONArray("translatedText")
+
+        return target.toString().trimEnd()
+
+    }
+
+}
diff --git a/TMessagesProj/src/main/res/values-zh-rCN/strings_na.xml b/TMessagesProj/src/main/res/values-zh-rCN/strings_na.xml
@@ -100,4 +100,5 @@
     <string name="ExternalStickerCacheRefresh">刷新外部缓存</string>
     <string name="ExternalStickerCacheDelete">删除外部缓存</string>
     <string name="ProviderTranSmartTranslate">腾讯交互翻译</string>
+    <string name="ProviderLibreTranslateTranslate">LibreTranslate</string>
 </resources>
diff --git a/TMessagesProj/src/main/res/values-zh-rTW/strings_na.xml b/TMessagesProj/src/main/res/values-zh-rTW/strings_na.xml
@@ -79,4 +79,5 @@
     <string name="HideBotButtonInInputField">隱藏輸入框中的機器人按鈕</string>
     <string name="ShowSendAsUnderMessageHint">在訊息提示下顯示傳送者</string>
     <string name="ProviderTranSmartTranslate">騰訊交互翻譯</string>
+    <string name="ProviderLibreTranslateTranslate">LibreTranslate</string>
 </resources>
diff --git a/TMessagesProj/src/main/res/values/strings_na.xml b/TMessagesProj/src/main/res/values/strings_na.xml
@@ -103,4 +103,5 @@
     <string name="ExternalStickerCacheRefresh">Refresh External Cache</string>
     <string name="ExternalStickerCacheDelete">Delete External Cache</string>
     <string name="ProviderTranSmartTranslate">TranSmart Translator</string>
+    <string name="ProviderLibreTranslateTranslate">LibreTranslate</string>
 </resources>