add proxy

Author: 虫子樱桃

deepl.go

@@ -9,6 +9,7 @@ import (
 	"math/big"
 	"math/rand"
 	"net/http"
+	"net/url"
 	"strings"
 	"time"
 )
@@ -100,7 +101,14 @@ func adjustJsonContent(id int64, jsonContent []byte) []byte {
 // SK: Slovakian
 // SL: Slovenian
 // SV: Swedish
-func Translate(sourceLanguage, targetLanguage, textToTranslate string) (jsonRpcResponse *JsonRpcResponse, err error) {
+func Translate(sourceLanguage, targetLanguage, textToTranslate string, options ...Option) (jsonRpcResponse *JsonRpcResponse, err error) {
+
+	clientOpt := &deepLClientOption{}
+	if len(options) > 0 {
+		for _, optFunc := range options {
+			optFunc(clientOpt)
+		}
+	}
 	if sourceLanguage == "" {
 		lang := whatlanggo.DetectLang(textToTranslate)
 		deepLLang := strings.ToUpper(lang.Iso6391())
@@ -147,6 +155,13 @@ func Translate(sourceLanguage, targetLanguage, textToTranslate string) (jsonRpcR
 	request.Header.Set("Connection", "keep-alive")
 
 	client := &http.Client{}
+	if clientOpt.httpProxy != "" {
+		httpProxy, _ := url.Parse(clientOpt.httpProxy)
+		if httpProxy != nil {
+			client.Transport = &http.Transport{Proxy: http.ProxyURL(httpProxy)}
+		}
+	}
+
 	resp, err := client.Do(request)
 	if err != nil {
 		return nil, err

deepl_test.go

@@ -10,3 +10,12 @@ func TestTranslate(t *testing.T) {
 	t.Log(translate.Result.Texts[0].Text)
 	t.Log(len(translate.Result.Texts[0].Alternatives))
 }
+func TestTranslateWithHttpProxy(t *testing.T) {
+	translate, err := Translate("", "zh", "I love Go programming language",
+		WithHttpProxy("http://127.0.0.1:10808"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Log(translate.Result.Texts[0].Text)
+	t.Log(len(translate.Result.Texts[0].Alternatives))
+}

options.go

@@ -0,0 +1,13 @@
+package deepl
+
+type deepLClientOption struct {
+	httpProxy string
+}
+
+type Option func(option *deepLClientOption)
+
+func WithHttpProxy(proxy string) Option {
+	return func(option *deepLClientOption) {
+		option.httpProxy = proxy
+	}
+}