package slacknotifier import ( "context" "fmt" "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)) notifier := &Notifier{ channel: channel, client: client, } for _, o := range options { o(notifier) } return notifier } func (n *Notifier) Notify(format string, args ...interface{}) error { return n.NotifyTo(n.channel, format, args...) } func (n *Notifier) NotifyTo(channel, format string, args ...interface{}) error { 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] } logrus.Infof(format, nonSlackArgs...) _, _, err := n.client.PostMessageContext(context.Background(), channel, slack.MsgOptionText(fmt.Sprintf(format, nonSlackArgs...), true), slack.MsgOptionAttachments(slackAttachments...)) if err != nil { logrus.WithError(err).Errorf("slack error: %s", err.Error()) } return err } /* 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") } } */