. */ 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\Export\ClassMetadataExporter; /** * 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 ConvertMappingTask { /** * @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 your entities.') )); $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(); $em = $this->getConfiguration()->getAttribute('em'); if ( ! isset($arguments['from']) || ! isset($arguments['dest'])) { throw new CliException('You must specify a value for --from and --dest'); } return true; } public function run() { $printer = $this->getPrinter(); $arguments = $this->getArguments(); $from = $arguments['from']; $dest = realpath($arguments['dest']); $generator = new EntityGenerator(); $generator->setGenerateAnnotations(false); $generator->setGenerateStubMethods(true); $generator->setRegenerateEntityIfExists(false); $generator->setUpdateEntityIfExists(true); if (isset($arguments['extend']) && $arguments['extend']) { $generator->setClassToExtend($arguments['extend']); } if (isset($arguments['num-spaces']) && $arguments['extend']) { $generator->setNumSpaces($arguments['num-spaces']); } $type = $this->_determineSourceType($from); if ( ! $type) { throw new CliException( "Invalid mapping source type '$sourceArg'." ); } $source = $this->_getSourceByType($type, $from); $cme = new ClassMetadataExporter(); $cme->addMappingSource($source, $type); $metadatas = $cme->getMetadatasForMappingSources(); $printer->writeln( sprintf( 'Generating entity stubs for "%s" mapping information located at "%s" to "%s"', $printer->format($type, 'KEYWORD'), $printer->format($from, 'KEYWORD'), $printer->format($dest, 'KEYWORD') ) ); $generator->generate($metadatas, $dest); } }