AddNoticeForUnassignedSalesChannels.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventorySalesAdminUi\Plugin\InventoryApi\StockRepository;

use Magento\Framework\Exception\CouldNotSaveException;
use Magento\Framework\Message\ManagerInterface;
use Magento\InventoryApi\Api\Data\StockInterface;
use Magento\InventoryApi\Api\StockRepositoryInterface;
use Magento\InventorySalesApi\Api\Data\SalesChannelInterface;
use Magento\InventorySalesApi\Model\GetAssignedSalesChannelsForStockInterface;

class AddNoticeForUnassignedSalesChannels
{
    private $messageManager;

    private $getAssignedSalesChannelsForStock;

    public function __construct(
        ManagerInterface $messageManager,
        GetAssignedSalesChannelsForStockInterface $getAssignedSalesChannelsForStock
    ) {
        $this->messageManager = $messageManager;
        $this->getAssignedSalesChannelsForStock = $getAssignedSalesChannelsForStock;
    }

    public function afterSave(
        StockRepositoryInterface $subject,
        int $stockId,
        StockInterface $stock
    ): int {
        $unAssignedSalesChannels = $this->getUnassignedSalesChannelsForStock($stock);

        if (count($unAssignedSalesChannels)) {
            $this->messageManager->addNoticeMessage(
                __('All unassigned sales channels will be assigned to the Default Stock')
            );
        }

        return $stockId;
    }

    private function getUnassignedSalesChannelsForStock(StockInterface $stock): array
    {
        $newWebsiteCodes = $result = [];
        $assignedSalesChannels = $this->getAssignedSalesChannelsForStock->execute((int)$stock->getStockId());
        $extensionAttributes = $stock->getExtensionAttributes();
        $newSalesChannels = $extensionAttributes->getSalesChannels() ?: [];

        foreach ($newSalesChannels as $salesChannel) {
            if ($salesChannel->getType() === SalesChannelInterface::TYPE_WEBSITE) {
                $newWebsiteCodes[] = $salesChannel->getCode();
            }
        }

        foreach ($assignedSalesChannels as $salesChannel) {
            if ($salesChannel->getType() === SalesChannelInterface::TYPE_WEBSITE
                && !in_array($salesChannel->getCode(), $newWebsiteCodes, true)
            ) {
                $result[] = $salesChannel;
            }
        }

        return $result;
    }
}