优化代码结构

plugin/coser/coser.go

@@ -2,7 +2,10 @@
 package coser
 
 import (
-	"regexp"
+	"errors"
+	"math/rand"
+	"os"
+	"time"
 
 	"github.com/tidwall/gjson"
 
@@ -10,6 +13,8 @@ import (
 	"github.com/wdvxdr1123/ZeroBot/message"
 	"github.com/wdvxdr1123/ZeroBot/utils/helper"
 
+	"github.com/FloatTech/AnimeAPI/setu"
+	"github.com/FloatTech/floatbox/file"
 	"github.com/FloatTech/floatbox/web"
 	ctrl "github.com/FloatTech/zbpctrl"
 	"github.com/FloatTech/zbputils/control"
@@ -19,35 +24,44 @@ import (
 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() {
+	p, err := setu.NewPool(setu.DefaultPoolDir,
+		func(s string) (string, error) {
+			if s != "coser" {
+				return "", errors.New("invalid call")
+			}
+			typ := setu.DefaultPoolDir + "/" + "coser"
+			if file.IsNotExist(typ) {
+				err := os.MkdirAll(typ, 0755)
+				if err != nil {
+					return "", err
+				}
+			}
+			data, err := web.RequestDataWith(web.NewDefaultClient(), coserURL, "GET", "", ua)
+			if err != nil {
+				return "", err
+			}
+			arr := gjson.Get(helper.BytesToString(data), "data.data").Array()
+			pic := arr[rand.Intn(len(arr))]
+			return pic.String(), nil
+		}, web.GetData, time.Minute)
+	if err != nil {
+		panic(err)
+	}
 	control.Register("coser", &ctrl.Options[*zero.Ctx]{
 		DisableOnDefault: false,
 		Brief:            "三次元coser",
 		Help:             "- coser",
 	}).ApplySingle(ctxext.DefaultSingle).OnFullMatch("coser").SetBlock(true).Limit(ctxext.LimitByGroup).
 		Handle(func(ctx *zero.Ctx) {
-			ctx.SendChain(message.Text("少女祈祷中......"))
-			data, err := web.RequestDataWith(web.NewDefaultClient(), coserURL, "GET", "", ua)
+			pic, err := p.Roll("coser")
 			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.Send(m).ID(); id == 0 {
+			if id := ctx.Send(message.Message{ctxext.FakeSenderForwardNode(ctx, message.Image(pic))}).ID(); id == 0 {
 				ctx.SendChain(message.Text("ERROR: 可能被风控或下载图片用时过长，请耐心等待"))
 			}
 		})