AdaptAddStockStatusToSelectPlugin.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventoryCatalog\Plugin\CatalogInventory\Model\ResourceModel\Stock\Status;

use Magento\CatalogInventory\Model\ResourceModel\Stock\Status;
use Magento\Framework\DB\Select;
use Magento\Framework\Exception\LocalizedException;
use Magento\InventoryCatalog\Model\ResourceModel\AddStockStatusToSelect;
use Magento\InventorySalesApi\Api\Data\SalesChannelInterface;
use Magento\InventorySalesApi\Api\StockResolverInterface;
use Magento\Store\Model\Website;

class AdaptAddStockStatusToSelectPlugin
{
    private $stockResolver;

    private $addStockStatusToSelect;

    public function __construct(
        StockResolverInterface $stockResolver,
        AddStockStatusToSelect $addStockStatusToSelect
    ) {
        $this->stockResolver = $stockResolver;
        $this->addStockStatusToSelect = $addStockStatusToSelect;
    }

    public function aroundAddStockStatusToSelect(
        Status $stockStatus,
        callable $proceed,
        Select $select,
        Website $website
    ) {
        $websiteCode = $website->getCode();
        if (null === $websiteCode) {
            throw new LocalizedException(__('Website code is empty'));
        }

        $stock = $this->stockResolver->execute(SalesChannelInterface::TYPE_WEBSITE, $websiteCode);
        $stockId = (int)$stock->getStockId();

        $this->addStockStatusToSelect->execute($select, $stockId);

        return $stockStatus;
    }
}