forked from mystiq/dex
111 lines
2.6 KiB
Go
111 lines
2.6 KiB
Go
package email
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"expvar"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
const (
|
|
FakeEmailerType = "fake"
|
|
)
|
|
|
|
var (
|
|
counterEmailSendErr = expvar.NewInt("email.send.err")
|
|
ErrorNoTemplate = errors.New("No HTML or Text template found for template name.")
|
|
)
|
|
|
|
func init() {
|
|
RegisterEmailerConfigType(FakeEmailerType, func() EmailerConfig { return &FakeEmailerConfig{} })
|
|
}
|
|
|
|
// Emailer is an object that sends emails.
|
|
type Emailer interface {
|
|
// SendMail queues an email to be sent to 1 or more recipients.
|
|
// At least one of "text" or "html" must not be blank. If text is blank, but
|
|
// html is not, then an html-only email should be sent and vice-versal.
|
|
SendMail(subject, text, html string, to ...string) error
|
|
}
|
|
|
|
//go:generate genconfig -o config.go email Emailer
|
|
type EmailerConfig interface {
|
|
EmailerID() string
|
|
EmailerType() string
|
|
|
|
// Because emailers can either be configured through command line flags or through
|
|
// JSON configs, we need a way to set the fromAddr when initializing the emailer.
|
|
//
|
|
// Values passed in the JSON config should override this value. Supplying neither
|
|
// should result in an error.
|
|
Emailer(fromAddress string) (Emailer, error)
|
|
}
|
|
|
|
func newEmailerConfigFromReader(r io.Reader) (EmailerConfig, error) {
|
|
var m map[string]interface{}
|
|
if err := json.NewDecoder(r).Decode(&m); err != nil {
|
|
return nil, err
|
|
}
|
|
cfg, err := newEmailerConfigFromMap(m)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return cfg, nil
|
|
}
|
|
|
|
func NewEmailerConfigFromFile(loc string) (EmailerConfig, error) {
|
|
cf, err := os.Open(loc)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer cf.Close()
|
|
|
|
cfg, err := newEmailerConfigFromReader(cf)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return cfg, nil
|
|
}
|
|
|
|
type FakeEmailerConfig struct {
|
|
FromAddr string `json:"from"`
|
|
}
|
|
|
|
func (cfg FakeEmailerConfig) EmailerType() string {
|
|
return FakeEmailerType
|
|
}
|
|
|
|
func (cfg FakeEmailerConfig) EmailerID() string {
|
|
return FakeEmailerType
|
|
}
|
|
|
|
func (cfg FakeEmailerConfig) Emailer(fromAddr string) (Emailer, error) {
|
|
from := cfg.FromAddr
|
|
if from == "" {
|
|
from = fromAddr
|
|
}
|
|
if from == "" {
|
|
// Since the emailer just prints to stdout, the actual value doesn't matter.
|
|
from = "noreply@example.com"
|
|
}
|
|
|
|
return FakeEmailer{from}, nil
|
|
}
|
|
|
|
// FakeEmailer is an Emailer that writes emails to stdout. Should only be used in development.
|
|
type FakeEmailer struct {
|
|
from string
|
|
}
|
|
|
|
func (f FakeEmailer) SendMail(subject, text, html string, to ...string) error {
|
|
fmt.Printf("From: %v\n", f.from)
|
|
fmt.Printf("Subject: %v\n", subject)
|
|
fmt.Printf("To: %v\n", strings.Join(to, ","))
|
|
fmt.Printf("Body(text): %v\n", text)
|
|
fmt.Printf("Body(html): %v\n", html)
|
|
return nil
|
|
}
|