MC-4242: Newsletter subscriptions per website

Author: serhii-balko

app/code/Magento/Customer/Controller/Adminhtml/Index/MassUnsubscribe.php

@@ -6,10 +6,13 @@
 namespace Magento\Customer\Controller\Adminhtml\Index;
 
 use Magento\Customer\Api\CustomerRepositoryInterface;
+use Magento\Customer\Api\Data\CustomerInterface;
+use Magento\Customer\Model\Config\Share;
 use Magento\Framework\App\Action\HttpPostActionInterface as HttpPostActionInterface;
 use Magento\Framework\Controller\ResultFactory;
 use Magento\Backend\App\Action\Context;
 use Magento\Newsletter\Model\SubscriptionManagerInterface;
+use Magento\Store\Model\StoreManagerInterface;
 use Magento\Ui\Component\MassAction\Filter;
 use Magento\Customer\Model\ResourceModel\Customer\CollectionFactory;
 use Magento\Eav\Model\Entity\Collection\AbstractCollection;
@@ -29,23 +32,38 @@ class MassUnsubscribe extends AbstractMassAction implements HttpPostActionInterf
      */
     private $subscriptionManager;
 
+    /**
+     * @var StoreManagerInterface
+     */
+    private $storeManager;
+
+    /**
+     * @var Share
+     */
+    private $shareConfig;
+
     /**
      * @param Context $context
      * @param Filter $filter
      * @param CollectionFactory $collectionFactory
      * @param CustomerRepositoryInterface $customerRepository
      * @param SubscriptionManagerInterface $subscriptionManager
+     * @param Share $shareConfig
      */
     public function __construct(
         Context $context,
         Filter $filter,
         CollectionFactory $collectionFactory,
         CustomerRepositoryInterface $customerRepository,
-        SubscriptionManagerInterface $subscriptionManager
+        SubscriptionManagerInterface $subscriptionManager,
+        StoreManagerInterface $storeManager,
+        Share $shareConfig
     ) {
         parent::__construct($context, $filter, $collectionFactory);
         $this->customerRepository = $customerRepository;
         $this->subscriptionManager = $subscriptionManager;
+        $this->storeManager = $storeManager;
+        $this->shareConfig = $shareConfig;
     }
 
     /**
@@ -60,8 +78,9 @@ protected function massAction(AbstractCollection $collection)
         foreach ($collection->getAllIds() as $customerId) {
             // Verify customer exists
             $customer = $this->customerRepository->getById($customerId);
-            $storeId = (int)$customer->getStoreId();
-            $this->subscriptionManager->unsubscribeCustomer($customerId, $storeId);
+            foreach ($this->getUnsubscribeStoreIds($customer) as $storeId) {
+                $this->subscriptionManager->unsubscribeCustomer((int)$customerId, $storeId);
+            }
             $customersUpdated++;
         }
 
@@ -74,4 +93,24 @@ protected function massAction(AbstractCollection $collection)
 
         return $resultRedirect;
     }
+
+    /**
+     * Get store ids to unsubscribe customer
+     *
+     * @param CustomerInterface $customer
+     * @return array
+     */
+    private function getUnsubscribeStoreIds(CustomerInterface $customer): array
+    {
+        $storeIds = [];
+        if ($this->shareConfig->isGlobalScope()) {
+            foreach ($this->storeManager->getStores() as $store) {
+                $storeIds[(int)$store->getWebsiteId()] = (int)$store->getId();
+            }
+        } else {
+            $storeIds = [(int)$customer->getStoreId()];
+        }
+
+        return $storeIds;
+    }
 }