domokits-NEW/domokits/local/modules/TheliaMigrateCountry/Controller/MigrateController.php

<?php

/*
 * This file is part of the Thelia package.
 * http://www.thelia.net
 *
 * (c) OpenStudio <info@thelia.net>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace TheliaMigrateCountry\Controller;

use Symfony\Component\Form\Extension\Core\Type\FormType;
use Symfony\Contracts\EventDispatcher\EventDispatcherInterface;
use Thelia\Controller\Admin\BaseAdminController;
use Thelia\Core\Security\AccessManager;
use Thelia\Core\Security\Resource\AdminResources;
use Thelia\Form\Exception\FormValidationException;
use Thelia\Model\ConfigQuery;
use Thelia\Model\Country;
use Thelia\Model\CountryQuery;
use Thelia\Model\Map\AddressTableMap;
use Thelia\Model\Map\CountryAreaTableMap;
use Thelia\Model\Map\TaxRuleCountryTableMap;
use TheliaMigrateCountry\Events\MigrateCountryEvent;
use TheliaMigrateCountry\Events\MigrateCountryEvents;

/**
 * Class MigrateController.
 *
 * @author Julien Chanséaume <julien@thelia.net>
 */
class MigrateController extends BaseAdminController
{
    protected $useFallbackTemplate = true;

    public function migrateSystemAction()
    {
        $response = $this->checkAuth(AdminResources::COUNTRY, [], AccessManager::UPDATE);
        if (null !== $response) {
            return $response;
        }

        // load country not migrated
        $dataForm = [];

        $migratedCountries = json_decode(ConfigQuery::read('thelia_country_state_migration', '[]'), true);

        $countries = CountryQuery::create()
            ->filterByHasStates(1)
        ;

        /** @var Country $country */
        foreach ($countries as $country) {
            $oldCountries = CountryQuery::create()
                ->filterByHasStates(0)
                ->filterByIsocode($country->getIsoCode())
                ->find()
            ;
            /** @var Country $oldCountry */
            foreach ($oldCountries as $oldCountry) {
                if (!isset($migratedCountries[$oldCountry->getId()])) {
                    $dataForm[] = [
                        'migrate' => false,
                        'country' => $oldCountry->getId(),
                        'new_country' => $country->getId(),
                        'new_state' => null,
                    ];
                }
            }
        }

        // prepare form
        $form = $this->createForm('thelia.admin.country.state.migration', FormType::class, ['migrations' => $dataForm]);
        $this->getParserContext()->addForm($form);

        return $this->render(
            'countries-migrate',
            [
                'countriesMigrated' => $migratedCountries,
                'showForm' => \count($dataForm) != 0,
            ]
        );
    }

    public function doMigrateSystemAction(EventDispatcherInterface $eventDispatcher)
    {
        $response = $this->checkAuth(AdminResources::COUNTRY, [], AccessManager::UPDATE);
        if (null !== $response) {
            return $response;
        }

        $changeForm = $this->createForm('thelia.admin.country.state.migration');
        $migratedCountries = [];

        try {
            // Check the form against constraints violations
            $form = $this->validateForm($changeForm, 'POST');

            // Get the form field values
            $data = $form->getData();

            foreach ($data['migrations'] as $migration) {
                if (!$migration['migrate']) {
                    continue;
                }

                $changeEvent = new MigrateCountryEvent(
                    $migration['country'],
                    $migration['new_country'],
                    $migration['new_state']
                );

                $eventDispatcher->dispatch($changeEvent, MigrateCountryEvents::MIGRATE_COUNTRY);

                // memorize the migration
                $migratedCountries = json_decode(ConfigQuery::read('thelia_country_state_migration', '[]'), true);
                $migratedCountries[$changeEvent->getCountry()] = [
                    'country' => $changeEvent->getNewCountry(),
                    'state' => $changeEvent->getNewState(),
                    'counter' => $changeEvent->getCounter(),
                ];
                ConfigQuery::write('thelia_country_state_migration', json_encode($migratedCountries));

                // message
                $message = $this->getTranslator()->trans(
                    'Country %id migrated to country (ID %country) and state (ID %state) (address: %address, tax rules: %tax, shipping zones: %zone)',
                    [
                        '%id' => $changeEvent->getCountry(),
                        '%country' => $changeEvent->getNewCountry(),
                        '%state' => $changeEvent->getNewState(),
                        '%address' => $changeEvent->getCounter()[AddressTableMap::TABLE_NAME],
                        '%tax' => $changeEvent->getCounter()[TaxRuleCountryTableMap::TABLE_NAME],
                        '%zone' => $changeEvent->getCounter()[CountryAreaTableMap::TABLE_NAME],
                    ]
                );

                // add flash message
                $this->getSession()->getFlashBag()->add('migrate', $message);

                // Log migration
                $this->adminLogAppend(
                    AdminResources::COUNTRY,
                    AccessManager::UPDATE,
                    $message,
                    $changeEvent->getCountry()
                );
            }

            return $this->generateSuccessRedirect($changeForm);
        } catch (FormValidationException $ex) {
            // Form cannot be validated
            $error_msg = $this->createStandardFormValidationErrorMessage($ex);
        }

        if (false !== $error_msg) {
            $this->setupFormErrorContext(
                $this->getTranslator()->trans('Country migration'),
                $error_msg,
                $changeForm,
                $ex
            );

            return $this->render(
                'countries-migrate',
                [
                    'countriesMigrated' => $migratedCountries,
                    'showForm' => true,
                ]
            );
        }
    }
}