AssignToDefaultStockDefaultSourceValidator.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventoryCatalog\Model\StockSourceLink\Validator;

use Magento\Framework\Validation\ValidationResult;
use Magento\Framework\Validation\ValidationResultFactory;
use Magento\InventoryApi\Model\StockSourceLinkValidatorInterface;
use Magento\InventoryApi\Api\Data\StockSourceLinkInterface;
use Magento\InventoryCatalogApi\Api\DefaultSourceProviderInterface;
use Magento\InventoryCatalogApi\Api\DefaultStockProviderInterface;

class AssignToDefaultStockDefaultSourceValidator implements StockSourceLinkValidatorInterface
{
    private $validationResultFactory;

    private $defaultSourceProvider;

    private $defaultStockProvider;

    public function __construct(
        ValidationResultFactory $validationResultFactory,
        DefaultSourceProviderInterface $defaultSourceProvider,
        DefaultStockProviderInterface $defaultStockProvider
    ) {
        $this->validationResultFactory = $validationResultFactory;
        $this->defaultSourceProvider = $defaultSourceProvider;
        $this->defaultStockProvider = $defaultStockProvider;
    }

    public function validate(StockSourceLinkInterface $link): ValidationResult
    {
        $initialAssignment = $this->isInitialAssignment($link);
        $linkContainDefaultSourceOrStock = $this->isLinkContainDefaultSourceOrStock($link);
        $errors = [];
        if (!$initialAssignment && $linkContainDefaultSourceOrStock) {
            $errors[] = __('Can not save link related to Default Source or Default Stock');
        }

        return $this->validationResultFactory->create(['errors' => $errors]);
    }

    private function isInitialAssignment(StockSourceLinkInterface $link)
    {
        $defaultStockId = $this->defaultStockProvider->getId();
        $defaultSourceCode = $this->defaultSourceProvider->getCode();
        $linkStockId = $link->getStockId();
        $linkSourceCode = $link->getSourceCode();
        $initialAssignment = false;
        if ($defaultStockId === $linkStockId && $defaultSourceCode === $linkSourceCode) {
            $initialAssignment = true;
        }
        return $initialAssignment;
    }

    private function isLinkContainDefaultSourceOrStock(StockSourceLinkInterface $link)
    {
        $defaultStockId = $this->defaultStockProvider->getId();
        $defaultSourceCode = $this->defaultSourceProvider->getCode();
        $linkStockId = $link->getStockId();
        $linkSourceCode = $link->getSourceCode();
        $linkContainDefaultSourceOrStock = false;
        if ($linkStockId === $defaultStockId || $linkSourceCode === $defaultSourceCode) {
            $linkContainDefaultSourceOrStock = true;
        }
        return $linkContainDefaultSourceOrStock;
    }
}