PopulateWithWebsiteSalesChannelsObserver.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventorySales\Observer\Stock;

use Magento\Framework\App\RequestInterface;
use Magento\Framework\Event\ObserverInterface;
use Magento\Framework\Event\Observer as EventObserver;
use Magento\InventoryApi\Api\Data\StockInterface;
use Magento\InventorySalesApi\Api\Data\SalesChannelInterface;
use Magento\InventorySalesApi\Api\Data\SalesChannelInterfaceFactory;

class PopulateWithWebsiteSalesChannelsObserver implements ObserverInterface
{
    private $salesChannelFactory;

    public function __construct(SalesChannelInterfaceFactory $salesChannelFactory)
    {
        $this->salesChannelFactory = $salesChannelFactory;
    }

    public function execute(EventObserver $observer)
    {
        $stock = $observer->getEvent()->getStock();
        $request = $observer->getEvent()->getRequest();
        $requestData = $request->getParams();

        $extensionAttributes = $stock->getExtensionAttributes();
        $assignedSalesChannels = $extensionAttributes->getSalesChannels();

        if (null !== $assignedSalesChannels) {
            foreach ($assignedSalesChannels as $key => $assignedSalesChannel) {
                if ($assignedSalesChannel->getType() === SalesChannelInterface::TYPE_WEBSITE) {
                    unset($assignedSalesChannels[$key]);
                }
            }
        }

        if (isset($requestData['sales_channels'][SalesChannelInterface::TYPE_WEBSITE])
            && is_array($requestData['sales_channels'][SalesChannelInterface::TYPE_WEBSITE])
        ) {
            foreach ($requestData['sales_channels'][SalesChannelInterface::TYPE_WEBSITE] as $websiteCode) {
                $assignedSalesChannels[] = $this->createSalesChannelByWebsiteCode($websiteCode);
            }
        }
        $extensionAttributes->setSalesChannels($assignedSalesChannels);
    }

    private function createSalesChannelByWebsiteCode(string $websiteCode): SalesChannelInterface
    {
        $salesChannel = $this->salesChannelFactory->create();
        $salesChannel->setCode($websiteCode);
        $salesChannel->setType(SalesChannelInterface::TYPE_WEBSITE);
        return $salesChannel;
    }
}