mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
57 lines
1.8 KiB
Go
57 lines
1.8 KiB
Go
// Package coser images
|
|
package coser
|
|
|
|
import (
|
|
"regexp"
|
|
|
|
"github.com/tidwall/gjson"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
"github.com/wdvxdr1123/ZeroBot/utils/helper"
|
|
|
|
control "github.com/FloatTech/zbputils/control"
|
|
"github.com/FloatTech/zbputils/ctxext"
|
|
"github.com/FloatTech/zbputils/web"
|
|
)
|
|
|
|
var (
|
|
ua = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.93 Safari/537.36"
|
|
coserURL = "http://ovooa.com/API/cosplay/api.php"
|
|
datestr = regexp.MustCompile(`/\d{4}-\d{2}-\d{2}/`)
|
|
)
|
|
|
|
func init() {
|
|
control.Register("coser", &control.Options{
|
|
DisableOnDefault: false,
|
|
Help: "三次元小姐姐\n- coser",
|
|
}).ApplySingle(ctxext.DefaultSingle).OnFullMatch("coser", zero.OnlyGroup).SetBlock(true).Limit(ctxext.LimitByGroup).
|
|
Handle(func(ctx *zero.Ctx) {
|
|
ctx.SendChain(message.Text("少女祈祷中......"))
|
|
data, err := web.RequestDataWith(web.NewDefaultClient(), coserURL, "GET", "", ua)
|
|
if err != nil {
|
|
log.Println("err为:", err)
|
|
}
|
|
|
|
text := gjson.Get(helper.BytesToString(data), "data.Title").String()
|
|
m := message.Message{ctxext.FakeSenderForwardNode(ctx, message.Text(text))}
|
|
ds := ""
|
|
gjson.Get(helper.BytesToString(data), "data.data").ForEach(func(_, value gjson.Result) bool {
|
|
if ds == "" {
|
|
ds = datestr.FindString(value.String())
|
|
} else if ds != datestr.FindString(value.String()) {
|
|
return false
|
|
}
|
|
m = append(m, ctxext.FakeSenderForwardNode(ctx, message.Image(value.String())))
|
|
return true
|
|
})
|
|
|
|
if id := ctx.SendGroupForwardMessage(
|
|
ctx.Event.GroupID,
|
|
m).Get("message_id").Int(); id == 0 {
|
|
ctx.SendChain(message.Text("ERROR:可能被风控或下载图片用时过长，请耐心等待"))
|
|
}
|
|
})
|
|
}
|