chore: 初始化 Cloudreve 源码仓库

2026-09-13 11:28:59 +08:00
commit 2a7b03ab1d
489 changed files with 163867 additions and 0 deletions
@@ -0,0 +1,147 @@
+package email
+
+import (
+	"context"
+	"fmt"
+	"html/template"
+	"net/url"
+	"strings"
+
+	"github.com/cloudreve/Cloudreve/v4/ent"
+	"github.com/cloudreve/Cloudreve/v4/pkg/setting"
+)
+
+type CommonContext struct {
+	SiteBasic *setting.SiteBasic
+	Logo      *setting.Logo
+	SiteUrl   string
+}
+
+// ResetContext used for variables in reset email
+type ResetContext struct {
+	*CommonContext
+	User *ent.User
+	Url  string
+}
+
+// NewResetEmail generates reset email from template
+func NewResetEmail(ctx context.Context, settings setting.Provider, user *ent.User, url string) (string, string, error) {
+	templates := settings.ResetEmailTemplate(ctx)
+	if len(templates) == 0 {
+		return "", "", fmt.Errorf("reset email template not configured")
+	}
+
+	selected := selectTemplate(templates, user)
+	resetCtx := ResetContext{
+		CommonContext: commonContext(ctx, settings),
+		User:          user,
+		Url:           url,
+	}
+
+	tmplTitle, err := template.New("resetTitle").Parse(selected.Title)
+	if err != nil {
+		return "", "", fmt.Errorf("failed to parse email title: %w", err)
+	}
+
+	var resTitle strings.Builder
+	err = tmplTitle.Execute(&resTitle, resetCtx)
+	if err != nil {
+		return "", "", fmt.Errorf("failed to execute email title: %w", err)
+	}
+
+	tmplBody, err := template.New("resetBody").Parse(selected.Body)
+	if err != nil {
+		return "", "", fmt.Errorf("failed to parse email template: %w", err)
+	}
+
+	var resBody strings.Builder
+	err = tmplBody.Execute(&resBody, resetCtx)
+	if err != nil {
+		return "", "", fmt.Errorf("failed to execute email template: %w", err)
+	}
+
+	return resTitle.String(), resBody.String(), nil
+}
+
+// ActivationContext used for variables in activation email
+type ActivationContext struct {
+	*CommonContext
+	User *ent.User
+	Url  string
+}
+
+// NewActivationEmail generates activation email from template
+func NewActivationEmail(ctx context.Context, settings setting.Provider, user *ent.User, url string) (string, string, error) {
+	templates := settings.ActivationEmailTemplate(ctx)
+	if len(templates) == 0 {
+		return "", "", fmt.Errorf("activation email template not configured")
+	}
+
+	selected := selectTemplate(templates, user)
+	activationCtx := ActivationContext{
+		CommonContext: commonContext(ctx, settings),
+		User:          user,
+		Url:           url,
+	}
+
+	tmplTitle, err := template.New("activationTitle").Parse(selected.Title)
+	if err != nil {
+		return "", "", fmt.Errorf("failed to parse email title: %w", err)
+	}
+
+	var resTitle strings.Builder
+	err = tmplTitle.Execute(&resTitle, activationCtx)
+	if err != nil {
+		return "", "", fmt.Errorf("failed to execute email title: %w", err)
+	}
+
+	tmplBody, err := template.New("activationBody").Parse(selected.Body)
+	if err != nil {
+		return "", "", fmt.Errorf("failed to parse email template: %w", err)
+	}
+
+	var resBody strings.Builder
+	err = tmplBody.Execute(&resBody, activationCtx)
+	if err != nil {
+		return "", "", fmt.Errorf("failed to execute email template: %w", err)
+	}
+
+	return resTitle.String(), resBody.String(), nil
+}
+
+func commonContext(ctx context.Context, settings setting.Provider) *CommonContext {
+	logo := settings.Logo(ctx)
+	siteUrl := settings.SiteURL(ctx)
+	res := &CommonContext{
+		SiteBasic: settings.SiteBasic(ctx),
+		Logo:      settings.Logo(ctx),
+		SiteUrl:   siteUrl.String(),
+	}
+
+	// Add site url if logo is not an url
+	if !strings.HasPrefix(logo.Light, "http") {
+		logoPath, _ := url.Parse(logo.Light)
+		res.Logo.Light = siteUrl.ResolveReference(logoPath).String()
+	}
+
+	if !strings.HasPrefix(logo.Normal, "http") {
+		logoPath, _ := url.Parse(logo.Normal)
+		res.Logo.Normal = siteUrl.ResolveReference(logoPath).String()
+	}
+
+	return res
+}
+
+func selectTemplate(templates []setting.EmailTemplate, u *ent.User) setting.EmailTemplate {
+	selected := templates[0]
+	if u != nil {
+		for _, t := range templates {
+			if strings.EqualFold(t.Language, u.Settings.Language) {
+				selected = t
+				break
+			}
+		}
+	}
+
+	return selected
+}