mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 05:30:07 +08:00
110 lines
2.8 KiB
Go
110 lines
2.8 KiB
Go
package diana
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/json"
|
||
"math"
|
||
"strconv"
|
||
"time"
|
||
|
||
"github.com/wdvxdr1123/ZeroBot/message"
|
||
|
||
"net/http"
|
||
"strings"
|
||
|
||
zero "github.com/wdvxdr1123/ZeroBot"
|
||
)
|
||
|
||
type zhiwang struct {
|
||
Code int `json:"code"`
|
||
Data struct {
|
||
EndTime int `json:"end_time"`
|
||
Rate float64 `json:"rate"`
|
||
Related [][]interface{} `json:"related"`
|
||
StartTime int `json:"start_time"`
|
||
} `json:"data"`
|
||
Message string `json:"message"`
|
||
}
|
||
|
||
// 小作文查重: 回复要查的消息 查重
|
||
func init() {
|
||
zero.OnMessage(fullmatch("查重")).
|
||
Handle(func(ctx *zero.Ctx) {
|
||
msg := ctx.Event.Message
|
||
if msg[0].Type == "reply" {
|
||
id, _ := strconv.Atoi(msg[0].Data["id"])
|
||
msg := ctx.GetMessage(int64(id)).Elements[0].Data["text"]
|
||
zhiwangjson := zhiwangapi(msg)
|
||
|
||
if zhiwangjson == nil || zhiwangjson.Code != 0 {
|
||
ctx.Send("api返回错误")
|
||
return
|
||
}
|
||
|
||
if len(zhiwangjson.Data.Related) == 0 {
|
||
ctx.Send("枝网没搜到，查重率为0%，我的评价是：一眼真")
|
||
return
|
||
}
|
||
|
||
related := zhiwangjson.Data.Related[0][1].(map[string]interface{})
|
||
ctx.SendChain(message.Text(
|
||
"枝网文本复制检测报告(简洁)", "\n",
|
||
"查重时间: ", time.Now().Format("2006-01-02 15:04:05"), "\n",
|
||
"总文字复制比: ", math.Floor(zhiwangjson.Data.Rate*100), "%", "\n",
|
||
"相似小作文:", "\n",
|
||
related["content"].(string)[:102]+".....", "\n",
|
||
"获赞数", related["like_num"], "\n",
|
||
zhiwangjson.Data.Related[0][2].(string), "\n",
|
||
"作者: ", related["m_name"], "\n",
|
||
"发表时间: ", time.Unix(int64(related["ctime"].(float64)), 0).Format("2006-01-02 15:04:05"), "\n",
|
||
"查重结果仅作参考，请注意辨别是否为原创", "\n",
|
||
"数据来源: https://asoulcnki.asia/",
|
||
))
|
||
}
|
||
})
|
||
}
|
||
|
||
// 发起api请求并把返回body交由json库解析
|
||
func zhiwangapi(text string) *zhiwang {
|
||
url := "https://asoulcnki.asia/v1/api/check"
|
||
post := "{\n\"text\":\"" + text + "\"\n}"
|
||
var jsonStr = []byte(post)
|
||
|
||
req, _ := http.NewRequest("POST", url, bytes.NewBuffer(jsonStr))
|
||
req.Header.Set("Content-Type", "application/json")
|
||
client := &http.Client{}
|
||
|
||
resp, err := client.Do(req)
|
||
|
||
if err != nil {
|
||
return nil
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
result := &zhiwang{}
|
||
if err1 := json.NewDecoder(resp.Body).Decode(result); err1 != nil {
|
||
return nil
|
||
}
|
||
return result
|
||
}
|
||
|
||
func fullmatch(src ...string) zero.Rule {
|
||
return func(ctx *zero.Ctx) bool {
|
||
msg := ctx.Event.Message
|
||
for _, elem := range msg {
|
||
if elem.Type == "text" {
|
||
text := elem.Data["text"]
|
||
text = strings.ReplaceAll(text, " ", "")
|
||
text = strings.ReplaceAll(text, "\r", "")
|
||
text = strings.ReplaceAll(text, "\n", "")
|
||
for _, s := range src {
|
||
if text == s {
|
||
return true
|
||
}
|
||
}
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
}
|