EntityRepositoryGenerator default repository

lib/Doctrine/ORM/Tools/Console/Command/GenerateRepositoriesCommand.php

@@ -73,6 +73,8 @@ protected function execute(InputInterface $input, OutputInterface $output)
         $metadatas = $em->getMetadataFactory()->getAllMetadata();
         $metadatas = MetadataFilter::filter($metadatas, $input->getOption('filter'));
 
+        $repositoryName = $em->getConfiguration()->getDefaultRepositoryClassName();
+
         // Process destination directory
         $destPath = realpath($input->getArgument('dest-path'));
 
@@ -92,6 +94,8 @@ protected function execute(InputInterface $input, OutputInterface $output)
             $numRepositories = 0;
             $generator = new EntityRepositoryGenerator();
 
+            $generator->setDefaultRepositoryName($repositoryName);
+
             foreach ($metadatas as $metadata) {
                 if ($metadata->customRepositoryClassName) {
                     $output->writeln(

lib/Doctrine/ORM/Tools/EntityRepositoryGenerator.php

@@ -32,20 +32,20 @@
  */
 class EntityRepositoryGenerator
 {
+    protected $_repositoryName = 'Doctrine\ORM\EntityRepository';
+
     protected static $_template =
 '<?php
 
 <namespace>
 
-use Doctrine\ORM\EntityRepository;
-
 /**
  * <className>
  *
  * This class was generated by the Doctrine ORM. Add your own custom
  * repository methods below.
  */
-class <className> extends EntityRepository
+class <className> extends <repositoryName>
 {
 }
 ';
@@ -57,16 +57,49 @@ class <className> extends EntityRepository
      */
     public function generateEntityRepositoryClass($fullClassName)
     {
-        $className = substr($fullClassName, strrpos($fullClassName, '\\') + 1, strlen($fullClassName));
-
         $variables = array(
-            '<namespace>' => $this->generateEntityRepositoryNamespace($fullClassName),
-            '<className>' => $className
+            '<namespace>'       => $this->generateEntityRepositoryNamespace($fullClassName),
+            '<repositoryName>'  => $this->generateEntityRepositoryName($fullClassName),
+            '<className>'       => $this->generateClassName($fullClassName)
         );
 
         return str_replace(array_keys($variables), array_values($variables), self::$_template);
     }
 
+    /**
+     * Generates the namespace, if class do not have namespace, return empty string instead.
+     *
+     * @param string $fullClassName
+     * 
+     * @return string $namespace
+     */
+    protected function getClassNamespace($fullClassName)
+    {
+        $namespace = substr($fullClassName, 0, strrpos($fullClassName, '\\'));
+
+        return $namespace;
+    }
+
+    /**
+     * Generates the class name
+     * 
+     * @param string $fullClassName
+     * 
+     * @return string
+     */
+    protected function generateClassName($fullClassName)
+    {
+        $namespace = $this->getClassNamespace($fullClassName);
+
+        $className = $fullClassName;
+
+        if ($namespace) {
+            $className = substr($fullClassName, strrpos($fullClassName, '\\') + 1, strlen($fullClassName));
+        }
+
+        return $className;
+    }
+
     /**
      * Generates the namespace statement, if class do not have namespace, return empty string instead.
      * 
@@ -76,11 +109,29 @@ public function generateEntityRepositoryClass($fullClassName)
      */
     private function generateEntityRepositoryNamespace($fullClassName)
     {
-        $namespace = substr($fullClassName, 0, strrpos($fullClassName, '\\'));
-        
+        $namespace = $this->getClassNamespace($fullClassName);
+
         return $namespace ? 'namespace ' . $namespace . ';' : '';
     }
 
+    /**
+     * @param string $fullClassName
+     * 
+     * @return string $repositoryName
+     */
+    protected function generateEntityRepositoryName($fullClassName)
+    {
+        $namespace = $this->getClassNamespace($fullClassName);
+
+        $repositoryName = $this->_repositoryName;
+
+        if ($namespace && $repositoryName[0] !== '\\') {
+            $repositoryName = '\\' . $repositoryName;
+        }
+
+        return $repositoryName;
+    }
+
     /**
      * @param string $fullClassName
      * @param string $outputDirectory
@@ -103,4 +154,17 @@ public function writeEntityRepositoryClass($fullClassName, $outputDirectory)
             file_put_contents($path, $code);
         }
     }
+
+    /**
+     * @param string $repositoryName
+     * 
+     * @return \Doctrine\ORM\Tools\EntityRepositoryGenerator
+     */
+    public function setDefaultRepositoryName($repositoryName)
+    {
+        $this->_repositoryName = $repositoryName;
+
+        return $this;
+    }
+
 }