package logic

import (
	"context"

	"lone-services/pkg/utils"
	"lone-services/pkg/validate"
	product "lone-services/rpc/product/pb"
	"lone-services/services/product/internal/dao"
	"lone-services/services/product/internal/model"
	"lone-services/services/product/internal/svc"
	"lone-services/services/product/validator"

	"github.com/zeromicro/go-zero/core/logx"
	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
)

type NumberAddLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewNumberAddLogic(ctx context.Context, svcCtx *svc.ServiceContext) *NumberAddLogic {
	return &NumberAddLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *NumberAddLogic) NumberAdd(in *product.NumberAddReq) (*product.Response, error) {
	var req validator.ProductNumberAddValidator
	if msg := validate.ValidateFromProto(in, &req); msg != utils.StringEmpty {
		return outResponse(utils.ErrorParams, msg), nil
	}

	adminInfo := utils.GetUserFromCtx(l.ctx)
	if adminInfo.ID < utils.NumberOne {
		return failResponse(utils.ErrorNoLoginInfo), nil
	}

	if err := changeProductNumber(l.Logger, model.ProductModel{}.Init(), req.Id, dao.NumberTypeAdd, req.Number); err != nil {
		return mapNumberErr(err), nil
	}
	return okResponse(nil), nil
}

func mapNumberErr(err error) *product.Response {
	st, ok := status.FromError(err)
	if !ok {
		return failResponse(utils.Fail)
	}
	switch st.Code() {
	case codes.InvalidArgument:
		return outResponse(utils.ErrorParams, st.Message())
	case codes.NotFound:
		return failResponse(utils.ErrorNotFund)
	case codes.FailedPrecondition:
		return failResponse(utils.ErrorStockNotEnough)
	default:
		return failResponse(utils.Fail)
	}
}