mirror of
https://github.com/c9s/bbgo.git
synced 2024-11-22 23:05:15 +00:00
120 lines
2.6 KiB
Go
120 lines
2.6 KiB
Go
package slacknotifier
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/slack-go/slack"
|
|
)
|
|
|
|
type SlackAttachmentCreator interface {
|
|
SlackAttachment() slack.Attachment
|
|
}
|
|
|
|
type Notifier struct {
|
|
client *slack.Client
|
|
channel string
|
|
}
|
|
|
|
type NotifyOption func(notifier *Notifier)
|
|
|
|
func New(token, channel string, options ...NotifyOption) *Notifier {
|
|
// var client = slack.New(token, slack.OptionDebug(true))
|
|
var client = slack.New(token)
|
|
|
|
notifier := &Notifier{
|
|
channel: channel,
|
|
client: client,
|
|
}
|
|
|
|
for _, o := range options {
|
|
o(notifier)
|
|
}
|
|
|
|
return notifier
|
|
}
|
|
|
|
func (n *Notifier) Notify(format string, args ...interface{}) {
|
|
n.NotifyTo(n.channel, format, args...)
|
|
}
|
|
|
|
func (n *Notifier) NotifyTo(channel, format string, args ...interface{}) {
|
|
if len(channel) == 0 {
|
|
channel = n.channel
|
|
}
|
|
|
|
var slackAttachments []slack.Attachment
|
|
var slackArgsOffset = -1
|
|
|
|
for idx, arg := range args {
|
|
switch a := arg.(type) {
|
|
|
|
// concrete type assert first
|
|
case slack.Attachment:
|
|
if slackArgsOffset == -1 {
|
|
slackArgsOffset = idx
|
|
}
|
|
|
|
slackAttachments = append(slackAttachments, a)
|
|
|
|
case SlackAttachmentCreator:
|
|
if slackArgsOffset == -1 {
|
|
slackArgsOffset = idx
|
|
}
|
|
|
|
slackAttachments = append(slackAttachments, a.SlackAttachment())
|
|
|
|
}
|
|
}
|
|
|
|
var nonSlackArgs = args
|
|
if slackArgsOffset > -1 {
|
|
nonSlackArgs = args[:slackArgsOffset]
|
|
}
|
|
|
|
log.Infof(format, nonSlackArgs...)
|
|
|
|
_, _, err := n.client.PostMessageContext(context.Background(), channel,
|
|
slack.MsgOptionText(fmt.Sprintf(format, nonSlackArgs...), true),
|
|
slack.MsgOptionAttachments(slackAttachments...))
|
|
if err != nil {
|
|
log.WithError(err).
|
|
WithField("channel", channel).
|
|
Errorf("slack error: %s", err.Error())
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
/*
|
|
func (n *Notifier) NotifyTrade(trade *types.Trade) {
|
|
_, _, err := n.client.PostMessageContext(context.Background(), n.TradeChannel,
|
|
slack.MsgOptionText(util.Render(`:handshake: {{ .Symbol }} {{ .Side }} Trade Execution @ {{ .Price }}`, trade), true),
|
|
slack.MsgOptionAttachments(trade.SlackAttachment()))
|
|
|
|
if err != nil {
|
|
logrus.WithError(err).Error("slack send error")
|
|
}
|
|
}
|
|
*/
|
|
|
|
/*
|
|
func (n *Notifier) NotifyPnL(report *pnl.AverageCostPnlReport) {
|
|
attachment := report.SlackAttachment()
|
|
|
|
_, _, err := n.client.PostMessageContext(context.Background(), n.PnlChannel,
|
|
slack.MsgOptionText(util.Render(
|
|
`:heavy_dollar_sign: Here is your *{{ .symbol }}* PnL report collected since *{{ .startTime }}*`,
|
|
map[string]interface{}{
|
|
"symbol": report.Symbol,
|
|
"startTime": report.StartTime.Format(time.RFC822),
|
|
}), true),
|
|
slack.MsgOptionAttachments(attachment))
|
|
|
|
if err != nil {
|
|
logrus.WithError(err).Errorf("slack send error")
|
|
}
|
|
}
|
|
*/
|