mirror of https://github.com/h44z/wg-portal.git
105 lines
2.9 KiB
Go
105 lines
2.9 KiB
Go
package mail
|
|
|
|
import (
|
|
"bytes"
|
|
"embed"
|
|
"fmt"
|
|
htmlTemplate "html/template"
|
|
"io"
|
|
"text/template"
|
|
|
|
"github.com/h44z/wg-portal/internal/domain"
|
|
)
|
|
|
|
//go:embed tpl_files/*
|
|
var TemplateFiles embed.FS
|
|
|
|
// TemplateHandler is a struct that holds the html and text templates.
|
|
type TemplateHandler struct {
|
|
portalUrl string
|
|
portalName string
|
|
htmlTemplates *htmlTemplate.Template
|
|
textTemplates *template.Template
|
|
}
|
|
|
|
func newTemplateHandler(portalUrl, portalName string) (*TemplateHandler, error) {
|
|
htmlTemplateCache, err := htmlTemplate.New("Html").ParseFS(TemplateFiles, "tpl_files/*.gohtml")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to parse html template files: %w", err)
|
|
}
|
|
|
|
txtTemplateCache, err := template.New("Txt").ParseFS(TemplateFiles, "tpl_files/*.gotpl")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to parse text template files: %w", err)
|
|
}
|
|
|
|
handler := &TemplateHandler{
|
|
portalUrl: portalUrl,
|
|
portalName: portalName,
|
|
htmlTemplates: htmlTemplateCache,
|
|
textTemplates: txtTemplateCache,
|
|
}
|
|
|
|
return handler, nil
|
|
}
|
|
|
|
// GetConfigMail returns the text and html template for the mail with a link.
|
|
func (c TemplateHandler) GetConfigMail(user *domain.User, link string) (io.Reader, io.Reader, error) {
|
|
var tplBuff bytes.Buffer
|
|
var htmlTplBuff bytes.Buffer
|
|
|
|
err := c.textTemplates.ExecuteTemplate(&tplBuff, "mail_with_link.gotpl", map[string]any{
|
|
"User": user,
|
|
"Link": link,
|
|
"PortalUrl": c.portalUrl,
|
|
})
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("failed to execute template mail_with_link.gotpl: %w", err)
|
|
}
|
|
|
|
err = c.htmlTemplates.ExecuteTemplate(&htmlTplBuff, "mail_with_link.gohtml", map[string]any{
|
|
"User": user,
|
|
"Link": link,
|
|
"PortalUrl": c.portalUrl,
|
|
})
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("failed to execute template mail_with_link.gohtml: %w", err)
|
|
}
|
|
|
|
return &tplBuff, &htmlTplBuff, nil
|
|
}
|
|
|
|
// GetConfigMailWithAttachment returns the text and html template for the mail with an attachment.
|
|
func (c TemplateHandler) GetConfigMailWithAttachment(user *domain.User, cfgName, qrName string) (
|
|
io.Reader,
|
|
io.Reader,
|
|
error,
|
|
) {
|
|
var tplBuff bytes.Buffer
|
|
var htmlTplBuff bytes.Buffer
|
|
|
|
err := c.textTemplates.ExecuteTemplate(&tplBuff, "mail_with_attachment.gotpl", map[string]any{
|
|
"User": user,
|
|
"ConfigFileName": cfgName,
|
|
"QrcodePngName": qrName,
|
|
"PortalUrl": c.portalUrl,
|
|
"PortalName": c.portalName,
|
|
})
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("failed to execute template mail_with_attachment.gotpl: %w", err)
|
|
}
|
|
|
|
err = c.htmlTemplates.ExecuteTemplate(&htmlTplBuff, "mail_with_attachment.gohtml", map[string]any{
|
|
"User": user,
|
|
"ConfigFileName": cfgName,
|
|
"QrcodePngName": qrName,
|
|
"PortalUrl": c.portalUrl,
|
|
"PortalName": c.portalName,
|
|
})
|
|
if err != nil {
|
|
return nil, nil, fmt.Errorf("failed to execute template mail_with_attachment.gohtml: %w", err)
|
|
}
|
|
|
|
return &tplBuff, &htmlTplBuff, nil
|
|
}
|