// Package coser images
package coser

import (
	"regexp"

	"github.com/tidwall/gjson"

	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 {
				ctx.SendChain(message.Text("ERROR:", err))
				return
			}
			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:可能被风控或下载图片用时过长，请耐心等待"))
			}
		})
}