Small tweaks

git-svn-id: file:///srv/svn/repo/mai/trunk@26 e410bdd4-646f-c54f-a7ce-fffcc4f439ae
2023-09-02 17:54:22 +00:00
parent 0377709f24
commit 9d12f6f17a
4 changed files with 20 additions and 5 deletions
--- a/engines/reverso.go
+++ b/engines/reverso.go
@@ -1,134 +0,0 @@
-package engines
-
-import (
-	"bytes"
-	"encoding/json"
-	"fmt"
-	"net/http"
-)
-
-// Reverso is an engine that fetches data from https://reverso.net.
-type Reverso struct{}
-
-func (_ *Reverso) DisplayName() string { return "Reverso" }
-
-var reversoLangs = Language{
-	"ara": "Arabic",
-	"chi": "Chinese (Simplified)", // marketed as just "Chinese"
-	"cze": "Czech",
-	"dan": "Danish",
-	"dut": "Dutch",
-	"eng": "English",
-	"fra": "French",
-	"ger": "German",
-	"heb": "Hebrew",
-	"hin": "Hindi",
-	"hun": "Hungarian",
-	"ita": "Italian",
-	"jpn": "Japanese",
-	"kor": "Korean",
-	"per": "Persian",
-	"pol": "Polish",
-	"por": "Portuguese",
-	"rum": "Romanian",
-	"rus": "Russian",
-	"slo": "Slovak",
-	"spa": "Spanish",
-	"swe": "Swedish",
-	"tha": "Thai",
-	"tur": "Turkish",
-	"ukr": "Ukrainian",
-}
-
-func (_ *Reverso) SourceLanguages() (Language, error) { return reversoLangs, nil }
-
-func (_ *Reverso) TargetLanguages() (Language, error) { return reversoLangs, nil }
-
-func (_ *Reverso) Tts(text, lang string) (string, error) { return "", nil }
-
-type reversoAPIResponse struct {
-	LanguageDetection struct {
-		DetectedLanguage string `json:"detectedLanguage"`
-	} `json:"languageDetection"`
-	Translation []string `json:"translation"`
-}
-
-func (e *Reverso) callAPI(text string, from, to string) (reversoAPIResponse, error) {
-	// `contextResults` must be false for language detection
-	formData := map[string]interface{}{
-		"format": "text",
-		"from":   from,
-		"to":     to,
-		"input":  text,
-		"options": map[string]interface{}{
-			"sentenceSplitter":  false,
-			"origin":            "translation.web",
-			"contextResults":    false,
-			"languageDetection": true,
-		},
-	}
-
-	formDataJSON, err := json.Marshal(formData)
-
-	if err != nil {
-		return reversoAPIResponse{}, err
-	}
-
-	request, err := http.NewRequest("POST", "https://api.reverso.net/translate/v1/translation", bytes.NewBuffer(formDataJSON))
-
-	if err != nil {
-		return reversoAPIResponse{}, err
-	}
-
-	request.Header.Set("Content-Type", "application/json")
-	request.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; rv:110.0) Gecko/20100101 Firefox/110.0")
-
-	client := &http.Client{}
-
-	response, err := client.Do(request)
-
-	if err != nil {
-		return reversoAPIResponse{}, err
-	}
-
-	defer response.Body.Close()
-
-	if response.StatusCode != 200 {
-		return reversoAPIResponse{}, fmt.Errorf("got status code %d from Reverso API", response.StatusCode)
-	}
-
-	var result reversoAPIResponse
-
-	if err := json.NewDecoder(response.Body).Decode(&result); err != nil {
-		return reversoAPIResponse{}, err
-	}
-
-	return result, nil
-}
-
-func (e *Reverso) Translate(text string, from, to string) (TranslationResult, error) {
-	if from == "auto" || from == "" {
-		from = "eng"
-	}
-
-	var translation string
-
-	r, err := e.callAPI(text, from, to)
-	if err != nil {
-		return TranslationResult{}, err
-	}
-
-	translation = r.Translation[0]
-	langCode := r.LanguageDetection.DetectedLanguage
-
-	for code := range reversoLangs {
-		if code == langCode {
-			from = code
-		}
-	}
-
-	return TranslationResult{
-		TranslatedText: translation,
-		SourceLanguage: from,
-	}, nil
-}