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

Timestamp:

Aug 30, 2023, 11:57:46 AM (22 months ago)

Author:

manerakai

Message:

Added gofiber

File:

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

trunk/engines/libretranslate.go

@@ -28 +28 @@
 func (_ *LibreTranslate) DisplayName() string { return "LibreTranslate" }
 
-type libreLanguagesResponse []struct {
-        Name string `json:"name"`
-        Code string `json:"code"`
-}
-
-func (e *LibreTranslate) getLangs() ([]Language, error) {
+func (e *LibreTranslate) getLangs() (Language, error) {
         response, err := http.Get(e.InstanceURL + "/languages")
 
@@ -46 +41 @@
         }
 
-        var langsResponse libreLanguagesResponse
+        var langsResponse []struct {
+                Name string `json:"name"`
+                Code string `json:"code"`
+        }
 
         if err := json.NewDecoder(response.Body).Decode(&langsResponse); err != nil {
@@ -52 +50 @@
         }
 
-        langs := make([]Language, len(langsResponse))
+        langs := Language{}
 
-        for i, lang := range langsResponse {
-                langs[i] = Language{Name: lang.Name, Code: lang.Code}
+        for _, lang := range langsResponse {
+                langs[lang.Code] = lang.Name
         }
 
@@ -62 +60 @@
 }
 
-func (e *LibreTranslate) SourceLanguages() ([]Language, error) { return e.getLangs() }
+func (e *LibreTranslate) SourceLanguages() (Language, error) { return e.getLangs() }
 
-func (e *LibreTranslate) TargetLanguages() ([]Language, error) { return e.getLangs() }
-
-func (_ *LibreTranslate) SupportsAutodetect() bool { return true }
+func (e *LibreTranslate) TargetLanguages() (Language, error) { return e.getLangs() }
 
 type libreDetectResponse []struct {
@@ -73 +69 @@
 }
 
-func (e *LibreTranslate) DetectLanguage(text string) (Language, error) {
+func (e *LibreTranslate) DetectLanguage(text string) (string, error) {
         formData := map[string]string{"q": text}
 
@@ -83 +79 @@
 
         if err != nil {
-                return Language{}, err
+                return "", err
         }
 
@@ -89 +85 @@
 
         if err != nil {
-                return Language{}, err
+                return "", err
         }
 
@@ -95 +91 @@
 
         if response.StatusCode != 200 {
-                return Language{}, fmt.Errorf("got status code %d from LibreTranslate API", response.StatusCode)
+                return "", fmt.Errorf("got status code %d from LibreTranslate API", response.StatusCode)
         }
 
@@ -101 +97 @@
 
         if err := json.NewDecoder(response.Body).Decode(&langs); err != nil {
-                return Language{}, err
+                return "", err
         }
 
@@ -115 +111 @@
 
         if err != nil {
-                return Language{}, err
+                return "", err
         }
 
-        for _, lang := range engineLangs {
-                if lang.Code == maxConfidenceLang.LanguageCode {
-                        return lang, nil
+        for code := range engineLangs {
+                if code == maxConfidenceLang.LanguageCode {
+                        return code, nil
                 }
         }
 
-        return Language{}, fmt.Errorf("language code \"%s\" is not in the instance's language list", maxConfidenceLang.LanguageCode)
+        return "", fmt.Errorf("language code \"%s\" is not in the instance's language list", maxConfidenceLang.LanguageCode)
 }
 
@@ -131 +127 @@
 }
 
-func (e *LibreTranslate) Translate(text string, from, to Language) (TranslationResult, error) {
+func (e *LibreTranslate) Translate(text string, from, to string) (TranslationResult, error) {
         formData := map[string]string{
                 "q":      text,
-                "source": from.Code,
-                "target": to.Code,
+                "source": from,
+                "target": to,
         }
 
@@ -166 +162 @@
         }
 
-        return TranslationResult{SourceLanguage: from, TranslatedText: responseJSON.TranslatedText}, nil
+        return TranslationResult{TranslatedText: responseJSON.TranslatedText}, nil
 }