119 lines
2.5 KiB
Go
119 lines
2.5 KiB
Go
package filters
|
|
|
|
import (
|
|
mcontext "github.com/abelith/gomax/context"
|
|
"github.com/abelith/gomax/core"
|
|
"github.com/abelith/gomax/state"
|
|
"github.com/max-messenger/max-bot-api-client-go/v2/model"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
type CallbackFilter core.Filter
|
|
|
|
func StartCommand() core.Filter {
|
|
return func(c *mcontext.Context) bool {
|
|
upd := c.Update()
|
|
if upd.UpdateType == model.UpdateBotStarted {
|
|
return true
|
|
}
|
|
|
|
return strings.HasPrefix(upd.GetCommand().Command, "/start")
|
|
}
|
|
}
|
|
|
|
func Command(cmd string) core.Filter {
|
|
return func(c *mcontext.Context) bool {
|
|
upd := c.Update()
|
|
return upd.UpdateType == model.UpdateMessageCreated && strings.HasPrefix(upd.GetCommand().Command, cmd)
|
|
}
|
|
}
|
|
|
|
func State(state state.State) core.Filter {
|
|
return func(c *mcontext.Context) bool {
|
|
return c.State().State() == state
|
|
}
|
|
}
|
|
|
|
func MessageText(s string) core.Filter {
|
|
return func(c *mcontext.Context) bool {
|
|
if msg := c.Update().Message; msg != nil {
|
|
return s == msg.Body.Text
|
|
}
|
|
return false
|
|
}
|
|
}
|
|
|
|
func MessageTextRe(re *regexp.Regexp) core.Filter {
|
|
return func(c *mcontext.Context) bool {
|
|
if msg := c.Update().Message; msg != nil {
|
|
s := strings.TrimSpace(msg.Body.Text)
|
|
return re.MatchString(s)
|
|
}
|
|
return false
|
|
}
|
|
}
|
|
|
|
func MessageGeo() core.Filter {
|
|
return func(c *mcontext.Context) bool {
|
|
if c.Update().Message == nil {
|
|
return false
|
|
}
|
|
|
|
for _, attachment := range c.Update().Message.Body.Attachments {
|
|
if attachment.Type == model.AttachLocation {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
}
|
|
|
|
func BotStopped() core.Filter {
|
|
return func(c *mcontext.Context) bool {
|
|
return c.Update().UpdateType == model.UpdateBotStopped
|
|
}
|
|
}
|
|
|
|
func Or(filters ...core.Filter) core.Filter {
|
|
return func(c *mcontext.Context) bool {
|
|
for _, filter := range filters {
|
|
if filter(c) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
}
|
|
|
|
func Message() core.Filter {
|
|
return func(c *mcontext.Context) bool {
|
|
return c.Update().UpdateType == model.UpdateMessageCreated && c.Update().Message != nil
|
|
}
|
|
}
|
|
|
|
func Callback() CallbackFilter {
|
|
return func(c *mcontext.Context) bool {
|
|
return c.Update().UpdateType == model.UpdateMessageCallback
|
|
}
|
|
}
|
|
|
|
func (cf CallbackFilter) StartsWith(prefix string) CallbackFilter {
|
|
return func(c *mcontext.Context) bool {
|
|
if !cf(c) {
|
|
return false
|
|
}
|
|
payload, ok := getCallbackPayload(c)
|
|
return ok && strings.HasPrefix(payload, prefix)
|
|
}
|
|
}
|
|
|
|
func getCallbackPayload(c *mcontext.Context) (string, bool) {
|
|
upd := c.Update()
|
|
if upd.Callback == nil {
|
|
return "", false
|
|
}
|
|
return upd.Callback.Payload, true
|
|
}
|