Server : Apache System : Linux server2.corals.io 4.18.0-348.2.1.el8_5.x86_64 #1 SMP Mon Nov 15 09:17:08 EST 2021 x86_64 User : corals ( 1002) PHP Version : 7.4.33 Disable Function : exec,passthru,shell_exec,system Directory :  /home/corals/old/app/code/Cnc/SerialNumber/Controller/Adminhtml/SerialNumber/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

<?php
/**
 * Copyright (c) 2020 Kaliop Digital Commerce (https://digitalcommerce.kaliop.com) All Rights Reserved.
 * https://opensource.org/licenses/OSL-3.0  Open Software License (OSL 3.0)
 * Krzysztof Majkowski <[email protected]>
 */

declare(strict_types=1);

namespace Cnc\SerialNumber\Controller\Adminhtml\SerialNumber;

use Cnc\Catalog\Model\ResourceModel\DataFactory as SourceDataFactory;
use Cnc\SerialNumber\Api\SerialNumberRepositoryInterface;
use Cnc\SerialNumber\Model\ResourceModel\SerialNumber\Collection;
use Cnc\SerialNumber\Model\ResourceModel\SerialNumber\CollectionFactory;
use Cnc\SerialNumber\Model\SerialNumber;
use Cnc\SerialNumber\Model\Source\SerialNumber\Status;
use Magento\Backend\App\Action;
use Magento\Backend\App\Action\Context;
use Magento\Backend\Model\View\Result\Redirect;
use Magento\Framework\App\Action\HttpPostActionInterface;
use Magento\Framework\App\ResponseInterface;
use Magento\Framework\Controller\Result\Json;
use Magento\Framework\Controller\ResultFactory;
use Magento\Framework\Controller\ResultInterface;
use Magento\Framework\Exception\AlreadyExistsException;
use Magento\Framework\Exception\LocalizedException;
use Magento\InventoryCatalog\Model\GetSourceItemsBySkuAndSourceCodesFactory;
use Magento\InventoryImportExport\Model\Import\SourcesFactory as Importer;
use Magento\Ui\Component\MassAction\Filter;

class MassBlock extends Action implements HttpPostActionInterface
{
    /**
     * Authorization level of a basic admin session
     */
    const ADMIN_RESOURCE = 'Cnc_SerialNumber::save';

    /**
     * @var SerialNumberRepositoryInterface
     */
    private $serialNumberRepository;

    /**
     * @var CollectionFactory
     */
    private $collectionFactory;

    /**
     * @var Filter
     */
    private $filter;

    /**
     * @var GetSourceItemsBySkuAndSourceCodesFactory
     */
    private $getSourceItemsFactory;

    /**
     * @var Importer
     */
    private $importer;

    /**
     * @var SourceDataFactory
     */
    private $sourceDataFactory;

    /**
     * InlineEdit constructor.
     * @param Context $context
     * @param SerialNumberRepositoryInterface $serialNumberRepository
     * @param CollectionFactory $collectionFactory
     * @param Filter $filter
     * @param GetSourceItemsBySkuAndSourceCodesFactory $getSourceItemsFactory
     * @param Importer $importer
     * @param SourceDataFactory $sourceDataFactory
     */
    public function __construct(
        Context $context,
        SerialNumberRepositoryInterface $serialNumberRepository,
        CollectionFactory $collectionFactory,
        Filter $filter,
        GetSourceItemsBySkuAndSourceCodesFactory $getSourceItemsFactory,
        Importer $importer,
        SourceDataFactory $sourceDataFactory
    ) {
        parent::__construct($context);
        $this->serialNumberRepository = $serialNumberRepository;
        $this->collectionFactory = $collectionFactory;
        $this->filter = $filter;
        $this->getSourceItemsFactory = $getSourceItemsFactory;
        $this->importer = $importer;
        $this->sourceDataFactory = $sourceDataFactory;
    }

    /**
     * @return ResponseInterface|Json|ResultInterface
     * @throws LocalizedException
     */
    public function execute()
    {
        try {
            $collection = $this->filter->getCollection($this->collectionFactory->create());
            $decreased = $this->decrease($collection);

            $this->messageManager->addSuccessMessage(
                __('A total of %1 record(s) have been blocked.', $decreased)
            );
        } catch (AlreadyExistsException $e) {
            $this->messageManager->addErrorMessage(__('An error occurred.'));
        }

        /** @var Redirect $resultRedirect */
        $resultRedirect = $this->resultFactory->create(ResultFactory::TYPE_REDIRECT);
        return $resultRedirect->setPath('*/*/grid');
    }

    /**
     * @param Collection $collection
     * @throws AlreadyExistsException
     */
    private function decrease(Collection $collection)
    {
        $toDecrease = [];

        /** @var SerialNumber $item */
        foreach ($collection as $item) {
            if ($item->getStatus() == Status::AVAILABLE) {
                $item->setStatus(Status::BLOCKED);

                if (!isset($toDecrease[$item->getSourceCode()])) {
                    $toDecrease[$item->getSourceCode()] = [];
                }
                if (!isset($toDecrease[$item->getSourceCode()][$item->getSku()])) {
                    $toDecrease[$item->getSourceCode()][$item->getSku()] = 0;
                }
                $toDecrease[$item->getSourceCode()][$item->getSku()]++;
                $this->serialNumberRepository->save($item);
            }
        }

        $this->decreaseStock($toDecrease);
    }

    /**
     * Decrease stock for given items.
     *
     * @param array $toDecrease
     */
    private function decreaseStock(array $toDecrease)
    {
        $getSourceItems = $this->getSourceItemsFactory->create();
        $updateSourceItems = [];

        foreach ($toDecrease as $source => $skus) {
            foreach ($skus as $sku => $count) {
                $sourceItems = $getSourceItems->execute($sku, [$source]);
                if (is_array($sourceItems) && count($sourceItems)) {
                    $sourceItem = $sourceItems[array_keys($sourceItems)[0]];
                    $qty = max($sourceItem->getQuantity() - $count, 0);

                    $updateSourceItems[] = [
                        'source_code' => $source,
                        'sku' => $sku,
                        'quantity' => $qty
                    ];
                }
            }
        }

        if (count($updateSourceItems)) {
            $sourceModel = $this->sourceDataFactory->create(['data' => [$updateSourceItems]]);
            $stock = $this->importer->create(['importData' => $sourceModel]);
            $stock->importData();

            return count($updateSourceItems);
        }

        return 0;
    }
}