package task

import (
	"strconv"
	"time"

	"github.com/bytedance/sonic"
	"go.uber.org/zap"
	"gorm.io/gorm"

	"github.com/break/junhong_cmp_fiber/internal/model"
	"github.com/break/junhong_cmp_fiber/pkg/constants"
)

// getIntervalByTaskType 从配置中提取指定任务类型的轮询间隔（秒）
func getIntervalByTaskType(cfg *model.PollingConfig, taskType string) int {
	switch taskType {
	case constants.TaskTypePollingRealname:
		if cfg.RealnameCheckInterval != nil && *cfg.RealnameCheckInterval > 0 {
			return *cfg.RealnameCheckInterval
		}
	case constants.TaskTypePollingCarddata:
		if cfg.CarddataCheckInterval != nil && *cfg.CarddataCheckInterval > 0 {
			return *cfg.CarddataCheckInterval
		}
	case constants.TaskTypePollingPackage:
		if cfg.PackageCheckInterval != nil && *cfg.PackageCheckInterval > 0 {
			return *cfg.PackageCheckInterval
		}
	case constants.TaskTypePollingProtect:
		if cfg.ProtectCheckInterval != nil && *cfg.ProtectCheckInterval > 0 {
			return *cfg.ProtectCheckInterval
		}
	}
	return 0
}

// shortTaskType 从完整任务类型中提取简短名称（如 polling:carddata → carddata）
func shortTaskType(fullTaskType string) string {
	for i := len(fullTaskType) - 1; i >= 0; i-- {
		if fullTaskType[i] == ':' {
			return fullTaskType[i+1:]
		}
	}
	return fullTaskType
}

// parseTaskPayload 解析轮询任务载荷，返回 cardID
func parseTaskPayload(payload []byte, logger *zap.Logger) (uint, bool) {
	var p struct {
		CardID string `json:"card_id"`
	}
	if err := sonic.Unmarshal(payload, &p); err != nil {
		logger.Error("解析任务载荷失败", zap.Error(err))
		return 0, false
	}
	id, parseErr := strconv.ParseUint(p.CardID, 10, 64)
	if parseErr != nil {
		logger.Error("解析卡ID失败", zap.String("card_id", p.CardID), zap.Error(parseErr))
		return 0, false
	}
	return uint(id), true
}

// boolToStr 布尔值转字符串（Redis hash 存储用）
func boolToStr(b bool) string {
	if b {
		return "1"
	}
	return "0"
}

// isNotFound 判断是否为 GORM 记录不存在错误
func isNotFound(err error) bool {
	return err == gorm.ErrRecordNotFound
}

// isResetWindow 判断今天是否在运营商流量重置日窗口内
// 窗口 = 重置日当天 + 前一天（容错网关数据延迟上报）
func isResetWindow(now time.Time, resetDay int) bool {
	today := now.Day()
	if today == resetDay {
		return true
	}
	resetDate := time.Date(now.Year(), now.Month(), resetDay, 0, 0, 0, 0, now.Location())
	prevDay := resetDate.AddDate(0, 0, -1).Day()
	return today == prevDay
}