custom/plugins/ProcAvailableStock/src/Subscriber/AvailableStockSubscriber.php line 64

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Proc\ProcAvailableStock\Subscriber;
  5. use Proc\ProcAvailableStock\Helper\ErrorHandler;
  6. use Shopware\Core\Content\Product\ProductEntity;
  7. use Shopware\Core\Framework\Context;
  8. use Shopware\Core\Framework\DataAbstractionLayer\EntityCollection;
  9. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepositoryInterface;
  10. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  11. use Shopware\Core\System\SystemConfig\SystemConfigService;
  12. use Shopware\Storefront\Page\PageLoadedEvent;
  13. use Shopware\Storefront\Page\Checkout\Cart\CheckoutCartPageLoadedEvent;
  14. use Shopware\Storefront\Page\Checkout\Offcanvas\OffcanvasCartPageLoadedEvent;
  15. use Shopware\Storefront\Page\Checkout\Confirm\CheckoutConfirmPageLoadedEvent;
  16. use Shopware\Storefront\Page\Checkout\Finish\CheckoutFinishPageLoadedEvent;
  17. use Shopware\Core\Checkout\Cart\LineItem\LineItem;
  18. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  20. /**
  21.  * Class AvailableStockSubscriber
  22.  * @package Proc\ProcAvailableStock\Subscriber
  23.  */
  24. class AvailableStockSubscriber implements EventSubscriberInterface
  25. {
  27.     /**
  28.      * @var PageLoadedEvent
  29.      */
  30.     private PageLoadedEvent $event;
  32.     /**
  33.      * @var EntityRepositoryInterface
  34.      */
  35.     private EntityRepositoryInterface $productRepository;
  38.     /**
  39.      * @inheritDoc
  40.      */
  41.     public static function getSubscribedEvents()
  42.     {
  43.         return [
  44. //            OffcanvasCartPageLoadedEvent::class => ['onCartLoaded', 1],
  45.             CheckoutCartPageLoadedEvent::class => ['onCartLoaded', 1],
  46.             CheckoutConfirmPageLoadedEvent::class => ['onCartLoaded', 1],
  47.             CheckoutFinishPageLoadedEvent::class => ['onFinishLoaded', 1]
  48.         ];
  49.     }
  52.     /**
  53.      * @param SystemConfigService $config
  54.      * @param EntityRepositoryInterface $repositoryInterface
  55.      */
  56.     function __construct(SystemConfigService $config, EntityRepositoryInterface $repositoryInterface)
  57.     {
  58.         $this->productRepository = $repositoryInterface;
  59.     }
  61.     /**
  62.      * @param PageLoadedEvent $event
  63.      */
  64.     public function onCartLoaded(PageLoadedEvent $event)
  65.     {
  66.         $this->event = $event;
  67.         if (
  68. //            !($this->event instanceof OffcanvasCartPageLoadedEvent) &&
  69.             !($this->event instanceof CheckoutCartPageLoadedEvent) &&
  70.             !($this->event instanceof CheckoutConfirmPageLoadedEvent)
  71.         ) return;
  72.         if (!$this->checkLogin()) return;
  74.         $lineItems = $this->event->getPage()->getCart()->getLineItems()->filterType(LineItem::PRODUCT_LINE_ITEM_TYPE);
  76.         if (count($lineItems) < 1) return;
  78.         foreach ($lineItems as $lineItem)
  79.         {
  80.             $product = $this->getProduct($lineItem->getReferencedId());
  81.             $lineItem->setExtensions(['availability' => $product->getStock() - $lineItem->getQuantity()]);
  82.         }
  83.     }
  85.     /**
  86.      * @param PageLoadedEvent $event
  87.      */
  88.     public function onFinishLoaded(PageLoadedEvent $event)
  89.     {
  90.         $this->event = $event;
  91.         if (!($this->event instanceof CheckoutFinishPageLoadedEvent)) return;
  92.         if (!$this->checkLogin()) return;
  94.         $lineItems = $this->event->getPage()->getOrder()->getLineItems()->filterByType(LineItem::PRODUCT_LINE_ITEM_TYPE);
  96.         if (count($lineItems) < 1) return;
  98.         foreach ($lineItems as $lineItem)
  99.         {
  100.             $product = $this->getProduct($lineItem->getReferencedId());
  101.             $lineItem->setExtensions(['availability' => $product->getStock() - $lineItem->getQuantity()]);
  102.         }
  103.     }
  106.     /**
  107.      * @param string $productID
  108.      * @return ProductEntity
  109.      */
  110.     public function getProduct(string $productID) : ProductEntity
  111.     {
  112.         $criteria = new Criteria([$productID]);
  114.         /** @var EntityCollection $products */
  115.         $products = $this->productRepository->search($criteria, Context::createDefaultContext());
  117.         $productList = $products->getElements();
  119.         /** @var ProductEntity $product */
  120.         $product = $productList[$productID];
  122.         return $product;
  123.     }
  126.     /**
  127.      * @return bool
  128.      */
  129.     private function checkLogin() : bool
  130.     {
  131.         if ($this->event->getSalesChannelContext()->getCustomer() === null)
  132.         {
  133.             $this->error = (new ErrorHandler)->error(ErrorHandler::NO_CUSTOMER_ERROR);
  134.             return false;
  135.         }
  136.         return true;
  137.     }
  139. }