修改加签逻辑 (#620)

* Update main.go

* Update main.go

* Update main.go

plugin/drawlots/main.go

@@ -2,8 +2,10 @@
 package drawlots
 
 import (
+	"bytes"
 	"errors"
 	"image"
+	"image/color"
 	"image/gif"
 	"math/rand"
 	"os"
@@ -11,6 +13,7 @@ import (
 	"strings"
 
 	"github.com/FloatTech/floatbox/file"
+	"github.com/FloatTech/floatbox/web"
 	"github.com/FloatTech/imgfactory"
 	ctrl "github.com/FloatTech/zbpctrl"
 	control "github.com/FloatTech/zbputils/control"
@@ -40,7 +43,9 @@ var (
 	en = control.Register("drawlots", &ctrl.Options[*zero.Ctx]{
 		DisableOnDefault: false,
 		Brief:            "多功能抽签",
-		Help:              "支持图包文件夹和gif抽签\n-------------\n- (刷新)抽签列表\n- 抽[签名]签\n- 看签[gif签名]\n- 加签[签名][gif图片]\n- 删签[gif签名]",
+		Help: "支持图包文件夹和gif抽签\n" +
+			"-------------\n" +
+			"- (刷新)抽签列表\n- 抽[签名]签\n- 看[gif签名]签\n- 加[签名]签[gif图片]\n- 删[gif签名]签",
 		PrivateDataFolder: "drawlots",
 	}).ApplySingle(ctxext.DefaultSingle)
 	datapath = file.BOTPATH + "/" + en.DataFolder()
@@ -96,9 +101,9 @@ func init() {
 		}
 		ctx.SendChain(message.Reply(ctx.Event.MessageID), message.ImageBytes(data))
 	})
-	en.OnPrefix("看签", zero.UserOrGrpAdmin).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
+	en.OnRegex(`^看(.+)签$`, zero.UserOrGrpAdmin).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
 		id := ctx.Event.MessageID
-		lotsName := strings.TrimSpace(ctx.State["args"].(string))
+		lotsName := ctx.State["regex_matched"].([]string)[1]
 		fileInfo, ok := lotsList[lotsName]
 		if !ok {
 			ctx.Send(message.ReplyWithMessage(id, message.Text("才...才没有", lotsName, "签这种东西啦")))
@@ -110,26 +115,25 @@ func init() {
 		}
 		ctx.Send(message.ReplyWithMessage(id, message.Image("file:///"+datapath+lotsName+"."+fileInfo.lotsType)))
 	})
-	en.OnPrefix("加签", zero.SuperUserPermission, zero.MustProvidePicture).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
+	en.OnRegex(`^加(.+)签.*`, zero.SuperUserPermission, zero.MustProvidePicture).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
 		id := ctx.Event.MessageID
-		lotsName := strings.TrimSpace(ctx.State["args"].(string))
+		lotsName := ctx.State["regex_matched"].([]string)[1]
 		if lotsName == "" {
 			ctx.Send(message.ReplyWithMessage(id, message.Text("请使用正确的指令形式哦~")))
 			return
 		}
 		picURL := ctx.State["image_url"].([]string)[0]
-		err := file.DownloadTo(picURL, datapath+"/"+lotsName+".gif")
+		gifdata, err := web.GetData(picURL)
+		if err != nil {
+			return
+		}
+		im, err := gif.DecodeAll(bytes.NewReader(gifdata))
 		if err != nil {
 			ctx.SendChain(message.Text("ERROR:", err))
 			return
 		}
-		file, err := os.Open(datapath + "/" + lotsName + ".gif")
+		fileName := datapath + "/" + lotsName + ".gif"
-		if err != nil {
+		err = file.DownloadTo(picURL, fileName)
-			ctx.SendChain(message.Text("ERROR:", err))
-			return
-		}
-		im, err := gif.DecodeAll(file)
-		_ = file.Close()
 		if err != nil {
 			ctx.SendChain(message.Text("ERROR:", err))
 			return
@@ -140,9 +144,9 @@ func init() {
 		}
 		ctx.Send(message.ReplyWithMessage(id, message.Text("成功！")))
 	})
-	en.OnPrefix("删签", zero.SuperUserPermission).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
+	en.OnRegex(`^删(.+)签$`, zero.SuperUserPermission).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
 		id := ctx.Event.MessageID
-		lotsName := strings.TrimSpace(ctx.State["args"].(string))
+		lotsName := ctx.State["regex_matched"].([]string)[1]
 		fileInfo, ok := lotsList[lotsName]
 		if !ok {
 			ctx.Send(message.ReplyWithMessage(id, message.Text("才...才没有", lotsName, "签这种东西啦")))
@@ -244,6 +248,6 @@ func randGif(gifName string) (image.Image, error) {
 	/*/
 	// 如果gif图片出现信息缺失请使用上面注释掉的代码，把下面注释了(上面代码部分图存在bug)
 	v := im.Image[rand.Intn(len(im.Image))]
-	return imgfactory.Size(v, v.Bounds().Max.X, v.Bounds().Max.Y).Image(), err
+	return imgfactory.NewFactoryBG(v.Rect.Max.X, v.Rect.Max.Y, color.NRGBA{0, 0, 0, 255}).InsertUp(v, 0, 0, 0, 0).Clone().Image(), err
 	// */
 }