From d4980da4bce3617d372a3cacacb047ffe9521746 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=A2=A8=E9=9B=AA?= <dzx1283@126.com>
Date: Mon, 17 Aug 2026 12:26:20 +0800
Subject: [PATCH] test: add persistent Telegram digest and wizard state helpers

---
 .../telegram/persistence_hardening.go         | 60 +++++++++++++++++++
 1 file changed, 60 insertions(+)
 create mode 100644 backend/internal/telegram/persistence_hardening.go

diff --git a/backend/internal/telegram/persistence_hardening.go b/backend/internal/telegram/persistence_hardening.go
new file mode 100644
index 0000000..4df5e00
--- /dev/null
+++ b/backend/internal/telegram/persistence_hardening.go
@@ -0,0 +1,60 @@
+package telegram
+
+import (
+	"encoding/json"
+	"strings"
+
+	"github.com/kazeyukiro/3m-ui/backend/internal/database/models"
+	"gorm.io/gorm"
+)
+
+const digestDatePrefix = "telegram_daily_digest_"
+const wizardPrefix = "telegram_wizard_"
+
+func dailyDigestSent(db *gorm.DB, day string) bool {
+	if db == nil || day == "" { return false }
+	var row models.PanelSetting
+	return db.Where("key = ?", digestDatePrefix+day).First(&row).Error == nil
+}
+
+func markDailyDigestSent(db *gorm.DB, day string) error {
+	if db == nil || day == "" { return nil }
+	key := digestDatePrefix + day
+	var row models.PanelSetting
+	err := db.Where("key = ?", key).First(&row).Error
+	if err == nil { return nil }
+	if err != gorm.ErrRecordNotFound { return err }
+	return db.Create(&models.PanelSetting{Key: key, Value: day}).Error
+}
+
+func saveWizardState(db *gorm.DB, chatID string, state addClientWizard) error {
+	if db == nil || strings.TrimSpace(chatID) == "" { return nil }
+	raw, err := json.Marshal(state)
+	if err != nil { return err }
+	key := wizardPrefix + strings.TrimSpace(chatID)
+	var row models.PanelSetting
+	err = db.Where("key = ?", key).First(&row).Error
+	if err == gorm.ErrRecordNotFound {
+		return db.Create(&models.PanelSetting{Key: key, Value: string(raw)}).Error
+	}
+	if err != nil { return err }
+	row.Value = string(raw)
+	return db.Save(&row).Error
+}
+
+func loadWizardState(db *gorm.DB, chatID string) (*addClientWizard, error) {
+	if db == nil || strings.TrimSpace(chatID) == "" { return nil, nil }
+	var row models.PanelSetting
+	if err := db.Where("key = ?", wizardPrefix+strings.TrimSpace(chatID)).First(&row).Error; err != nil {
+		if err == gorm.ErrRecordNotFound { return nil, nil }
+		return nil, err
+	}
+	var state addClientWizard
+	if err := json.Unmarshal([]byte(row.Value), &state); err != nil { return nil, err }
+	return &state, nil
+}
+
+func clearWizardState(db *gorm.DB, chatID string) error {
+	if db == nil || strings.TrimSpace(chatID) == "" { return nil }
+	return db.Where("key = ?", wizardPrefix+strings.TrimSpace(chatID)).Delete(&models.PanelSetting{}).Error
+}