Your IP : 216.73.216.97
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
declare(strict_types=1);
namespace Magento\Quote\Model;
use Magento\Customer\Api\AccountManagementInterface as AccountManagement;
use Magento\Customer\Api\AddressRepositoryInterface as CustomerAddressRepository;
use Magento\Customer\Api\CustomerRepositoryInterface as CustomerRepository;
use Magento\Customer\Model\AddressFactory;
use Magento\Framework\App\ObjectManager;
use Magento\Framework\Validator\Exception as ValidatorException;
use Magento\Framework\Validator\Factory as ValidatorFactory;
use Magento\Quote\Model\Quote as QuoteEntity;
/**
* Class Customer
*/
class CustomerManagement
{
/**
* @var CustomerRepository
*/
protected $customerRepository;
/**
* @var CustomerAddressRepository
*/
protected $customerAddressRepository;
/**
* @var AccountManagement
*/
protected $accountManagement;
/**
* @var ValidatorFactory
*/
private $validatorFactory;
/**
* @var AddressFactory
*/
private $addressFactory;
/**
* CustomerManagement constructor.
* @param CustomerRepository $customerRepository
* @param CustomerAddressRepository $customerAddressRepository
* @param AccountManagement $accountManagement
* @param ValidatorFactory|null $validatorFactory
* @param AddressFactory|null $addressFactory
*/
public function __construct(
CustomerRepository $customerRepository,
CustomerAddressRepository $customerAddressRepository,
AccountManagement $accountManagement,
ValidatorFactory $validatorFactory = null,
AddressFactory $addressFactory = null
) {
$this->customerRepository = $customerRepository;
$this->customerAddressRepository = $customerAddressRepository;
$this->accountManagement = $accountManagement;
$this->validatorFactory = $validatorFactory ?: ObjectManager::getInstance()
->get(ValidatorFactory::class);
$this->addressFactory = $addressFactory ?: ObjectManager::getInstance()
->get(AddressFactory::class);
}
/**
* Populate customer model
*
* @param Quote $quote
* @return void
*/
public function populateCustomerInfo(QuoteEntity $quote)
{
$customer = $quote->getCustomer();
if (!$customer->getId()) {
$customer = $this->accountManagement->createAccountWithPasswordHash(
$customer,
$quote->getPasswordHash()
);
$quote->setCustomer($customer);
$this->fillCustomerAddressId($quote);
}
if (!$quote->getBillingAddress()->getId() && $customer->getDefaultBilling()) {
$quote->getBillingAddress()->importCustomerAddressData(
$this->customerAddressRepository->getById($customer->getDefaultBilling())
);
$quote->getBillingAddress()->setCustomerAddressId($customer->getDefaultBilling());
}
if (!$quote->getShippingAddress()->getSameAsBilling()
&& !$quote->getBillingAddress()->getId()
&& $customer->getDefaultShipping()
) {
$quote->getShippingAddress()->importCustomerAddressData(
$this->customerAddressRepository->getById($customer->getDefaultShipping())
);
$quote->getShippingAddress()->setCustomerAddressId($customer->getDefaultShipping());
}
}
/**
* Filling 'CustomerAddressId' in quote for a newly created customer.
*
* @param QuoteEntity $quote
* @return void
*/
private function fillCustomerAddressId(QuoteEntity $quote): void
{
$customer = $quote->getCustomer();
$customer->getDefaultBilling() ?
$quote->getBillingAddress()->setCustomerAddressId($customer->getDefaultBilling()) :
$quote->getBillingAddress()->setCustomerAddressId(0);
if ($customer->getDefaultShipping() || $customer->getDefaultBilling()) {
if ($quote->getShippingAddress()->getSameAsBilling()) {
$quote->getShippingAddress()->setCustomerAddressId($customer->getDefaultBilling());
} else {
$quote->getShippingAddress()->setCustomerAddressId($customer->getDefaultShipping());
}
} else {
$quote->getShippingAddress()->setCustomerAddressId(0);
}
}
/**
* Validate Quote Addresses
*
* @param Quote $quote
* @throws ValidatorException
* @return void
*/
public function validateAddresses(QuoteEntity $quote)
{
$addresses = [];
if ($quote->getBillingAddress()->getCustomerAddressId()) {
$addresses[] = $this->customerAddressRepository->getById(
$quote->getBillingAddress()->getCustomerAddressId()
);
}
if ($quote->getShippingAddress()->getCustomerAddressId()) {
$addresses[] = $this->customerAddressRepository->getById(
$quote->getShippingAddress()->getCustomerAddressId()
);
}
if (!empty($addresses)) {
foreach ($addresses as $address) {
$validator = $this->validatorFactory->createValidator('customer_address', 'save');
$addressModel = $this->addressFactory->create();
$addressModel->updateData($address);
if (!$validator->isValid($addressModel)) {
throw new ValidatorException(
null,
null,
$validator->getMessages()
);
}
}
}
}
}