chore: restore useful image pool

FloatTech · Oct 13, 2024 · 84964a8 · 84964a8
1 parent 2b2983e
commit 84964a8
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/img/pool/sender.go b/img/pool/sender.go
@@ -0,0 +1,31 @@
+package pool
+
+import (
+	"errors"
+
+	"github.com/wdvxdr1123/ZeroBot/message"
+
+	"github.com/FloatTech/floatbox/file"
+
+	"github.com/FloatTech/zbputils/ctxext"
+)
+
+// SendImageFromPool ...
+func SendImageFromPool(imgname, imgpath string, genimg func() error, send ctxext.NoCtxSendMsg, get ctxext.NoCtxGetMsg) error {
+	if file.IsNotExist(imgpath) {
+		err := genimg()
+		if err != nil {
+			return err
+		}
+	}
+	// 发送图片
+	img := message.Image(file.BOTPATH + "/" + imgpath)
+	id := send(message.Message{img})
+	if id == 0 {
+		id = send(message.Message{img.Add("cache", "0")})
+		if id == 0 {
+			return errors.New("图片发送失败, 可能被风控了~")
+		}
+	}
+	return nil
+}