AC-11432: Improve address validation

Author: eug123

app/code/Magento/Quote/Model/BillingAddressManagement.php

@@ -77,10 +77,6 @@ public function assign($cartId, AddressInterface $address, $useForShipping = fal
     {
         /** @var \Magento\Quote\Model\Quote $quote */
         $quote = $this->quoteRepository->getActive($cartId);
-
-        // validate the address
-        $this->addressValidator->validateWithExistingAddress($quote, $address);
-
         $address->setCustomerId($quote->getCustomerId());
         $quote->removeAddress($quote->getBillingAddress()->getId());
         $quote->setBillingAddress($address);

app/code/Magento/Quote/Model/QuoteAddressValidator.php

@@ -121,27 +121,6 @@ public function validate(AddressInterface $addressData): bool
         return true;
     }
 
-    /**
-     * Validate Quest Address for guest user
-     *
-     * @param AddressInterface $address
-     * @param CartInterface $cart
-     * @return void
-     * @throws NoSuchEntityException
-     */
-    private function doValidateForGuestQuoteAddress(AddressInterface $address, CartInterface $cart): void
-    {
-        //validate guest cart address
-        if ($address->getId() !== null) {
-            $old = $cart->getAddressById($address->getId());
-            if ($old === false) {
-                throw new NoSuchEntityException(
-                    __('Invalid quote address id %1', $address->getId())
-                );
-            }
-        }
-    }
-
     /**
      * Validate address to be used for cart.
      *
@@ -153,9 +132,6 @@ private function doValidateForGuestQuoteAddress(AddressInterface $address, CartI
      */
     public function validateForCart(CartInterface $cart, AddressInterface $address): void
     {
-        if ($cart->getCustomerIsGuest()) {
-            $this->doValidateForGuestQuoteAddress($address, $cart);
-        }
         $this->doValidate($address, $cart->getCustomerIsGuest() ? null : (int) $cart->getCustomer()->getId());
     }
 
@@ -171,8 +147,8 @@ public function validateWithExistingAddress(CartInterface $cart, AddressInterfac
     {
         // check if address belongs to quote.
         if ($address->getId() !== null) {
-            $old = $cart->getAddressesCollection()->getItemById($address->getId());
-            if ($old === null) {
+            $old = $cart->getAddressById($address->getId());
+            if ($old === false) {
                 throw new NoSuchEntityException(
                     __('Invalid quote address id %1', $address->getId())
                 );

app/code/Magento/Quote/Plugin/QuoteAddress.php

@@ -0,0 +1,67 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\Quote\Plugin;
+
+use Magento\Framework\Exception\NoSuchEntityException;
+use Magento\Quote\Model\Quote;
+use Magento\Quote\Api\Data\AddressInterface;
+use Magento\Quote\Model\QuoteAddressValidator;
+
+/**
+ * Quote address plugin
+ */
+class QuoteAddress
+{
+    /**
+     * @var QuoteAddressValidator
+     */
+    protected QuoteAddressValidator $addressValidator;
+
+    /**
+     * @param QuoteAddressValidator $addressValidator
+     */
+    public function __construct(
+        QuoteAddressValidator $addressValidator
+    ) {
+        $this->addressValidator = $addressValidator;
+    }
+
+    /**
+     * Validate address before setting billing address
+     *
+     * @param Quote $subject
+     * @param AddressInterface|null $address
+     * @return array
+     * @throws NoSuchEntityException
+     */
+    public function beforeSetBillingAddress(Quote $subject, AddressInterface $address = null): array
+    {
+        if ($address !== null) {
+            $this->addressValidator->validateWithExistingAddress($subject, $address);
+        }
+
+        return [$address];
+    }
+
+    /**
+     * Validate address before setting shipping address
+     *
+     * @param Quote $subject
+     * @param AddressInterface|null $address
+     * @return array
+     * @throws NoSuchEntityException
+     */
+    public function beforeSetShippingAddress(Quote $subject, AddressInterface $address = null): array
+    {
+        if ($address !== null) {
+            $this->addressValidator->validateWithExistingAddress($subject, $address);
+        }
+
+        return [$address];
+    }
+}

app/code/Magento/Quote/etc/webapi_rest/di.xml

@@ -18,6 +18,7 @@
     </type>
     <type name="Magento\Quote\Model\Quote">
         <plugin name="updateQuoteStoreId" type="Magento\Quote\Model\Quote\Plugin\UpdateQuoteStoreId" />
+        <plugin name="validateQuoteAddress" type="Magento\Quote\Plugin\QuoteAddress" />
     </type>
     <type name="Magento\Webapi\Controller\Rest\ParamsOverrider">
         <plugin name="validateQuoteData" type="Magento\Quote\Plugin\Webapi\Controller\Rest\ValidateQuoteData" sortOrder="1" disabled="false" />