Sku.php

<?php
 
namespace Bitrix\Catalog\v2\Sku;
 
use Bitrix\Catalog\ProductTable;
use Bitrix\Catalog\v2\Barcode\BarcodeRepositoryContract;
use Bitrix\Catalog\v2\BaseCollection;
use Bitrix\Catalog\v2\BaseEntity;
use Bitrix\Catalog\v2\Iblock\IblockInfo;
use Bitrix\Catalog\v2\Image\ImageRepositoryContract;
use Bitrix\Catalog\v2\MeasureRatio\MeasureRatioRepositoryContract;
use Bitrix\Catalog\v2\Price\PriceRepositoryContract;
use Bitrix\Catalog\v2\Product\BaseProduct;
use Bitrix\Catalog\v2\Property\Property;
use Bitrix\Catalog\v2\Property\PropertyRepositoryContract;
use Bitrix\Catalog\v2\StoreProduct\StoreProductRepositoryContract;
use Bitrix\Main\Result;
 
class Sku extends BaseSku
{
    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,
            $priceRepository,
            $measureRatioRepository,
            $barcodeRepository,
            $storeProductRepository
        );
 
        $this->setIblockId($this->iblockInfo->getSkuIblockId());
        $this->setType(ProductTable::TYPE_FREE_OFFER);
    }
 
    public function getDetailUrl(): string
    {
        $detailUrl = parent::getDetailUrl();
 
        if (!$detailUrl && $product = $this->getParent())
        {
            $detailUrl = $product->getDetailUrl();
        }
 
        return $detailUrl;
    }
 
    public function setParentCollection(?BaseCollection $collection): BaseEntity
    {
        // ToDo check for correct parent iblock for iblockelemententities!
        parent::setParentCollection($collection);
 
        if ($this->isNew())
        {
            $this->checkProductLink();
        }
 
        return $this;
    }
 
    public function checkProductLink(): self
    {
        $product = $this->getParent();
 
        if ($product)
        {
            $this->linkProduct($product);
        }
        else
        {
            $this->unlinkProduct();
        }
 
        return $this;
    }
 
    protected function linkProduct(BaseProduct $product): self
    {
        if (!$this->isNew())
        {
            $this->setProductLinkProperty($product->getId());
        }
 
        $this->setType(ProductTable::TYPE_OFFER);
 
        return $this;
    }
 
    protected function unlinkProduct(): self
    {
        $this->unsetProductLinkProperty();
        $this->setType(ProductTable::TYPE_FREE_OFFER);
 
        return $this;
    }
 
    protected function getProductLinkProperty(): ?Property
    {
        $skuPropertyId = $this->iblockInfo->getSkuPropertyId();
 
        if ($skuPropertyId)
        {
            return $this->getPropertyCollection()->findBySetting('ID', $skuPropertyId);
        }
 
        return null;
    }
 
    protected function setProductLinkProperty(int $productId): self
    {
        $property = $this->getProductLinkProperty();
 
        if ($property)
        {
            // ToDo do we need property values type casting?
            $property->getPropertyValueCollection()->setValues((string)$productId);
        }
 
        return $this;
    }
 
    protected function unsetProductLinkProperty(): self
    {
        $property = $this->getProductLinkProperty();
 
        if ($property)
        {
            $property->getPropertyValueCollection()->setValues(null);
        }
 
        return $this;
    }
 
    public function saveInternalEntity(): Result
    {
        $isNeedCheckProductLinkAfterSaving = $this->isNew();
        $result = parent::saveInternalEntity();
 
        if ($isNeedCheckProductLinkAfterSaving && $result->isSuccess())
        {
            $product = $this->getParent();
            if ($product && $this->iblockInfo->getSkuPropertyId())
            {
                \CIBlockElement::SetPropertyValuesEx(
                    $this->getId(),
                    $this->iblockInfo->getSkuIblockId(),
                    [$this->iblockInfo->getSkuPropertyId() => $product->getId()]
                );
            }
 
            $this->checkProductLink();
        }
 
        return $result;
    }
}