add caiyun translator

Author: xwjdsh

caiyun.go

@@ -0,0 +1,132 @@
+package fy
+
+import (
+	"bytes"
+	"context"
+	"encoding/base64"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"strings"
+
+	"github.com/google/uuid"
+	"github.com/tidwall/gjson"
+)
+
+type caiyunTranslator struct {
+	token      string
+	seq1, seq2 string
+}
+
+var caiyun translator = &caiyunTranslator{
+	token: "token:qgemv4jr1y38jyq6vhvi",
+	seq1:  "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
+	seq2:  "NOPQRSTUVWXYZABCDEFGHIJKLMnopqrstuvwxyzabcdefghijklm",
+}
+
+func (b *caiyunTranslator) Desc() (string, string) {
+	return "caiyun", "https://fanyi.caiyunapp.com"
+}
+
+func CaiyunTranslate(ctx context.Context, req Request) *Response {
+	return caiyun.translate(ctx, req)
+}
+
+func (t *caiyunTranslator) translate(ctx context.Context, req Request) (resp *Response) {
+	resp = newResp(t)
+	req.ToLang, resp.Err = t.convertLanguage(req.ToLang)
+	if resp.Err != nil {
+		return
+	}
+
+	browserId := strings.ReplaceAll(uuid.New().String(), "-", "")
+	jwtBody := strings.NewReader(fmt.Sprintf(`{"browser_id":"%s"}`, browserId))
+
+	var jwtResponse struct {
+		Rc  int    `json:"rc"`
+		Jwt string `json:"jwt"`
+	}
+
+	_, data, err := sendRequest(ctx, http.MethodPost, "https://api.interpreter.caiyunai.com/v1/user/jwt/generate", jwtBody, func(req *http.Request) error {
+		req.Header.Set("Content-Type", "application/json")
+		req.Header.Set("User-Agent", UserAgent)
+		req.Header.Set("X-Authorization", t.token)
+		return nil
+	})
+	if err := json.Unmarshal(data, &jwtResponse); err != nil {
+		resp.Err = fmt.Errorf("get jwt error: %v", err)
+		return
+	}
+	if jwtResponse.Rc != 0 {
+		resp.Err = fmt.Errorf("unexpected jwt response: %s", string(data))
+		return
+	}
+
+	param := map[string]interface{}{
+		"browser_id": browserId,
+		"cached":     true,
+		"detect":     true,
+		"dict":       true,
+		"media":      "text",
+		"os_type":    "web",
+		"replaced":   true,
+		"request_id": "web_fanyi",
+		"source":     req.Text,
+		"trans_type": "auto2" + req.ToLang,
+	}
+	reqData, _ := json.Marshal(param)
+	body := bytes.NewReader(reqData)
+
+	_, data, err = sendRequest(ctx, http.MethodPost, "https://api.interpreter.caiyunai.com/v1/translator", body, func(req *http.Request) error {
+		req.Header.Set("Content-Type", "application/json")
+		req.Header.Set("User-Agent", UserAgent)
+		req.Header.Set("X-Authorization", t.token)
+		req.Header.Set("T-Authorization", jwtResponse.Jwt)
+		return nil
+	})
+	if err != nil {
+		resp.Err = fmt.Errorf("sendRequest error: %v", err)
+		return
+	}
+
+	jr := gjson.Parse(string(data))
+	dit := jr.Get("target").String()
+	if dit == "" {
+		resp.Err = fmt.Errorf("cannot get translate result, resp: %s", string(data))
+		return
+	}
+
+	encoded := ""
+	for _, s := range dit {
+		if i := strings.IndexRune(t.seq1, s); i > -1 {
+			encoded += string(t.seq2[i])
+		} else {
+			encoded += string(s)
+		}
+	}
+
+	result, err := base64.StdEncoding.DecodeString(encoded)
+	if err != nil {
+		resp.Err = fmt.Errorf("base64 decode error: %v", err)
+		return
+	}
+
+	resp.Result = string(result)
+	return
+}
+
+func (*caiyunTranslator) convertLanguage(language string) (string, error) {
+	var l string
+	switch language {
+	case Chinese:
+		l = "zh"
+	case Japanese, English, Russian, French, Spanish:
+		l = language
+	}
+
+	if l == "" {
+		return "", fmt.Errorf("unsupported target language: %s", language)
+	}
+
+	return l, nil
+}

caiyun_test.go

@@ -0,0 +1,32 @@
+package fy
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCaiyunTranslate(t *testing.T) {
+	{
+		resp := CaiyunTranslate(context.Background(), Request{
+			ToLang: Chinese,
+			Text:   "One",
+		})
+		assert.Nil(t, resp.Err)
+		expectedResp := newResp(caiyun)
+		expectedResp.Result = "一个"
+		assert.Equal(t, expectedResp, resp)
+	}
+
+	{
+		resp := CaiyunTranslate(context.Background(), Request{
+			ToLang: English,
+			Text:   "一",
+		})
+		assert.Nil(t, resp.Err)
+		expectedResp := newResp(caiyun)
+		expectedResp.Result = "One"
+		assert.Equal(t, expectedResp, resp)
+	}
+}

deepl.go

@@ -52,7 +52,7 @@ func (t *deeplTranslator) translate(ctx context.Context, req Request) (resp *Res
 }
 
 func (*deeplTranslator) convertLanguage(language string) (string, error) {
-	l := language
+	var l string
 	switch language {
 	case Chinese:
 		l = "ZH"
@@ -70,7 +70,7 @@ func (*deeplTranslator) convertLanguage(language string) (string, error) {
 		l = "ES"
 	}
 	if l == "" {
-		fmt.Println("unsupported target language")
+		return "", fmt.Errorf("unsupported target language: %s", language)
 	}
 
 	return l, nil

fy.go

@@ -20,7 +20,7 @@ const (
 )
 
 var Translators = []translator{
-	baidu, bing, google, sogou, tencent, youdao, deepl,
+	baidu, bing, google, sogou, tencent, youdao, deepl, caiyun,
 }
 
 // Request translation request

go.mod

@@ -3,6 +3,7 @@ module github.com/xwjdsh/fy
 require (
 	github.com/atotto/clipboard v0.1.2
 	github.com/fatih/color v1.7.0
+	github.com/google/uuid v1.3.0 // indirect
 	github.com/mattn/go-colorable v0.1.1 // indirect
 	github.com/mattn/go-isatty v0.0.7 // indirect
 	github.com/robertkrimen/otto v0.0.0-20180617131154-15f95af6e78d

go.sum

@@ -4,6 +4,8 @@ github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/fatih/color v1.7.0 h1:DkWD4oS2D8LGGgTQ6IvwJJXSL5Vp2ffcQg58nFV38Ys=
 github.com/fatih/color v1.7.0/go.mod h1:Zm6kSWBoL9eyXnKyktHP6abPY2pDugNf5KwzbycvMj4=
+github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
+github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
 github.com/mattn/go-colorable v0.1.1 h1:G1f5SKeVxmagw/IyvzvtZE4Gybcc4Tr1tf7I8z0XgOg=
 github.com/mattn/go-colorable v0.1.1/go.mod h1:FuOcm+DKB9mbwrcAfNl7/TZVBZ6rcnceauSikq3lYCQ=
 github.com/mattn/go-isatty v0.0.5/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=