package mail

import (
	"context"
	"log/slog"

	"gorm.io/gorm"

	"ws_usernode/internal/model"
)

// 邮件状态（mail_logs.status）。
const (
	StatusPending = "pending" // 已入队，等待直发
	StatusSent    = "sent"    // 发送成功
	StatusFailed  = "failed"  // 发送失败，等待重试
)

// MaxRetries 单封邮件最大重试次数；超出后停止重试，保留失败记录（PLAN F5）。
const MaxRetries = 5

// Retryable 可由 cron 定时调用的邮件队列重试接口。
type Retryable interface {
	Retry(ctx context.Context, limit int) error
}

// QueuedMailer 邮件队列实现（PLAN F5）：每封邮件先落 mail_logs 再同步尝试直发；
// 失败记录 error 与 retry_count，由 cron 定时重试；成功标记 sent。
// Send 恒返回 nil（除 DB 落库失败）：邮件失败不阻断业务
// （OTP 双通道、审批结果等，见 AuthService.UserOTPSend 的既有约定）。
type QueuedMailer struct {
	base Mailer
	db   *gorm.DB
	log  *slog.Logger
}

// NewQueuedMailer 创建邮件队列，包装底层直发实现（SMTP 或 LogMailer）。
func NewQueuedMailer(base Mailer, db *gorm.DB, log *slog.Logger) *QueuedMailer {
	return &QueuedMailer{base: base, db: db, log: log}
}

// Send 入队并尝试直发。返回 nil 表示已受理；实际送达与否见 mail_logs。
func (m *QueuedMailer) Send(ctx context.Context, to, subject, body string) error {
	entry := model.MailLog{To: to, Subject: subject, Body: body, Status: StatusPending}
	if err := m.db.WithContext(ctx).Create(&entry).Error; err != nil {
		m.log.Error("mail: 记录 mail_log 失败", "err", err)
		return err
	}
	if err := m.base.Send(ctx, to, subject, body); err != nil {
		_ = m.db.WithContext(ctx).Model(&entry).Updates(map[string]any{
			"status": StatusFailed, "error": err.Error(), "retry_count": 1,
		}).Error
		m.log.Warn("mail: 直发失败，进入重试队列", "to", to, "subject", subject, "err", err)
		return nil // 非阻断
	}
	_ = m.db.WithContext(ctx).Model(&entry).Update("status", StatusSent).Error
	return nil
}

// Retry 重试失败邮件（retry_count < limit，按原内容重发）。成功标记 sent，
// 失败累加 retry_count 并刷新 error；达到上限后不再挑选，保留失败记录待人工处理。
func (m *QueuedMailer) Retry(ctx context.Context, limit int) error {
	if limit <= 0 {
		return nil
	}
	var rows []model.MailLog
	if err := m.db.WithContext(ctx).
		Where("status = ? AND retry_count < ?", StatusFailed, limit).
		Find(&rows).Error; err != nil {
		return err
	}
	for _, r := range rows {
		if err := m.base.Send(ctx, r.To, r.Subject, r.Body); err != nil {
			_ = m.db.WithContext(ctx).Model(&model.MailLog{}).Where("id = ?", r.ID).Updates(map[string]any{
				"retry_count": r.RetryCount + 1, "error": err.Error(),
			}).Error
			m.log.Warn("mail: 重试仍失败", "id", r.ID, "to", r.To, "retry_count", r.RetryCount+1, "err", err)
			continue
		}
		_ = m.db.WithContext(ctx).Model(&model.MailLog{}).Where("id = ?", r.ID).Update("status", StatusSent).Error
		m.log.Info("mail: 重试成功", "id", r.ID, "to", r.To)
	}
	return nil
}