ScopedTierPriceManagement.php

Go to the documentation of this file.

<?php
namespace Magento\Catalog\Model\Product;

use Magento\Catalog\Api\Data\ProductTierPriceInterface;
use Magento\Catalog\Api\ScopedProductTierPriceManagementInterface;
use Magento\Store\Model\ScopeInterface;

class ScopedTierPriceManagement implements ScopedProductTierPriceManagementInterface
{
    private $productRepository;

    private $storeManager;

    private $config;

    private $priceModifier;

    private $tierPriceManagement;

    public function __construct(
        \Magento\Catalog\Api\ProductRepositoryInterface $productRepository,
        \Magento\Store\Model\StoreManagerInterface $storeManager,
        \Magento\Framework\App\Config\ScopeConfigInterface $config,
        PriceModifier $priceModifier,
        TierPriceManagement $tierPriceManagement
    ) {
        $this->productRepository = $productRepository;
        $this->storeManager = $storeManager;
        $this->priceModifier = $priceModifier;
        $this->config = $config;
        $this->tierPriceManagement = $tierPriceManagement;
    }

    public function add($sku, ProductTierPriceInterface $tierPrice)
    {
        $product = $this->productRepository->get($sku, ['edit_mode' => true]);
        $product->setTierPrices(
            $this->prepareTierPrices($product->getTierPrices(), $tierPrice)
        );
        try {
            $this->productRepository->save($product);
        } catch (\Exception $e) {
            throw new \Magento\Framework\Exception\CouldNotSaveException(__("The group price couldn't be saved."));
        }
        return true;
    }

    private function prepareTierPrices(array $tierPrices, ProductTierPriceInterface $tierPrice)
    {
        $this->validate($tierPrice);
        $websiteId = $this->getWebsiteId();

        foreach ($tierPrices as $index => $item) {
            $tierPriceWebsite = $tierPrice->getExtensionAttributes()
                ? $tierPrice->getExtensionAttributes()->getWebsiteId()
                : 0;

            if ($item->getCustomerGroupId() == $tierPrice->getCustomerGroupId()
                && $websiteId == $tierPriceWebsite
                && $item->getQty() == $tierPrice->getQty()
            ) {
                unset($tierPrices[$index]);
                break;
            }
        }

        $tierPrices[] = $tierPrice;
        return $tierPrices;
    }

    private function getWebsiteId()
    {
        $websiteIdentifier = 0;
        $value = $this->config->getValue('catalog/price/scope', ScopeInterface::SCOPE_WEBSITE);
        if ($value != 0) {
            $websiteIdentifier = $this->storeManager->getWebsite()->getId();
        }

        return $websiteIdentifier;
    }

    private function validate(ProductTierPriceInterface $tierPrice)
    {
        $data = ['qty' => $tierPrice->getQty(), 'price' => $tierPrice->getValue()];
        foreach ($data as $value) {
            if (!is_float($value) || $value <= 0) {
                throw new \Magento\Framework\Exception\InputException(
                    __('The data was invalid. Verify the data and try again.')
                );
            }
        }
    }

    public function remove($sku, ProductTierPriceInterface $tierPrice)
    {
        $product = $this->productRepository->get($sku, ['edit_mode' => true]);
        $this->priceModifier->removeTierPrice(
            $product,
            $tierPrice->getCustomerGroupId(),
            $tierPrice->getQty(),
            $this->getWebsiteId()
        );
        return true;
    }

    public function getList($sku, $customerGroupId)
    {
        return $this->tierPriceManagement->getList($sku, $customerGroupId);
    }
}