Server : Apache System : Linux server2.corals.io 4.18.0-348.2.1.el8_5.x86_64 #1 SMP Mon Nov 15 09:17:08 EST 2021 x86_64 User : corals ( 1002) PHP Version : 7.4.33 Disable Function : exec,passthru,shell_exec,system Directory :  /home/corals/cartforge.co/vendor/magento/module-quote/Model/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

<?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()
                    );
                }
            }
        }
    }
}