wakapi/services/mail/mail.go

package mail

import (
	"bytes"
	"fmt"
	"github.com/markbates/pkger"
	conf "github.com/muety/wakapi/config"
	"github.com/muety/wakapi/services"
	"io/ioutil"
	"text/template"
)

const (
	tplPath              = "/views/mail"
	tplNamePasswordReset = "reset_password"
	subjectPasswordReset = "Wakapi – Password Reset"
)

type passwordResetLinkTplData struct {
	ResetLink string
}

// Factory
func NewMailService() services.IMailService {
	config := conf.Get()
	if config.Mail.Enabled {
		if config.Mail.Provider == conf.MailProviderMailWhale {
			return NewMailWhaleService(config.Mail.MailWhale)
		} else if config.Mail.Provider == conf.MailProviderSmtp {
			return NewSMTPMailService(config.Mail.Smtp)
		}
	}
	return &NoopMailService{}
}

func getPasswordResetTemplate(data passwordResetLinkTplData) (*bytes.Buffer, error) {
	tpl, err := loadTemplate(tplNamePasswordReset)
	if err != nil {
		return nil, err
	}
	var rendered bytes.Buffer
	if err := tpl.Execute(&rendered, data); err != nil {
		return nil, err
	}
	return &rendered, nil
}

func loadTemplate(tplName string) (*template.Template, error) {
	tplFile, err := pkger.Open(fmt.Sprintf("%s/%s.tpl.html", tplPath, tplName))
	if err != nil {
		return nil, err
	}
	defer tplFile.Close()

	tplData, err := ioutil.ReadAll(tplFile)
	if err != nil {
		return nil, err
	}

	return template.New(tplName).Parse(string(tplData))
}