mirror of
https://github.com/tucnak/telebot
synced 2024-11-15 06:13:01 +00:00
29 lines
531 B
Go
29 lines
531 B
Go
package layout
|
|
|
|
import (
|
|
tele "gopkg.in/tucnak/telebot.v3"
|
|
)
|
|
|
|
type LocaleFunc func(tele.Recipient) string
|
|
|
|
func (lt *Layout) Middleware(defaultLocale string, localeFunc ...LocaleFunc) tele.MiddlewareFunc {
|
|
var f LocaleFunc
|
|
if len(localeFunc) > 0 {
|
|
f = localeFunc[0]
|
|
}
|
|
|
|
return func(next tele.HandlerFunc) tele.HandlerFunc {
|
|
return func(c tele.Context) error {
|
|
locale := defaultLocale
|
|
if f != nil {
|
|
if l := f(c.Sender()); l != "" {
|
|
locale = l
|
|
}
|
|
}
|
|
|
|
lt.SetLocale(c, locale)
|
|
return next(c)
|
|
}
|
|
}
|
|
}
|