mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package utils
|
||
|
||
import (
|
||
"fmt"
|
||
"io/ioutil"
|
||
"net/http"
|
||
"net/url"
|
||
|
||
"github.com/tidwall/gjson"
|
||
"github.com/wdvxdr1123/ZeroBot/message"
|
||
)
|
||
|
||
// SauceNaoSearch SauceNao 以图搜图 需要链接 返回错误和信息
|
||
func SauceNaoSearch(pic string) (message.Message, error) {
|
||
var (
|
||
api = "https://saucenao.com/search.php"
|
||
apiKey = "2cc2772ca550dbacb4c35731a79d341d1a143cb5"
|
||
|
||
minSimilarity = 70.0 // 返回图片结果的最小相似度
|
||
)
|
||
|
||
// 包装请求参数
|
||
link, _ := url.Parse(api)
|
||
link.RawQuery = url.Values{
|
||
"url": []string{pic},
|
||
"api_key": []string{apiKey},
|
||
"db": []string{"5"},
|
||
"numres": []string{"1"},
|
||
"output_type": []string{"2"},
|
||
}.Encode()
|
||
|
||
// 网络请求
|
||
client := &http.Client{}
|
||
req, err := http.NewRequest("GET", link.String(), nil)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
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")
|
||
resp, err := client.Do(req)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
defer resp.Body.Close()
|
||
body, err := ioutil.ReadAll(resp.Body)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
if resp.StatusCode != http.StatusOK {
|
||
// 如果返回不是200则立刻抛出错误
|
||
return nil, fmt.Errorf("SauceNAO not found, code %d", resp.StatusCode)
|
||
}
|
||
content := gjson.ParseBytes(body)
|
||
if status := content.Get("header.status").Int(); status != 0 {
|
||
// 如果json信息返回status不为0则立刻抛出错误
|
||
return nil, fmt.Errorf("SauceNAO not found, status %d", status)
|
||
}
|
||
if content.Get("results.0.header.similarity").Float() < minSimilarity {
|
||
return nil, fmt.Errorf("SauceNAO not found")
|
||
}
|
||
result := content.Get("results.0")
|
||
// 正常发送
|
||
return message.Message{
|
||
message.Text("[SetuTime] 我有把握是这个！"),
|
||
message.Image(result.Get("header.thumbnail").Str),
|
||
message.Text(
|
||
"\n",
|
||
"相似度：", result.Get("header.similarity").Str, "\n",
|
||
"标题：", result.Get("data.title").Str, "\n",
|
||
"插画ID：", result.Get("data.pixiv_id").Int(), "\n",
|
||
"画师：", result.Get("data.member_name").Str, "\n",
|
||
"画师ID：", result.Get("data.member_id").Int(), "\n",
|
||
"直链：", "https://pixivel.moe/detail?id=", result.Get("data.pixiv_id").Int(),
|
||
),
|
||
}, nil
|
||
}
|