DoctrineDataFixtureModule\Command\ImportCommand

Complexity

Total Complexity

12

Size/Duplication

Total Lines	128
Duplicated Lines	0 %

Importance

Changes	4
Bugs	0	Features	1

9 Methods

Rating	Name	Duplication	Size	Complexity
A	setPaths()	0	3	1
A	setEntityManager()	0	3	1
A	getEntityManager()	0	3	1
A	setServiceLocator()	0	3	1
A	getServiceLocator()	0	3	1
A	configure()	0	9	1
A	getPaths()	0	3	1
A	__construct()	0	5	1
A	execute()	0	27	4

<?php

namespace DoctrineDataFixtureModule\Command;

use Doctrine\Common\DataFixtures\Executor\AbstractExecutor;
use Doctrine\Common\DataFixtures\Loader;
use Doctrine\ORM\EntityManager;
use Symfony\Component\Console\Command\Command;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Output\OutputInterface;
use Symfony\Component\Console\Input\InputOption;
use Doctrine\Common\DataFixtures\Executor\ORMExecutor;
use Doctrine\Common\DataFixtures\Purger\ORMPurger;
use Zend\ServiceManager\ServiceLocatorInterface;

/**
 * Command for generate migration classes by comparing your current database schema
 * to your mapping information.
 *
 * @license http://www.opensource.org/licenses/lgpl-license.php LGPL
 * @link    www.doctrine-project.org
 * @since   2.0
 * @author  Jonathan Wage <jonwage@gmail.com>
 * @author  Rob van der Lee <r.vdlee@salesupply.com>
 */
class ImportCommand extends Command
{
    /**
     * @var array
     */
    protected $paths;

    /**
     * @var EntityManager
     */
    protected $entityManager;
    
    /**
     * The Zend ServiceManager
     * @var ServiceLocatorInterface
     */
    protected $serviceLocator;

    /**
     * @var int
     */
    const PURGE_MODE_TRUNCATE = 2;

    /**
     * ImportCommand constructor.
     * @param ServiceLocatorInterface $serviceLocator
     */
    public function __construct(ServiceLocatorInterface $serviceLocator)
    {
        $this->setServiceLocator($serviceLocator);

        parent::__construct();
    }

    /**
     * Configure the command
     */
    protected function configure()
    {
        parent::configure();

        $this->setName('data-fixture:import')
            ->setDescription('Import Data Fixtures')
            ->setHelp('The import command Imports data-fixtures')
            ->addOption('append', null, InputOption::VALUE_NONE, 'Append data to existing data.')
            ->addOption('purge-with-truncate', null, InputOption::VALUE_NONE, 'Truncate tables before inserting data');
    }

    /**
     * @param InputInterface $input
     * @param OutputInterface $output
     * @return int|null|void
     */
    public function execute(InputInterface $input, OutputInterface $output)
    {
        /** @var array $config */
        $config = $this->getServiceLocator()->get('config');
        /** @var Loader $loader */
        $loader = new Loader();
        /** @var ORMPurger $purger */
        $purger = new ORMPurger();

        // Accept option purge-with-truncate
        if ($input->getOption('purge-with-truncate')) {
            $purger->setPurgeMode(self::PURGE_MODE_TRUNCATE);
        }

        if ( ! isset($config['rvdlee']['doctrine-data-fixture']['executor'])) {
            /** @var ORMExecutor $executor */
            $executor = new ORMExecutor($this->getEntityManager(), $purger);
        } else {
            /** @var AbstractExecutor $executor */
            $executor = $this->getServiceLocator()->get($config['rvdlee']['doctrine-data-fixture']['executor']);
        }

        foreach ($this->getPaths() as $key => $value) {
            $loader->loadFromDirectory($value);
        }

        $executor->execute($loader->getFixtures(), $input->getOption('append'));
    }

    /**
     * @return array
     */
    public function getPaths(): array
    {
        return $this->paths;
    }

    /**
     * @param array $paths
     */
    public function setPaths(array $paths): void
    {
        $this->paths = $paths;
    }

    /**
     * @return EntityManager
     */
    public function getEntityManager(): EntityManager
    {
        return $this->entityManager;
    }

    /**
     * @param EntityManager $entityManager
     */
    public function setEntityManager(EntityManager $entityManager): void
    {
        $this->entityManager = $entityManager;
    }

    /**
     * @return ServiceLocatorInterface
     */
    public function getServiceLocator(): ServiceLocatorInterface
    {
        return $this->serviceLocator;
    }

    /**
     * @param ServiceLocatorInterface $serviceLocator
     */
    public function setServiceLocator(ServiceLocatorInterface $serviceLocator): void
    {
        $this->serviceLocator = $serviceLocator;
    }
}