fumiama/ReiBot
1
0
Fork 0
mirror of https://github.com/fumiama/ReiBot.git synced 2026-06-09 12:30:47 +08:00
Code Activity

fix MustProvidePhoto

Browse Source
This commit is contained in:
源文雨
2022-10-04 18:26:47 +08:00
parent 63a0752c61
commit 6bc5f89ed3
1 changed files with 24 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
rules.go
View File

@@ -367,29 +367,31 @@ func IsPhoto(ctx *Ctx) bool {
} }
// MustProvidePhoto 消息不存在图片阻塞120秒至有图片，超时返回 false // MustProvidePhoto 消息不存在图片阻塞120秒至有图片，超时返回 false
func MustProvidePhoto(ctx *Ctx, needphohint, failhint string) bool { func MustProvidePhoto(ctx *Ctx, needphohint, failhint string) Rule {
msg, ok := ctx.Value.(*tgba.Message) return func(ctx *Ctx) bool {
if ok && len(msg.Photo) > 0 { // 确保无空 msg, ok := ctx.Value.(*tgba.Message)
ctx.State["photos"] = msg.Photo if ok && len(msg.Photo) > 0 { // 确保无空
return true ctx.State["photos"] = msg.Photo
} return true
// 没有图片就索取 }
if needphohint != "" { // 没有图片就索取
_, err := ctx.Caller.Send(tgba.NewMessage(msg.Chat.ID, needphohint)) if needphohint != "" {
if err != nil { _, err := ctx.Caller.Send(tgba.NewMessage(msg.Chat.ID, needphohint))
if err != nil {
return false
}
}
next := NewFutureEvent("Message", 999, false, ctx.CheckSession(), IsPhoto).Next()
select {
case <-time.After(time.Second * 120):
if failhint != "" {
_, _ = ctx.Caller.Send(tgba.NewMessage(msg.Chat.ID, failhint))
}
return false return false
case newCtx := <-next:
ctx.State["photos"] = newCtx.State["photos"]
ctx.Event = newCtx.Event
return true
} }
} }
next := NewFutureEvent("message", 999, false, ctx.CheckSession(), IsPhoto).Next()
select {
case <-time.After(time.Second * 120):
if failhint != "" {
_, _ = ctx.Caller.Send(tgba.NewMessage(msg.Chat.ID, failhint))
}
return false
case newCtx := <-next:
ctx.State["photos"] = newCtx.State["photos"]
ctx.Event = newCtx.Event
return true
}
} }