51 lines
911 B
Go
51 lines
911 B
Go
|
package mailer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
hTemplate "html/template"
|
||
|
|
"text/template"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Renderer for easy template of TXT or HTML
|
||
|
|
type Renderer struct {
|
||
|
|
tmpl *template.Template
|
||
|
|
hTmpl *hTemplate.Template
|
||
|
|
}
|
||
|
|
|
||
|
|
// TemplateTXT - create template render
|
||
|
|
func TemplateTXT(temp string) *Renderer {
|
||
|
|
tmpl, err := template.New("").Parse(temp)
|
||
|
|
if err != nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return &Renderer{
|
||
|
|
tmpl: tmpl,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// TemplateHTML - create template render for html
|
||
|
|
func TemplateHTML(temp string) *Renderer {
|
||
|
|
tmpl, err := hTemplate.New("").Parse(temp)
|
||
|
|
if err != nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return &Renderer{
|
||
|
|
hTmpl: tmpl,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Render template
|
||
|
|
func (r *Renderer) Render(data interface{}) string {
|
||
|
|
var buf bytes.Buffer
|
||
|
|
if r.hTmpl != nil {
|
||
|
|
if err := r.hTmpl.Execute(&buf, data); err != nil {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if err := r.tmpl.Execute(&buf, data); err != nil {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return string(buf.Bytes())
|
||
|
|
}
|