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"
	"gorm.io/gorm"
)

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

func NewSetPriceLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SetPriceLogic {
	return &SetPriceLogic{
		Logger: logger.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *SetPriceLogic) SetPrice(req *types.SetPromoPriceRequest) error {
	if _, err := l.svcCtx.PromoModel.FindRule(l.ctx, req.PromoRuleId); err != nil {
		l.Errorw("[SetPromoPrice] Find Rule Error", logger.Field("error", err.Error()))
		return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "find promo rule error: %v", err.Error())
	}
	items := make([]*promomodel.SubscribePromo, 0, len(req.Items))
	cacheKeys := make([]string, 0, len(req.Items))
	for _, item := range req.Items {
		sub, err := l.svcCtx.SubscribeModel.FindOne(l.ctx, item.SubscribeId)
		if err != nil {
			if errors.Is(err, gorm.ErrRecordNotFound) {
				return errors.Wrapf(xerr.NewErrCode(xerr.InvalidParams), "subscribe plan not found")
			}
			l.Errorw("[SetPromoPrice] Find Subscribe Error", logger.Field("error", err.Error()))
			return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseQueryError), "find subscribe error: %v", err.Error())
		}
		originPrice := sub.UnitPrice * item.Quantity
		if item.PromoPrice >= originPrice {
			return errors.Wrapf(xerr.NewErrCode(xerr.InvalidParams), "promo_price must be less than unit_price * quantity")
		}
		items = append(items, &promomodel.SubscribePromo{
			SubscribeId: item.SubscribeId,
			PromoRuleId: req.PromoRuleId,
			Quantity:    item.Quantity,
			PromoPrice:  item.PromoPrice,
		})
		cacheKeys = append(cacheKeys, subscribePromoCacheKey(item.SubscribeId))
	}
	if err := l.svcCtx.PromoModel.UpsertPrices(l.ctx, req.PromoRuleId, items); err != nil {
		l.Errorw("[SetPromoPrice] Database Error", logger.Field("error", err.Error()))
		return errors.Wrapf(xerr.NewErrCode(xerr.DatabaseUpdateError), "set promo price error: %v", err.Error())
	}
	if len(cacheKeys) > 0 {
		if err := l.svcCtx.Redis.Del(l.ctx, cacheKeys...).Err(); err != nil {
			l.Errorw("[SetPromoPrice] Delete Cache Error", logger.Field("error", err.Error()))
		}
	}
	return nil
}