mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-28 05:46:36 +08:00
ryan
55db1c01ec
Result: {"status":"keep","total_issues":8,"eslint_errors":0,"eslint_problems":0,"eslint_warnings":0,"golint_canonicalheader":0,"golint_errname":0,"golint_errorlint":1,"golint_exhaustive":0,"golint_forcetypeassert":0,"golint_gosec":0,"golint_intrange":0,"golint_modernize":3,"golint_nilnil":3,"golint_perfsprint":0,"golint_prealloc":0,"golint_recvcheck":1,"golint_test_testifylint":0,"golint_test_thelper":0,"golint_test_total":0,"golint_test_usetesting":0,"golint_total":8,"golint_usestdlibvars":0,"golint_vetx_total":0,"golint_wastedassign":0,"measure_s":112,"tsc_errors":0,"vitest_failed":0,"vitest_total":126}
120 lines
3.0 KiB
Go
120 lines
3.0 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package push
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"net/smtp"
|
|
"strings"
|
|
|
|
"github.com/Rain-kl/Wavelet/pkg/util"
|
|
)
|
|
|
|
func init() {
|
|
Register("email", &EmailPusher{})
|
|
}
|
|
|
|
// EmailPusher 极简 SMTP 邮件推送实现 (静态、解耦)
|
|
type EmailPusher struct{}
|
|
|
|
// sanitizeEmailHeader removes CR/LF bytes so untrusted values cannot inject
|
|
// additional email headers (email header injection).
|
|
func sanitizeEmailHeader(v string) string {
|
|
v = strings.ReplaceAll(v, "\r", "")
|
|
v = strings.ReplaceAll(v, "\n", "")
|
|
return v
|
|
}
|
|
|
|
// Send 发送邮件
|
|
func (p *EmailPusher) Send(ctx context.Context, cfg Config, target string, body map[string]any, _ string, ext map[string]any) (string, error) {
|
|
if cfg.URL == "" || cfg.Key == "" || cfg.Secret == "" {
|
|
return "", errors.New("email: SMTP configuration (url, key, secret) is incomplete")
|
|
}
|
|
if target == "" {
|
|
return "", errors.New("email: target email address is required")
|
|
}
|
|
|
|
title := defaultTitle
|
|
if t, ok := body["title"].(string); ok && t != "" {
|
|
title = t
|
|
}
|
|
|
|
var content string
|
|
if c, ok := body["content"].(string); ok && c != "" {
|
|
content = c
|
|
} else {
|
|
// 自动格式化 map
|
|
var parts []string
|
|
for k, v := range body {
|
|
parts = append(parts, fmt.Sprintf("<p><b>%s</b>: %v</p>", k, v))
|
|
}
|
|
content = strings.Join(parts, "")
|
|
}
|
|
|
|
// 邮件头和体
|
|
from := cfg.Key
|
|
to := target
|
|
|
|
// 如果 ext 中指定了 from_name，我们在 From 头部包含它
|
|
fromName := "System Notification"
|
|
if ext != nil {
|
|
if fn, ok := ext["from_name"].(string); ok && fn != "" {
|
|
fromName = fn
|
|
}
|
|
}
|
|
|
|
subjectHeader := fmt.Sprintf("Subject: %s\r\n", sanitizeEmailHeader(title))
|
|
fromHeader := fmt.Sprintf("From: %s <%s>\r\n", sanitizeEmailHeader(fromName), sanitizeEmailHeader(from))
|
|
toHeader := fmt.Sprintf("To: %s\r\n", sanitizeEmailHeader(to))
|
|
mimeHeader := "MIME-version: 1.0;\r\nContent-Type: text/html; charset=\"UTF-8\";\r\n\r\n"
|
|
|
|
// 拼装完整的邮件报文
|
|
// 简单的 HTML 正文渲染
|
|
htmlBody := fmt.Sprintf(`<html><body><h2>%s</h2><div>%s</div></body></html>`, title, content)
|
|
msg := []byte(fromHeader + toHeader + subjectHeader + mimeHeader + htmlBody + "\r\n")
|
|
|
|
// 解析 Host 和 Port
|
|
host, port, err := net.SplitHostPort(cfg.URL)
|
|
if err != nil {
|
|
host = cfg.URL
|
|
port = "25" // 默认 SMTP 端口
|
|
}
|
|
|
|
auth := smtp.PlainAuth("", cfg.Key, cfg.Secret, host)
|
|
|
|
// 异步超时处理
|
|
errChan := make(chan error, 1)
|
|
util.Go(func() {
|
|
errChan <- smtp.SendMail(host+":"+port, auth, from, []string{to}, msg)
|
|
})
|
|
|
|
select {
|
|
case <-ctx.Done():
|
|
return "", ctx.Err()
|
|
case err := <-errChan:
|
|
if err != nil {
|
|
return "", fmt.Errorf("email: send smtp mail failed: %w", err)
|
|
}
|
|
}
|
|
|
|
return "", nil
|
|
}
|
|
|
|
// ValidateConfig 校验邮件 SMTP 配置
|
|
func (p *EmailPusher) ValidateConfig(cfg Config) error {
|
|
if cfg.URL == "" {
|
|
return errors.New("SMTP host:port is required")
|
|
}
|
|
if cfg.Key == "" {
|
|
return errors.New("SMTP username is required")
|
|
}
|
|
if cfg.Secret == "" {
|
|
return errors.New("SMTP password is required")
|
|
}
|
|
return nil
|
|
}
|