package stock

import (
	"context"

	"muyu-apiserver/gateway/internal/svc"
	"muyu-apiserver/gateway/internal/types"
	"muyu-apiserver/rpc/inventory/pb"

	"github.com/zeromicro/go-zero/core/logx"
)

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

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

func (l *GetStockByLocationLogic) GetStockByLocation() (resp *types.StockGroupResp, err error) {
	rpcResp, err := l.svcCtx.InventoryRpc.GetStockGroup(l.ctx, &pb.StockGroupReq{
		GroupBy: "location",
	})
	if err != nil {
		return nil, err
	}

	list := make([]types.StockGroupItem, 0, len(rpcResp.List))
	for _, item := range rpcResp.List {
		list = append(list, types.StockGroupItem{
			Name:     item.Name,
			Count:    item.Count,
			Quantity: item.Quantity,
		})
	}

	return &types.StockGroupResp{
		List: list,
	}, nil
}