bbgo_origin/pkg/notifier/telegramnotifier/telegram.go

84 lines
1.6 KiB
Go
Raw Normal View History

2020-12-05 06:20:27 +00:00
package telegramnotifier
import (
"fmt"
2020-12-11 07:58:05 +00:00
"github.com/c9s/bbgo/pkg/types"
2020-12-05 06:20:27 +00:00
)
type Notifier struct {
interaction *Interaction
2020-12-05 06:20:27 +00:00
}
type NotifyOption func(notifier *Notifier)
// New
// TODO: register interaction with channel, so that we can route message to the specific telegram bot
func New(interaction *Interaction, options ...NotifyOption) *Notifier {
notifier := &Notifier{
interaction: interaction,
}
for _, o := range options {
o(notifier)
}
2020-12-05 06:20:27 +00:00
return notifier
}
func (n *Notifier) Notify(obj interface{}, args ...interface{}) {
n.NotifyTo("", obj, args...)
2020-12-05 06:20:27 +00:00
}
func filterPlaintextMessages(args []interface{}) (texts []string, pureArgs []interface{}) {
2020-12-11 07:58:05 +00:00
var textArgsOffset = -1
for idx, arg := range args {
switch a := arg.(type) {
case types.PlainText:
texts = append(texts, a.PlainText())
textArgsOffset = idx
case types.Stringer:
texts = append(texts, a.String())
textArgsOffset = idx
2020-12-11 07:58:05 +00:00
}
}
pureArgs = args
2020-12-11 07:58:05 +00:00
if textArgsOffset > -1 {
pureArgs = args[:textArgsOffset]
2020-12-05 06:20:27 +00:00
}
return texts, pureArgs
}
2020-12-05 06:20:27 +00:00
func (n *Notifier) NotifyTo(channel string, obj interface{}, args ...interface{}) {
var texts, pureArgs = filterPlaintextMessages(args)
var message string
switch a := obj.(type) {
case string:
log.Infof(a, pureArgs...)
message = fmt.Sprintf(a, pureArgs...)
2020-12-11 07:58:05 +00:00
case types.Stringer:
message = a.String()
case types.PlainText:
message = a.PlainText()
default:
log.Errorf("unsupported notification format: %T %+v", a, a)
}
n.interaction.SendToOwner(message)
2020-12-11 07:58:05 +00:00
for _, text := range texts {
n.interaction.SendToOwner(text)
}
2020-12-05 06:20:27 +00:00
}