98 lines
2.2 KiB
Go
98 lines
2.2 KiB
Go
package mailer
|
|
|
|
import (
|
|
"embed"
|
|
"errors"
|
|
"io/fs"
|
|
"strings"
|
|
"text/template"
|
|
)
|
|
|
|
const EMAIL_FS_PREFIX = "templates/"
|
|
|
|
const TPL_PASSWORD_RESET = "password_reset"
|
|
const TPL_VERIFY_EMAIL = "verify_email"
|
|
|
|
type EmailTemplate struct {
|
|
Subject *template.Template
|
|
Body *template.Template
|
|
}
|
|
|
|
//go:embed templates/*
|
|
var templateFS embed.FS
|
|
|
|
var ErrTemplateNotFound = errors.New("template not found")
|
|
|
|
type TemplateRegistry struct {
|
|
templates map[string]EmailTemplate
|
|
}
|
|
|
|
func NewTemplateRegistry() *TemplateRegistry {
|
|
return &TemplateRegistry{
|
|
templates: make(map[string]EmailTemplate),
|
|
}
|
|
}
|
|
|
|
func (tr *TemplateRegistry) loadTemplates() error {
|
|
subjectTemplates, err := fs.Glob(templateFS, EMAIL_FS_PREFIX+"*_subject*")
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, subjectTemplate := range subjectTemplates {
|
|
templateName := strings.TrimPrefix(subjectTemplate, EMAIL_FS_PREFIX)
|
|
templateName = strings.TrimSuffix(templateName, "_subject.tpl")
|
|
bodyTemplate := strings.TrimSuffix(subjectTemplate, "_subject.tpl") + "_body.tpl"
|
|
bodyTemplate = strings.TrimPrefix(bodyTemplate, EMAIL_FS_PREFIX)
|
|
|
|
subjectContent, err := fs.ReadFile(templateFS, EMAIL_FS_PREFIX+templateName+"_subject.tpl")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
subjectTmpl, err := template.New(templateName).Parse(string(subjectContent))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
bodyContent, err := fs.ReadFile(templateFS, EMAIL_FS_PREFIX+bodyTemplate)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
bodyTmpl, err := template.New(templateName).Parse(string(bodyContent))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
tr.templates[templateName] = EmailTemplate{
|
|
Subject: subjectTmpl,
|
|
Body: bodyTmpl,
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (tr *TemplateRegistry) RenderTemplate(templateName string, subjectVars TemplateData, bodyVars TemplateData) (*Email, error) {
|
|
tmpl, ok := tr.templates[templateName]
|
|
if !ok {
|
|
return nil, ErrTemplateNotFound
|
|
}
|
|
|
|
var subjectBuilder strings.Builder
|
|
err := tmpl.Subject.Execute(&subjectBuilder, subjectVars)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var bodyBuilder strings.Builder
|
|
err = tmpl.Body.Execute(&bodyBuilder, bodyVars)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return NewEmail(subjectBuilder.String(), bodyBuilder.String()), nil
|
|
}
|