kkyy
/
tgbot_misaka_5882f7
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactor: use Sender() instead of Update() to get user info

This commit is contained in:
Yiyang Kang 2022-11-21 13:22:22 +08:00
parent 09f9b1ab68
commit a78802a7dd
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
bot.go
View File

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