mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 14:06:36 +08:00
ryan
63cd906cfc
- Merge all files inside openflare-server to the repository root directory. - Relocate agent, relay, and flared subprojects from internal/ to internal/apps/. - Combine docker-compose files and update build context paths to root. - Update GitHub workflows and Dockerfiles to refer to new directories and package names. - Rewrite Go package imports across all files. - Resolve database renew test race condition and clean up docs.
89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package model
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"strings"
|
|
"text/template"
|
|
"time"
|
|
)
|
|
|
|
// 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"`
|
|
}
|
|
|
|
// TableName 表名
|
|
func (Template) TableName() string {
|
|
return "w_templates"
|
|
}
|
|
|
|
// 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
|
|
}
|