providerbuilder.php

<?php
 
namespace Bitrix\Sale\Internals;
 
use Bitrix\Main;
use Bitrix\Sale;
 
class ProviderBuilder extends ProviderBuilderBase
{
    public function addProductByBasketItem(Sale\BasketItemBase $basketItem)
    {
        $fields = array(
            'ITEM_CODE' => $basketItem->getProductId(),
            'BASKET_ID' => $basketItem->getId(),
            'BASKET_CODE' => $basketItem->getBasketCode(),
            'PRODUCT_ID' => $basketItem->getProductId(),
            'QUANTITY' => $basketItem->getNotPurchasedQuantity(),
            'RESERVED_QUANTITY' => $basketItem->getReservedQuantity(),
            'IS_BARCODE_MULTI' => $basketItem->isBarcodeMulti(),
            'BUNDLE_CHILD' => false,
            'SUBSCRIBE' => $basketItem->getField('SUBSCRIBE') === 'Y',
        );
 
        if ($basketItem instanceof Sale\BasketItem)
        {
            $fields['BUNDLE_CHILD'] = $basketItem->isBundleChild();
        }
 
        $this->addItem($basketItem->getProductId(), $fields);
    }
 
    public function addProductByShipmentItem(Sale\ShipmentItem $shipmentItem)
    {
        $basketItem = $shipmentItem->getBasketItem();
        $productId = $basketItem->getProductId();
 
        $fields = [
            'ITEM_CODE' => $productId,
            'BASKET_CODE' => $basketItem->getBasketCode(),
            'PRODUCT_ID' => $productId,
            'QUANTITY' => $shipmentItem->getQuantity(),
            'IS_BARCODE_MULTI' => $basketItem->isBarcodeMulti(),
            'BUNDLE_CHILD' => $basketItem->isBundleChild(),
            'SHIPMENT_ITEM_DATA' => [
                $shipmentItem->getInternalIndex() => $shipmentItem->getQuantity(),
            ],
            'SHIPMENT_ITEM' => $shipmentItem,
        ];
 
        if (Sale\Configuration::useStoreControl())
        {
            $storeData = Sale\Internals\Catalog\Provider::createMapShipmentItemStoreData($shipmentItem);
 
            $reservedQuantity = 0;
            $needReserveByStore = [];
 
            if ($storeData)
            {
                foreach ($storeData as $item)
                {
                    $reservedQuantity += $item['RESERVED_QUANTITY'];
 
                    $needReserveByStore[$item['STORE_ID']] = $item['RESERVED_QUANTITY'] > 0;
                }
            }
 
            $fields['STORE_DATA'] = array(
                $shipmentItem->getInternalIndex() => $storeData,
            );
 
            $fields['NEED_RESERVE_BY_STORE'] = [
                $shipmentItem->getInternalIndex() => $needReserveByStore,
            ];
        }
        else
        {
            $reservedQuantity = $basketItem->getReservedQuantity();
        }
 
        $fields['RESERVED_QUANTITY'] = $reservedQuantity;
        $fields['NEED_RESERVE'] = [
            $shipmentItem->getInternalIndex() => $reservedQuantity > 0,
        ];
 
        $this->addItem($productId, $fields);
    }
 
    public function addProductData(array $productData)
    {
        if ($productData['QUANTITY'] == 0)
        {
            return;
        }
 
        $shipmentItem = $productData['SHIPMENT_ITEM'] ?? null;
        $basketItem = $productData['BASKET_ITEM'];
        $productId = $productData['PRODUCT_ID'] ?? $basketItem->getProductId();
 
        $fields = [
            'ITEM_CODE' => $productId,
            'BASKET_CODE' => $basketItem->getBasketCode(),
            'PRODUCT_ID' => $productId,
            'QUANTITY' => $productData['QUANTITY'],
            'BUNDLE_PARENT' => $basketItem->isBundleParent(),
            'BUNDLE_CHILD' => $basketItem->isBundleChild(),
            'IS_BARCODE_MULTI' => $basketItem->isBarcodeMulti(),
            'RESERVED_QUANTITY' => $productData['RESERVED_QUANTITY'] ?? 0.0,
        ];
 
        if ($shipmentItem)
        {
            $fields['SHIPMENT_ITEM'] = $shipmentItem;
            $fields['SHIPMENT_ITEM_DATA'] = [$shipmentItem->getInternalIndex() => $shipmentItem->getQuantity()];
            $fields['NEED_RESERVE'] = [$shipmentItem->getInternalIndex() => $productData["NEED_RESERVE"] ?? null];
        }
 
        if (Sale\Configuration::useStoreControl())
        {
            if ($shipmentItem)
            {
                $storeData = Sale\Internals\Catalog\Provider::createMapShipmentItemStoreData($shipmentItem);
 
                if (!empty($storeData))
                {
                    $fields['STORE_DATA'] = [
                        $shipmentItem->getInternalIndex() => $storeData,
                    ];
                }
            }
        }
 
        if (isset($productData['NEED_RESERVE_BY_STORE']))
        {
            $fields['NEED_RESERVE_BY_STORE'] = $productData['NEED_RESERVE_BY_STORE'];
        }
 
        if (isset($productData['QUANTITY_BY_STORE']))
        {
            $fields['QUANTITY_BY_STORE'] = $productData['QUANTITY_BY_STORE'];
        }
 
        if (isset($productData['RESERVED_QUANTITY_BY_STORE']))
        {
            $fields['RESERVED_QUANTITY_BY_STORE'] = $productData['RESERVED_QUANTITY_BY_STORE'];
        }
 
        $this->addItem($productId, $fields);
    }
 
    public function setItemsResultAfterTryShip(PoolQuantity $pool, array $productTryShipList)
    {
        $products = $this->getItems();
 
        if (empty($products))
        {
            return new Sale\Result();
        }
 
        foreach ($products as $productId => $productData)
        {
            if (!isset($productTryShipList[$productId]))
            {
                continue;
            }
 
            if (empty($productData['SHIPMENT_ITEM_DATA_LIST']))
            {
                continue;
            }
 
            if (empty($productData['SHIPMENT_ITEM_LIST']))
            {
                continue;
            }
 
            foreach ($productData['SHIPMENT_ITEM_DATA_LIST'] as $shipmentItemIndex => $shipmentItemQuantity)
            {
                $shipmentItem = $productData['SHIPMENT_ITEM_LIST'][$shipmentItemIndex] ?? null;
                if ($shipmentItem === null)
                {
                    continue;
                }
 
                $shipment = $shipmentItem->getCollection()->getShipment();
 
                $coefficient = -1;
                if ($shipment->needShip() === Sale\Internals\Catalog\Provider::SALE_TRANSFER_PROVIDER_SHIPMENT_NEED_NOT_SHIP)
                {
                    $coefficient = 1;
                }
 
                $shipmentItemQuantity = $shipmentItem->getQuantity();
 
                $shipmentItemStoreCollection = $shipmentItem->getShipmentItemStoreCollection();
                if ($shipmentItemStoreCollection)
                {
                    foreach ($shipmentItemStoreCollection as $shipmentItemStore)
                    {
                        $quantity = $coefficient * $shipmentItemStore->getQuantity();
                        $pool->addByStore(PoolQuantity::POOL_QUANTITY_TYPE, $productId, $shipmentItemStore->getStoreId(), $quantity);
 
                        $shipmentItemQuantity -= $shipmentItemStore->getQuantity();
                    }
                }
 
                if ($shipmentItemQuantity > 0)
                {
                    $pool->add(PoolQuantity::POOL_QUANTITY_TYPE, $productId, $shipmentItemQuantity);
                }
 
                $foundItem = false;
                $poolItems = Sale\Internals\ItemsPool::get($shipment->getOrder()->getInternalId(), $productId);
                if (!empty($poolItems))
                {
                    foreach ($poolItems as $poolItem)
                    {
                        if (
                            $poolItem instanceof Sale\ShipmentItem
                            && $poolItem->getInternalIndex() == $shipmentItem->getInternalIndex()
                        )
                        {
                            $foundItem = true;
                            break;
                        }
                    }
                }
 
                if (!$foundItem)
                {
                    Sale\Internals\ItemsPool::add($shipment->getOrder()->getInternalId(), $productId, $shipmentItem);
                }
            }
        }
 
        return new Sale\Result();
    }
 
    public function getTransferClassName()
    {
        return '\Bitrix\Sale\Internals\TransferProvider';
    }
 
}