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

move methods to telegramnotifier

Browse Source
This commit is contained in:
c9s 2022-01-14 11:37:23 +08:00
parent 832faf91f8
commit dd93ee4fd3
2 changed files with 51 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pkg/bbgo/environment.go
View File

@ -623,7 +623,7 @@ func (environ *Environment) ConfigureNotificationSystem(userConfig *Config) erro
opts = append(opts, telegramnotifier.UseBroadcast()) opts = append(opts, telegramnotifier.UseBroadcast())
} }
var notifier = telegramnotifier.New(interaction, opts...) var notifier = telegramnotifier.New(opts...)
environ.Notifiability.AddNotifier(notifier) environ.Notifiability.AddNotifier(notifier)
} }

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

@ -2,13 +2,26 @@ package telegramnotifier
import ( import (
"fmt" "fmt"
"strconv"
"time"
"gopkg.in/tucnak/telebot.v2"
"github.com/c9s/bbgo/pkg/types" "github.com/c9s/bbgo/pkg/types"
) )
type Notifier struct { type Notifier struct {
interaction *Interaction Bot *telebot.Bot
broadcast bool
// Chats stores the Chat objects for broadcasting
Chats map[int64]time.Time `json:"chats"`
// Owner
// when owner and owner chat is not nil, notification will send to the owner chat
Owner *telebot.User `json:"owner"`
OwnerChat *telebot.Chat `json:"chat"`
broadcast bool
} }
type Option func(notifier *Notifier) type Option func(notifier *Notifier)
@ -20,11 +33,8 @@ func UseBroadcast() Option {
} }
// New // New
// TODO: register interaction with channel, so that we can route message to the specific telegram bot func New(options ...Option) *Notifier {
func New(interaction *Interaction, options ...Option) *Notifier { notifier := &Notifier{}
notifier := &Notifier{
interaction: interaction,
}
for _, o := range options { for _, o := range options {
o(notifier) o(notifier)
@ -85,14 +95,42 @@ func (n *Notifier) NotifyTo(channel string, obj interface{}, args ...interface{}
} }
if n.broadcast { if n.broadcast {
n.interaction.Broadcast(message) n.Broadcast(message)
for _, text := range texts { for _, text := range texts {
n.interaction.Broadcast(text) n.Broadcast(text)
} }
} else { } else if n.OwnerChat != nil {
n.interaction.SendToOwner(message) n.SendToOwner(message)
for _, text := range texts { for _, text := range texts {
n.interaction.SendToOwner(text) n.SendToOwner(text)
}
}
}
func (n *Notifier) AddSubscriber(m *telebot.Message) {
if n.Chats == nil {
n.Chats = make(map[int64]time.Time)
}
n.Chats[m.Chat.ID] = m.Time()
}
func (n *Notifier) SendToOwner(message string) {
if _, err := n.Bot.Send(n.OwnerChat, message); err != nil {
log.WithError(err).Error("telegram send error")
}
}
func (n *Notifier) Broadcast(message string) {
for chatID := range n.Chats {
chat, err := n.Bot.ChatByID(strconv.FormatInt(chatID, 10))
if err != nil {
log.WithError(err).Error("can not get chat by ID")
continue
}
if _, err := n.Bot.Send(chat, message); err != nil {
log.WithError(err).Error("failed to send message")
} }
} }
} }