Add virtual property on cart api model

thelia-modules · Mar 26, 2021 · bc1714a · bc1714a
1 parent 60a263c
commit bc1714a
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 4 deletions.
diff --git a/Config/module.xml b/Config/module.xml
@@ -20,7 +20,7 @@
         <language>en_US</language>
         <language>fr_FR</language>
     </languages>
-    <version>1.0.1</version>
+    <version>1.0.2</version>
     <authors>
         <author>
             <name>Vincent Lopes-Vicente</name>

diff --git a/Model/Api/Cart.php b/Model/Api/Cart.php
@@ -98,6 +98,14 @@ class Cart extends BaseApiModel
      */
     protected $currency;
 
+    /**
+     * @var boolean
+     * @OA\Property(
+     *    type="boolean"
+     * )
+     */
+    protected $virtual;
+
     /**
      * @var array
      * @OA\Property(
@@ -154,7 +162,8 @@ function ($theliaCartItem) use ($modelFactory, $deliveryCountry) {
             ->setCoupons($coupons)
             ->setTotal($theliaModel->getTaxedAmount($deliveryCountry, false, null))
             ->setCurrency($theliaModel->getCurrency()->getSymbol())
-            ->setItems($cartItems);
+            ->setItems($cartItems)
+            ->setVirtual($theliaModel->isVirtual());
     }
 
     /**
@@ -247,6 +256,24 @@ public function setDiscount($discount)
         return $this;
     }
 
+    /**
+     * @return boolean
+     */
+    public function getVirtual()
+    {
+        return $this->virtual;
+    }
+
+    /**
+     * @param boolean $virtual
+     * @return Cart
+     */
+    public function setVirtual($virtual)
+    {
+        $this->virtual = $virtual;
+        return $this;
+    }
+
     /**
      * @return float
      */
@@ -334,8 +361,7 @@ protected function getEstimatedPostageForCountry(\Thelia\Model\Cart $cart, Count
         $deliveryModules = ModuleQuery::create()
             ->filterByActivate(1)
             ->filterByType(BaseModule::DELIVERY_MODULE_TYPE, Criteria::EQUAL)
-            ->find()
-        ;
+            ->find();
 
         $virtual = $cart->isVirtual();
         $postage = null;