forked from mystiq/dex
90 lines
2.5 KiB
Go
90 lines
2.5 KiB
Go
package email
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
htmltemplate "html/template"
|
|
"text/template"
|
|
)
|
|
|
|
// NewTemplatizedEmailerFromGlobs creates a new TemplatizedEmailer, parsing the templates found in the given filepattern globs.
|
|
func NewTemplatizedEmailerFromGlobs(textGlob, htmlGlob string, emailer Emailer, fromAddr string) (*TemplatizedEmailer, error) {
|
|
textTemplates, err := template.ParseGlob(textGlob)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
htmlTemplates, err := htmltemplate.ParseGlob(htmlGlob)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return NewTemplatizedEmailerFromTemplates(textTemplates, htmlTemplates, emailer, fromAddr), nil
|
|
}
|
|
|
|
// NewTemplatizedEmailerFromTemplates creates a new TemplatizedEmailer, given root text and html templates.
|
|
func NewTemplatizedEmailerFromTemplates(textTemplates *template.Template, htmlTemplates *htmltemplate.Template, emailer Emailer, fromAddr string) *TemplatizedEmailer {
|
|
return &TemplatizedEmailer{
|
|
emailer: emailer,
|
|
textTemplates: textTemplates,
|
|
htmlTemplates: htmlTemplates,
|
|
fromAddr: fromAddr,
|
|
}
|
|
}
|
|
|
|
// TemplatizedEmailer sends emails based on templates, using any Emailer.
|
|
type TemplatizedEmailer struct {
|
|
textTemplates *template.Template
|
|
htmlTemplates *htmltemplate.Template
|
|
emailer Emailer
|
|
globalCtx map[string]interface{}
|
|
fromAddr string
|
|
}
|
|
|
|
func (t *TemplatizedEmailer) SetGlobalContext(ctx map[string]interface{}) {
|
|
t.globalCtx = ctx
|
|
}
|
|
|
|
// SendMail queues an email to be sent to a recipient.
|
|
// SendMail has similar semantics to Emailer.SendMail, except that you provide
|
|
// the template names you want to base the message on instead of the actual
|
|
// text. "to", "from" and "subject" will be added into the data map regardless
|
|
// of if they are used.
|
|
func (t *TemplatizedEmailer) SendMail(subject, tplName string, data map[string]interface{}, to string) error {
|
|
if tplName == "" {
|
|
return errors.New("Must provide a template name")
|
|
}
|
|
|
|
textTpl := t.textTemplates.Lookup(tplName + ".txt")
|
|
htmlTpl := t.htmlTemplates.Lookup(tplName + ".html")
|
|
|
|
if textTpl == nil && htmlTpl == nil {
|
|
return ErrorNoTemplate
|
|
}
|
|
|
|
data["to"] = to
|
|
data["from"] = t.fromAddr
|
|
data["subject"] = subject
|
|
|
|
for k, v := range t.globalCtx {
|
|
data[k] = v
|
|
}
|
|
|
|
var textBuffer bytes.Buffer
|
|
if textTpl != nil {
|
|
err := textTpl.Execute(&textBuffer, data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
var htmlBuffer bytes.Buffer
|
|
if htmlTpl != nil {
|
|
err := htmlTpl.Execute(&htmlBuffer, data)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return t.emailer.SendMail(subject, textBuffer.String(), htmlBuffer.String(), to)
|
|
}
|