7999d24f81
this commit updates our monkit dependency to the v3 version where it outputs in an influx style. this makes discovery much easier as many tools are built to look at it this way. graphite and rothko will suffer some due to no longer being a tree based on dots. hopefully time will exist to update rothko to index based on the new metric format. it adds an influx output for the statreceiver so that we can write to influxdb v1 or v2 directly. Change-Id: Iae9f9494a6d29cfbd1f932a5e71a891b490415ff
110 lines
2.1 KiB
Go
110 lines
2.1 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information
|
|
|
|
package post
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"io"
|
|
"net"
|
|
"net/mail"
|
|
"net/smtp"
|
|
|
|
"github.com/spacemonkeygo/monkit/v3"
|
|
"github.com/zeebo/errs"
|
|
)
|
|
|
|
// Address is alias of net/mail.Address
|
|
type Address = mail.Address
|
|
|
|
var mon = monkit.Package()
|
|
|
|
// SMTPSender is smtp sender
|
|
type SMTPSender struct {
|
|
ServerAddress string
|
|
|
|
From Address
|
|
Auth smtp.Auth
|
|
}
|
|
|
|
// FromAddress implements satellite/mail.SMTPSender
|
|
func (sender *SMTPSender) FromAddress() Address {
|
|
return sender.From
|
|
}
|
|
|
|
// SendEmail sends email message to the given recipient
|
|
func (sender *SMTPSender) SendEmail(ctx context.Context, msg *Message) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
client, err := smtp.Dial(sender.ServerAddress)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err = sender.communicate(ctx, client, msg); err != nil {
|
|
return errs.Combine(err, client.Close())
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// communicate sends mail via SMTP using provided client and message
|
|
func (sender *SMTPSender) communicate(ctx context.Context, client *smtp.Client, msg *Message) error {
|
|
// suppress error because address should be validated
|
|
// before creating SMTPSender
|
|
host, _, _ := net.SplitHostPort(sender.ServerAddress)
|
|
|
|
// send smtp hello or ehlo msg and establish connection over tls
|
|
err := client.StartTLS(&tls.Config{ServerName: host})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = client.Auth(sender.Auth)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = client.Mail(sender.From.Address)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// add recipients
|
|
for _, to := range msg.To {
|
|
err = client.Rcpt(to.Address)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
mess, err := msg.Bytes()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
data, err := client.Data()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = writeData(data, mess)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// send quit msg to stop gracefully
|
|
return client.Quit()
|
|
}
|
|
|
|
// writeData ensures that writer will be closed after data is written
|
|
func writeData(writer io.WriteCloser, data []byte) (err error) {
|
|
defer func() {
|
|
err = errs.Combine(err, writer.Close())
|
|
}()
|
|
|
|
_, err = writer.Write(data)
|
|
return
|
|
}
|