Inital commit

core/lib/Thelia/Action/Area.php

@@ -11,52 +11,73 @@
 /*************************************************************************************/
 
 namespace Thelia\Action;
+
+use Symfony\Component\EventDispatcher\EventDispatcherInterface;
 use Symfony\Component\EventDispatcher\EventSubscriberInterface;
 use Thelia\Core\Event\Area\AreaAddCountryEvent;
 use Thelia\Core\Event\Area\AreaCreateEvent;
 use Thelia\Core\Event\Area\AreaDeleteEvent;
 use Thelia\Core\Event\Area\AreaRemoveCountryEvent;
+use Thelia\Core\Event\Area\AreaUpdateEvent;
 use Thelia\Core\Event\Area\AreaUpdatePostageEvent;
 use Thelia\Core\Event\TheliaEvents;
-use Thelia\Model\AreaQuery;
-use Thelia\Model\CountryQuery;
-
 use Thelia\Model\Area as AreaModel;
+use Thelia\Model\AreaQuery;
+use Thelia\Model\CountryArea;
+use Thelia\Model\CountryAreaQuery;
 
 /**
  * Class Area
  * @package Thelia\Action
- * @author Manuel Raynaud <mraynaud@openstudio.fr>
+ * @author Manuel Raynaud <manu@raynaud.io>
  */
 class Area extends BaseAction implements EventSubscriberInterface
 {
-
     public function addCountry(AreaAddCountryEvent $event)
     {
-        if (null !== $country = CountryQuery::create()->findPk($event->getCountryId())) {
-            $country->setDispatcher($event->getDispatcher());
-            $country->setAreaId($event->getAreaId())
-                ->save();
+        $countryIds = $event->getCountryId();
 
-            $event->setArea($country->getArea());
+        $areaId = $event->getAreaId();
+
+        foreach ($countryIds as $countryId) {
+            $countryArea = new CountryArea();
+
+            $country = explode('-', $countryId);
+            if (count($country) === 1) {
+                $country[1] = null;
+            }
+            if ($country[1] == 0) {
+                $country[1] = null;
+            }
+
+            $countryArea
+                ->setAreaId($areaId)
+                ->setCountryId($country[0])
+                ->setStateId($country[1])
+                ->save()
+            ;
         }
+
+        $event->setArea(AreaQuery::create()->findPk($areaId));
     }
 
     public function removeCountry(AreaRemoveCountryEvent $event)
     {
-        if (null !== $country = CountryQuery::create()->findPk($event->getCountryId())) {
-            $event->setArea($country->getArea());
+        CountryAreaQuery::create()
+                ->filterByCountryId($event->getCountryId())
+                ->filterByStateId($event->getStateId())
+                ->filterByAreaId($event->getAreaId())
+                ->delete();
 
-            $country->setDispatcher($event->getDispatcher());
-            $country->setAreaId(null)
-                ->save();
+        if (null !== $area = AreaQuery::create()->findPk($event->getAreaId())) {
+            $event->setArea($area);
         }
     }
 
-    public function updatePostage(AreaUpdatePostageEvent $event)
+    public function updatePostage(AreaUpdatePostageEvent $event, $eventName, EventDispatcherInterface $dispatcher)
     {
         if (null !== $area = AreaQuery::create()->findPk($event->getAreaId())) {
-            $area->setDispatcher($event->getDispatcher());
+            $area->setDispatcher($dispatcher);
             $area
                 ->setPostage($event->getPostage())
                 ->save();
@@ -65,47 +86,42 @@ class Area extends BaseAction implements EventSubscriberInterface
         }
     }
 
-    public function delete(AreaDeleteEvent $event)
+    public function delete(AreaDeleteEvent $event, $eventName, EventDispatcherInterface $dispatcher)
     {
         if (null !== $area = AreaQuery::create()->findPk($event->getAreaId())) {
-            $area->setDispatcher($event->getDispatcher());
+            $area->setDispatcher($dispatcher);
             $area->delete();
 
             $event->setArea($area);
         }
     }
 
-    public function create(AreaCreateEvent $event)
+    public function create(AreaCreateEvent $event, $eventName, EventDispatcherInterface $dispatcher)
     {
         $area = new AreaModel();
 
         $area
-            ->setDispatcher($event->getDispatcher())
+            ->setDispatcher($dispatcher)
             ->setName($event->getAreaName())
             ->save();
 
         $event->setArea($area);
     }
 
+    public function update(AreaUpdateEvent $event, $eventName, EventDispatcherInterface $dispatcher)
+    {
+        if (null !== $area = AreaQuery::create()->findPk($event->getAreaId())) {
+            $area
+                ->setDispatcher($dispatcher)
+                ->setName($event->getAreaName())
+                ->save();
+
+            $event->setArea($area);
+        }
+    }
+
     /**
-     * Returns an array of event names this subscriber wants to listen to.
-     *
-     * The array keys are event names and the value can be:
-     *
-     *  * The method name to call (priority defaults to 0)
-     *  * An array composed of the method name to call and the priority
-     *  * An array of arrays composed of the method names to call and respective
-     *    priorities, or 0 if unset
-     *
-     * For instance:
-     *
-     *  * array('eventName' => 'methodName')
-     *  * array('eventName' => array('methodName', $priority))
-     *  * array('eventName' => array(array('methodName1', $priority), array('methodName2'))
-     *
-     * @return array The event names to listen to
-     *
-     * @api
+     * {@inheritdoc}
      */
     public static function getSubscribedEvents()
     {
@@ -114,7 +130,8 @@ class Area extends BaseAction implements EventSubscriberInterface
             TheliaEvents::AREA_REMOVE_COUNTRY => array('removeCountry', 128),
             TheliaEvents::AREA_POSTAGE_UPDATE => array('updatePostage', 128),
             TheliaEvents::AREA_DELETE => array('delete', 128),
-            TheliaEvents::AREA_CREATE => array('create', 128)
+            TheliaEvents::AREA_CREATE => array('create', 128),
+            TheliaEvents::AREA_UPDATE => array('update', 128)
         );
     }
 }