package services

import (
	"gpt-load/internal/config"
	"gpt-load/internal/db"
	"gpt-load/internal/models"
	"time"

	"github.com/sirupsen/logrus"
)

// LogCleanupService 负责清理过期的请求日志
type LogCleanupService struct {
	stopCh chan struct{}
}

// NewLogCleanupService 创建新的日志清理服务
func NewLogCleanupService() *LogCleanupService {
	return &LogCleanupService{
		stopCh: make(chan struct{}),
	}
}

// Start 启动日志清理服务
func (s *LogCleanupService) Start() {
	go s.run()
	logrus.Info("Log cleanup service started")
}

// Stop 停止日志清理服务
func (s *LogCleanupService) Stop() {
	close(s.stopCh)
	logrus.Info("Log cleanup service stopped")
}

// run 运行日志清理的主循环
func (s *LogCleanupService) run() {
	// 每天凌晨2点执行清理任务
	ticker := time.NewTicker(24 * time.Hour)
	defer ticker.Stop()

	// 启动时先执行一次清理
	s.cleanupExpiredLogs()

	for {
		select {
		case <-ticker.C:
			s.cleanupExpiredLogs()
		case <-s.stopCh:
			return
		}
	}
}

// cleanupExpiredLogs 清理过期的请求日志
func (s *LogCleanupService) cleanupExpiredLogs() {
	if db.DB == nil {
		logrus.Error("Database connection is not available for log cleanup")
		return
	}

	// 获取日志保留天数配置
	settingsManager := config.GetSystemSettingsManager()
	if settingsManager == nil {
		logrus.Error("System settings manager is not available for log cleanup")
		return
	}

	settings := settingsManager.GetSettings()
	retentionDays := settings.RequestLogRetentionDays

	if retentionDays <= 0 {
		logrus.Debug("Log retention is disabled (retention_days <= 0)")
		return
	}

	// 计算过期时间点
	cutoffTime := time.Now().AddDate(0, 0, -retentionDays).UTC()

	// 执行删除操作
	result := db.DB.Where("timestamp < ?", cutoffTime).Delete(&models.RequestLog{})
	if result.Error != nil {
		logrus.WithError(result.Error).Error("Failed to cleanup expired request logs")
		return
	}

	if result.RowsAffected > 0 {
		logrus.WithFields(logrus.Fields{
			"deleted_count":  result.RowsAffected,
			"cutoff_time":    cutoffTime.Format(time.RFC3339),
			"retention_days": retentionDays,
		}).Info("Successfully cleaned up expired request logs")
	} else {
		logrus.Debug("No expired request logs found to cleanup")
	}
}

// CleanupNow 立即执行一次日志清理
func (s *LogCleanupService) CleanupNow() {
	logrus.Info("Manual log cleanup triggered")
	s.cleanupExpiredLogs()
}