middleware: change Restrict middleware chats type to int64

pull/428/head
Demian 3 years ago
parent 8f725d434e
commit 802d480e10

@ -3,7 +3,7 @@ package middleware
import tele "gopkg.in/tucnak/telebot.v3" import tele "gopkg.in/tucnak/telebot.v3"
type RestrictConfig struct { type RestrictConfig struct {
Chats []tele.Recipient Chats []int64
In, Out tele.HandlerFunc In, Out tele.HandlerFunc
} }
@ -17,7 +17,7 @@ func Restrict(v RestrictConfig) tele.MiddlewareFunc {
} }
return func(c tele.Context) error { return func(c tele.Context) error {
for _, chat := range v.Chats { for _, chat := range v.Chats {
if chat.Recipient() == c.Sender().Recipient() { if chat == c.Sender().ID {
return v.In(c) return v.In(c)
} }
} }
@ -26,7 +26,7 @@ func Restrict(v RestrictConfig) tele.MiddlewareFunc {
} }
} }
func Whitelist(chats ...tele.Recipient) tele.MiddlewareFunc { func Whitelist(chats ...int64) tele.MiddlewareFunc {
return func(next tele.HandlerFunc) tele.HandlerFunc { return func(next tele.HandlerFunc) tele.HandlerFunc {
return Restrict(RestrictConfig{ return Restrict(RestrictConfig{
Chats: chats, Chats: chats,
@ -36,7 +36,7 @@ func Whitelist(chats ...tele.Recipient) tele.MiddlewareFunc {
} }
} }
func Blacklist(chats ...tele.Recipient) tele.MiddlewareFunc { func Blacklist(chats ...int64) tele.MiddlewareFunc {
return func(next tele.HandlerFunc) tele.HandlerFunc { return func(next tele.HandlerFunc) tele.HandlerFunc {
return Restrict(RestrictConfig{ return Restrict(RestrictConfig{
Chats: chats, Chats: chats,

Loading…
Cancel
Save