doctrine2/lib/Doctrine/ORM/Tools/Console/Command/SchemaTool/AbstractCommand.php

<?php
/*
 *  $Id$
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * This software consists of voluntary contributions made by many individuals
 * and is licensed under the LGPL. For more information, see
 * <http://www.doctrine-project.org>.
*/

namespace Doctrine\ORM\Tools\Console\Command\SchemaTool;

use Symfony\Components\Console\Input\InputArgument,
    Symfony\Components\Console\Input\InputOption,
    Symfony\Components\Console\Input\InputInterface,
    Symfony\Components\Console\Output\OutputInterface,
    Symfony\Components\Console\Command\Command,
    Doctrine\ORM\Tools\Console\Helper\EntityManagerHelper,
    Doctrine\ORM\Tools\SchemaTool,
    Doctrine\ORM\Mapping\Driver\AbstractFileDriver;

abstract class AbstractCommand extends Command
{
    /**
     * @param InputInterface $input
     * @param OutputInterface $output
     * @param SchemaTool $schemaTool
     * @param array $metadatas
     */
    abstract protected function executeSchemaCommand(InputInterface $input, OutputInterface $output, SchemaTool $schemaTool, array $metadatas);

    /**
     * @see Console\Command\Command
     */
    protected function execute(InputInterface $input, OutputInterface $output)
    {
        $emHelper = $this->getHelper('em');

        /* @var $em \Doctrine\ORM\EntityManager */
        $em = $emHelper->getEntityManager();

        $reader = new \Doctrine\ORM\Tools\ClassMetadataReader();
        $reader->setEntityManager($em);

        $metadataDriver = $em->getConfiguration()->getMetadataDriverImpl();
        if ($metadataDriver instanceof AbstractFileDriver) {
            $reader->addMappingSource($metadataDriver);
        }

        // Process source directories
        if ($emHelper->hasAdditionalMappingPathInformation()) {

            foreach ($emHelper->getMappingPaths() as $dirName) {
                $dirName = realpath($dirName);

                if ( ! file_exists($dirName)) {
                    throw new \InvalidArgumentException(
                        sprintf("Mapping directory '<info>%s</info>' does not exist.", $dirName)
                    );
                } else if ( ! is_readable($dirName)) {
                    throw new \InvalidArgumentException(
                        sprintf("Mapping directory '<info>%s</info>' does not have read permissions.", $dirName)
                    );
                }

                $reader->addMappingSource($dirName);
            }
        }

        // Retrieving ClassMetadatas, autoloading required since we need access to the Reflection stuff.
        $metadatas = $reader->getMetadatas(true);

        if ( ! empty($metadatas)) {
            // Create SchemaTool
            $tool = new \Doctrine\ORM\Tools\SchemaTool($em);

            $this->executeSchemaCommand($input, $output, $tool, $metadatas);
        } else {
            $output->write('No Metadata Classes to process.' . PHP_EOL);
        }
    }
}