mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2025-12-19 22:00:11 +08:00
59 lines
1.3 KiB
Go
59 lines
1.3 KiB
Go
package context
|
||
|
||
import (
|
||
"strconv"
|
||
"time"
|
||
|
||
zero "github.com/wdvxdr1123/ZeroBot"
|
||
)
|
||
|
||
// HasPicture 消息含有图片返回 true
|
||
func HasPicture() zero.Rule {
|
||
return func(ctx *zero.Ctx) bool {
|
||
msg := ctx.Event.Message
|
||
url := []string{}
|
||
// 如果是回复信息则将信息替换成被回复的那条
|
||
if msg[0].Type == "reply" {
|
||
id, _ := strconv.Atoi(msg[0].Data["id"])
|
||
msg = ctx.GetMessage(int64(id)).Elements
|
||
}
|
||
// 遍历信息中所有图片
|
||
for _, elem := range msg {
|
||
if elem.Type == "image" {
|
||
url = append(url, elem.Data["url"])
|
||
}
|
||
}
|
||
// 如果有图片就返回true
|
||
if len(url) > 0 {
|
||
ctx.State["image_url"] = url
|
||
return true
|
||
}
|
||
return false
|
||
}
|
||
}
|
||
|
||
// MustHasPicture 消息不存在图片阻塞60秒至有图片，超时返回 false
|
||
func MustHasPicture() zero.Rule {
|
||
return func(ctx *zero.Ctx) bool {
|
||
if HasPicture()(ctx) {
|
||
return true
|
||
}
|
||
// 没有图片就索取
|
||
ctx.Send("请发送一张图片")
|
||
next := zero.NewFutureEvent("message", 999, false, zero.CheckUser(ctx.Event.UserID), HasPicture())
|
||
recv, cancel := next.Repeat()
|
||
select {
|
||
case e := <-recv:
|
||
cancel()
|
||
newCtx := &zero.Ctx{Event: e, State: zero.State{}}
|
||
if HasPicture()(newCtx) {
|
||
ctx.State["image_url"] = newCtx.State["image_url"]
|
||
return true
|
||
}
|
||
return false
|
||
case <-time.After(time.Second * 60):
|
||
return false
|
||
}
|
||
}
|
||
}
|