diff --git a/bot.go b/bot.go index 5b0a49c..5142d73 100644 --- a/bot.go +++ b/bot.go @@ -11,12 +11,12 @@ import ( "git.gensokyo.cafe/kkyy/tgbot_misaka_5882f7/stats" ) -func isFromAdmin(upd tele.Update) bool { - if upd.Message == nil || upd.Message.Sender == nil { +func isFromAdmin(sender *tele.User) bool { + if sender == nil { return false } - _, ok := config.AdminUIDs[upd.Message.Sender.ID] + _, ok := config.AdminUIDs[sender.ID] return ok } @@ -39,8 +39,7 @@ func initBot() (*tele.Bot, error) { } func handleStartCmd(c tele.Context) error { - upd := c.Update() - if !isFromAdmin(upd) { + if !isFromAdmin(c.Sender()) { return c.Send("Hello, stranger :)") } @@ -48,8 +47,7 @@ func handleStartCmd(c tele.Context) error { } func handleTrafficCmd(c tele.Context) error { - upd := c.Update() - if !isFromAdmin(upd) { + if !isFromAdmin(c.Sender()) { return nil }