<?php

/*

 * This file is part of the Symfony package.

 *

 * (c) Fabien Potencier <fabien@symfony.com>

 *

 * For the full copyright and license information, please view the LICENSE

 * file that was distributed with this source code.

 */

namespace Symfony\Bundle\DoctrineBundle\Command;

use Symfony\Bundle\FrameworkBundle\Command\ContainerAwareCommand;

use Doctrine\ORM\Tools\EntityGenerator;

/**

 * Base class for Doctrine console commands to extend from.

 *

 * @author Fabien Potencier <fabien@symfony.com>

 */

abstract class DoctrineCommand extends ContainerAwareCommand

{

    protected function getEntityGenerator()

    {

        $entityGenerator = new EntityGenerator();

        $entityGenerator->setGenerateAnnotations(false);

        $entityGenerator->setGenerateStubMethods(true);

        $entityGenerator->setRegenerateEntityIfExists(false);

        $entityGenerator->setUpdateEntityIfExists(true);

        $entityGenerator->setNumSpaces(4);

        $entityGenerator->setAnnotationPrefix('ORM\\');

        return $entityGenerator;

    }

    protected function getEntityManager($name)

    {

        return $this->getContainer()->get('doctrine')->getEntityManager($name);

    }

    /**

     * Get a doctrine dbal connection by symfony name.

     *

     * @param string $name

     * @return Doctrine\DBAL\Connection

     */

    protected function getDoctrineConnection($name)

    {

        return $this->getContainer()->get('doctrine')->getConnection($name);

    }

}