package promo

import (
	"context"

	promoModel "github.com/perfect-panel/server/internal/model/promo"
	"github.com/perfect-panel/server/internal/svc"
	"github.com/perfect-panel/server/internal/types"
	"github.com/perfect-panel/server/pkg/logger"
	"github.com/perfect-panel/server/pkg/xerr"
	"github.com/pkg/errors"
)

type UpdatePromoRuleLogic struct {
	logger.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

// Update promo rule
func NewUpdatePromoRuleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdatePromoRuleLogic {
	return &UpdatePromoRuleLogic{
		Logger: logger.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *UpdatePromoRuleLogic) UpdatePromoRule(req *types.UpdatePromoRuleRequest) (*types.PromoRule, error) {
	if err := validateRulePayload(req.Type, req.Params, req.Priority, req.StartTime, req.EndTime); err != nil {
		return nil, err
	}

	data, err := l.svcCtx.PromoModel.FindRule(l.ctx, req.Id)
	if err != nil {
		if promoModel.IsNotFound(err) {
			return nil, xerr.NewErrCodeMsg(xerr.InvalidParams, "promo rule not found")
		}
		l.Errorw("[UpdatePromoRule] Database Query Error", logger.Field("error", err.Error()))
		return nil, wrapQueryError("get promo rule failed", err)
	}

	start, end := ruleTimes(req.StartTime, req.EndTime)
	data.Name = req.Name
	data.Type = req.Type
	data.Params = string(req.Params)
	data.Priority = req.Priority
	data.Enabled = *req.Enabled
	data.StartTime = start
	data.EndTime = end

	if err = l.svcCtx.PromoModel.UpdateRule(l.ctx, data); err != nil {
		l.Errorw("[UpdatePromoRule] Database Update Error", logger.Field("error", err.Error()))
		return nil, errors.Wrapf(xerr.NewErrCode(xerr.DatabaseUpdateError), "update promo rule failed: %v", err.Error())
	}
	clearRuleCache(l.ctx, l.svcCtx)
	resp := toRuleResponse(data)
	return &resp, nil
}