Add availibility check for product

Sylius · May 20, 2020 · d3b2357 · d3b2357
1 parent d5c273a
commit d3b2357
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 4 deletions.
diff --git a/src/Handler/Cart/PutSimpleItemToCartHandler.php b/src/Handler/Cart/PutSimpleItemToCartHandler.php
@@ -8,6 +8,7 @@
 use Sylius\Component\Core\Model\ProductInterface;
 use Sylius\Component\Core\Repository\OrderRepositoryInterface;
 use Sylius\Component\Core\Repository\ProductRepositoryInterface;
+use Sylius\Component\Inventory\Checker\AvailabilityCheckerInterface;
 use Sylius\ShopApiPlugin\Checker\ProductInCartChannelCheckerInterface;
 use Sylius\ShopApiPlugin\Command\Cart\PutSimpleItemToCart;
 use Sylius\ShopApiPlugin\Modifier\OrderModifierInterface;
@@ -27,16 +28,21 @@ final class PutSimpleItemToCartHandler
     /** @var ProductInCartChannelCheckerInterface */
     private $channelChecker;
 
+    /** @var AvailabilityCheckerInterface */
+    private $availabilityChecker;
+
     public function __construct(
         OrderRepositoryInterface $cartRepository,
         ProductRepositoryInterface $productRepository,
         OrderModifierInterface $orderModifier,
-        ProductInCartChannelCheckerInterface $channelChecker
+        ProductInCartChannelCheckerInterface $channelChecker,
+        AvailabilityCheckerInterface $availabilityChecker
     ) {
         $this->cartRepository = $cartRepository;
         $this->productRepository = $productRepository;
         $this->orderModifier = $orderModifier;
         $this->channelChecker = $channelChecker;
+        $this->availabilityChecker = $availabilityChecker;
     }
 
     public function __invoke(PutSimpleItemToCart $putSimpleItemToCart): void
@@ -54,6 +60,9 @@ public function __invoke(PutSimpleItemToCart $putSimpleItemToCart): void
 
         $productVariant = $product->getVariants()[0];
 
-        $this->orderModifier->modify($cart, $productVariant, $putSimpleItemToCart->quantity());
+        $quantity = $putSimpleItemToCart->quantity();
+        Assert::true($this->availabilityChecker->isStockSufficient($productVariant, $quantity), 'Product variant is not available in stock');
+
+        $this->orderModifier->modify($cart, $productVariant, $quantity);
     }
 }
diff --git a/src/Handler/Cart/PutVariantBasedConfigurableItemToCartHandler.php b/src/Handler/Cart/PutVariantBasedConfigurableItemToCartHandler.php
@@ -8,6 +8,7 @@
 use Sylius\Component\Core\Model\ProductVariantInterface;
 use Sylius\Component\Core\Repository\OrderRepositoryInterface;
 use Sylius\Component\Core\Repository\ProductVariantRepositoryInterface;
+use Sylius\Component\Inventory\Checker\AvailabilityCheckerInterface;
 use Sylius\ShopApiPlugin\Checker\ProductInCartChannelCheckerInterface;
 use Sylius\ShopApiPlugin\Command\Cart\PutVariantBasedConfigurableItemToCart;
 use Sylius\ShopApiPlugin\Modifier\OrderModifierInterface;
@@ -27,16 +28,21 @@ final class PutVariantBasedConfigurableItemToCartHandler
     /** @var ProductInCartChannelCheckerInterface */
     private $channelChecker;
 
+    /** @var AvailabilityCheckerInterface */
+    private $availabilityChecker;
+
     public function __construct(
         OrderRepositoryInterface $cartRepository,
         ProductVariantRepositoryInterface $productVariantRepository,
         OrderModifierInterface $orderModifier,
-        ProductInCartChannelCheckerInterface $channelChecker
+        ProductInCartChannelCheckerInterface $channelChecker,
+        AvailabilityCheckerInterface $availabilityChecker
     ) {
         $this->cartRepository = $cartRepository;
         $this->productVariantRepository = $productVariantRepository;
         $this->orderModifier = $orderModifier;
         $this->channelChecker = $channelChecker;
+        $this->availabilityChecker = $availabilityChecker;
     }
 
     public function __invoke(PutVariantBasedConfigurableItemToCart $putConfigurableItemToCart): void
@@ -55,6 +61,9 @@ public function __invoke(PutVariantBasedConfigurableItemToCart $putConfigurableI
 
         Assert::true($this->channelChecker->isProductInCartChannel($product, $cart), 'Product is not in same channel as cart');
 
-        $this->orderModifier->modify($cart, $productVariant, $putConfigurableItemToCart->quantity());
+        $quantity = $putConfigurableItemToCart->quantity();
+        Assert::true($this->availabilityChecker->isStockSufficient($productVariant, $quantity), 'Product variant is not available in stock');
+
+        $this->orderModifier->modify($cart, $productVariant, $quantity);
     }
 }
diff --git a/src/Resources/config/services/handler/cart.xml b/src/Resources/config/services/handler/cart.xml
@@ -31,6 +31,7 @@
             <argument type="service" id="sylius.repository.product"/>
             <argument type="service" id="sylius.shop_api_plugin.modifier.order_modifier"/>
             <argument type="service" id="sylius.shop_api_plugin.checker.product_in_cart_channel_checker" />
+            <argument type="service" id="sylius.availability_checker"/>
             <tag name="messenger.message_handler"  bus="sylius_shop_api_plugin.command_bus"/>
         </service>
 
@@ -40,6 +41,7 @@
             <argument type="service" id="sylius.repository.product_variant"/>
             <argument type="service" id="sylius.shop_api_plugin.modifier.order_modifier"/>
             <argument type="service" id="sylius.shop_api_plugin.checker.product_in_cart_channel_checker" />
+            <argument type="service" id="sylius.availability_checker"/>
             <tag name="messenger.message_handler"  bus="sylius_shop_api_plugin.command_bus"/>
         </service>