package task

import (
	"context"
	"fmt"
	"time"

	"github.com/bytedance/sonic"
	"github.com/hibiken/asynq"
	"go.uber.org/zap"

	"github.com/break/junhong_cmp_fiber/internal/application/auditarchive"
	"github.com/break/junhong_cmp_fiber/pkg/constants"
)

// IntegrationDailyArchivePayload 是人工补档时可选的任务载荷。
type IntegrationDailyArchivePayload struct {
	ArchiveDate string `json:"archive_date"`
}

// IntegrationArchiveHandler 处理 Integration Log 每日归档。
type IntegrationArchiveHandler struct {
	service *auditarchive.Service
	logger  *zap.Logger
	enabled bool
}

// NewIntegrationArchiveHandler 创建 Integration Log 归档任务处理器。
func NewIntegrationArchiveHandler(service *auditarchive.Service, logger *zap.Logger, enabled bool) *IntegrationArchiveHandler {
	return &IntegrationArchiveHandler{service: service, logger: logger, enabled: enabled}
}

// HandleDaily 执行前一完整自然日归档，或按任务载荷补档指定自然日。
func (h *IntegrationArchiveHandler) HandleDaily(ctx context.Context, task *asynq.Task) error {
	if !h.enabled {
		h.logger.Info("Integration Log 日归档已停用，跳过任务")
		return nil
	}
	if h.service == nil {
		return fmt.Errorf("Integration Log 归档服务未配置")
	}
	var err error
	if len(task.Payload()) == 0 {
		err = h.service.ArchivePreviousIntegrationDay(ctx)
	} else {
		var payload IntegrationDailyArchivePayload
		if unmarshalErr := sonic.Unmarshal(task.Payload(), &payload); unmarshalErr != nil {
			return fmt.Errorf("解析 Integration Log 每日归档任务载荷失败: %w", unmarshalErr)
		}
		date, parseErr := parseArchiveDate(payload.ArchiveDate)
		if parseErr != nil {
			return parseErr
		}
		err = h.service.ArchiveIntegrationDate(ctx, date)
	}
	if err != nil {
		h.logger.Error("Integration Log 每日冷归档失败", zap.Error(err))
		return err
	}
	h.logger.Info("Integration Log 每日冷归档完成")
	return nil
}

func parseArchiveDate(value string) (time.Time, error) {
	location, err := time.LoadLocation(constants.AuditArchiveTimezone)
	if err != nil {
		return time.Time{}, fmt.Errorf("加载 Integration Log 归档时区失败: %w", err)
	}
	date, err := time.ParseInLocation(time.DateOnly, value, location)
	if err != nil {
		return time.Time{}, fmt.Errorf("解析 Integration Log 归档日期失败: %w", err)
	}
	return date, nil
}