lychiyu/bbgo_origin
1
0
Fork 0
mirror of https://github.com/c9s/bbgo.git synced 2024-11-10 09:11:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: telegram bot send message to correct user

Browse Source
This commit is contained in:
David Chang 2020-12-06 12:11:27 +08:00
parent 7caf986d11
commit f9124aa907
1 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
pkg/notifier/telegramnotifier/telegram.go
View File

@ -33,21 +33,34 @@ func New(botToken, initToken string, options ...NotifyOption) *Notifier {
chatUser := &tb.User{}
bot.Handle("/help", func(m *tb.Message) {
helpMsg := `
help - print help message
init - initialize telegram bot with initToken. ex. /init my-token
hi - print welcome message to authorized user
`
bot.Send(m.Sender, helpMsg)
})
// init check initToken and then set sender id
bot.Handle("/init", func(m *tb.Message) {
if m.Text == initToken {
bot.Send(m.Sender, "Bot initialized")
log.Info("Receive message: ", m) //debug
if m.Payload == initToken {
chatUser = m.Sender
bot.Send(m.Sender, "Bot initialized")
} else {
bot.Send(m.Sender, "Error: bot intialize failed. Init token not match!")
}
})
bot.Handle("/bbgo", func(m *tb.Message) {
if m.Sender == chatUser {
bot.Send(chatUser, "bbgo!")
bot.Handle("/hi", func(m *tb.Message) {
if m.Sender.ID == chatUser.ID {
bot.Send(chatUser,
fmt.Sprintf("Welcome! user: %s, ID: %s"),
chatUser.Username,
chatUser.ID)
} else {
log.Warningf("Incorrect user tried to access bot! sender id: %s", m.Sender.Username)
log.Warningf("Incorrect user tried to access bot! sender id: %s", m.Sender)
}
})