coupon management in order and cart process

local/modules/Front/Controller/CartController.php

@@ -24,11 +24,13 @@ namespace Front\Controller;
 
 use Propel\Runtime\Exception\PropelException;
 use Thelia\Controller\Front\BaseFrontController;
+use Thelia\Core\Event\Order\OrderEvent;
 use Thelia\Form\Exception\FormValidationException;
 use Thelia\Core\Event\Cart\CartEvent;
 use Thelia\Core\Event\TheliaEvents;
 use Symfony\Component\HttpFoundation\Request;
 use Thelia\Form\CartAdd;
+use Thelia\Model\AddressQuery;
 
 class CartController extends BaseFrontController
 {
@@ -53,6 +55,8 @@ class CartController extends BaseFrontController
 
             $this->getDispatcher()->dispatch(TheliaEvents::CART_ADDITEM, $cartEvent);
 
+            $this->afterModifyCart();
+
             $this->redirectSuccess();
 
         } catch (PropelException $e) {
@@ -83,6 +87,8 @@ class CartController extends BaseFrontController
         try {
             $this->dispatch(TheliaEvents::CART_UPDATEITEM, $cartEvent);
 
+            $this->afterModifyCart();
+
             $this->redirectSuccess();
         } catch (PropelException $e) {
             $this->getParserContext()->setGeneralError($e->getMessage());
@@ -98,6 +104,8 @@ class CartController extends BaseFrontController
         try {
             $this->getDispatcher()->dispatch(TheliaEvents::CART_DELETEITEM, $cartEvent);
 
+            $this->afterModifyCart();
+
             $this->redirectSuccess();
         } catch (PropelException $e) {
             \Thelia\Log\Tlog::getInstance()->error(sprintf("error during deleting cartItem with message : %s", $e->getMessage()));
@@ -142,4 +150,25 @@ class CartController extends BaseFrontController
         return $cartAdd;
     }
 
+    protected function afterModifyCart()
+    {
+        /* recalculate postage amount */
+        $order = $this->getSession()->getOrder();
+        if(null !== $order) {
+            $deliveryModule = $order->getModuleRelatedByDeliveryModuleId();
+            $deliveryAddress = AddressQuery::create()->findPk($order->chosenDeliveryAddress);
+
+            if(null !== $deliveryModule && null !== $deliveryAddress) {
+                $moduleInstance = $this->container->get(sprintf('module.%s', $deliveryModule->getCode()));
+                $postage = $moduleInstance->getPostage($deliveryAddress->getCountry());
+
+                $orderEvent = new OrderEvent($order);
+                $orderEvent->setPostage($postage);
+
+                $this->getDispatcher()->dispatch(TheliaEvents::ORDER_SET_POSTAGE, $orderEvent);
+            }
+        }
+    }
+
+
 }