mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 05:30:07 +08:00
90 lines
2.7 KiB
Go
90 lines
2.7 KiB
Go
// Package diana 嘉然相关
|
||
package diana
|
||
|
||
import (
|
||
"bytes"
|
||
"math"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/FloatTech/floatbox/binary"
|
||
"github.com/FloatTech/floatbox/web"
|
||
"github.com/FloatTech/zbputils/ctxext"
|
||
"github.com/tidwall/gjson"
|
||
"github.com/wdvxdr1123/ZeroBot/message"
|
||
|
||
zero "github.com/wdvxdr1123/ZeroBot"
|
||
)
|
||
|
||
// 小作文查重: 回复要查的消息 查重
|
||
func init() {
|
||
engine.OnMessage(func(ctx *zero.Ctx) bool {
|
||
msg := ctx.Event.Message
|
||
if msg[0].Type != "reply" {
|
||
return false
|
||
}
|
||
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", "")
|
||
if text == "查重" {
|
||
return true
|
||
}
|
||
}
|
||
}
|
||
return false
|
||
}).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
|
||
msg := ctx.GetMessage(message.NewMessageIDFromString(ctx.Event.Message[0].Data["id"])).Elements[0].Data["text"]
|
||
result, err := zhiwangapi(msg)
|
||
if err != nil {
|
||
ctx.SendChain(message.Text("ERROR:", err))
|
||
return
|
||
}
|
||
if result.Get("code").Int() != 0 {
|
||
ctx.SendChain(message.Text("api返回错误:", result.Get("code").Int()))
|
||
return
|
||
}
|
||
if result.Get("data.related.#").Int() == 0 {
|
||
ctx.Send(message.ReplyWithMessage(ctx.Event.MessageID, message.Text("枝网没搜到，查重率为0%，鉴定为原创")))
|
||
return
|
||
}
|
||
related := result.Get("data.related.0.reply").Map()
|
||
rate := result.Get("data.related.0.rate").Float()
|
||
relatedcontent := related["content"].String()
|
||
if len(relatedcontent) > 102 {
|
||
relatedcontent = relatedcontent[:102] + "....."
|
||
}
|
||
ctx.Send(message.ReplyWithMessage(ctx.Event.MessageID, message.Text(
|
||
"枝网文本复制检测报告(简洁)", "\n",
|
||
"查重时间: ", time.Now().Format("2006-01-02 15:04:05"), "\n",
|
||
"总文字复制比: ", math.Floor(rate*100), "%", "\n",
|
||
"相似小作文：", "\n", relatedcontent, "\n",
|
||
"获赞数：", related["like_num"].String(), "\n",
|
||
result.Get("data.related.0.reply_url").String(), "\n",
|
||
"作者: ", related["m_name"].String(), "\n",
|
||
"发表时间: ", time.Unix(int64(related["ctime"].Float()), 0).Format("2006-01-02 15:04:05"), "\n",
|
||
"查重结果仅作参考，请注意辨别是否为原创", "\n",
|
||
"数据来源: https://asoulcnki.asia/",
|
||
)))
|
||
})
|
||
}
|
||
|
||
func zhiwangapi(text string) (*gjson.Result, error) {
|
||
b, cl := binary.OpenWriterF(func(w *binary.Writer) {
|
||
w.WriteString("{\n\"text\":\"")
|
||
w.WriteString(text)
|
||
w.WriteString("\"\n}")
|
||
})
|
||
|
||
data, err := web.PostData("https://asoulcnki.asia/v1/api/check", "application/json", bytes.NewReader(b))
|
||
cl()
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
result := gjson.ParseBytes(data)
|
||
return &result, nil
|
||
}
|