doctrine2/lib/Doctrine/ORM/Tools/Export/ClassMetadataExporter.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\Export;

use Doctrine\ORM\EntityManager;
use Doctrine\ORM\Mapping\ClassMetadataInfo;
use Doctrine\ORM\Mapping\ClassMetadata;

/**
 * Class used for converting your mapping information between the 
 * supported formats: yaml, xml, and php/annotation.
 *
 *     [php]
 *     // Unify all your mapping information which is written in php, xml, yml
 *     // and convert it to a single set of yaml files.
 *     
 *     $cme = new Doctrine\ORM\Tools\Export\ClassMetadataExporter();
 *     $cme->addMappingDirectory(__DIR__ . '/Entities', 'php');
 *     $cme->addMappingDirectory(__DIR__ . '/xml', 'xml');
 *     $cme->addMappingDirectory(__DIR__ . '/yaml', 'yaml');
 *     
 *     $exporter = $cme->getExporter('yaml');
 *     $exporter->setOutputDir(__DIR__ . '/new_yaml');
 *
 *     $exporter->setMetadatas($cme->getMetadatasForMappingDirectories());
 *     $exporter->export();
 *
 * @license http://www.opensource.org/licenses/lgpl-license.php LGPL
 * @link    www.doctrine-project.org
 * @since   2.0
 * @version $Revision$
 * @author  Jonathan Wage <jonwage@gmail.com>
 */
class ClassMetadataExporter
{
    private $_exporterDrivers = array(
        'xml' => 'Doctrine\ORM\Tools\Export\Driver\XmlExporter',
        'yaml' => 'Doctrine\ORM\Tools\Export\Driver\YamlExporter',
        'yml' => 'Doctrine\ORM\Tools\Export\Driver\YamlExporter',
        'php' => 'Doctrine\ORM\Tools\Export\Driver\PhpExporter',
        'annotation' => 'Doctrine\ORM\Tools\Export\Driver\AnnotationExporter'
    );

    private $_mappingDrivers = array(
        'annotation' => 'Doctrine\ORM\Mapping\Driver\AnnotationDriver',
        'yaml' => 'Doctrine\ORM\Mapping\Driver\YamlDriver',
        'yml' => 'Doctrine\ORM\Mapping\Driver\YamlDriver',
        'xml'  => 'Doctrine\ORM\Mapping\Driver\XmlDriver'
    );

    private $_mappingDirectories = array();

    /**
     * Add a new mapping directory to the array of directories to convert and export
     * to another format
     *
     *     [php]
     *     $cme = new Doctrine\ORM\Tools\Export\ClassMetadataExporter();
     *     $cme->addMappingDirectory(__DIR__ . '/yaml', 'yaml');
     *
     * @param string $dir   The path to the mapping files
     * @param string $type  The type of mapping files (yml, xml, etc.)
     * @return void
     */
    public function addMappingDirectory($dir, $type)
    {
        if ($type === 'php') {
            $this->_mappingDirectories[] = array($dir, $type);
        } else {
            if ( ! isset($this->_mappingDrivers[$type])) {
                throw DoctrineException::invalidMappingDriverType($type);
            }

            $driver = $this->getMappingDriver($type, $dir);
            $this->_mappingDirectories[] = array($dir, $driver);
        }
    }

    /**
     * Get an instance of a mapping driver
     *
     * @param string $type   The type of mapping driver (yaml, xml, annotation, etc.)
     * @param string $dir    The directory to configure the driver to look in. Only required for file drivers (yml, xml).
     * @return AbstractDriver $driver
     */
    public function getMappingDriver($type, $dir = null)
    {
        if ( ! isset($this->_mappingDrivers[$type])) {
            return false;
        }
        $class = $this->_mappingDrivers[$type];
        if (is_subclass_of($class, 'Doctrine\ORM\Mapping\Driver\AbstractFileDriver')) {
            if (is_null($dir)) {
                throw DoctrineException::fileMappingDriversRequireDirectoryPath();
            }
            $driver = new $class($dir, constant($class . '::PRELOAD'));
        } else {
            $reader = new \Doctrine\Common\Annotations\AnnotationReader(new \Doctrine\Common\Cache\ArrayCache);
            $reader->setDefaultAnnotationNamespace('Doctrine\ORM\Mapping\\');
            $driver = new $class($reader);
        }
        return $driver;
    }

    /**
     * Get the array of added mapping directories
     *
     * @return array $mappingDirectories
     */
    public function getMappingDirectories()
    {
        return $this->_mappingDirectories;
    }

    /**
     * Get an array of ClassMetadataInfo instances for all the configured mapping
     * directories. Reads the mapping directories and populates ClassMetadataInfo
     * instances.
     *
     * @return array $classes
     */
    public function getMetadatasForMappingDirectories()
    {
        $classes = array();

        foreach ($this->_mappingDirectories as $d) {
            list($dir, $driver) = $d;
            if ($driver == 'php') {
                $iter = new \RecursiveIteratorIterator(new \RecursiveDirectoryIterator($dir),
                                                      \RecursiveIteratorIterator::LEAVES_ONLY);

                foreach ($iter as $item) {
                    $info = pathinfo($item->getPathName());
                    if (! isset($info['extension']) || $info['extension'] != 'php') {
                        continue;
                    }
                    include $item->getPathName();
                    $vars = get_defined_vars();
                    foreach ($vars as $var) {
                        if ($var instanceof \Doctrine\ORM\Mapping\ClassMetadataInfo) {
                            $classes[$var->name] = $var;
                        }
                    }
                }
            } else {
                if ($driver instanceof \Doctrine\ORM\Mapping\Driver\AnnotationDriver) {
                    $iter = new \RecursiveIteratorIterator(new \RecursiveDirectoryIterator($dir),
                                                          \RecursiveIteratorIterator::LEAVES_ONLY);

                    $declared = get_declared_classes();          
                    foreach ($iter as $item) {
                        $info = pathinfo($item->getPathName());
                        if (! isset($info['extension']) || $info['extension'] != 'php') {
                            continue;
                        }
                        require_once $item->getPathName();
                    }
                    $declared = array_diff(get_declared_classes(), $declared);

                    foreach ($declared as $className) {                 
                        if ( ! $driver->isTransient($className)) {
                            $metadata = new ClassMetadata($className);  
                            $driver->loadMetadataForClass($className, $metadata);
                            $classes[$metadata->name] = $metadata;
                        }
                    }
                } else {
                    $preloadedClasses = $driver->preload(true);
                    foreach ($preloadedClasses as $className) {
                        $metadata = new ClassMetadataInfo($className);    
                        $driver->loadMetadataForClass($className, $metadata);
                        $classes[$metadata->name] = $metadata;
                    }
                }
            }
        }

        foreach ($classes as $key => $class) {
            if ($class->isMappedSuperclass) {
                unset($classes[$key]);
            }
        }
        $classes = array_values($classes);
        return $classes;
    }

    /**
     * Get a exporter driver instance
     *
     * @param string $type   The type to get (yml, xml, etc.)
     * @param string $dir    The directory where the exporter will export to
     * @return AbstractExporter $exporter
     */
    public function getExporter($type, $dir = null)
    {
        if ( ! isset($this->_exporterDrivers[$type])) {
            throw DoctrineException::invalidExporterDriverType($type);
        }

        $class = $this->_exporterDrivers[$type];
        return new $class($dir);
    }
}