v/GTT
1
0
Fork 0
mirror of https://github.com/eeeXun/GTT.git synced 2025-05-16 07:40:44 -07:00
Code Issues Projects Releases Wiki Activity

refactor: APIKey in struct

Browse Source
This commit is contained in:
eeeXun 2023-05-01 11:11:30 +08:00
parent 4cb525a97e
commit 76ae888f29
7 changed files with 27 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
internal/translate/apertium/translator.go
View File

@ -16,6 +16,7 @@ const (
)
type Translator struct {
*core.APIKey
*core.Language
*core.TTSLock
core.EngineName
@ -23,15 +24,13 @@ type Translator struct {
func NewTranslator() *Translator {
return &Translator{
APIKey: new(core.APIKey),
Language: new(core.Language),
TTSLock: core.NewTTSLock(),
EngineName: core.NewEngineName("Apertium"),
}
}
func (t *Translator) SetAPIKey(key string) {
}
func (t *Translator) GetAllLang() []string {
return lang
}

5
internal/translate/argos/translator.go
View File

@ -16,6 +16,7 @@ const (
)
type Translator struct {
*core.APIKey
*core.Language
*core.TTSLock
core.EngineName
@ -23,15 +24,13 @@ type Translator struct {
func NewTranslator() *Translator {
return &Translator{
APIKey: new(core.APIKey),
Language: new(core.Language),
TTSLock: core.NewTTSLock(),
EngineName: core.NewEngineName("Argos"),
}
}
func (t *Translator) SetAPIKey(key string) {
}
func (t *Translator) GetAllLang() []string {
return lang
}

5
internal/translate/bing/translator.go
View File

@ -25,6 +25,7 @@ const (
)
type Translator struct {
*core.APIKey
*core.Language
*core.TTSLock
core.EngineName
@ -39,15 +40,13 @@ type setUpData struct {
func NewTranslator() *Translator {
return &Translator{
APIKey: new(core.APIKey),
Language: new(core.Language),
TTSLock: core.NewTTSLock(),
EngineName: core.NewEngineName("Bing"),
}
}
func (t *Translator) SetAPIKey(key string) {
}
func (t *Translator) GetAllLang() []string {
return lang
}

11
internal/translate/chatgpt/translator.go
View File

@ -16,24 +16,21 @@ const (
)
type Translator struct {
*core.APIKey
*core.Language
*core.TTSLock
core.EngineName
apiKey string
}
func NewTranslator() *Translator {
return &Translator{
APIKey: new(core.APIKey),
Language: new(core.Language),
TTSLock: core.NewTTSLock(),
EngineName: core.NewEngineName("ChatGPT"),
}
}
func (t *Translator) SetAPIKey(key string) {
t.apiKey = key
}
func (t *Translator) GetAllLang() []string {
return lang
}
@ -42,7 +39,7 @@ func (t *Translator) Translate(message string) (translation *core.Translation, e
translation = new(core.Translation)
var data map[string]interface{}
if len(t.apiKey) <= 0 {
if len(t.GetAPIKey()) <= 0 {
return nil, errors.New("Please write your API Key in config file for " + t.GetEngineName())
}
@ -64,7 +61,7 @@ func (t *Translator) Translate(message string) (translation *core.Translation, e
bytes.NewBuffer(userData),
)
req.Header.Add("Content-Type", "application/json")
req.Header.Add("Authorization", "Bearer "+t.apiKey)
req.Header.Add("Authorization", "Bearer "+t.GetAPIKey())
res, err := http.DefaultClient.Do(req)
if err != nil {
return nil, err

13
internal/translate/core/apikey.go Normal file
View File

@ -0,0 +1,13 @@
package core
type APIKey struct {
key string
}
func (k *APIKey) SetAPIKey(key string) {
k.key = key
}
func (k *APIKey) GetAPIKey() string {
return k.key
}

5
internal/translate/google/translator.go
View File

@ -20,6 +20,7 @@ const (
)
type Translator struct {
*core.APIKey
*core.Language
*core.TTSLock
core.EngineName
@ -27,15 +28,13 @@ type Translator struct {
func NewTranslator() *Translator {
return &Translator{
APIKey: new(core.APIKey),
Language: new(core.Language),
TTSLock: core.NewTTSLock(),
EngineName: core.NewEngineName("Google"),
}
}
func (t *Translator) SetAPIKey(key string) {
}
func (t *Translator) GetAllLang() []string {
return lang
}

5
internal/translate/reverso/translator.go
View File

@ -22,6 +22,7 @@ const (
)
type Translator struct {
*core.APIKey
*core.Language
*core.TTSLock
core.EngineName
@ -29,15 +30,13 @@ type Translator struct {
func NewTranslator() *Translator {
return &Translator{
APIKey: new(core.APIKey),
Language: new(core.Language),
TTSLock: core.NewTTSLock(),
EngineName: core.NewEngineName("Reverso"),
}
}
func (t *Translator) SetAPIKey(key string) {
}
func (t *Translator) GetAllLang() []string {
return lang
}