Skip to content

Commit 01e6fd2

Browse files
committed
Added more methods to Updates
1 parent 2df616c commit 01e6fd2

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

types.go

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,6 +99,30 @@ func (u *Update) SentFrom() *User {
9999
}
100100
}
101101

102+
func (u *Update) CallbackData() string {
103+
if u.CallbackQuery != nil {
104+
return u.CallbackQuery.Data
105+
}
106+
return ""
107+
}
108+
109+
func (u *Update) FromChat() *Chat {
110+
switch {
111+
case u.Message != nil:
112+
return u.Message.Chat
113+
case u.EditedMessage != nil:
114+
return u.EditedMessage.Chat
115+
case u.ChannelPost != nil:
116+
return u.ChannelPost.Chat
117+
case u.EditedChannelPost != nil:
118+
return u.EditedChannelPost.Chat
119+
case u.CallbackQuery != nil:
120+
return u.CallbackQuery.Message.Chat
121+
default:
122+
return nil
123+
}
124+
}
125+
102126
// UpdatesChannel is the channel for getting updates.
103127
type UpdatesChannel <-chan Update
104128

0 commit comments

Comments
 (0)