BaseSku.php

<?php
 
namespace Bitrix\Catalog\v2\Sku;
 
use Bitrix\Catalog\v2\Barcode\BarcodeCollection;
use Bitrix\Catalog\v2\Barcode\BarcodeRepositoryContract;
use Bitrix\Catalog\v2\Barcode\HasBarcodeCollection;
use Bitrix\Catalog\v2\BaseIblockElementEntity;
use Bitrix\Catalog\v2\Iblock\IblockInfo;
use Bitrix\Catalog\v2\Image\ImageCollection;
use Bitrix\Catalog\v2\Image\ImageRepositoryContract;
use Bitrix\Catalog\v2\MeasureRatio\HasMeasureRatioCollection;
use Bitrix\Catalog\v2\MeasureRatio\MeasureRatioCollection;
use Bitrix\Catalog\v2\MeasureRatio\MeasureRatioRepositoryContract;
use Bitrix\Catalog\v2\Price\HasPriceCollection;
use Bitrix\Catalog\v2\Price\PriceCollection;
use Bitrix\Catalog\v2\Price\PriceRepositoryContract;
use Bitrix\Catalog\v2\Property\PropertyRepositoryContract;
use Bitrix\Catalog\v2\StoreProduct\StoreProductCollection;
use Bitrix\Catalog\v2\StoreProduct\StoreProductRepositoryContract;
use Bitrix\Catalog\v2\StoreProduct\HasStoreProductCollection;
use Bitrix\Catalog\v2\StoreProduct\StoreProduct;
 
abstract class BaseSku extends BaseIblockElementEntity
    implements HasPriceCollection, HasMeasureRatioCollection, HasBarcodeCollection, HasStoreProductCollection
{
    protected $priceCollection;
    protected $priceRepository;
 
    protected $measureRatioRepository;
    protected $measureRatioCollection;
 
    protected $barcodeRepository;
    protected $barcodeCollection;
 
    protected $storeProductRepository;
    protected $storeProductCollection;
 
    public function __construct(
        IblockInfo $iblockInfo,
        SkuRepositoryContract $skuRepository,
        PropertyRepositoryContract $propertyRepository,
        ImageRepositoryContract $imageRepository,
        PriceRepositoryContract $priceRepository,
        MeasureRatioRepositoryContract $measureRatioRepository,
        BarcodeRepositoryContract $barcodeRepository,
        StoreProductRepositoryContract $storeProductRepository
    )
    {
        parent::__construct($iblockInfo, $skuRepository, $propertyRepository, $imageRepository);
        $this->priceRepository = $priceRepository;
        $this->measureRatioRepository = $measureRatioRepository;
        $this->barcodeRepository = $barcodeRepository;
        $this->storeProductRepository = $storeProductRepository;
    }
 
    public function getFrontImageCollection(): ImageCollection
    {
        $collection = $this->getImageCollection();
        if ($collection->isEmpty() && $this->getParent())
        {
            $parentCollection = $this->getParent()->getImageCollection();
            if (!$parentCollection->isEmpty())
            {
                return $parentCollection;
            }
        }
 
        return $collection;
    }
 
    public function getPriceCollection(): PriceCollection
    {
        if ($this->priceCollection === null)
        {
            // ToDo make lazy load like sku collection with iterator callback?
            $this->setPriceCollection($this->loadPriceCollection());
        }
 
        return $this->priceCollection;
    }
 
    protected function loadPriceCollection(): PriceCollection
    {
        return $this->priceRepository->getCollectionByParent($this);
    }
 
    public function setPriceCollection(PriceCollection $priceCollection): self
    {
        $priceCollection->setParent($this);
 
        $this->priceCollection = $priceCollection;
 
        return $this;
    }
 
    public function getMeasureRatioCollection(): MeasureRatioCollection
    {
        if ($this->measureRatioCollection === null)
        {
            $this->setMeasureRatioCollection($this->loadMeasureRatioCollection());
        }
 
        return $this->measureRatioCollection;
    }
 
    protected function loadMeasureRatioCollection(): MeasureRatioCollection
    {
        return $this->measureRatioRepository->getCollectionByParent($this);
    }
 
    public function setMeasureRatioCollection(MeasureRatioCollection $measureRatioCollection): self
    {
        $measureRatioCollection->setParent($this);
 
        $this->measureRatioCollection = $measureRatioCollection;
 
        return $this;
    }
 
    public function getBarcodeCollection(): BarcodeCollection
    {
        if ($this->barcodeCollection === null)
        {
            $this->setBarcodeCollection($this->loadBarcodeCollection());
        }
 
        return $this->barcodeCollection;
    }
 
    protected function loadBarcodeCollection(): BarcodeCollection
    {
        return $this->barcodeRepository->getCollectionByParent($this);
    }
 
    public function setBarcodeCollection(BarcodeCollection $barcodeCollection): self
    {
        $barcodeCollection->setParent($this);
 
        $this->barcodeCollection = $barcodeCollection;
 
        return $this;
    }
 
    public function getStoreProductCollection(): StoreProductCollection
    {
        if ($this->storeProductCollection === null)
        {
            $this->setStoreProductCollection($this->loadStoreProductCollection());
        }
 
        return $this->storeProductCollection;
    }
 
    protected function loadStoreProductCollection(): StoreProductCollection
    {
        return $this->storeProductRepository->getCollectionByParent($this);
    }
 
    public function setStoreProductCollection(StoreProductCollection $storeProductCollection): self
    {
        $storeProductCollection->setParent($this);
 
        $this->storeProductCollection = $storeProductCollection;
 
        return $this;
    }
}