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)
}