mirror of
https://github.com/eeeXun/GTT.git
synced 2025-05-19 09:10:42 -07:00
cbd985b599
GetSrcLang, GetDstLang, SetSrcLang, SetDstLang, SwapLang are reusable functions from translator to translator. So make Language struct. GetEngineName is also a reusable function. So make EngineName struct. Finally, let Translator to inherit those functions in these three (Language, TTSLock, EngineName) structs. No need to write the duplicated functions for each Translator.
70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package argostranslate
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
"github.com/eeeXun/gtt/internal/translate/core"
|
|
)
|
|
|
|
const (
|
|
textURL = "https://translate.argosopentech.com/translate"
|
|
)
|
|
|
|
type ArgosTranslate struct {
|
|
*core.Language
|
|
*core.TTSLock
|
|
core.EngineName
|
|
}
|
|
|
|
func NewArgosTranslate() *ArgosTranslate {
|
|
return &ArgosTranslate{
|
|
Language: core.NewLanguage(),
|
|
TTSLock: core.NewTTSLock(),
|
|
EngineName: core.NewEngineName("ArgosTranslate"),
|
|
}
|
|
}
|
|
|
|
func (t *ArgosTranslate) GetAllLang() []string {
|
|
return lang
|
|
}
|
|
|
|
func (t *ArgosTranslate) Translate(message string) (translation, definition, partOfSpeech string, err error) {
|
|
var data map[string]interface{}
|
|
|
|
res, err := http.PostForm(textURL,
|
|
url.Values{
|
|
"q": {message},
|
|
"source": {langCode[t.GetSrcLang()]},
|
|
"target": {langCode[t.GetDstLang()]},
|
|
})
|
|
if err != nil {
|
|
return "", "", "", err
|
|
}
|
|
body, err := ioutil.ReadAll(res.Body)
|
|
if err != nil {
|
|
return "", "", "", err
|
|
}
|
|
if err = json.Unmarshal(body, &data); err != nil {
|
|
return "", "", "", err
|
|
}
|
|
|
|
if len(data) <= 0 {
|
|
return "", "", "", errors.New("Translation not found")
|
|
}
|
|
|
|
translation += fmt.Sprintf("%v", data["translatedText"])
|
|
|
|
return translation, definition, partOfSpeech, nil
|
|
}
|
|
|
|
func (t *ArgosTranslate) PlayTTS(lang, message string) error {
|
|
defer t.ReleaseLock()
|
|
|
|
return errors.New(t.GetEngineName() + " does not support text to speech")
|
|
}
|