Changeset 23 in code for trunk/engines/libretranslate.go

Timestamp:

Sep 2, 2023, 12:15:30 PM (21 months ago)

Author:

manerakai

Message:

Improved code

File:

• trunk/engines/libretranslate.go (modified) (5 diffs)

trunk/engines/libretranslate.go

@@ -23 +23 @@
         APIKey string
 }
-
-func (_ *LibreTranslate) InternalName() string { return "libre" }
 
 func (_ *LibreTranslate) DisplayName() string { return "LibreTranslate" }
@@ -71 +69 @@
 }
 
-func (e *LibreTranslate) DetectLanguage(text string) (string, error) {
+func (e *LibreTranslate) detectLanguage(text string) (string, error) {
         formData := map[string]string{"q": text}
 
@@ -125 +123 @@
 }
 
-type libreTranslateResponse struct {
-        TranslatedText string `json:"translatedText"`
-}
-
 func (e *LibreTranslate) Translate(text string, from, to string) (TranslationResult, error) {
         formData := map[string]string{
@@ -158 +152 @@
         }
 
-        var responseJSON libreTranslateResponse
+        var responseJSON struct {
+                TranslatedText string `json:"translatedText"`
+        }
 
         if err := json.NewDecoder(response.Body).Decode(&responseJSON); err != nil {
@@ -164 +160 @@
         }
 
-        return TranslationResult{TranslatedText: responseJSON.TranslatedText}, nil
+        if r, err := e.detectLanguage(text); err == nil {
+                from = r
+        }
+
+        return TranslationResult{
+                TranslatedText: responseJSON.TranslatedText,
+                SourceLanguage: from,
+        }, nil
 }