. */ namespace Doctrine\ORM\Tools\CLI\Tasks; use Doctrine\Common\CLI\Tasks\AbstractTask, Doctrine\Common\CLI\Option, Doctrine\Common\CLI\OptionGroup, Doctrine\Common\CLI\CLIException, Doctrine\ORM\Tools\EntityGenerator, Doctrine\ORM\Tools\ClassMetadataReader; /** * CLI Task to generate entity classes and method stubs from your mapping information. * * @license http://www.opensource.org/licenses/lgpl-license.php LGPL * @link www.doctrine-project.org * @since 2.0 * @version $Revision$ * @author Benjamin Eberlei * @author Guilherme Blanco * @author Jonathan Wage * @author Roman Borschel */ class GenerateEntitiesTask extends AbstractTask { /** * @inheritdoc */ public function buildDocumentation() { $options = new OptionGroup(OptionGroup::CARDINALITY_N_N, array( new Option('from', '', 'The path to mapping information.'), new Option('dest', '', 'The path to generate your entity classes.') )); $doc = $this->getDocumentation(); $doc->setName('generate-entities') ->setDescription('Generate entity classes and method stubs from your mapping information.') ->getOptionGroup() ->addOption($options); } /** * @inheritdoc */ public function validate() { $arguments = $this->getArguments(); if ( ! isset($arguments['from']) || ! isset($arguments['dest'])) { throw new CLIException('You must specify a value for --from and --dest'); } return true; } /** * @inheritdoc */ public function run() { $printer = $this->getPrinter(); $arguments = $this->getArguments(); $from = $arguments['from']; $dest = realpath($arguments['dest']); $entityGenerator = new EntityGenerator(); $entityGenerator->setGenerateAnnotations(false); $entityGenerator->setGenerateStubMethods(true); $entityGenerator->setRegenerateEntityIfExists(false); $entityGenerator->setUpdateEntityIfExists(true); if (isset($arguments['extend']) && $arguments['extend']) { $entityGenerator->setClassToExtend($arguments['extend']); } if (isset($arguments['num-spaces']) && $arguments['extend']) { $entityGenerator->setNumSpaces($arguments['num-spaces']); } $reader = new ClassMetadataReader(); $reader->setEntityManager($this->getConfiguration()->getAttribute('em')); $reader->addMappingSource($from); $metadatas = $reader->getMetadatas(); foreach ($metadatas as $metadata) { $printer->writeln( sprintf('Processing entity "%s"', $printer->format($metadata->name, 'KEYWORD')) ); } $entityGenerator->generate($metadatas, $dest); $printer->write(PHP_EOL); $printer->writeln( sprintf('Entity classes generated to "%s"', $printer->format($dest, 'KEYWORD') ) ); } }