mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 22:06:38 +08:00
ryan
102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package model
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
"text/template"
|
|
"time"
|
|
|
|
"github.com/Rain-kl/Wavelet/internal/db"
|
|
)
|
|
|
|
// Template 邮件/消息模板实体
|
|
type Template struct {
|
|
ID uint64 `json:"id" gorm:"primaryKey;autoIncrement"`
|
|
Key string `json:"key" gorm:"uniqueIndex;size:80;not null"`
|
|
Name string `json:"name" gorm:"size:100;not null"`
|
|
Type string `json:"type" gorm:"size:20;not null;default:'email'"`
|
|
Subject string `json:"subject" gorm:"size:255"`
|
|
Content string `json:"content" gorm:"type:text;not null"`
|
|
Description string `json:"description" gorm:"size:255"`
|
|
IsSystem bool `json:"is_system" gorm:"index;not null;default:false"`
|
|
CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime;index"`
|
|
UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime;index"`
|
|
}
|
|
|
|
// Normalize 规范化模板字段
|
|
func (t *Template) Normalize() {
|
|
t.Key = strings.TrimSpace(t.Key)
|
|
t.Name = strings.TrimSpace(t.Name)
|
|
t.Type = strings.ToLower(strings.TrimSpace(t.Type))
|
|
t.Subject = strings.TrimSpace(t.Subject)
|
|
t.Content = strings.TrimSpace(t.Content)
|
|
t.Description = strings.TrimSpace(t.Description)
|
|
if t.Type == "" {
|
|
t.Type = "email"
|
|
}
|
|
}
|
|
|
|
// Validate 校验模板必填字段
|
|
func (t *Template) Validate() error {
|
|
t.Normalize()
|
|
if t.Key == "" {
|
|
return errors.New(errTemplateKeyRequired)
|
|
}
|
|
if t.Name == "" {
|
|
return errors.New(errTemplateNameRequired)
|
|
}
|
|
if t.Content == "" {
|
|
return errors.New(errTemplateContentRequired)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Render 渲染模板的 Subject 和 Content
|
|
func (t *Template) Render(data any) (string, string, error) {
|
|
// Render Subject
|
|
var subject string
|
|
if t.Subject != "" {
|
|
tmplSubject, err := template.New(t.Key + "_subject").Parse(t.Subject)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
var subBuf bytes.Buffer
|
|
if err := tmplSubject.Execute(&subBuf, data); err != nil {
|
|
return "", "", err
|
|
}
|
|
subject = subBuf.String()
|
|
}
|
|
|
|
// Render Content
|
|
tmplContent, err := template.New(t.Key + "_content").Parse(t.Content)
|
|
if err != nil {
|
|
return "", "", err
|
|
}
|
|
var bodyBuf bytes.Buffer
|
|
if err := tmplContent.Execute(&bodyBuf, data); err != nil {
|
|
return "", "", err
|
|
}
|
|
|
|
return subject, bodyBuf.String(), nil
|
|
}
|
|
|
|
// RenderTemplate 渲染指定模板。模板不存在或渲染失败时返回错误，由调用方决定如何处理。
|
|
func RenderTemplate(ctx context.Context, key string, data any) (string, string, error) {
|
|
var t Template
|
|
if err := db.DB(ctx).Where("key = ?", key).First(&t).Error; err != nil {
|
|
return "", "", fmt.Errorf(errTemplateUnavailable, key, err)
|
|
}
|
|
|
|
subject, body, err := t.Render(data)
|
|
if err != nil {
|
|
return "", "", fmt.Errorf(errTemplateRenderFailed, key, err)
|
|
}
|
|
return subject, body, nil
|
|
}
|