Northern_Lights/ZeroBot-Plugin
1
0
Fork 0
mirror of https://github.com/FloatTech/ZeroBot-Plugin.git synced 2025-12-20 06:20:08 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update: drawlots (#613)

Browse Source
This commit is contained in:
苜蓿紫 2023-03-09 22:14:26 +08:00 committed by GitHub
parent 90efebf02f
commit d8cb5206e8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
plugin/drawlots/main.go
View File

@ -71,7 +71,7 @@ func init() {
lotsType := ctx.State["regex_matched"].([]string)[1] lotsType := ctx.State["regex_matched"].([]string)[1]
fileInfo, ok := lotsList[lotsType] fileInfo, ok := lotsList[lotsType]
if !ok { if !ok {
ctx.SendChain(message.Text("签名[", lotsType, "]不存在")) ctx.Send(message.ReplyWithMessage(ctx.Event.MessageID, message.Text("才...才没有", lotsType, "签这种东西啦")))
return return
} }
if fileInfo.lotsType == "folder" { if fileInfo.lotsType == "folder" {
@ -101,11 +101,11 @@ func init() {
lotsName := strings.TrimSpace(ctx.State["args"].(string)) lotsName := strings.TrimSpace(ctx.State["args"].(string))
fileInfo, ok := lotsList[lotsName] fileInfo, ok := lotsList[lotsName]
if !ok { if !ok {
ctx.Send(message.ReplyWithMessage(id, message.Text("签名[", lotsName, "]不存在"))) ctx.Send(message.ReplyWithMessage(id, message.Text("才...才没有", lotsName, "签这种东西啦")))
return return
} }
if fileInfo.lotsType == "folder" { if fileInfo.lotsType == "folder" {
ctx.Send(message.ReplyWithMessage(id, message.Text("仅支持查看gif抽签"))) ctx.Send(message.ReplyWithMessage(id, message.Text("只能查看gif签哦~")))
return return
} }
ctx.Send(message.ReplyWithMessage(id, message.Image("file:///"+datapath+lotsName+"."+fileInfo.lotsType))) ctx.Send(message.ReplyWithMessage(id, message.Image("file:///"+datapath+lotsName+"."+fileInfo.lotsType)))
@ -114,7 +114,7 @@ func init() {
id := ctx.Event.MessageID id := ctx.Event.MessageID
lotsName := strings.TrimSpace(ctx.State["args"].(string)) lotsName := strings.TrimSpace(ctx.State["args"].(string))
if lotsName == "" { if lotsName == "" {
ctx.Send(message.ReplyWithMessage(id, message.Text("请使用正确的指令形式"))) ctx.Send(message.ReplyWithMessage(id, message.Text("请使用正确的指令形式哦~")))
return return
} }
picURL := ctx.State["image_url"].([]string)[0] picURL := ctx.State["image_url"].([]string)[0]
@ -138,18 +138,18 @@ func init() {
lotsType: "gif", lotsType: "gif",
quantity: len(im.Image), quantity: len(im.Image),
} }
ctx.Send(message.ReplyWithMessage(id, message.Text("成功"))) ctx.Send(message.ReplyWithMessage(id, message.Text("成功！")))
}) })
en.OnPrefix("删签", zero.SuperUserPermission).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) { en.OnPrefix("删签", zero.SuperUserPermission).SetBlock(true).Limit(ctxext.LimitByUser).Handle(func(ctx *zero.Ctx) {
id := ctx.Event.MessageID id := ctx.Event.MessageID
lotsName := strings.TrimSpace(ctx.State["args"].(string)) lotsName := strings.TrimSpace(ctx.State["args"].(string))
fileInfo, ok := lotsList[lotsName] fileInfo, ok := lotsList[lotsName]
if !ok { if !ok {
ctx.Send(message.ReplyWithMessage(id, message.Text("签名[", lotsName, "]不存在"))) ctx.Send(message.ReplyWithMessage(id, message.Text("才...才没有", lotsName, "签这种东西啦")))
return return
} }
if fileInfo.lotsType == "folder" { if fileInfo.lotsType == "folder" {
ctx.Send(message.ReplyWithMessage(id, message.Text("图包请手动移除(保护图源误删),谢谢"))) ctx.Send(message.ReplyWithMessage(id, message.Text("为了防止误删图源，图包请手动移除哦~")))
return return
} }
err := os.Remove(datapath + lotsName + "." + fileInfo.lotsType) err := os.Remove(datapath + lotsName + "." + fileInfo.lotsType)
@ -158,7 +158,7 @@ func init() {
return return
} }
delete(lotsList, lotsName) delete(lotsList, lotsName)
ctx.Send(message.ReplyWithMessage(id, message.Text("成功"))) ctx.Send(message.ReplyWithMessage(id, message.Text("成功！")))
}) })
} }
@ -169,7 +169,7 @@ func getList() (list map[string]info, err error) {
return return
} }
if len(files) == 0 { if len(files) == 0 {
err = errors.New("不存在任何抽签") err = errors.New("什么签也没有哦~")
return return
} }
for _, lots := range files { for _, lots := range files {
@ -214,13 +214,13 @@ func randFile(path string, indexMax int) (string, error) {
if drawFile.IsDir() { if drawFile.IsDir() {
indexMax-- indexMax--
if indexMax <= 0 { if indexMax <= 0 {
return "", errors.New("图包[" + path + "]存在太多非图片文件,请清理") return "", errors.New("图包[" + path + "]存在太多非图片文件,请清理~")
} }
return randFile(path, indexMax) return randFile(path, indexMax)
} }
return picPath + "/" + drawFile.Name(), err return picPath + "/" + drawFile.Name(), err
} }
return "", errors.New("图包[" + path + "]不存在签内容") return "", errors.New("图包[" + path + "]不存在签内容！")
} }
func randGif(gifName string) (image.Image, error) { func randGif(gifName string) (image.Image, error) {