使用go构建彩云翻译的Alfred Workflow | 青训营笔记
2022-5-8 16:0:8 Author: www.gem-love.com(查看原文) 阅读量:2 收藏

在Day1的『Go语言上手-基础语言』中，老师给了三个项目，其中一个是基于彩云翻译使用Go来实现一个建议的词典查询功能，代码支持英译汉。我发现彩云翻译很好用，不需要自己申请api就可以使用，于是我在此基础上加以改造，将其做成了一个Alfred Workflow，并且添加了中译英的功能。

package main
import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"os"
)
type DictRequest struct {
	TransType string `json:"trans_type"`
	Source    string `json:"source"`
	UserID    string `json:"user_id"`
}
type DictResponse struct {
	Rc   int `json:"rc"`
	Wiki struct {
		KnownInLaguages int `json:"known_in_laguages"`
		Description     struct {
			Source string      `json:"source"`
			Target interface{} `json:"target"`
		} `json:"description"`
		ID   string `json:"id"`
		Item struct {
			Source string `json:"source"`
			Target string `json:"target"`
		} `json:"item"`
		ImageURL  string `json:"image_url"`
		IsSubject string `json:"is_subject"`
		Sitelink  string `json:"sitelink"`
	} `json:"wiki"`
	Dictionary struct {
		Prons struct {
			EnUs string `json:"en-us"`
			En   string `json:"en"`
		} `json:"prons"`
		Explanations []string      `json:"explanations"`
		Synonym      []string      `json:"synonym"`
		Antonym      []string      `json:"antonym"`
		WqxExample   [][]string    `json:"wqx_example"`
		Entry        string        `json:"entry"`
		Type         string        `json:"type"`
		Related      []interface{} `json:"related"`
		Source       string        `json:"source"`
	} `json:"dictionary"`
}
func query(word string) {
	client := &http.Client{}
	request := DictRequest{TransType: "en2zh", Source: word}
	buf, err := json.Marshal(request)
	if err != nil {
		log.Fatal(err)
	}
	var data = bytes.NewReader(buf)
	req, err := http.NewRequest("POST", "https://api.interpreter.caiyunai.com/v1/dict", data)
	if err != nil {
		log.Fatal(err)
	}
	req.Header.Set("Connection", "keep-alive")
	req.Header.Set("DNT", "1")
	req.Header.Set("os-version", "")
	req.Header.Set("sec-ch-ua-mobile", "?0")
	req.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.51 Safari/537.36")
	req.Header.Set("app-name", "xy")
	req.Header.Set("Content-Type", "application/json;charset=UTF-8")
	req.Header.Set("Accept", "application/json, text/plain, */*")
	req.Header.Set("device-id", "")
	req.Header.Set("os-type", "web")
	req.Header.Set("X-Authorization", "token:qgemv4jr1y38jyq6vhvi")
	req.Header.Set("Origin", "https://fanyi.caiyunapp.com")
	req.Header.Set("Sec-Fetch-Site", "cross-site")
	req.Header.Set("Sec-Fetch-Mode", "cors")
	req.Header.Set("Sec-Fetch-Dest", "empty")
	req.Header.Set("Referer", "https://fanyi.caiyunapp.com/")
	req.Header.Set("Accept-Language", "zh-CN,zh;q=0.9")
	req.Header.Set("Cookie", "_ym_uid=16456948721020430059; _ym_d=1645694872")
	resp, err := client.Do(req)
	if err != nil {
		log.Fatal(err)
	}
	defer resp.Body.Close()
	bodyText, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		log.Fatal(err)
	}
	if resp.StatusCode != 200 {
		log.Fatal("bad StatusCode:", resp.StatusCode, "body", string(bodyText))
	}
	var dictResponse DictResponse
	err = json.Unmarshal(bodyText, &dictResponse)
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println(word, "UK:", dictResponse.Dictionary.Prons.En, "US:", dictResponse.Dictionary.Prons.EnUs)
	for _, item := range dictResponse.Dictionary.Explanations {
		fmt.Println(item)
	}
}
func main() {
	if len(os.Args) != 2 {
		fmt.Fprintf(os.Stderr, `usage: simpleDict WORD
example: simpleDict hello
		`)
		os.Exit(1)
	}
	word := os.Args[1]
	query(word)
}
package main
import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"os"
	"strconv"
	"strings"
	"unicode"
)
type DictRequest struct {
	TransType string `json:"trans_type"`
	Source    string `json:"source"`
	UserID    string `json:"user_id"`
}
type DictResponse struct {
	Rc   int `json:"rc"`
	Wiki struct {
		KnownInLaguages int `json:"known_in_laguages"`
		Description     struct {
			Source string      `json:"source"`
			Target interface{} `json:"target"`
		} `json:"description"`
		ID   string `json:"id"`
		Item struct {
			Source string `json:"source"`
			Target string `json:"target"`
		} `json:"item"`
		ImageURL  string `json:"image_url"`
		IsSubject string `json:"is_subject"`
		Sitelink  string `json:"sitelink"`
	} `json:"wiki"`
	Dictionary struct {
		Prons struct {
			EnUs string `json:"en-us"`
			En   string `json:"en"`
		} `json:"prons"`
		Explanations []string      `json:"explanations"`
		Synonym      []string      `json:"synonym"`
		Antonym      []string      `json:"antonym"`
		WqxExample   [][]string    `json:"wqx_example"`
		Entry        string        `json:"entry"`
		Type         string        `json:"type"`
		Related      []interface{} `json:"related"`
		Source       string        `json:"source"`
	} `json:"dictionary"`
}
// 用于翻译结果的图标显示
type icon struct {
	Type string `json:"type"`
	Path string `json:"path"`
}
// 用于单条结果
type item struct {
	Subtitle string `json:"subtitle"` // 固定的小字的标题
	Title    string `json:"title"`
	Arg      string `json:"arg"`
	Icon     icon   `json:"icon"`
}
// 结果集
type items struct {
	Items []item `json:"items"`
}
func queryEnglish(word string, icon icon) {
	client := &http.Client{}
	request := DictRequest{TransType: "en2zh", Source: word}
	buf, err := json.Marshal(request)
	if err != nil {
		log.Fatal(err)
	}
	var data = bytes.NewReader(buf)
	req, err := http.NewRequest("POST", "https://api.interpreter.caiyunai.com/v1/dict", data)
	if err != nil {
		log.Fatal(err)
	}
	req.Header.Set("Connection", "keep-alive")
	req.Header.Set("DNT", "1")
	req.Header.Set("os-version", "")
	req.Header.Set("sec-ch-ua-mobile", "?0")
	req.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.51 Safari/537.36")
	req.Header.Set("app-name", "xy")
	req.Header.Set("Content-Type", "application/json;charset=UTF-8")
	req.Header.Set("Accept", "application/json, text/plain, */*")
	req.Header.Set("device-id", "")
	req.Header.Set("os-type", "web")
	req.Header.Set("X-Authorization", "token:qgemv4jr1y38jyq6vhvi")
	req.Header.Set("Origin", "https://fanyi.caiyunapp.com")
	req.Header.Set("Sec-Fetch-Site", "cross-site")
	req.Header.Set("Sec-Fetch-Mode", "cors")
	req.Header.Set("Sec-Fetch-Dest", "empty")
	req.Header.Set("Referer", "https://fanyi.caiyunapp.com/")
	req.Header.Set("Accept-Language", "zh-CN,zh;q=0.9")
	req.Header.Set("Cookie", "_ym_uid=16456948721020430059; _ym_d=1645694872")
	resp, err := client.Do(req)
	if err != nil {
		log.Fatal(err)
	}
	defer resp.Body.Close()
	bodyText, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		log.Fatal(err)
	}
	// 发生错误时的结果
	if resp.StatusCode != 200 {
		//log.Fatal("bad StatusCode:", resp.StatusCode, "body", string(bodyText))
		var errItem []item
		errItem = []item{
			{
				Subtitle: "error",
				Title:    "bad StatusCode: " + strconv.FormatInt(int64(resp.StatusCode), 10),
				Arg:      "bad StatusCode: " + strconv.FormatInt(int64(resp.StatusCode), 10),
				Icon:     icon,
			},
		}
		errItems := items{Items: errItem}
		itemsJson, _ := json.Marshal(errItems)
		fmt.Println(string(itemsJson))
		return
	}
	var dictResponse DictResponse
	err = json.Unmarshal(bodyText, &dictResponse)
	if err != nil {
		log.Fatal(err)
	}
	//fmt.Println(word, "UK:", dictResponse.Dictionary.Prons.En, "US:", dictResponse.Dictionary.Prons.EnUs)
	//for _, item := range dictResponse.Dictionary.Explanations {
	//	fmt.Println(item)
	//}
	var resultItem []item
	resultItem = []item{
		{
			Subtitle: "音标",
			Title:    "EN:" + dictResponse.Dictionary.Prons.En + " US:" + dictResponse.Dictionary.Prons.EnUs,
			Arg:      "EN:" + dictResponse.Dictionary.Prons.En + " US:" + dictResponse.Dictionary.Prons.EnUs,
			Icon:     icon,
		},
	}
	for i, paraphrase := range dictResponse.Dictionary.Explanations {
		paraphraseItem := item{Subtitle: "解释" + strconv.FormatInt(int64(i+1), 10), Title: paraphrase, Arg: "解释" + strconv.FormatInt(int64(i+1), 10), Icon: icon}
		resultItem = append(resultItem, paraphraseItem)
	}
	resultItems := items{Items: resultItem}
	resultJson, _ := json.Marshal(resultItems)
	fmt.Println(string(resultJson))
}
func queryChinese(word string, icon icon) {
	client := &http.Client{}
	request := DictRequest{TransType: "zh2en", Source: word}
	buf, err := json.Marshal(request)
	if err != nil {
		log.Fatal(err)
	}
	var data = bytes.NewReader(buf)
	req, err := http.NewRequest("POST", "https://api.interpreter.caiyunai.com/v1/dict", data)
	if err != nil {
		log.Fatal(err)
	}
	req.Header.Set("Connection", "keep-alive")
	req.Header.Set("DNT", "1")
	req.Header.Set("os-version", "")
	req.Header.Set("sec-ch-ua-mobile", "?0")
	req.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/99.0.4844.51 Safari/537.36")
	req.Header.Set("app-name", "xy")
	req.Header.Set("Content-Type", "application/json;charset=UTF-8")
	req.Header.Set("Accept", "application/json, text/plain, */*")
	req.Header.Set("device-id", "")
	req.Header.Set("os-type", "web")
	req.Header.Set("X-Authorization", "token:qgemv4jr1y38jyq6vhvi")
	req.Header.Set("Origin", "https://fanyi.caiyunapp.com")
	req.Header.Set("Sec-Fetch-Site", "cross-site")
	req.Header.Set("Sec-Fetch-Mode", "cors")
	req.Header.Set("Sec-Fetch-Dest", "empty")
	req.Header.Set("Referer", "https://fanyi.caiyunapp.com/")
	req.Header.Set("Accept-Language", "zh-CN,zh;q=0.9")
	req.Header.Set("Cookie", "_ym_uid=16456948721020430059; _ym_d=1645694872")
	resp, err := client.Do(req)
	if err != nil {
		log.Fatal(err)
	}
	defer resp.Body.Close()
	bodyText, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		log.Fatal(err)
	}
	// 发生错误时的结果
	if resp.StatusCode != 200 {
		//log.Fatal("bad StatusCode:", resp.StatusCode, "body", string(bodyText))
		var errItem []item
		errItem = []item{
			{
				Subtitle: "error",
				Title:    "bad StatusCode: " + strconv.FormatInt(int64(resp.StatusCode), 10),
				Arg:      "bad StatusCode: " + strconv.FormatInt(int64(resp.StatusCode), 10),
				Icon:     icon,
			},
		}
		errItems := items{Items: errItem}
		itemsJson, _ := json.Marshal(errItems)
		fmt.Println(string(itemsJson))
		return
	}
	var dictResponse DictResponse
	err = json.Unmarshal(bodyText, &dictResponse)
	if err != nil {
		log.Fatal(err)
	}
	//fmt.Println(word, "UK:", dictResponse.Dictionary.Prons.En, "US:", dictResponse.Dictionary.Prons.EnUs)
	//for _, item := range dictResponse.Dictionary.Explanations {
	//	fmt.Println(item)
	//}
	var resultItem []item
	resultItem = []item{}
	//fmt.Printf("[+] %+v\n", dictResponse.Dictionary)
	i := 1
	for _, paraphrase := range dictResponse.Dictionary.Explanations {
		paraphraseArr := strings.Split(paraphrase, "; ")
		for _, singleParapharase := range paraphraseArr {
			paraphraseItem := item{Subtitle: "解释" + strconv.FormatInt(int64(i), 10), Title: singleParapharase, Arg: "解释" + strconv.FormatInt(int64(i), 10), Icon: icon}
			resultItem = append(resultItem, paraphraseItem)
			i += 1
		}
	}
	resultItems := items{Items: resultItem}
	resultJson, _ := json.Marshal(resultItems)
	fmt.Println(string(resultJson))
}
func IsChinese(str string) bool {
	var count int
	for _, v := range str {
		if unicode.Is(unicode.Han, v) {
			count++
			break
		}
	}
	return count > 0
}
func main() {
	if len(os.Args) != 2 {
		//		fmt.Fprintf(os.Stderr, `usage: simpleDict WORD
		//example: simpleDict hello
		//		`)
		os.Exit(1)
	}
	word := os.Args[1]
	icon := icon{Type: "file", Path: "logo2.png"}
	if !IsChinese(word) {
		queryEnglish(word, icon)
	} else {
		queryChinese(word, icon)
	}
}

文章来源: https://www.gem-love.com/2022/05/08/%E4%BD%BF%E7%94%A8go%E6%9E%84%E5%BB%BA%E5%BD%A9%E4%BA%91%E7%BF%BB%E8%AF%91%E7%9A%84Alfred-Workflow/
如有侵权请联系:admin#unsafe.sh