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

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"`
+}