feat: 新增 SafeW 出站通知渠道

通过 Bot API sendMessage 投递渲染后的通知，MarkdownV2 自动转义标题与正文。 Co-authored-by: Cursor <cursoragent@cursor.com>
2026-08-14 00:58:44 +08:00
parent 39f3774940
commit 0d0cd0c510
7 changed files with 1003 additions and 3 deletions
@@ -21,6 +21,8 @@ func NewSender(channelType string, smtpCfg *config.SMTPConfig, dingtalkLimiter *
 		return &WeComSender{}, nil
 	case "bark":
 		return &BarkSender{}, nil
+	case "safew":
+		return &SafeWSender{}, nil
 	case "email":
 		return NewEmailSender(*smtpCfg), nil
 	default:
@@ -0,0 +1,138 @@
+package adapter
+
+import (
+	"bytes"
+	"encoding/json"
+	"fmt"
+	"io"
+	"net/http"
+	"strings"
+)
+
+const safewAPIBase = "https://api.safew.bot"
+
+type SafeWSender struct {
+	apiBase string
+}
+
+type safewConfig struct {
+	Token  string          `json:"token"`
+	ChatID json.RawMessage `json:"chat_id"`
+}
+
+type safewMessage struct {
+	ChatID    string `json:"chat_id"`
+	Text      string `json:"text"`
+	ParseMode string `json:"parse_mode"`
+}
+
+type safewAPIResponse struct {
+	OK          bool   `json:"ok"`
+	Description string `json:"description"`
+}
+
+func (s *SafeWSender) Type() string { return "safew" }
+
+func (s *SafeWSender) Send(title, content string, config json.RawMessage) error {
+	var cfg safewConfig
+	if err := json.Unmarshal(config, &cfg); err != nil {
+		return fmt.Errorf("parse safew config: %w", err)
+	}
+	token := strings.TrimSpace(cfg.Token)
+	if token == "" {
+		return fmt.Errorf("safew: token is required")
+	}
+	chatID, err := parseSafewChatID(cfg.ChatID)
+	if err != nil {
+		return err
+	}
+
+	payload := safewMessage{
+		ChatID:    chatID,
+		Text:      "*" + escapeMarkdownV2(title) + "*\n" + escapeMarkdownV2(content),
+		ParseMode: "MarkdownV2",
+	}
+	body, err := json.Marshal(payload)
+	if err != nil {
+		return fmt.Errorf("safew marshal: %w", err)
+	}
+
+	resp, err := http.Post(s.endpoint(token), "application/json", bytes.NewReader(body))
+	if err != nil {
+		return fmt.Errorf("safew send: %w", err)
+	}
+	defer resp.Body.Close()
+
+	respBody, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return fmt.Errorf("safew read: %w", err)
+	}
+	if resp.StatusCode >= 400 {
+		desc := safewErrorDescription(respBody)
+		if desc != "" {
+			return fmt.Errorf("safew returned status %d: %s", resp.StatusCode, desc)
+		}
+		return fmt.Errorf("safew returned status %d", resp.StatusCode)
+	}
+
+	var api safewAPIResponse
+	if err := json.Unmarshal(respBody, &api); err != nil {
+		return fmt.Errorf("safew decode: %w", err)
+	}
+	if !api.OK {
+		desc := api.Description
+		if desc == "" {
+			desc = "ok=false"
+		}
+		return fmt.Errorf("safew: %s", desc)
+	}
+	return nil
+}
+
+func (s *SafeWSender) endpoint(token string) string {
+	base := s.apiBase
+	if base == "" {
+		base = safewAPIBase
+	}
+	return strings.TrimRight(base, "/") + "/bot" + token + "/sendMessage"
+}
+
+func parseSafewChatID(raw json.RawMessage) (string, error) {
+	raw = bytes.TrimSpace(raw)
+	if len(raw) == 0 || string(raw) == "null" {
+		return "", fmt.Errorf("safew: chat_id is required")
+	}
+	if raw[0] == '"' {
+		var id string
+		if err := json.Unmarshal(raw, &id); err != nil {
+			return "", fmt.Errorf("safew: invalid chat_id: %w", err)
+		}
+		id = strings.TrimSpace(id)
+		if id == "" {
+			return "", fmt.Errorf("safew: chat_id is required")
+		}
+		return id, nil
+	}
+	return string(raw), nil
+}
+
+func safewErrorDescription(body []byte) string {
+	var api safewAPIResponse
+	if err := json.Unmarshal(body, &api); err != nil {
+		return strings.TrimSpace(string(body))
+	}
+	return api.Description
+}
+
+func escapeMarkdownV2(s string) string {
+	var b strings.Builder
+	b.Grow(len(s))
+	for _, r := range s {
+		switch r {
+		case '_', '*', '[', ']', '(', ')', '~', '`', '>', '#', '+', '-', '=', '|', '{', '}', '.', '!', '\\':
+			b.WriteByte('\\')
+		}
+		b.WriteRune(r)
+	}
+	return b.String()
+}
@@ -0,0 +1,172 @@
+package adapter
+
+import (
+	"encoding/json"
+	"io"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	"aiaa-notification-service/internal/config"
+)
+
+func TestEscapeMarkdownV2(t *testing.T) {
+	tests := []struct {
+		name string
+		in   string
+		want string
+	}{
+		{name: "underscore and dot", in: "hello_world.", want: `hello\_world\.`},
+		{name: "empty", in: "", want: ""},
+		{name: "no specials", in: "hello", want: "hello"},
+		{name: "backslash first", in: `a\b`, want: `a\\b`},
+		{
+			name: "all specials",
+			in:   "_*[]()~`>#+-=|{}.!\\",
+			want: "\\_\\*\\[\\]\\(\\)\\~\\`\\>\\#\\+\\-\\=\\|\\{\\}\\.\\!\\\\",
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := escapeMarkdownV2(tt.in)
+			if got != tt.want {
+				t.Fatalf("escapeMarkdownV2(%q) = %q, want %q", tt.in, got, tt.want)
+			}
+		})
+	}
+}
+
+func TestNewSenderSafew(t *testing.T) {
+	s, err := NewSender("safew", &config.SMTPConfig{}, nil)
+	if err != nil {
+		t.Fatalf("NewSender(safew): %v", err)
+	}
+	if s.Type() != "safew" {
+		t.Fatalf("Type() = %q, want safew", s.Type())
+	}
+}
+
+func TestSafeWSenderSendSuccess(t *testing.T) {
+	var gotPath string
+	var gotBody map[string]any
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		gotPath = r.URL.Path
+		raw, _ := io.ReadAll(r.Body)
+		if err := json.Unmarshal(raw, &gotBody); err != nil {
+			t.Errorf("unmarshal request: %v", err)
+		}
+		w.Header().Set("Content-Type", "application/json")
+		_, _ = w.Write([]byte(`{"ok":true,"result":{}}`))
+	}))
+	defer srv.Close()
+
+	sender := &SafeWSender{apiBase: srv.URL}
+	cfg, _ := json.Marshal(map[string]any{"token": "tok-1", "chat_id": "123456789"})
+	if err := sender.Send("hello_world.", "price=1.5", cfg); err != nil {
+		t.Fatalf("Send: %v", err)
+	}
+	if gotPath != "/bottok-1/sendMessage" {
+		t.Fatalf("path = %q, want /bottok-1/sendMessage", gotPath)
+	}
+	if gotBody["chat_id"] != "123456789" {
+		t.Fatalf("chat_id = %#v, want \"123456789\"", gotBody["chat_id"])
+	}
+	if gotBody["parse_mode"] != "MarkdownV2" {
+		t.Fatalf("parse_mode = %#v, want MarkdownV2", gotBody["parse_mode"])
+	}
+	wantText := "*" + escapeMarkdownV2("hello_world.") + "*\n" + escapeMarkdownV2("price=1.5")
+	if gotBody["text"] != wantText {
+		t.Fatalf("text = %#v, want %#v", gotBody["text"], wantText)
+	}
+}
+
+func TestSafeWSenderSendNumericChatID(t *testing.T) {
+	var gotBody map[string]any
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		raw, _ := io.ReadAll(r.Body)
+		_ = json.Unmarshal(raw, &gotBody)
+		w.WriteHeader(http.StatusOK)
+		_, _ = w.Write([]byte(`{"ok":true}`))
+	}))
+	defer srv.Close()
+
+	sender := &SafeWSender{apiBase: srv.URL}
+	cfg := json.RawMessage(`{"token":"tok","chat_id":123456789}`)
+	if err := sender.Send("t", "c", cfg); err != nil {
+		t.Fatalf("Send: %v", err)
+	}
+	if gotBody["chat_id"] != "123456789" {
+		t.Fatalf("chat_id = %#v, want string \"123456789\"", gotBody["chat_id"])
+	}
+}
+
+func TestSafeWSenderSendOKFalse(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusOK)
+		_, _ = w.Write([]byte(`{"ok":false,"description":"chat not found"}`))
+	}))
+	defer srv.Close()
+
+	sender := &SafeWSender{apiBase: srv.URL}
+	cfg, _ := json.Marshal(map[string]string{"token": "tok", "chat_id": "1"})
+	err := sender.Send("t", "c", cfg)
+	if err == nil {
+		t.Fatal("expected error")
+	}
+	if !strings.Contains(err.Error(), "chat not found") {
+		t.Fatalf("error %q should contain description", err)
+	}
+}
+
+func TestSafeWSenderSendHTTPError(t *testing.T) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		w.WriteHeader(http.StatusUnauthorized)
+		_, _ = w.Write([]byte(`{"ok":false,"description":"Unauthorized"}`))
+	}))
+	defer srv.Close()
+
+	sender := &SafeWSender{apiBase: srv.URL}
+	cfg, _ := json.Marshal(map[string]string{"token": "bad", "chat_id": "1"})
+	err := sender.Send("t", "c", cfg)
+	if err == nil {
+		t.Fatal("expected error")
+	}
+	if !strings.Contains(err.Error(), "Unauthorized") {
+		t.Fatalf("error %q should contain description", err)
+	}
+}
+
+func TestSafeWSenderSendMissingFields(t *testing.T) {
+	hits := 0
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		hits++
+		w.WriteHeader(http.StatusOK)
+		_, _ = w.Write([]byte(`{"ok":true}`))
+	}))
+	defer srv.Close()
+
+	sender := &SafeWSender{apiBase: srv.URL}
+	cases := []json.RawMessage{
+		json.RawMessage(`{"chat_id":"1"}`),
+		json.RawMessage(`{"token":"tok"}`),
+		json.RawMessage(`{"token":"  ","chat_id":"1"}`),
+		json.RawMessage(`{"token":"tok","chat_id":""}`),
+		json.RawMessage(`{"token":"tok","chat_id":null}`),
+	}
+	for _, cfg := range cases {
+		if err := sender.Send("t", "c", cfg); err == nil {
+			t.Fatalf("expected error for config %s", cfg)
+		}
+	}
+	if hits != 0 {
+		t.Fatalf("unexpected HTTP calls: %d", hits)
+	}
+
+	if err := sender.Send("t", "c", json.RawMessage(`{`)); err == nil {
+		t.Fatal("expected error for invalid json")
+	}
+	if hits != 0 {
+		t.Fatalf("unexpected HTTP calls after invalid json: %d", hits)
+	}
+}