54 lines
974 B
Go
54 lines
974 B
Go
package cachet
|
|
|
|
import (
|
|
"bytes"
|
|
"text/template"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
)
|
|
|
|
type MessageTemplate struct {
|
|
Subject string `json:"subject"`
|
|
Message string `json:"message"`
|
|
|
|
subjectTpl *template.Template
|
|
messageTpl *template.Template
|
|
}
|
|
|
|
func (t *MessageTemplate) SetDefault(d MessageTemplate) {
|
|
if len(t.Subject) == 0 {
|
|
t.Subject = d.Subject
|
|
}
|
|
if len(t.Message) == 0 {
|
|
t.Message = d.Message
|
|
}
|
|
}
|
|
|
|
func (t *MessageTemplate) Compile() error {
|
|
var err error
|
|
|
|
if len(t.Subject) > 0 {
|
|
t.subjectTpl, err = compileTemplate(t.Subject)
|
|
}
|
|
|
|
if err != nil && len(t.Message) > 0 {
|
|
t.messageTpl, err = compileTemplate(t.Message)
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
func (t *MessageTemplate) Exec(data interface{}) (string, string) {
|
|
buf := new(bytes.Buffer)
|
|
|
|
logrus.Warnf("%#v", t.subjectTpl)
|
|
t.subjectTpl.Execute(buf, data)
|
|
subject := buf.String()
|
|
|
|
return subject, ""
|
|
}
|
|
|
|
func compileTemplate(text string) (*template.Template, error) {
|
|
return template.New("").Parse(text)
|
|
}
|