vendor/doctrine/orm/lib/Doctrine/ORM/Tools/EntityRepositoryGenerator.php

   1 <?php
   2 /*
   3  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
   4  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
   5  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
   6  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
   7  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
   8  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
   9  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  10  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  11  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  12  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  13  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  14  *
  15  * This software consists of voluntary contributions made by many individuals
  16  * and is licensed under the MIT license. For more information, see
  17  * <http://www.doctrine-project.org>.
  18  */
  19
  20 namespace Doctrine\ORM\Tools;
  21
  22 /**
  23  * Class to generate entity repository classes
  24  *
  25  *
  26  * @link    www.doctrine-project.org
  27  * @since   2.0
  28  * @author  Benjamin Eberlei <kontakt@beberlei.de>
  29  * @author  Guilherme Blanco <guilhermeblanco@hotmail.com>
  30  * @author  Jonathan Wage <jonwage@gmail.com>
  31  * @author  Roman Borschel <roman@code-factory.org>
  32  */
  33 class EntityRepositoryGenerator
  34 {
  35     protected static $_template =
  36 '<?php
  37
  38 namespace <namespace>;
  39
  40 use Doctrine\ORM\EntityRepository;
  41
  42 /**
  43  * <className>
  44  *
  45  * This class was generated by the Doctrine ORM. Add your own custom
  46  * repository methods below.
  47  */
  48 class <className> extends EntityRepository
  49 {
  50 }
  51 ';
  52
  53     public function generateEntityRepositoryClass($fullClassName)
  54     {
  55         $namespace = substr($fullClassName, 0, strrpos($fullClassName, '\\'));
  56         $className = substr($fullClassName, strrpos($fullClassName, '\\') + 1, strlen($fullClassName));
  57
  58         $variables = array(
  59             '<namespace>' => $namespace,
  60             '<className>' => $className
  61         );
  62         return str_replace(array_keys($variables), array_values($variables), self::$_template);
  63     }
  64
  65     public function writeEntityRepositoryClass($fullClassName, $outputDirectory)
  66     {
  67         $code = $this->generateEntityRepositoryClass($fullClassName);
  68
  69         $path = $outputDirectory . DIRECTORY_SEPARATOR
  70               . str_replace('\\', \DIRECTORY_SEPARATOR, $fullClassName) . '.php';
  71         $dir = dirname($path);
  72
  73         if ( ! is_dir($dir)) {
  74             mkdir($dir, 0777, true);
  75         }
  76
  77         if ( ! file_exists($path)) {
  78             file_put_contents($path, $code);
  79         }
  80     }
  81 }