082a0130c2
Signed-off-by: Gergely POLONKAI (W00d5t0ck) <polesz@w00d5t0ck.info>
135 lines
5.6 KiB
PHP
135 lines
5.6 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Doctrine Bundle
|
|
*
|
|
* The code was originally distributed inside the Symfony framework.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
* (c) Doctrine Project, Benjamin Eberlei <kontakt@beberlei.de>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Doctrine\Bundle\DoctrineBundle\Command;
|
|
|
|
use Symfony\Component\Console\Input\InputArgument;
|
|
use Symfony\Component\Console\Input\InputOption;
|
|
use Symfony\Component\Console\Input\InputInterface;
|
|
use Symfony\Component\Console\Output\OutputInterface;
|
|
use Doctrine\ORM\Tools\EntityRepositoryGenerator;
|
|
use Doctrine\Bundle\DoctrineBundle\Mapping\DisconnectedMetadataFactory;
|
|
|
|
/**
|
|
* Generate entity classes from mapping information
|
|
*
|
|
* @author Fabien Potencier <fabien@symfony.com>
|
|
* @author Jonathan H. Wage <jonwage@gmail.com>
|
|
*/
|
|
class GenerateEntitiesDoctrineCommand extends DoctrineCommand
|
|
{
|
|
protected function configure()
|
|
{
|
|
$this
|
|
->setName('doctrine:generate:entities')
|
|
->setAliases(array('generate:doctrine:entities'))
|
|
->setDescription('Generates entity classes and method stubs from your mapping information')
|
|
->addArgument('name', InputArgument::REQUIRED, 'A bundle name, a namespace, or a class name')
|
|
->addOption('path', null, InputOption::VALUE_REQUIRED, 'The path where to generate entities when it cannot be guessed')
|
|
->addOption('no-backup', null, InputOption::VALUE_NONE, 'Do not backup existing entities files.')
|
|
->setHelp(<<<EOT
|
|
The <info>doctrine:generate:entities</info> command generates entity classes
|
|
and method stubs from your mapping information:
|
|
|
|
You have to limit generation of entities:
|
|
|
|
* To a bundle:
|
|
|
|
<info>php app/console doctrine:generate:entities MyCustomBundle</info>
|
|
|
|
* To a single entity:
|
|
|
|
<info>php app/console doctrine:generate:entities MyCustomBundle:User</info>
|
|
<info>php app/console doctrine:generate:entities MyCustomBundle/Entity/User</info>
|
|
|
|
* To a namespace
|
|
|
|
<info>php app/console doctrine:generate:entities MyCustomBundle/Entity</info>
|
|
|
|
If the entities are not stored in a bundle, and if the classes do not exist,
|
|
the command has no way to guess where they should be generated. In this case,
|
|
you must provide the <comment>--path</comment> option:
|
|
|
|
<info>php app/console doctrine:generate:entities Blog/Entity --path=src/</info>
|
|
|
|
By default, the unmodified version of each entity is backed up and saved
|
|
(e.g. Product.php~). To prevent this task from creating the backup file,
|
|
pass the <comment>--no-backup</comment> option:
|
|
|
|
<info>php app/console doctrine:generate:entities Blog/Entity --no-backup</info>
|
|
|
|
<error>Important:</error> Even if you specified Inheritance options in your
|
|
XML or YAML Mapping files the generator cannot generate the base and
|
|
child classes for you correctly, because it doesn't know which
|
|
class is supposed to extend which. You have to adjust the entity
|
|
code manually for inheritance to work!
|
|
|
|
EOT
|
|
);
|
|
}
|
|
|
|
protected function execute(InputInterface $input, OutputInterface $output)
|
|
{
|
|
$manager = new DisconnectedMetadataFactory($this->getContainer()->get('doctrine'));
|
|
|
|
try {
|
|
$bundle = $this->getApplication()->getKernel()->getBundle($input->getArgument('name'));
|
|
|
|
$output->writeln(sprintf('Generating entities for bundle "<info>%s</info>"', $bundle->getName()));
|
|
$metadata = $manager->getBundleMetadata($bundle);
|
|
} catch (\InvalidArgumentException $e) {
|
|
$name = strtr($input->getArgument('name'), '/', '\\');
|
|
|
|
if (false !== $pos = strpos($name, ':')) {
|
|
$name = $this->getContainer()->get('doctrine')->getEntityNamespace(substr($name, 0, $pos)).'\\'.substr($name, $pos + 1);
|
|
}
|
|
|
|
if (class_exists($name)) {
|
|
$output->writeln(sprintf('Generating entity "<info>%s</info>"', $name));
|
|
$metadata = $manager->getClassMetadata($name, $input->getOption('path'));
|
|
} else {
|
|
$output->writeln(sprintf('Generating entities for namespace "<info>%s</info>"', $name));
|
|
$metadata = $manager->getNamespaceMetadata($name, $input->getOption('path'));
|
|
}
|
|
}
|
|
|
|
$generator = $this->getEntityGenerator();
|
|
|
|
$backupExisting = !$input->getOption('no-backup');
|
|
$generator->setBackupExisting($backupExisting);
|
|
|
|
$repoGenerator = new EntityRepositoryGenerator();
|
|
foreach ($metadata->getMetadata() as $m) {
|
|
if ($backupExisting) {
|
|
$basename = substr($m->name, strrpos($m->name, '\\') + 1);
|
|
$output->writeln(sprintf(' > backing up <comment>%s.php</comment> to <comment>%s.php~</comment>', $basename, $basename));
|
|
}
|
|
// Getting the metadata for the entity class once more to get the correct path if the namespace has multiple occurrences
|
|
try {
|
|
$entityMetadata = $manager->getClassMetadata($m->getName(), $input->getOption('path'));
|
|
} catch (\RuntimeException $e) {
|
|
// fall back to the bundle metadata when no entity class could be found
|
|
$entityMetadata = $metadata;
|
|
}
|
|
|
|
$output->writeln(sprintf(' > generating <comment>%s</comment>', $m->name));
|
|
$generator->generate(array($m), $entityMetadata->getPath());
|
|
|
|
if ($m->customRepositoryClassName && false !== strpos($m->customRepositoryClassName, $metadata->getNamespace())) {
|
|
$repoGenerator->writeEntityRepositoryClass($m->customRepositoryClassName, $metadata->getPath());
|
|
}
|
|
}
|
|
}
|
|
}
|