🎨 改进代码

setutime/utils/ascii2d.go

@@ -24,12 +24,12 @@ func Ascii2dSearch(pic string) (text string, err error) {
 		Transport: &transport,
 	}
 
-	// TODO 包装请求参数
+	// 包装请求参数
 	data := url.Values{}
 	data.Set("uri", pic) // 图片链接
 	fromData := strings.NewReader(data.Encode())
 
-	// TODO 网络请求
+	// 网络请求
 	req, _ := http.NewRequest("POST", api, fromData)
 	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
 	req.Header.Set("Accept", "Mozilla/5.0 (Windows NT 6.1; WOW64; rv:6.0) Gecko/20100101 Firefox/6.0")
@@ -37,7 +37,7 @@ func Ascii2dSearch(pic string) (text string, err error) {
 	if err != nil {
 		return "", err
 	}
-	// TODO 色合检索改变到特征检索
+	// 色合检索改变到特征检索
 	var bovwUrl = strings.ReplaceAll(resp.Request.URL.String(), "color", "bovw")
 	bovwReq, _ := http.NewRequest("POST", bovwUrl, nil)
 	bovwReq.Header.Set("Content-Type", "application/x-www-form-urlencoded")
@@ -47,15 +47,15 @@ func Ascii2dSearch(pic string) (text string, err error) {
 		return "", err
 	}
 	defer bovwResp.Body.Close()
-	// TODO 解析XPATH
+	// 解析XPATH
 	doc, err := xpath.Parse(resp.Body)
 	if err != nil {
 		return "", err
 	}
-	// TODO 取出每个返回的结果
+	// 取出每个返回的结果
 	list := xpath.Find(doc, `//div[@class="row item-box"]`)
 	var link string
-	// TODO 遍历取出第一个返回的PIXIV结果
+	// 遍历取出第一个返回的PIXIV结果
 	for _, n := range list {
 		linkPath := xpath.Find(n, `//div[2]/div[3]/h6/a[1]`)
 		picPath := xpath.Find(n, `//div[1]/img`)
@@ -66,7 +66,7 @@ func Ascii2dSearch(pic string) (text string, err error) {
 			}
 		}
 	}
-	// TODO 链接取出PIXIV id
+	// 链接取出PIXIV id
 	var index = strings.LastIndex(link, "/")
 	if link == "" || index == -1 {
 		return "", errors.New("Ascii2d not found")
@@ -75,7 +75,7 @@ func Ascii2dSearch(pic string) (text string, err error) {
 	if id == 0 {
 		return "", errors.New("convert to pid error")
 	}
-	// TODO 根据PID查询插图信息
+	// 根据PID查询插图信息
 	var illust = &Illust{}
 	if err := illust.IllustInfo(id); err != nil {
 		return "", err
@@ -83,7 +83,7 @@ func Ascii2dSearch(pic string) (text string, err error) {
 	if illust.AgeLimit != "all-age" {
 		return "", errors.New("Ascii2d not found")
 	}
-	// TODO 返回插图信息文本
+	// 返回插图信息文本
 	return fmt.Sprintf(
 		`[SetuTime] emmm大概是这个？
 标题：%s

setutime/utils/download.go

@@ -18,13 +18,13 @@ func (this *Illust) PixivPicDown(path string) (savePath string, err error) {
 	url = strings.ReplaceAll(url, "img-original", "img-master")
 	url = strings.ReplaceAll(url, "_p0", "_p0_master1200")
 	url = strings.ReplaceAll(url, ".png", ".jpg")
-	// TODO 文件名为url的hash值
+	// 文件名为url的hash值
 	savePath = path + Int2Str(pid) + ".jpg"
-	// TODO 文件存在或文件大小大于10kb
+	// 文件存在或文件大小大于10kb
 	if PathExists(savePath) && FileSize(savePath) > 10240 {
 		return savePath, nil
 	}
-	// TODO 模拟QQ客户端请求
+	// 模拟QQ客户端请求
 	client := &http.Client{}
 	reqest, _ := http.NewRequest("GET", url, nil)
 	reqest.Header.Add("User-Agent", "QQ/8.2.0.1296 CFNetwork/1126")
@@ -39,7 +39,7 @@ func (this *Illust) PixivPicDown(path string) (savePath string, err error) {
 		return "", errors.New(fmt.Sprintf("Download failed, code %d", code))
 	}
 	defer resp.Body.Close()
-	// TODO 写入文件
+	// 写入文件
 	data, _ := ioutil.ReadAll(resp.Body)
 	f, _ := os.OpenFile(savePath, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
 	defer f.Close()

setutime/utils/pixiv_api.go

@@ -30,12 +30,12 @@ func (this *Illust) IllustInfo(id int64) (err error) {
 	api := fmt.Sprintf("https://pixiv.net/ajax/illust/%d", id)
 	transport := http.Transport{
 		DisableKeepAlives: true,
-		// TODO 绕过sni审查
+		// 绕过sni审查
 		TLSClientConfig: &tls.Config{
 			ServerName:         "-",
 			InsecureSkipVerify: true,
 		},
-		// TODO 更改dns
+		// 更改dns
 		Dial: func(network, addr string) (net.Conn, error) {
 			return net.Dial("tcp", "210.140.131.223:443")
 		},
@@ -44,7 +44,7 @@ func (this *Illust) IllustInfo(id int64) (err error) {
 		Transport: &transport,
 	}
 
-	// TODO 网络请求
+	// 网络请求
 	req, err := http.NewRequest("GET", api, nil)
 	if err != nil {
 		return err
@@ -67,7 +67,7 @@ func (this *Illust) IllustInfo(id int64) (err error) {
 	}
 	json := gjson.ParseBytes(body).Get("body")
 
-	// TODO 如果有"R-18"tag则判断为R-18（暂时）
+	// 如果有"R-18"tag则判断为R-18（暂时）
 	var ageLimit = "all-age"
 	for _, tag := range json.Get("tags.tags.#.tag").Array() {
 		if tag.Str == "R-18" {
@@ -75,12 +75,12 @@ func (this *Illust) IllustInfo(id int64) (err error) {
 			break
 		}
 	}
-	// TODO 解决json返回带html格式
+	// 解决json返回带html格式
 	var caption = strings.ReplaceAll(json.Get("illustComment").Str, "<br />", "\n")
 	if index := strings.Index(caption, "<"); index != -1 {
 		caption = caption[:index]
 	}
-	// TODO 解析返回插画信息
+	// 解析返回插画信息
 	this.Pid = json.Get("illustId").Int()
 	this.Title = json.Get("illustTitle").Str
 	this.Caption = caption

setutime/utils/saucenao.go

@@ -27,7 +27,7 @@ func SauceNaoSearch(pic string) (text string, err error) {
 		Transport: &transport,
 	}
 
-	// TODO 包装请求参数
+	// 包装请求参数
 	data := url.Values{}
 	data.Set("url", pic)         // 图片链接
 	data.Set("api_key", apiKey)  // api_key
@@ -36,7 +36,7 @@ func SauceNaoSearch(pic string) (text string, err error) {
 	data.Set("output_type", "2") // 返回JSON格式数据
 	fromData := strings.NewReader(data.Encode())
 
-	// TODO 网络请求
+	// 网络请求
 	req, err := http.NewRequest("POST", api, fromData)
 	if err != nil {
 		return "", err
@@ -65,7 +65,7 @@ func SauceNaoSearch(pic string) (text string, err error) {
 	if content.Get("results.0.header.similarity").Float() < minSimilarity {
 		return "", errors.New("SauceNAO not found")
 	}
-	// TODO 正常发送
+	// 正常发送
 	return fmt.Sprintf(
 		`[SetuTime] 我有把握是这个！[CQ:image,file=%s]相似度：%s%%
 标题：%s