实现飞书/微信/邮件发送通知

internal/infra/notify/lark.go

@@ -0,0 +1,194 @@
+package notify
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"strings"
+	"sync"
+	"time"
+)
+
+const (
+	// 飞书获取 tenant_access_token 的 API 地址
+	larkGetTokenURL = "https://open.feishu.cn/open-apis/auth/v3/tenant_access_token/internal"
+	// 飞书发送消息的 API 地址
+	larkSendMessageURL = "https://open.feishu.cn/open-apis/im/v1/messages"
+)
+
+// larkNotifier 实现了 Notifier 接口，用于通过飞书自建应用发送私聊消息。
+type larkNotifier struct {
+	appID     string // 应用 ID
+	appSecret string // 应用密钥
+
+	// 用于线程安全地管理 tenant_access_token
+	mu             sync.Mutex
+	accessToken    string
+	tokenExpiresAt time.Time
+}
+
+// NewLarkNotifier 创建一个新的 larkNotifier 实例。
+// 调用者需要注入飞书应用的 AppID 和 AppSecret。
+func NewLarkNotifier(appID, appSecret string) Notifier {
+	return &larkNotifier{
+		appID:     appID,
+		appSecret: appSecret,
+	}
+}
+
+// Send 向指定用户发送一条飞书消息卡片。
+// toAddr 参数是接收者的邮箱地址。
+func (l *larkNotifier) Send(content AlarmContent, toAddr string) error {
+	// 1. 获取有效的 tenant_access_token
+	token, err := l.getAccessToken()
+	if err != nil {
+		return err
+	}
+
+	// 2. 构建消息卡片 JSON
+	// 飞书消息卡片结构复杂，这里构建一个简单的 Markdown 文本卡片
+	cardContent := map[string]interface{}{
+		"config": map[string]bool{
+			"wide_screen_mode": true,
+		},
+		"elements": []map[string]interface{}{
+			{
+				"tag": "div",
+				"text": map[string]string{
+					"tag": "lark_md",
+					"content": fmt.Sprintf("## %s\n**级别**: %s\n**时间**: %s\n\n%s",
+						content.Title,
+						content.Level.String(),
+						content.Timestamp.Format(DefaultTimeFormat),
+						content.Message,
+					),
+				},
+			},
+		},
+	}
+
+	cardJSON, err := json.Marshal(cardContent)
+	if err != nil {
+		return fmt.Errorf("序列化飞书卡片内容失败: %w", err)
+	}
+
+	// 3. 构建请求的 JSON Body
+	payload := larkMessagePayload{
+		ReceiveID:     toAddr,
+		ReceiveIDType: "email",       // 指定接收者类型为邮箱
+		MsgType:       "interactive", // 消息卡片类型
+		Content:       string(cardJSON),
+	}
+
+	jsonBytes, err := json.Marshal(payload)
+	if err != nil {
+		return fmt.Errorf("序列化飞书消息失败: %w", err)
+	}
+
+	// 4. 发送 HTTP POST 请求
+	url := fmt.Sprintf("%s?receive_id_type=email", larkSendMessageURL) // 在 URL 中指定 receive_id_type
+	req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonBytes))
+	if err != nil {
+		return fmt.Errorf("创建飞书请求失败: %w", err)
+	}
+	req.Header.Set("Content-Type", "application/json")
+	req.Header.Set("Authorization", "Bearer "+token) // 携带 access_token
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		return fmt.Errorf("发送飞书通知失败: %w", err)
+	}
+	defer resp.Body.Close()
+
+	// 5. 检查响应
+	var response larkResponse
+	if err := json.NewDecoder(resp.Body).Decode(&response); err != nil {
+		return fmt.Errorf("解析飞书响应失败: %w", err)
+	}
+
+	if response.Code != 0 {
+		return fmt.Errorf("飞书API返回错误: code=%d, msg=%s", response.Code, response.Msg)
+	}
+
+	return nil
+}
+
+// getAccessToken 获取并缓存 tenant_access_token，处理了线程安全和自动刷新。
+func (l *larkNotifier) getAccessToken() (string, error) {
+	l.mu.Lock()
+	defer l.mu.Unlock()
+
+	// 如果 token 存在且有效期还有5分钟以上，则直接返回缓存的 token
+	if l.accessToken != "" && time.Now().Before(l.tokenExpiresAt.Add(-5*time.Minute)) {
+		return l.accessToken, nil
+	}
+
+	// 否则，重新获取 token
+	payload := map[string]string{
+		"app_id":     l.appID,
+		"app_secret": l.appSecret,
+	}
+	jsonBytes, err := json.Marshal(payload)
+	if err != nil {
+		return "", fmt.Errorf("序列化获取 token 请求体失败: %w", err)
+	}
+
+	req, err := http.NewRequest("POST", larkGetTokenURL, bytes.NewBuffer(jsonBytes))
+	if err != nil {
+		return "", fmt.Errorf("创建获取 token 请求失败: %w", err)
+	}
+	req.Header.Set("Content-Type", "application/json")
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		return "", fmt.Errorf("获取 tenant_access_token 请求失败: %w", err)
+	}
+	defer resp.Body.Close()
+
+	var tokenResp larkTokenResponse
+	if err := json.NewDecoder(resp.Body).Decode(&tokenResp); err != nil {
+		return "", fmt.Errorf("解析 tenant_access_token 响应失败: %w", err)
+	}
+
+	if tokenResp.Code != 0 {
+		return "", fmt.Errorf("获取 tenant_access_token API 返回错误: code=%d, msg=%s", tokenResp.Code, tokenResp.Msg)
+	}
+
+	// 缓存新的 token 和过期时间
+	l.accessToken = tokenResp.TenantAccessToken
+	l.tokenExpiresAt = time.Now().Add(time.Duration(tokenResp.Expire) * time.Second)
+
+	return l.accessToken, nil
+}
+
+// Type 返回通知器的类型
+func (l *larkNotifier) Type() NotifierType {
+	return NotifierTypeLark
+}
+
+// --- API 数据结构 ---
+
+// larkTokenResponse 是获取 tenant_access_token API 的响应结构体
+type larkTokenResponse struct {
+	Code              int    `json:"code"`
+	Msg               string `json:"msg"`
+	TenantAccessToken string `json:"tenant_access_token"`
+	Expire            int    `json:"expire"` // 有效期，单位秒
+}
+
+// larkMessagePayload 是发送消息 API 的请求体结构
+type larkMessagePayload struct {
+	ReceiveID     string `json:"receive_id"`
+	ReceiveIDType string `json:"receive_id_type"`
+	MsgType       string `json:"msg_type"`
+	Content       string `json:"content"` // 对于 interactive 消息，这里是卡片的 JSON 字符串
+}
+
+// larkResponse 是飞书 API 的通用响应结构体
+type larkResponse struct {
+	Code int    `json:"code"`
+	Msg  string `json:"msg"`
+}

internal/infra/notify/notify.go

@@ -0,0 +1,42 @@
+package notify
+
+import (
+	"time"
+
+	"go.uber.org/zap/zapcore"
+)
+
+// DefaultTimeFormat 定义了所有通知中统一使用的时间格式。
+const DefaultTimeFormat = "2006-01-02 15:04:05"
+
+// NotifierType 定义了通知器的类型。
+type NotifierType string
+
+const (
+	// NotifierTypeSMTP 表示 SMTP 邮件通知器。
+	NotifierTypeSMTP NotifierType = "邮件"
+	// NotifierTypeWeChat 表示企业微信通知器。
+	NotifierTypeWeChat NotifierType = "企业微信"
+	// NotifierTypeLark 表示飞书通知器。
+	NotifierTypeLark NotifierType = "飞书"
+)
+
+// AlarmContent 定义了通知的内容
+type AlarmContent struct {
+	// 通知标题
+	Title string
+	// 通知信息
+	Message string
+	// 通知级别
+	Level zapcore.Level
+	// 通知时间
+	Timestamp time.Time
+}
+
+// Notifier 定义了通知发送器的接口
+type Notifier interface {
+	// Send 发送通知
+	Send(content AlarmContent, toAddr string) error
+	// Type 返回通知器的类型
+	Type() NotifierType
+}

internal/infra/notify/smtp.go

@@ -0,0 +1,73 @@
+package notify
+
+import (
+	"fmt"
+	"net/smtp"
+	"strings"
+)
+
+// smtpNotifier 实现了 Notifier 接口，用于通过 SMTP 发送邮件通知。
+type smtpNotifier struct {
+	host     string // SMTP 服务器地址
+	port     int    // SMTP 服务器端口
+	username string // 发件人邮箱地址
+	password string // 发件人邮箱授权码
+	sender   string // 发件人名称或地址，显示在邮件的 \"From\" 字段
+}
+
+// NewSMTPNotifier 创建一个新的 smtpNotifier 实例。
+// 调用者需要注入 SMTP 相关的配置。
+func NewSMTPNotifier(host string, port int, username, password, sender string) Notifier {
+	return &smtpNotifier{
+		host:     host,
+		port:     port,
+		username: username,
+		password: password,
+		sender:   sender,
+	}
+}
+
+// Send 使用 net/smtp 包发送一封邮件。
+func (s *smtpNotifier) Send(content AlarmContent, toAddr string) error {
+	// 1. 设置认证信息
+	auth := smtp.PlainAuth("", s.username, s.password, s.host)
+
+	// 2. 构建邮件内容
+	// 邮件头
+	subject := content.Title
+	contentType := "Content-Type: text/plain; charset=UTF-8"
+	fromHeader := fmt.Sprintf("From: %s", s.sender)
+	toHeader := fmt.Sprintf("To: %s", toAddr)
+	subjectHeader := fmt.Sprintf("Subject: %s", subject)
+
+	// 邮件正文
+	body := fmt.Sprintf("级别: %s\n时间: %s\n\n%s",
+		content.Level.String(),
+		content.Timestamp.Format(DefaultTimeFormat),
+		content.Message,
+	)
+
+	// 拼接完整的邮件报文
+	msg := strings.Join([]string{
+		fromHeader,
+		toHeader,
+		subjectHeader,
+		contentType,
+		"", // 邮件头和正文之间的空行
+		body,
+	}, "\r\n")
+
+	// 3. 发送邮件
+	addr := fmt.Sprintf("%s:%d", s.host, s.port)
+	err := smtp.SendMail(addr, auth, s.username, []string{toAddr}, []byte(msg))
+	if err != nil {
+		return fmt.Errorf("发送邮件失败: %w", err)
+	}
+
+	return nil
+}
+
+// Type 返回通知器的类型
+func (s *smtpNotifier) Type() NotifierType {
+	return NotifierTypeSMTP
+}

internal/infra/notify/wechat.go

@@ -0,0 +1,169 @@
+package notify
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"strings"
+	"sync"
+	"time"
+)
+
+const (
+	// 获取 access_token 的 API 地址
+	getTokenURL = "https://qyapi.weixin.qq.com/cgi-bin/gettoken"
+	// 发送应用消息的 API 地址
+	sendMessageURL = "https://qyapi.weixin.qq.com/cgi-bin/message/send"
+)
+
+// wechatNotifier 实现了 Notifier 接口，用于通过企业微信自建应用发送私聊消息。
+type wechatNotifier struct {
+	corpID  string // 企业ID (CorpID)
+	agentID string // 应用ID (AgentID)
+	secret  string // 应用密钥 (Secret)
+
+	// 用于线程安全地管理 access_token
+	mu             sync.Mutex
+	accessToken    string
+	tokenExpiresAt time.Time
+}
+
+// NewWechatNotifier 创建一个新的 wechatNotifier 实例。
+// 调用者需要注入企业微信应用的 CorpID, AgentID 和 Secret。
+func NewWechatNotifier(corpID, agentID, secret string) Notifier {
+	return &wechatNotifier{
+		corpID:  corpID,
+		agentID: agentID,
+		secret:  secret,
+	}
+}
+
+// Send 向指定用户发送一条 markdown 格式的私聊消息。
+// toAddr 参数是接收者的 UserID 列表，用逗号或竖线分隔。
+func (w *wechatNotifier) Send(content AlarmContent, toAddr string) error {
+	// 1. 获取有效的 access_token
+	token, err := w.getAccessToken()
+	if err != nil {
+		return err
+	}
+
+	// 2. 构建 markdown 内容
+	markdownContent := fmt.Sprintf("## %s\n> 级别: <font color=\"warning\">%s</font>\n> 时间: %s\n\n%s",
+		content.Title,
+		content.Level.String(),
+		content.Timestamp.Format(DefaultTimeFormat),
+		content.Message,
+	)
+
+	// 3. 构建请求的 JSON Body
+	// 将逗号分隔的 toAddr 转换为竖线分隔，以符合 API 要求
+	userList := strings.ReplaceAll(toAddr, ",", "|")
+	payload := wechatMessagePayload{
+		ToUser:  userList,
+		MsgType: "markdown",
+		AgentID: w.agentID,
+		Markdown: struct {
+			Content string `json:"content"`
+		}{
+			Content: markdownContent,
+		},
+	}
+
+	jsonBytes, err := json.Marshal(payload)
+	if err != nil {
+		return fmt.Errorf("序列化企业微信消息失败: %w", err)
+	}
+
+	// 4. 发送 HTTP POST 请求
+	url := fmt.Sprintf("%s?access_token=%s", sendMessageURL, token)
+	req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonBytes))
+	if err != nil {
+		return fmt.Errorf("创建企业微信请求失败: %w", err)
+	}
+	req.Header.Set("Content-Type", "application/json")
+
+	client := &http.Client{}
+	resp, err := client.Do(req)
+	if err != nil {
+		return fmt.Errorf("发送企业微信通知失败: %w", err)
+	}
+	defer resp.Body.Close()
+
+	// 5. 检查响应
+	var response wechatResponse
+	if err := json.NewDecoder(resp.Body).Decode(&response); err != nil {
+		return fmt.Errorf("解析企业微信响应失败: %w", err)
+	}
+
+	if response.ErrCode != 0 {
+		return fmt.Errorf("企业微信API返回错误: code=%d, msg=%s", response.ErrCode, response.ErrMsg)
+	}
+
+	return nil
+}
+
+// getAccessToken 获取并缓存 access_token，处理了线程安全和自动刷新。
+func (w *wechatNotifier) getAccessToken() (string, error) {
+	w.mu.Lock()
+	defer w.mu.Unlock()
+
+	// 如果 token 存在且有效期还有5分钟以上，则直接返回缓存的 token
+	if w.accessToken != "" && time.Now().Before(w.tokenExpiresAt.Add(-5*time.Minute)) {
+		return w.accessToken, nil
+	}
+
+	// 否则，重新获取 token
+	url := fmt.Sprintf("%s?corpid=%s&corpsecret=%s", getTokenURL, w.corpID, w.secret)
+	resp, err := http.Get(url)
+	if err != nil {
+		return "", fmt.Errorf("获取 access_token 请求失败: %w", err)
+	}
+	defer resp.Body.Close()
+
+	var tokenResp tokenResponse
+	if err := json.NewDecoder(resp.Body).Decode(&tokenResp); err != nil {
+		return "", fmt.Errorf("解析 access_token 响应失败: %w", err)
+	}
+
+	if tokenResp.ErrCode != 0 {
+		return "", fmt.Errorf("获取 access_token API 返回错误: code=%d, msg=%s", tokenResp.ErrCode, tokenResp.ErrMsg)
+	}
+
+	// 缓存新的 token 和过期时间
+	w.accessToken = tokenResp.AccessToken
+	w.tokenExpiresAt = time.Now().Add(time.Duration(tokenResp.ExpiresIn) * time.Second)
+
+	return w.accessToken, nil
+}
+
+// Type 返回通知器的类型
+func (w *wechatNotifier) Type() NotifierType {
+	return NotifierTypeWeChat
+}
+
+// --- API 数据结构 ---
+
+// tokenResponse 是获取 access_token API 的响应结构体
+type tokenResponse struct {
+	ErrCode     int    `json:"errcode"`
+	ErrMsg      string `json:"errmsg"`
+	AccessToken string `json:"access_token"`
+	ExpiresIn   int    `json:"expires_in"`
+}
+
+// wechatMessagePayload 是发送应用消息 API 的请求体结构
+type wechatMessagePayload struct {
+	ToUser   string `json:"touser"`
+	MsgType  string `json:"msgtype"`
+	AgentID  string `json:"agentid"`
+	Markdown struct {
+		Content string `json:"content"`
+	} `json:"markdown"`
+}
+
+// wechatResponse 是企业微信 API 的通用响应结构体
+type wechatResponse struct {
+	ErrCode int    `json:"errcode"`
+	ErrMsg  string `json:"errmsg"`
+}

internal/infra/repository/user_repository.go

@@ -13,6 +13,7 @@ type UserRepository interface {
 	FindByUsername(username string) (*models.User, error)
 	FindByID(id uint) (*models.User, error)
 	FindUserForLogin(identifier string) (*models.User, error)
+	FindAll() ([]*models.User, error)
 }
 
 // gormUserRepository 是 UserRepository 的 GORM 实现
@@ -64,3 +65,12 @@ func (r *gormUserRepository) FindByID(id uint) (*models.User, error) {
 	}
 	return &user, nil
 }
+
+// FindAll 返回数据库中的所有用户
+func (r *gormUserRepository) FindAll() ([]*models.User, error) {
+	var users []*models.User
+	if err := r.db.Where("1 = 1").Find(&users).Error; err != nil {
+		return nil, err
+	}
+	return users, nil
+}