arya/mozhi
1
0
Fork 0
mirror of https://codeberg.org/aryak/mozhi synced 2025-06-04 00:41:07 +05:30
Code Issues Actions Packages Projects Releases Wiki Activity

add support for disablin engines

Browse Source
This commit is contained in:
Arya
2023-08-30 16:05:16 +05:30
parent 0e2be0cd32
commit fc6f689fb6
5 changed files with 62 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
utils/main.go
View File

@@ -17,32 +17,39 @@ type LangOut struct {
TargetLang string `json:"target_language"`
}
func EnvTrueNoExist(env string) bool {
if _, ok := os.LookupEnv(env); ok == false || os.Getenv(env) == "true" {
return true
}
return false
}
func LangList(engine string, listType string) ([]List, error) {
var data []List
if listType != "sl" && listType != "tl" {
return []List{}, errors.New("list type invalid: either give tl for target languages or sl for source languages.")
}
if engine == "google" {
if engine == "google" && EnvTrueNoExist("MOZHI_GOOGLE_ENABLED") {
data = LangListGoogle(listType)
} else if engine == "libre" {
if os.Getenv("MOZHI_LIBRETRANSLATE_URL") == "" {
} else if engine == "libre" && EnvTrueNoExist("MOZHI_LIBRETRANSLATE_ENABLED") {
if EnvTrueNoExist("MOZHI_LIBRETRANSLATE_URL") {
return []List{}, errors.New("Please set MOZHI_LIBRETRANSLATE_URL if you want to use libretranslate. Example: MOZHI_LIBRETRANSLATE_URL=https://lt.psf.lt")
}
data = LangListLibreTranslate(listType)
} else if engine == "reverso" {
} else if engine == "reverso" && EnvTrueNoExist("MOZHI_REVERSO_ENABLED") {
data = LangListReverso(listType)
} else if engine == "deepl" {
} else if engine == "deepl" && EnvTrueNoExist("MOZHI_DEEPL_ENABLED") {
data = LangListDeepl(listType)
} else if engine == "watson" {
} else if engine == "watson" && EnvTrueNoExist("MOZHI_WATSON_ENABLED") {
data = LangListWatson(listType)
} else if engine == "yandex" {
} else if engine == "yandex" && EnvTrueNoExist("MOZHI_YANDEX_ENABLED") {
data = LangListYandex(listType)
} else if engine == "mymemory" {
} else if engine == "mymemory" && EnvTrueNoExist("MOZHI_MYMEMORY_ENABLED") {
data = LangListMyMemory(listType)
} else if engine == "duckduckgo" {
} else if engine == "duckduckgo" && EnvTrueNoExist("MOZHI_DUCKDUCKGO_ENABLED") {
data = LangListDuckDuckGo(listType)
} else {
return []List{}, errors.New("Engine does not exist.")
return []List{}, errors.New("Engine does not exist or has been disabled.")
}
return data, nil
}
@@ -51,27 +58,27 @@ func LangList(engine string, listType string) ([]List, error) {
func Translate(engine string, to string, from string, text string) (LangOut, error) {
var err error
var data LangOut
if engine == "google" {
if engine == "google" && EnvTrueNoExist("MOZHI_GOOGLE_ENABLED") {
data, err = TranslateGoogle(to, from, text)
} else if engine == "libre" {
} else if engine == "libre" && EnvTrueNoExist("MOZHI_LIBRETRANSLATE_ENABLED") {
if os.Getenv("MOZHI_LIBRETRANSLATE_URL") == "" {
return LangOut{}, errors.New("Please set MOZHI_LIBRETRANSLATE_URL if you want to use libretranslate. Example: MOZHI_LIBRETRANSLATE_URL=https://lt.psf.lt")
}
data, err = TranslateLibreTranslate(to, from, text)
} else if engine == "reverso" {
} else if engine == "reverso" && EnvTrueNoExist("MOZHI_REVERSO_ENABLED") {
data, err = TranslateReverso(to, from, text)
} else if engine == "deepl" {
} else if engine == "deepl" && EnvTrueNoExist("MOZHI_DEEPL_ENABLED") {
data, err = TranslateDeepl(to, from, text)
} else if engine == "watson" {
} else if engine == "watson" && EnvTrueNoExist("MOZHI_WATSON_ENABLED") {
data, err = TranslateWatson(to, from, text)
} else if engine == "yandex" {
} else if engine == "yandex" && EnvTrueNoExist("MOZHI_YANDEX_ENABLED") {
data, err = TranslateYandex(to, from, text)
} else if engine == "mymemory" {
} else if engine == "mymemory" && EnvTrueNoExist("MOZHI_MYMEMORY_ENABLED") {
data, err = TranslateMyMemory(to, from, text)
} else if engine == "duckduckgo" {
} else if engine == "duckduckgo" && EnvTrueNoExist("MOZHI_DUCKDUCKGO_ENABLED") {
data, err = TranslateDuckDuckGo(to, from, text)
} else {
return LangOut{}, errors.New("Engine does not exist.")
return LangOut{}, errors.New("Engine does not exist or has been disabled.")
}
if err != nil {
return LangOut{}, err
@@ -82,12 +89,12 @@ func Translate(engine string, to string, from string, text string) (LangOut, err
func TTS(engine string, lang string, text string) ([]byte, error) {
var err error
var data []byte
if engine == "google" {
if engine == "google" && EnvTrueNoExist("MOZHI_GOOGLE_ENABLED") {
data, err = TTSGoogle(lang, text)
} else if engine == "reverso" {
} else if engine == "reverso" && EnvTrueNoExist("MOZHI_REVERSO_ENABLED") {
data, err = TTSReverso(lang, text)
} else {
return []byte(""), errors.New("Engine does not exist and/or doesn't support TTS.")
return []byte(""), errors.New("Engine does not exist and/or doesn't support TTS and/or has been disabled.")
}
if err != nil {
return []byte(""), err