mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 13:59:39 +08:00
277 lines
7.6 KiB
Go
277 lines
7.6 KiB
Go
// Package drawlots 多功能抽签插件
|
|
package drawlots
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"image"
|
|
"image/draw"
|
|
"image/gif"
|
|
"io"
|
|
"math/rand"
|
|
"os"
|
|
"strconv"
|
|
"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"
|
|
"github.com/FloatTech/zbputils/ctxext"
|
|
"github.com/FloatTech/zbputils/img/text"
|
|
"github.com/fumiama/jieba/util/helper"
|
|
"github.com/sirupsen/logrus"
|
|
zero "github.com/wdvxdr1123/ZeroBot"
|
|
"github.com/wdvxdr1123/ZeroBot/message"
|
|
)
|
|
|
|
type info struct {
|
|
lotsType string // 文件后缀
|
|
quantity int // 签数
|
|
}
|
|
|
|
var (
|
|
lotsList = func() map[string]info {
|
|
lotsList, err := getList()
|
|
if err != nil {
|
|
logrus.Infoln("[drawlots]加载失败:", err)
|
|
} else {
|
|
logrus.Infoln("[drawlots]加载", len(lotsList), "个抽签")
|
|
}
|
|
return lotsList
|
|
}()
|
|
en = control.AutoRegister(&ctrl.Options[*zero.Ctx]{
|
|
DisableOnDefault: false,
|
|
Brief: "多功能抽签",
|
|
Help: "支持图包文件夹和gif抽签\n" +
|
|
"-------------\n" +
|
|
"- (刷新)抽签列表\n- 抽[签名]签\n- 看[gif签名]签\n- 加[签名]签[gif图片]\n- 删[gif签名]签",
|
|
PrivateDataFolder: "drawlots",
|
|
}).ApplySingle(ctxext.DefaultSingle)
|
|
datapath = file.BOTPATH + "/" + en.DataFolder()
|
|
)
|
|
|
|
func init() {
|
|
en.OnFullMatchGroup([]string{"抽签列表", "刷新抽签列表"}).SetBlock(true).Handle(func(ctx *zero.Ctx) {
|
|
var err error
|
|
lotsList, err = getList() // 刷新列表
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
messageText := &strings.Builder{}
|
|
messageText.WriteString(" 签 名 [ 类 型 ]----签数\n")
|
|
messageText.WriteString("———————————\n")
|
|
for name, fileInfo := range lotsList {
|
|
messageText.WriteString(name + "[" + fileInfo.lotsType + "]----" + strconv.Itoa(fileInfo.quantity) + "\n")
|
|
messageText.WriteString("----------\n")
|
|
}
|
|
textPic, err := text.RenderToBase64(messageText.String(), text.BoldFontFile, 400, 50)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
ctx.SendChain(message.Image("base64://" + helper.BytesToString(textPic)))
|
|
})
|
|
en.OnRegex(`^抽(.+)签$`).SetBlock(true).Handle(func(ctx *zero.Ctx) {
|
|
lotsType := ctx.State["regex_matched"].([]string)[1]
|
|
fileInfo, ok := lotsList[lotsType]
|
|
if !ok {
|
|
ctx.Send(message.ReplyWithMessage(ctx.Event.MessageID, message.Text("才...才没有", lotsType, "签这种东西啦")))
|
|
return
|
|
}
|
|
if fileInfo.lotsType == "folder" {
|
|
picPath, err := randFile(lotsType, 3)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
ctx.SendChain(message.Reply(ctx.Event.MessageID), message.Image("file:///"+picPath))
|
|
return
|
|
}
|
|
lotsImg, err := randGif(lotsType + "." + fileInfo.lotsType)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
// 生成图片
|
|
data, err := imgfactory.ToBytes(lotsImg)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
ctx.SendChain(message.Reply(ctx.Event.MessageID), message.ImageBytes(data))
|
|
})
|
|
en.OnRegex(`^看(.+)签$`, zero.UserOrGrpAdmin).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
|
|
id := ctx.Event.MessageID
|
|
lotsName := ctx.State["regex_matched"].([]string)[1]
|
|
fileInfo, ok := lotsList[lotsName]
|
|
if !ok {
|
|
ctx.Send(message.ReplyWithMessage(id, message.Text("才...才没有", lotsName, "签这种东西啦")))
|
|
return
|
|
}
|
|
if fileInfo.lotsType == "folder" {
|
|
ctx.Send(message.ReplyWithMessage(id, message.Text("只能查看gif签哦~")))
|
|
return
|
|
}
|
|
ctx.Send(message.ReplyWithMessage(id, message.Image("file:///"+datapath+lotsName+"."+fileInfo.lotsType)))
|
|
})
|
|
en.OnRegex(`^加(.+)签.*`, zero.SuperUserPermission, zero.MustProvidePicture).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
|
|
id := ctx.Event.MessageID
|
|
lotsName := ctx.State["regex_matched"].([]string)[1]
|
|
if lotsName == "" {
|
|
ctx.Send(message.ReplyWithMessage(id, message.Text("请使用正确的指令形式哦~")))
|
|
return
|
|
}
|
|
picURL := ctx.State["image_url"].([]string)[0]
|
|
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
|
|
}
|
|
fileName := datapath + "/" + lotsName + ".gif"
|
|
err = file.DownloadTo(picURL, fileName)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
lotsList[lotsName] = info{
|
|
lotsType: "gif",
|
|
quantity: len(im.Image),
|
|
}
|
|
ctx.Send(message.ReplyWithMessage(id, message.Text("成功！")))
|
|
})
|
|
en.OnRegex(`^删(.+)签$`, zero.SuperUserPermission).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
|
|
id := ctx.Event.MessageID
|
|
lotsName := ctx.State["regex_matched"].([]string)[1]
|
|
fileInfo, ok := lotsList[lotsName]
|
|
if !ok {
|
|
ctx.Send(message.ReplyWithMessage(id, message.Text("才...才没有", lotsName, "签这种东西啦")))
|
|
return
|
|
}
|
|
if fileInfo.lotsType == "folder" {
|
|
ctx.Send(message.ReplyWithMessage(id, message.Text("为了防止误删图源，图包请手动移除哦~")))
|
|
return
|
|
}
|
|
err := os.Remove(datapath + lotsName + "." + fileInfo.lotsType)
|
|
if err != nil {
|
|
ctx.SendChain(message.Text("ERROR: ", err))
|
|
return
|
|
}
|
|
delete(lotsList, lotsName)
|
|
ctx.Send(message.ReplyWithMessage(id, message.Text("成功！")))
|
|
})
|
|
}
|
|
|
|
func getList() (list map[string]info, err error) {
|
|
list = make(map[string]info, 100)
|
|
files, err := os.ReadDir(datapath)
|
|
if err != nil {
|
|
return
|
|
}
|
|
if len(files) == 0 {
|
|
err = errors.New("什么签也没有哦~")
|
|
return
|
|
}
|
|
for _, lots := range files {
|
|
if lots.IsDir() {
|
|
files, _ := os.ReadDir(datapath + "/" + lots.Name())
|
|
list[lots.Name()] = info{
|
|
lotsType: "folder",
|
|
quantity: len(files),
|
|
}
|
|
continue
|
|
}
|
|
before, after, ok := strings.Cut(lots.Name(), ".")
|
|
if !ok || before == "" {
|
|
continue
|
|
}
|
|
file, err := os.Open(datapath + "/" + lots.Name())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
im, err := gif.DecodeAll(file)
|
|
_ = file.Close()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
list[before] = info{
|
|
lotsType: after,
|
|
quantity: len(im.Image),
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func randFile(path string, indexMax int) (string, error) {
|
|
picPath := datapath + path
|
|
files, err := os.ReadDir(picPath)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if len(files) > 0 {
|
|
drawFile := files[rand.Intn(len(files))]
|
|
// 如果是文件夹就递归
|
|
if drawFile.IsDir() {
|
|
indexMax--
|
|
if indexMax <= 0 {
|
|
return "", errors.New("图包[" + path + "]存在太多非图片文件,请清理~")
|
|
}
|
|
return randFile(path, indexMax)
|
|
}
|
|
return picPath + "/" + drawFile.Name(), err
|
|
}
|
|
return "", errors.New("图包[" + path + "]不存在签内容！")
|
|
}
|
|
|
|
func randGif(gifName string) (image.Image, error) {
|
|
name := datapath + gifName
|
|
file, err := os.Open(name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
im, err := gif.DecodeAll(file)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
_, err = file.Seek(0, io.SeekStart)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
config, err := gif.DecodeConfig(file)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
_ = file.Close()
|
|
// https://zhuanlan.zhihu.com/p/27718135
|
|
rect := image.Rect(0, 0, config.Width, config.Height)
|
|
if rect.Min == rect.Max {
|
|
var max image.Point
|
|
for _, frame := range im.Image {
|
|
maxF := frame.Bounds().Max
|
|
if max.X < maxF.X {
|
|
max.X = maxF.X
|
|
}
|
|
if max.Y < maxF.Y {
|
|
max.Y = maxF.Y
|
|
}
|
|
}
|
|
rect.Max = max
|
|
}
|
|
img := image.NewRGBA(rect)
|
|
b := rand.Intn(len(im.Image)) + 1
|
|
a := 0
|
|
if b > 8 {
|
|
a = b - 8
|
|
}
|
|
for _, srcimg := range im.Image[a:b] {
|
|
draw.Draw(img, srcimg.Bounds(), srcimg, srcimg.Rect.Min, draw.Src)
|
|
}
|
|
return img, err
|
|
}
|