local/modules/TheliaMigrateCountry/EventListeners/MigrateCountryListener.php line 35

  1. <?php
  3. /*
  4.  * This file is part of the Thelia package.
  5.  * http://www.thelia.net
  6.  *
  7.  * (c) OpenStudio <info@thelia.net>
  8.  *
  9.  * For the full copyright and license information, please view the LICENSE
  10.  * file that was distributed with this source code.
  11.  */
  13. namespace TheliaMigrateCountry\EventListeners;
  15. use Propel\Runtime\Exception\PropelException;
  16. use Propel\Runtime\Propel;
  17. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  18. use Thelia\Model\AddressQuery;
  19. use Thelia\Model\Base\CountryQuery;
  20. use Thelia\Model\CountryAreaQuery;
  21. use Thelia\Model\Map\AddressTableMap;
  22. use Thelia\Model\Map\CountryAreaTableMap;
  23. use Thelia\Model\Map\TaxRuleCountryTableMap;
  24. use Thelia\Model\TaxRuleCountryQuery;
  25. use TheliaMigrateCountry\Events\MigrateCountryEvent;
  26. use TheliaMigrateCountry\Events\MigrateCountryEvents;
  28. /**
  29.  * Class MigrateCountryListener.
  30.  *
  31.  * @author Julien Chanséaume <julien@thelia.net>
  32.  */
  33. class MigrateCountryListener implements EventSubscriberInterface
  34. {
  35.     public function migrateCountry(MigrateCountryEvent $event): void
  36.     {
  37.         $counter = [];
  39.         // update address
  40.         $counter[AddressTableMap::TABLE_NAME] = $this->migrateAddress($event);
  42.         // tax rules
  43.         $counter[TaxRuleCountryTableMap::TABLE_NAME] = $this->migrateAddress($event);
  45.         // shipping zone
  46.         $counter[CountryAreaTableMap::TABLE_NAME] = $this->migrateAddress($event);
  48.         // if it succeeds we toggle the visibility of old country and new
  49.         $this->setCountriesVisibility($event);
  51.         $event->setCounter($counter);
  52.     }
  54.     protected function migrateAddress(MigrateCountryEvent $event)
  55.     {
  56.         $con = Propel::getWriteConnection(AddressTableMap::DATABASE_NAME);
  57.         $con->beginTransaction();
  58.         try {
  59.             $updatedRows = AddressQuery::create()
  60.                 ->filterByCountryId($event->getCountry())
  61.                 ->update(
  62.                     [
  63.                         'CountryId' => $event->getNewCountry(),
  64.                         'StateId' => $event->getNewState(),
  65.                     ]
  66.                 );
  68.             $con->commit();
  70.             return $updatedRows;
  71.         } catch (PropelException $e) {
  72.             $con->rollback();
  73.             throw $e;
  74.         }
  75.     }
  77.     protected function migrateTaxRules(MigrateCountryEvent $event)
  78.     {
  79.         $con = Propel::getWriteConnection(TaxRuleCountryTableMap::DATABASE_NAME);
  80.         $con->beginTransaction();
  81.         try {
  82.             $updatedRows = TaxRuleCountryQuery::create()
  83.                 ->filterByCountryId($event->getCountry())
  84.                 ->update(
  85.                     [
  86.                         'CountryId' => $event->getNewCountry(),
  87.                         'StateId' => $event->getNewState(),
  88.                     ]
  89.                 );
  91.             $con->commit();
  93.             return $updatedRows;
  94.         } catch (PropelException $e) {
  95.             $con->rollback();
  96.             throw $e;
  97.         }
  98.     }
  100.     protected function migrateShippingZones(MigrateCountryEvent $event)
  101.     {
  102.         $con = Propel::getWriteConnection(CountryAreaTableMap::DATABASE_NAME);
  103.         $con->beginTransaction();
  104.         try {
  105.             $updatedRows = CountryAreaQuery::create()
  106.                 ->filterByCountryId($event->getCountry())
  107.                 ->update(
  108.                     [
  109.                         'CountryId' => $event->getNewCountry(),
  110.                         'StateId' => $event->getNewState(),
  111.                     ]
  112.                 );
  114.             $con->commit();
  116.             return $updatedRows;
  117.         } catch (PropelException $e) {
  118.             $con->rollback();
  119.             throw $e;
  120.         }
  121.     }
  123.     private function setCountriesVisibility(MigrateCountryEvent $event): void
  124.     {
  125.         $oldCountry = CountryQuery::create()->findPk($event->getCountry());
  127.         if (null !== $oldCountry) {
  128.             $oldCountry
  129.                 ->setVisible(0)
  130.                 ->save()
  131.             ;
  132.         }
  134.         $newCountry = CountryQuery::create()->findPk($event->getNewCountry());
  135.         if (null !== $newCountry) {
  136.             $newCountry
  137.                 ->setVisible(1)
  138.                 ->save()
  139.             ;
  140.         }
  141.     }
  143.     /**
  144.      * {@inheritdoc}
  145.      */
  146.     public static function getSubscribedEvents()
  147.     {
  148.         return [
  149.             MigrateCountryEvents::MIGRATE_COUNTRY => 'migrateCountry',
  150.         ];
  151.     }
  152. }