AllowedCountryValidationRule.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\Quote\Model\ValidationRules;

use Magento\Directory\Model\AllowedCountries;
use Magento\Framework\Validation\ValidationResultFactory;
use Magento\Quote\Model\Quote;
use Magento\Store\Model\ScopeInterface;

class AllowedCountryValidationRule implements QuoteValidationRuleInterface
{
    private $generalMessage;

    private $allowedCountryReader;

    private $validationResultFactory;

    public function __construct(
        AllowedCountries $allowedCountryReader,
        ValidationResultFactory $validationResultFactory,
        string $generalMessage = ''
    ) {
        $this->allowedCountryReader = $allowedCountryReader;
        $this->validationResultFactory = $validationResultFactory;
        $this->generalMessage = $generalMessage;
    }

    public function validate(Quote $quote): array
    {
        $validationErrors = [];

        if (!$quote->isVirtual()) {
            $shippingAddress = $quote->getShippingAddress();
            $shippingAddress->setStoreId($quote->getStoreId());
            $validationResult =
                in_array(
                    $shippingAddress->getCountryId(),
                    $this->allowedCountryReader->getAllowedCountries(
                        ScopeInterface::SCOPE_STORE,
                        $quote->getStoreId()
                    )
                );
            if (!$validationResult) {
                $validationErrors = [__($this->generalMessage)];
            }
        }

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