"+html.EscapeString(body)+"") + } + return strings.Join(lines, "\n") +} + +func (a *App) sendTelegramMessage(ctx context.Context, token, chatID, text string) error { + return a.callTelegram(ctx, token, "sendMessage", map[string]any{ + "chat_id": chatID, + "text": text, + "parse_mode": "HTML", + "disable_web_page_preview": true, + }, nil) +} + +func (a *App) callTelegram(ctx context.Context, token, method string, payload any, result any) error { + token = strings.TrimSpace(token) + if token == "" || strings.ContainsAny(token, "/\\\r\n") { + return errors.New("Telegram Bot Token 无效") + } + base := strings.TrimRight(strings.TrimSpace(a.telegramURL), "/") + endpoint := base + "/bot" + url.PathEscape(token) + "/" + method + body, err := json.Marshal(payload) + if err != nil { + return err + } + req, err := http.NewRequestWithContext(ctx, http.MethodPost, endpoint, bytes.NewReader(body)) + if err != nil { + return err + } + req.Header.Set("Content-Type", "application/json") + req.Header.Set("User-Agent", "NewSzxcn-Email-Telegram/1.0") + client := &http.Client{Timeout: 12 * time.Second, CheckRedirect: func(*http.Request, []*http.Request) error { return http.ErrUseLastResponse }} + resp, err := client.Do(req) + if err != nil { + return errors.New("Telegram 请求失败，请检查网络连接和机器人配置") + } + defer resp.Body.Close() + raw, err := io.ReadAll(io.LimitReader(resp.Body, 1<<20)) + if err != nil { + return err + } + var apiResponse telegramAPIResponse + if err := json.Unmarshal(raw, &apiResponse); err != nil { + return fmt.Errorf("Telegram 返回了无效响应（HTTP %d）", resp.StatusCode) + } + if resp.StatusCode < 200 || resp.StatusCode >= 300 || !apiResponse.OK { + description := strings.TrimSpace(apiResponse.Description) + if description == "" { + description = fmt.Sprintf("HTTP %d", resp.StatusCode) + } + return fmt.Errorf("Telegram 发送失败: %s", description) + } + if result != nil && len(apiResponse.Result) > 0 { + if err := json.Unmarshal(apiResponse.Result, result); err != nil { + return err + } + } + return nil +} diff --git a/apps/api/internal/app/telegram_test.go b/apps/api/internal/app/telegram_test.go new file mode 100644 index 0000000..1a03b39 --- /dev/null +++ b/apps/api/internal/app/telegram_test.go @@ -0,0 +1,147 @@ +package app + +import ( + "context" + "encoding/json" + "net/http" + "net/http/httptest" + "strings" + "testing" + "time" +) + +func TestTelegramSettingsDiscoveryTestAndMailQueue(t *testing.T) { + type sentMessage struct { + ChatID string `json:"chat_id"` + Text string `json:"text"` + } + var sent []sentMessage + telegramServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + w.Header().Set("Content-Type", "application/json") + switch r.URL.Path { + case "/bottest-token/getUpdates": + _, _ = w.Write([]byte(`{"ok":true,"result":[{"update_id":7,"message":{"chat":{"id":123456789,"type":"private","first_name":"Zhenxi","last_name":"Shen"}}}]}`)) + case "/bottest-token/sendMessage": + var message sentMessage + if err := json.NewDecoder(r.Body).Decode(&message); err != nil { + t.Fatalf("decode Telegram message: %v", err) + } + sent = append(sent, message) + _, _ = w.Write([]byte(`{"ok":true,"result":{"message_id":8}}`)) + default: + http.NotFound(w, r) + } + })) + defer telegramServer.Close() + + a := newTestApp(t) + stopTestWorkers(a) + a.telegramURL = telegramServer.URL + server := httptest.NewServer(a.Router()) + defer server.Close() + admin := &testClient{t: t, server: server} + var login map[string]any + if code := admin.do("POST", "/api/auth/login", map[string]string{"email": "admin@lanqin.local", "password": "ChangeMe123!"}, &login); code != http.StatusOK { + t.Fatalf("login code=%d body=%v", code, login) + } + + var settings SystemSettings + if code := admin.do("GET", "/api/admin/settings", nil, &settings); code != http.StatusOK { + t.Fatalf("get settings code=%d", code) + } + payload := systemSettingsPayload(settings) + payload["telegramMailEnabled"] = true + payload["telegramBotToken"] = "test-token" + payload["telegramPrivateChatId"] = "123456789" + payload["telegramBodyMode"] = "full" + if code := admin.do("POST", "/api/admin/settings", payload, &settings); code != http.StatusOK { + t.Fatalf("save Telegram settings code=%d settings=%+v", code, settings) + } + if !settings.TelegramMailEnabled || !settings.TelegramBotTokenSet || settings.TelegramPrivateChatID != "123456789" || settings.TelegramBodyMode != "full" { + t.Fatalf("unexpected Telegram settings: %+v", settings) + } + if a.config().TelegramBotToken != "test-token" { + t.Fatal("Telegram token was not persisted in runtime config") + } + + var discovered map[string]string + if code := admin.do("POST", "/api/admin/settings/telegram/discover", map[string]string{"botToken": ""}, &discovered); code != http.StatusOK { + t.Fatalf("discover chat code=%d response=%v", code, discovered) + } + if discovered["chatId"] != "123456789" || discovered["displayName"] != "Zhenxi Shen" { + t.Fatalf("unexpected discovered chat: %v", discovered) + } + var testResult map[string]any + if code := admin.do("POST", "/api/admin/settings/telegram/test", map[string]string{"botToken": "", "chatId": ""}, &testResult); code != http.StatusOK { + t.Fatalf("test Telegram code=%d response=%v", code, testResult) + } + if len(sent) != 1 || sent[0].ChatID != "123456789" || !strings.Contains(sent[0].Text, "通知测试") { + t.Fatalf("unexpected Telegram test message: %+v", sent) + } + + sent = nil + receivedAt := time.Date(2026, 8, 6, 9, 30, 0, 0, time.UTC) + a.enqueueTelegramMailNotification(context.Background(), "mail_test_telegram", storedMessage{ + RecipientAddr: "admin@example.com", + Subject: "账单 <已生成>", + From: "billing@example.net", + FromName: "Billing & Support", + ReceivedAt: receivedAt, + BodyText: "这是邮件正文，包含