stock_website_sales_channels.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

use Magento\InventoryApi\Api\StockRepositoryInterface;
use Magento\InventorySalesApi\Api\Data\SalesChannelInterface;
use Magento\InventorySalesApi\Api\Data\SalesChannelInterfaceFactory;
use Magento\TestFramework\Helper\Bootstrap;

$stockRepository = Bootstrap::getObjectManager()->get(StockRepositoryInterface::class);
$salesChannelFactory = Bootstrap::getObjectManager()->get(SalesChannelInterfaceFactory::class);

$salesChannelData = [10 => 'eu_website', 20 => 'us_website', 30 => 'global_website'];

foreach ($salesChannelData as $stockId => $websiteCode) {
    $stock = $stockRepository->get($stockId);
    $extensionAttributes = $stock->getExtensionAttributes();
    $salesChannels = $extensionAttributes->getSalesChannels();

    $salesChannel = $salesChannelFactory->create();
    $salesChannel->setCode($websiteCode);
    $salesChannel->setType(SalesChannelInterface::TYPE_WEBSITE);
    $salesChannels[] = $salesChannel;

    $extensionAttributes->setSalesChannels($salesChannels);
    $stockRepository->save($stock);
}