082a0130c2
Signed-off-by: Gergely POLONKAI (W00d5t0ck) <polesz@w00d5t0ck.info>
120 lines
4.4 KiB
PHP
120 lines
4.4 KiB
PHP
<?php
|
|
|
|
/*
|
|
* This file is part of the Symfony package.
|
|
*
|
|
* (c) Fabien Potencier <fabien@symfony.com>
|
|
*
|
|
* For the full copyright and license information, please view the LICENSE
|
|
* file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace Sensio\Bundle\GeneratorBundle\Generator;
|
|
|
|
use Symfony\Component\Filesystem\Filesystem;
|
|
use Symfony\Component\HttpKernel\Bundle\BundleInterface;
|
|
use Symfony\Bridge\Doctrine\RegistryInterface;
|
|
use Doctrine\ORM\Mapping\ClassMetadataInfo;
|
|
use Doctrine\ORM\Tools\EntityGenerator;
|
|
use Doctrine\ORM\Tools\EntityRepositoryGenerator;
|
|
use Doctrine\ORM\Tools\Export\ClassMetadataExporter;
|
|
|
|
/**
|
|
* Generates a form class based on a Doctrine entity.
|
|
*
|
|
* @author Fabien Potencier <fabien@symfony.com>
|
|
* @author Jonathan H. Wage <jonwage@gmail.com>
|
|
*/
|
|
class DoctrineEntityGenerator extends Generator
|
|
{
|
|
private $filesystem;
|
|
private $registry;
|
|
|
|
public function __construct(Filesystem $filesystem, RegistryInterface $registry)
|
|
{
|
|
$this->filesystem = $filesystem;
|
|
$this->registry = $registry;
|
|
}
|
|
|
|
public function generate(BundleInterface $bundle, $entity, $format, array $fields, $withRepository)
|
|
{
|
|
// configure the bundle (needed if the bundle does not contain any Entities yet)
|
|
$config = $this->registry->getEntityManager(null)->getConfiguration();
|
|
$config->setEntityNamespaces(array_merge(
|
|
array($bundle->getName() => $bundle->getNamespace().'\\Entity'),
|
|
$config->getEntityNamespaces()
|
|
));
|
|
|
|
$entityClass = $this->registry->getEntityNamespace($bundle->getName()).'\\'.$entity;
|
|
$entityPath = $bundle->getPath().'/Entity/'.str_replace('\\', '/', $entity).'.php';
|
|
if (file_exists($entityPath)) {
|
|
throw new \RuntimeException(sprintf('Entity "%s" already exists.', $entityClass));
|
|
}
|
|
|
|
$class = new ClassMetadataInfo($entityClass);
|
|
if ($withRepository) {
|
|
$class->customRepositoryClassName = $entityClass.'Repository';
|
|
}
|
|
$class->mapField(array('fieldName' => 'id', 'type' => 'integer', 'id' => true));
|
|
$class->setIdGeneratorType(ClassMetadataInfo::GENERATOR_TYPE_AUTO);
|
|
foreach ($fields as $field) {
|
|
$class->mapField($field);
|
|
}
|
|
|
|
$entityGenerator = $this->getEntityGenerator();
|
|
if ('annotation' === $format) {
|
|
$entityGenerator->setGenerateAnnotations(true);
|
|
$entityCode = $entityGenerator->generateEntityClass($class);
|
|
$mappingPath = $mappingCode = false;
|
|
} else {
|
|
$cme = new ClassMetadataExporter();
|
|
$exporter = $cme->getExporter('yml' == $format ? 'yaml' : $format);
|
|
$mappingPath = $bundle->getPath().'/Resources/config/doctrine/'.str_replace('\\', '.', $entity).'.orm.'.$format;
|
|
|
|
if (file_exists($mappingPath)) {
|
|
throw new \RuntimeException(sprintf('Cannot generate entity when mapping "%s" already exists.', $mappingPath));
|
|
}
|
|
|
|
$mappingCode = $exporter->exportClassMetadata($class);
|
|
$entityGenerator->setGenerateAnnotations(false);
|
|
$entityCode = $entityGenerator->generateEntityClass($class);
|
|
}
|
|
|
|
$this->filesystem->mkdir(dirname($entityPath));
|
|
file_put_contents($entityPath, $entityCode);
|
|
|
|
if ($mappingPath) {
|
|
$this->filesystem->mkdir(dirname($mappingPath));
|
|
file_put_contents($mappingPath, $mappingCode);
|
|
}
|
|
|
|
if ($withRepository) {
|
|
$path = $bundle->getPath().str_repeat('/..', substr_count(get_class($bundle), '\\'));
|
|
$this->getRepositoryGenerator()->writeEntityRepositoryClass($class->customRepositoryClassName, $path);
|
|
}
|
|
}
|
|
|
|
public function isReservedKeyword($keyword)
|
|
{
|
|
return $this->registry->getConnection()->getDatabasePlatform()->getReservedKeywordsList()->isKeyword($keyword);
|
|
}
|
|
|
|
protected function getEntityGenerator()
|
|
{
|
|
$entityGenerator = new EntityGenerator();
|
|
$entityGenerator->setGenerateAnnotations(false);
|
|
$entityGenerator->setGenerateStubMethods(true);
|
|
$entityGenerator->setRegenerateEntityIfExists(false);
|
|
$entityGenerator->setUpdateEntityIfExists(true);
|
|
$entityGenerator->setNumSpaces(4);
|
|
$entityGenerator->setAnnotationPrefix('ORM\\');
|
|
|
|
return $entityGenerator;
|
|
}
|
|
|
|
protected function getRepositoryGenerator()
|
|
{
|
|
return new EntityRepositoryGenerator();
|
|
}
|
|
}
|