Extracting repository generation code to standalone class so it can be re-used

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

@@ -24,7 +24,8 @@ namespace Doctrine\ORM\Tools\Console\Command;
 use Symfony\Components\Console\Input\InputArgument,
     Symfony\Components\Console\Input\InputOption,
     Symfony\Components\Console,
-    Doctrine\ORM\Tools\Console\MetadataFilter;
+    Doctrine\ORM\Tools\Console\MetadataFilter,
+    Doctrine\ORM\Tools\EntityRepositoryGenerator;
 
 /**
  * Command to generate repository classes for mapping information.
@@ -40,23 +41,6 @@ use Symfony\Components\Console\Input\InputArgument,
  */
 class GenerateRepositoriesCommand extends Console\Command\Command
 {
-    protected static $_template =
-'<?php
-
-namespace <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
-{
-}';
-
     /**
      * @see Console\Command\Command
      */
@@ -105,6 +89,7 @@ EOT
 
         if (count($metadatas)) {
             $numRepositories = 0;
+            $generator = new EntityRepositoryGenerator();
 
             foreach ($metadatas as $metadata) {
                 if ($metadata->customRepositoryClassName) {
@@ -112,7 +97,7 @@ EOT
                         sprintf('Processing repository "<info>%s</info>"', $metadata->customRepositoryClassName) . PHP_EOL
                     );
 
-                    $this->_writeRepositoryClass($metadata, $destPath);
+                    $generator->writeEntityRepositoryClass($metadata->customRepositoryClassName, $destPath);
 
                     $numRepositories++;
                 }
@@ -128,33 +113,4 @@ EOT
             $output->write('No Metadata Classes to process.' . PHP_EOL);
         }
     }
-
-    protected function _generateRepositoryClass($metadata)
-    {
-        fullClassName = $metadata->customRepositoryClassName;
-        $namespace = substr($fullClassName, 0, strrpos($fullClassName, '\\'));
-        $className = substr($fullClassName, strrpos($fullClassName, '\\') + 1, strlen($fullClassName));
-
-        $variables = array(
-            '<namespace>' => $namespace,
-            '<className>' => $className
-        );
-        $code = str_replace(array_keys($variables), array_values($variables), self::$_template);
-        return $code;
-    }
-
-    protected function _writeRepositoryClass($metadata, $destPath)
-    {
-        $code = $this->_generateRepositoryClass($metadata);
-
-        $path = $destPath . DIRECTORY_SEPARATOR
-              . str_replace('\\', \DIRECTORY_SEPARATOR, $metadata->customRepositoryClassName) . '.php';
-        $dir = dirname($path);
-
-        if ( ! is_dir($dir)) {
-            mkdir($dir, 0777, true);
-        }
-
-        file_put_contents($path, $code);
-    }
 }
\ No newline at end of file

lib/Doctrine/ORM/Tools/EntityRepositoryGenerator.php

+<?php
+/*
+ *  $Id$
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ * This software consists of voluntary contributions made by many individuals
+ * and is licensed under the LGPL. For more information, see
+ * <http://www.doctrine-project.org>.
+ */
+
+namespace Doctrine\ORM\Tools;
+
+/**
+ * Class to generate entity repository classes
+ *
+ * @license http://www.opensource.org/licenses/lgpl-license.php LGPL
+ * @link    www.doctrine-project.org
+ * @since   2.0
+ * @version $Revision$
+ * @author  Benjamin Eberlei <kontakt@beberlei.de>
+ * @author  Guilherme Blanco <guilhermeblanco@hotmail.com>
+ * @author  Jonathan Wage <jonwage@gmail.com>
+ * @author  Roman Borschel <roman@code-factory.org>
+ */
+class EntityRepositoryGenerator
+{
+    protected static $_template =
+'<?php
+
+namespace <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
+{
+}';
+
+    public function generateEntityRepositoryClass($fullClassName)
+    {
+        $namespace = substr($fullClassName, 0, strrpos($fullClassName, '\\'));
+        $className = substr($fullClassName, strrpos($fullClassName, '\\') + 1, strlen($fullClassName));
+
+        $variables = array(
+            '<namespace>' => $namespace,
+            '<className>' => $className
+        );
+        return str_replace(array_keys($variables), array_values($variables), self::$_template);
+    }
+
+    public function writeEntityRepositoryClass($fullClassName, $outputDirectory)
+    {
+        $code = $this->generateEntityRepositoryClass($fullClassName);
+
+        $path = $outputDirectory . DIRECTORY_SEPARATOR
+              . str_replace('\\', \DIRECTORY_SEPARATOR, $fullClassName) . '.php';
+        $dir = dirname($path);
+
+        if ( ! is_dir($dir)) {
+            mkdir($dir, 0777, true);
+        }
+
+        file_put_contents($path, $code);
+    }
+}
\ No newline at end of file