package cron_service

import (
	"fmt"
	"github.com/robfig/cron/v3"
	"github.com/sirupsen/logrus"
	"github.com/unknwon/com"
	"message-nest/models"
	"message-nest/pkg/constant"
	"message-nest/service/send_message_service"
)

var ClearLogsTaskId cron.EntryID

// ClearLogs 清除日志的定时任务
func ClearLogs() {
	var errStr string
	sm := send_message_service.SendMessageService{
		TaskID: constant.CleanLogsTaskId,
		DefaultLogger: logrus.WithFields(logrus.Fields{
			"prefix": "[Cron Clear Logs]",
		}),
	}
	sm.Status = send_message_service.SendSuccess

	sm.LogsAndStatusMark("开始清除日志", sm.Status)

	setting, err := models.GetSettingByKey(constant.LogsCleanSectionName, constant.LogsCleanKeepKeyName)
	if err != nil {
		errStr = fmt.Sprintf("获取日志的保留数失败，原因：%s", err)
		sm.LogsAndStatusMark(errStr, send_message_service.SendFail)
	}

	keepNum := com.StrTo(setting.Value).MustInt()
	affectedRows, err := models.DeleteOutDateLogs(keepNum)
	if err != nil {
		errStr = fmt.Sprintf("删除日志失败，原因：%s", err)
		sm.LogsAndStatusMark(errStr, send_message_service.SendFail)
	} else {
		errStr = fmt.Sprintf("删除日志成功，删除条数：%d，保留数目：%d", affectedRows, keepNum)
		sm.LogsAndStatusMark(errStr, sm.Status)
	}

	sm.RecordSendLog()
}

type CronService struct {
}

// StartLogsCronRun 启动注册清除任务定时任务
func (cs *CronService) StartLogsCronRun() {
	// 注册任务
	setting, err := models.GetSettingByKey(constant.LogsCleanSectionName, constant.LogsCleanCronKeyName)
	if err != nil {
		logrus.Error(fmt.Sprintf("获取日志的cron失败，原因：%s", err))
	}
	ClearLogsTaskId = AddTask(ScheduledTask{
		Schedule: setting.Value,
		Job:      ClearLogs,
	})

	// 添加任务
	err = models.AddSendTaskWithID("日志定时清除", constant.CleanLogsTaskId, "admin")
	if err != nil {
		logrus.Error(fmt.Sprintf("添加日志定时清除任务失败，原因：%s", err))
	}
}

// UpdateLogsCronRun 更新清除任务定时任务
func (cs *CronService) UpdateLogsCronRun(cron string) {
	RemoveTask(ClearLogsTaskId)
	ClearLogsTaskId = AddTask(ScheduledTask{
		Schedule: cron,
		Job:      ClearLogs,
	})
	logrus.Error(fmt.Sprintf("更新日志的cron成功，%s", cron))
	logrus.Error(fmt.Sprintf("所有的定时任务： %s", TaskList))

}

// StartLogsCronRun 启动的时候开启定时任务
func StartLogsCronRun() {
	logrus.Infof("开始注册定时清除日志任务...")
	cs := CronService{}
	cs.StartLogsCronRun()
}