UpdateSalesChannelWebsiteCodePlugin.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventorySales\Plugin\Store\Model\ResourceModel\Website;

use Magento\Framework\Exception\CouldNotSaveException;
use Magento\Framework\Exception\NoSuchEntityException;
use Magento\Framework\Model\AbstractModel;
use Magento\Framework\Validation\ValidationException;
use Magento\InventorySales\Model\ResourceModel\GetWebsiteCodeByWebsiteId;
use Magento\InventorySales\Model\ResourceModel\UpdateSalesChannelWebsiteCode;
use Magento\Store\Api\Data\WebsiteInterface;
use Magento\Store\Model\ResourceModel\Website as WebsiteResourceModel;
use Magento\Store\Model\Website;

class UpdateSalesChannelWebsiteCodePlugin
{
    private $updateSalesChannelWebsiteCode;

    private $getWebsiteCodeByWebsiteId;

    public function __construct(
        UpdateSalesChannelWebsiteCode $updateSalesChannelWebsiteCode,
        GetWebsiteCodeByWebsiteId $getWebsiteCodeByWebsiteId
    ) {
        $this->updateSalesChannelWebsiteCode = $updateSalesChannelWebsiteCode;
        $this->getWebsiteCodeByWebsiteId = $getWebsiteCodeByWebsiteId;
    }

    public function aroundSave(
        WebsiteResourceModel $subject,
        callable $proceed,
        AbstractModel $website
    ) {
        $newCode = $website->getCode();
        $oldCode = null;

        if (null !== $website->getId()) {
            $oldCode = $this->getWebsiteCodeByWebsiteId->execute((int)$website->getId());
        }

        $result = $proceed($website);

        if (($oldCode !== null) && ($oldCode !== WebsiteInterface::ADMIN_CODE) && ($oldCode !== $newCode)) {
            $this->updateSalesChannelWebsiteCode->execute($oldCode, $newCode);
        }
        return $result;
    }
}